package service

import (
	"aps_crm/model"
	"aps_crm/model/request"
	"aps_crm/pkg/ecode"
)

type CourierCompanyService struct{}

func NewCourierCompanyService() CourierCompanyService {
	return CourierCompanyService{}
}

func (CourierCompanyService) AddCourierCompany(courierCompany *model.CourierCompany) int {
	err := model.NewCourierCompanySearch().Create(courierCompany)
	if err != nil {
		return ecode.DBErr
	}

	return ecode.OK
}

func (CourierCompanyService) DeleteCourierCompany(id int) int {
	err := model.NewCourierCompanySearch().SetId(id).Delete()
	if err != nil {
		return ecode.DBErr
	}
	return ecode.OK
}

func (CourierCompanyService) GetCourierCompanyList() ([]*model.CourierCompany, int64, int) {
	list, total, err := model.NewCourierCompanySearch().Find()
	if err != nil {
		return nil, 0, ecode.DBErr
	}

	return list, total, ecode.OK
}

func (CourierCompanyService) UpdateCourierCompanys(CourierCompanys []*request.UpdateCourierCompany) int {
	for _, v := range CourierCompanys {
		// check CourierCompany exist
		_, err := model.NewCourierCompanySearch().SetId(v.Id).First()
		if err != nil {
			return ecode.DBErr
		}

		err = model.NewCourierCompanySearch().SetId(v.Id).Updates(map[string]interface{}{})
		if err != nil {
			return ecode.DBErr
		}
	}

	return ecode.OK
}

func (CourierCompanyService) UpdateCourierCompany(courierCompany *model.CourierCompany) int {
	err := model.NewCourierCompanySearch().SetId(courierCompany.Id).Update(courierCompany)
	if err != nil {
		return ecode.DBErr
	}
	return ecode.OK
}