package mcp

import (
	"context"
	"yunyan/lego/core"
	"yunyan/lego/core/cbase"
	"yunyan/lego/sys/log"
	"yunyan/sys/juhe"
	"fmt"

	"github.com/mark3labs/mcp-go/mcp"
)

// 股票信息卡片
type FinanceCard struct {
	CompanyName string `json:"companyName"`
	CurrPrice   string `json:"currPrice"`
	OpenPrice   string `json:"openPrice"`
	TodayLow    string `json:"todayLow"`
	TodayHigh   string `json:"todayHigh"`
	Volume      string `json:"volume"`
}

// 股票查询
type tool_finance struct {
	cbase.ModuleCompBase
	module *Mcp
}

// 组件初始化接口
func (this *tool_finance) Init(service core.IService, module core.IModule, comp core.IModuleComp, opt core.IModuleOptions) (err error) {
	this.ModuleCompBase.Init(service, module, comp, opt)
	this.module = module.(*Mcp)
	return
}
func (this *tool_finance) Start() (err error) {
	err = this.ModuleCompBase.Start()
	if this.module.AddTool(ToolGroup_CHINA, this.Tool(), this.Handl) {
		if err := juhe.OnInit(this.module.service.GetSettings().Sys["juhe"]); err != nil {
			panic(fmt.Sprintf("init sys.juhe err: %s", err.Error()))
		} else {
			log.Infof("init sys.juhe success!")
		}
	}
	return
}
func (this *tool_finance) Tool() (tool mcp.Tool) {
	return mcp.NewTool("finance_query",
		mcp.WithDescription("根据提供的股票代码查询股票信息"),
		mcp.WithString("code",
			mcp.Description("股票代码"),
			mcp.Required(),
		),
	)
}

// Handl 工具处理接口
// 参数:
//   - ctx: 上下文
//   - request: MCP 工具调用请求
//
// 返回值:
//   - result: MCP 工具调用结果
//   - err: 处理过程中的错误
//
// 异常:
//   - 无（不主动 panic）
func (this *tool_finance) Handl(ctx context.Context, request mcp.CallToolRequest) (result *mcp.CallToolResult, err error) {
	var (
		code    string
		data    *juhe.StockResponse
		card    *FinanceCard
		context string
	)
	if code, err = request.RequireString("code"); err != nil {
		return
	}
	if data, err = juhe.Financebygid(code); err != nil {
		context = "获取天气信息失败"
	} else {
		context = fmt.Sprintf("当前%s股票情况:今日开盘价:%s,昨日收盘价:%s,当前价格:%s,今日最高价:%s,今日最低价:%s,成交量:%s,成交金额:%s",
			data.FinanceResult[0].Data.Name,
			data.FinanceResult[0].Data.TodayStartPri,
			data.FinanceResult[0].Data.YestodEndPri,
			data.FinanceResult[0].Data.NowPri,
			data.FinanceResult[0].Data.TodayMax,
			data.FinanceResult[0].Data.TodayMin,
			data.FinanceResult[0].Data.TraNumber,
			data.FinanceResult[0].Data.TraAmount,
		)
		card = &FinanceCard{
			CompanyName: code,
			CurrPrice:   data.FinanceResult[0].Data.NowPri,
			OpenPrice:   data.FinanceResult[0].Data.TodayStartPri,
			TodayLow:    data.FinanceResult[0].Data.TodayMin,
			TodayHigh:   data.FinanceResult[0].Data.TodayMax,
			Volume:      data.FinanceResult[0].Data.TraNumber,
		}
	}

	result = &mcp.CallToolResult{
		Content: []mcp.Content{
			mcp.TextContent{
				Type: "text",
				Text: context,
			},
		},
	}
	result.Meta = map[string]interface{}{
		"tool_finance": card,
	}
	return
}