package biz

import (
	"ai_scheduler/internal/biz/do"
	"ai_scheduler/internal/entitys"
	"context"

	"github.com/open-dingtalk/dingtalk-stream-sdk-go/chatbot"
)

// AiRouterBiz 智能路由服务
type DingTalkBotBiz struct {
	do     *do.Do
	handle *do.Handle
}

// NewDingTalkBotBiz
func NewDingTalkBotBiz(
	do *do.Do,
	handle *do.Handle,
) *DingTalkBotBiz {
	return &DingTalkBotBiz{
		do:     do,
		handle: handle,
	}
}

func (d *DingTalkBotBiz) InitRequire(ctx context.Context, data *chatbot.BotCallbackDataModel) (requireData *entitys.RequireDataDingTalkBot, err error) {
	requireData = &entitys.RequireDataDingTalkBot{
		Req: data,
		Ch:  make(chan entitys.Response, 2),
	}
	entitys.ResLog(requireData.Ch, "recognize_start", "收到消息，正在处理中，请稍等")

	requireData.Sys, err = d.do.GetSysInfoForDingTalkBot(requireData)
	requireData.Tasks, err = d.do.GetTasks(requireData.Sys.SysID)
	return
}

func (d *DingTalkBotBiz) Recognize(ctx context.Context, rec *entitys.Recognize, ch chan entitys.Response) (match *entitys.Match, err error) {
	return d.handle.RecognizeBot(ctx, rec, ch)
}