package service

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

type ContractService struct{}

func (ContractService) AddContract(contract *model.Contract) int {
	err := model.NewContractSearch().Create(contract)
	if err != nil {
		return ecode.ContractExist
	}

	return ecode.OK
}

func (ContractService) DeleteContract(id int) int {
	_, err := model.NewContractSearch().SetId(id).Find()
	if err != nil {
		return ecode.ContractNotExist
	}

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

func (ContractService) GetContractList() ([]*model.Contract, int) {
	list, err := model.NewContractSearch().FindAll()
	if err != nil {
		return nil, ecode.ContractListErr
	}

	return list, ecode.OK
}

func (ContractService) UpdateContract(contract *model.Contract) int {
	// check contract exist
	_, err := model.NewContractSearch().SetId(contract.Id).Find()
	if err != nil {
		return ecode.ContractNotExist
	}

	err = model.NewContractSearch().SetId(contract.Id).Update(contract)
	if err != nil {
		return ecode.ContractSetErr
	}

	return ecode.OK
}