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
}

// 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, data map[string]interface{}) ([]*model.FollowRecord, int64, int) {
	// get contact list
	contacts, total, err := model.NewFollowRecordSearch().SetPage(page, pageSize).SetSearchMap(data).FindAll()
	if err != nil {
		return nil, 0, ecode.FollowRecordListErr
	}
	return contacts, total, ecode.OK
}

func (FollowRecordService) DeleteFollowRecord(ids []int) int {
	// delete client
	err := model.NewFollowRecordSearch().SetIds(ids).Delete()
	if err != nil {
		return ecode.FollowRecordDeleteErr
	}
	return ecode.OK
}

func (FollowRecordService) Assign(ids []int, memberId int) int {
	// check followRecord exist
	//errCode := checkFollowRecordExist(id)
	//if errCode != ecode.OK {
	//	return errCode
	//}

	// assign followRecord
	err := model.NewFollowRecordSearch().SetIds(ids).UpdateMap(map[string]interface{}{
		"member_id": memberId,
	})
	if err != nil {
		return ecode.FollowRecordAssignErr
	}

	return ecode.OK
}