package service

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

type PlcBrandService struct{}

func NewPlcBrandService() PlcBrandService {
	return PlcBrandService{}
}

func (PlcBrandService) AddPlcBrand(plcBrand *model.PlcBrand) int {
	err := model.NewPlcBrandSearch().Create(plcBrand)
	if err != nil {
		return ecode.DBErr
	}

	return ecode.OK
}

func (PlcBrandService) GetPlcBrand(id uint) (*model.PlcBrand, int) {
	plcBrand, err := model.NewPlcBrandSearch().SetId(id).First()
	if err != nil {
		return nil, ecode.DBErr
	}

	return plcBrand, ecode.OK
}

func (PlcBrandService) DeletePlcBrand(id uint) int {
	err := model.NewPlcBrandSearch().SetId(id).Delete()
	if err != nil {
		return ecode.DBErr
	}
	return ecode.OK
}

func (PlcBrandService) GetPlcBrandList() ([]*model.PlcBrand, int64, int) {
	list, total, err := model.NewPlcBrandSearch().Find()
	if err != nil {
		return nil, 0, ecode.DBErr
	}

	return list, total, ecode.OK
}

func (PlcBrandService) UpdatePlcBrands(PlcBrands []*request.UpdatePlcBrand) int {
	for _, v := range PlcBrands {
		// check PlcBrand exist
		_, err := model.NewPlcBrandSearch().SetId(v.Id).First()
		if err != nil {
			return ecode.DBErr
		}

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

	return ecode.OK
}

func (PlcBrandService) UpdatePlcBrand(plcBrand *model.PlcBrand) int {
	err := model.NewPlcBrandSearch().SetId(plcBrand.ID).Save(plcBrand)
	if err != nil {
		return ecode.DBErr
	}
	return ecode.OK
}