package service

import (
	"context"
	"eino-project/internal/domain/agent"
	"eino-project/internal/domain/llm"
	"eino-project/internal/pkg/adkutil"
	"encoding/json"
	"net/http"

	"github.com/cloudwego/eino/adk"
	"github.com/go-kratos/kratos/v2/log"
	kratoshttp "github.com/go-kratos/kratos/v2/transport/http"
)

type AgentService struct {
	productAgent adk.Agent
	log          *log.Helper
}

func NewAgentService(
    logger log.Logger,
    models llm.LLM,
) *AgentService {
    return &AgentService{
        log: log.NewHelper(logger),
        // 构建商品查询 ChatModelAgent，绑定工具并让模型自动选择调用
        productAgent: agent.NewProductChatAgent(context.Background(), models),
    }
}

type ProductQueryRes struct {
	Items  []*ProductItem `json:"items"`
	Source string         `json:"source"`
	Count  int            `json:"count"`
}
type ProductItem struct {
	ID          string `json:"id"`
	Name        string `json:"name"`
	Price       int    `json:"price"`
	Description string `json:"description"`
}

type CommonResp struct {
	Code int         `json:"code"`
	Msg  string      `json:"msg"`
	Data interface{} `json:"data"`
}

// HandleProductQuery 商品查询处理
func (a *AgentService) HandleProductQuery(ctx kratoshttp.Context) error {
	type reqType struct {
		Message string `json:"message"`
		Session string `json:"session_id"`
	}

	var req reqType
	if err := json.NewDecoder(ctx.Request().Body).Decode(&req); err != nil {
		return ctx.JSON(http.StatusBadRequest, &CommonResp{
			Code: http.StatusBadRequest,
			Msg:  "Invalid request body",
		})
	}
	if a.productAgent == nil {
		return ctx.JSON(http.StatusInternalServerError, &CommonResp{Code: http.StatusInternalServerError, Msg: "product agent not configured"})
	}

	out, err := adkutil.QueryJSONWithLogger[ProductQueryRes](ctx, a.productAgent, req.Message, a.log)
	if err != nil {
		return ctx.JSON(http.StatusInternalServerError, &CommonResp{Code: http.StatusInternalServerError, Msg: err.Error()})
	}

	return ctx.JSON(http.StatusOK, &CommonResp{Code: http.StatusOK, Msg: "success", Data: out})
}