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) 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
}

func (PlanService) GetPlanList(page, pageSize int, keyword string) ([]*model.Plan, int64, int) {
	// get contact list
	contacts, total, err := model.NewPlanSearch().SetKeyword(keyword).SetPage(page, pageSize).FindAll()
	if err != nil {
		return nil, 0, ecode.PlanListErr
	}
	return contacts, total, ecode.OK
}