package impl

import (
	"ai_scheduler/internal/data/model"
	"ai_scheduler/tmpl/dataTemp"
	"ai_scheduler/utils"
	"database/sql"
)

type BotUserImpl struct {
	dataTemp.DataTemp
}

func NewBotUserImpl(db *utils.Db) *BotUserImpl {
	return &BotUserImpl{
		DataTemp: *dataTemp.NewDataTemp(db, new(model.AiBotUser)),
	}
}

func (k BotUserImpl) GetByStaffId(staffId string) (*model.AiBotUser, error) {
	var data model.AiBotUser
	err := k.Db.Model(k.Model).Where("staff_id = ?", staffId).Find(&data).Error
	if data.UserID == 0 {
		err = sql.ErrNoRows
	}
	return &data, err
}