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
}

func (FollowRecordService) GetFollowRecordList() (int, []*model.FollowRecord) {
	// get followRecord list
	followRecords, err := model.NewFollowRecordSearch().Find()
	if err != nil {
		return ecode.FollowRecordListErr, nil
	}

	return ecode.OK, followRecords
}

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