package service

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

type PlanService struct{}

func (PlanService) AddPlan(plan *model.Plan) int {
	err := model.NewPlanSearch().Create(plan)
	if err != nil {
		return ecode.PlanExist
	}

	return ecode.OK
}

func (PlanService) DeletePlan(id int) int {
	_, err := model.NewPlanSearch().SetId(id).Find()
	if err != nil {
		return ecode.PlanNotExist
	}

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

func (PlanService) GetPlanList() ([]*model.Plan, int) {
	list, err := model.NewPlanSearch().FindAll()
	if err != nil {
		return nil, ecode.PlanListErr
	}

	return list, ecode.OK
}

func (PlanService) UpdatePlan(plan *model.Plan) int {
	// check plan exist
	_, err := model.NewPlanSearch().SetId(plan.Id).Find()
	if err != nil {
		return ecode.PlanNotExist
	}

	err = model.NewPlanSearch().SetId(plan.Id).Update(plan)
	if err != nil {
		return ecode.PlanSetErr
	}

	return ecode.OK
}