package service

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

type FollowRecordService struct{}

func (FollowRecordService) AddFollowRecord(followRecord *model.FollowRecord) int {
	err := model.NewFollowRecordSearch().Create(followRecord)
	if err != nil {
		return ecode.FollowRecordExist
	}
	return ecode.OK
}

func (FollowRecordService) DeleteFollowRecord(id int) int {
	// check followRecord exist
	_, err := model.NewFollowRecordSearch().SetId(id).First()
	if err != nil {
		return ecode.FollowRecordNotExist
	}

	// delete followRecord
	err = model.NewFollowRecordSearch().SetId(id).Delete()
	if err != nil {
		return ecode.FollowRecordDeleteErr
	}
	return ecode.OK
}

// check followRecord exist
func checkFollowRecordExist(id int) int {
	_, err := model.NewFollowRecordSearch().SetId(id).First()
	if err != nil {
		return ecode.FollowRecordNotExist
	}

	return ecode.OK
}

func (FollowRecordService) UpdateFollowRecord(followRecord *model.FollowRecord) int {
	// check followRecord exist
	errCode := checkFollowRecordExist(followRecord.Id)
	if errCode != ecode.OK {
		return errCode
	}

	// update followRecord
	err := model.NewFollowRecordSearch().SetId(followRecord.Id).Update(followRecord)
	if err != nil {
		return ecode.FollowRecordUpdateErr
	}
	return ecode.OK
}

func (FollowRecordService) GetFollowRecordList(page, pageSize int, keyword string) ([]*model.FollowRecord, int64, int) {
	// get contact list
	contacts, total, err := model.NewFollowRecordSearch().SetKeyword(keyword).SetPage(page, pageSize).FindAll()
	if err != nil {
	return nil, 0, ecode.FollowRecordListErr
	}
	return contacts, total, ecode.OK
}