package service

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

type SubOrderService struct{}

func (SubOrderService) AddSubOrder(subOrder *model.SubOrder) int {
	err := model.NewSubOrderSearch().Create(subOrder)
	if err != nil {
		return ecode.SubOrderExist
	}

	return ecode.OK
}

func (SubOrderService) DeleteSubOrder(id int) int {
	_, err := model.NewSubOrderSearch().SetId(id).Find()
	if err != nil {
		return ecode.SubOrderNotExist
	}

	err = model.NewSubOrderSearch().SetId(id).Delete()
	if err != nil {
		return ecode.SubOrderNotExist
	}
	return ecode.OK
}

func (SubOrderService) GetSubOrderList() ([]*model.SubOrder, int) {
	list, err := model.NewSubOrderSearch().FindAll()
	if err != nil {
		return nil, ecode.SubOrderListErr
	}

	return list, ecode.OK
}

func (SubOrderService) UpdateSubOrder(subOrder *model.SubOrder) int {
	// check subOrder exist
	_, err := model.NewSubOrderSearch().SetId(subOrder.Id).Find()
	if err != nil {
		return ecode.SubOrderNotExist
	}

	err = model.NewSubOrderSearch().SetId(subOrder.Id).Update(subOrder)
	if err != nil {
		return ecode.SubOrderSetErr
	}

	return ecode.OK
}