package mcp

import (
	"context"
	"yunyan/lego/core"
	"yunyan/lego/core/cbase"
	"yunyan/lego/sys/log"
	lgspotify "yunyan/sys/spotify"
	"yunyan/utils"
	"fmt"

	"github.com/mark3labs/mcp-go/mcp"
	"github.com/zmb3/spotify/v2"
)

// 音乐卡片 结构统一化输出
type SpotifyMusicCard struct {
	Name   string `json:"name"`
	Images string `json:"image"`
	Url    string `json:"url"`
	Sgener string `json:"sgener"`
}

// 天气查询
type tool_spotify_music_play struct {
	cbase.ModuleCompBase
	module *Mcp
}

// 组件初始化接口
func (this *tool_spotify_music_play) 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_spotify_music_play) Start() (err error) {
	err = this.ModuleCompBase.Start()
	if this.module.AddTool(ToolGroup_GLOBAL, this.Tool(), this.Handl) {
		if err := lgspotify.OnInit(this.module.service.GetSettings().Sys["spotify"]); err != nil {
			panic(fmt.Sprintf("init sys.spotify err: %s", err.Error()))
		} else {
			log.Infof("init sys.spotify success!")
		}
	}
	return
}

func (this *tool_spotify_music_play) Tool() (tool mcp.Tool) {
	return mcp.NewTool("spotify_music_play",
		mcp.WithDescription("为用户提供音乐播放服务"),
		mcp.WithString("query",
			mcp.Description("Music keyword information"),
			mcp.Required(),
		),
		mcp.WithString("offset",
			mcp.Description("music offset (max 9, default 0)"),
			mcp.DefaultNumber(0),
		),
	)
}

// Handl 工具处理接口
// 参数:
//   - ctx: 上下文
//   - request: MCP 工具调用请求
//
// 返回值:
//   - result: MCP 工具调用结果
//   - err: 处理过程中的错误
//
// 异常:
//   - 无（不主动 panic）
func (this *tool_spotify_music_play) Handl(ctx context.Context, request mcp.CallToolRequest) (result *mcp.CallToolResult, err error) {
	var (
		offset  int
		query   string
		data    []spotify.FullTrack
		card    *SpotifyMusicCard
		context string
	)
	offset = request.GetInt("offset", 0)
	if query, err = request.RequireString("query"); err != nil {
		return
	}
	data, err = lgspotify.SearchMusic(ctx, query, 1, offset)

	if err != nil {
		context = "播放音乐失败"
		log.Errorf("获取天气信息失败:%v", err)
		return
	} else {
		card = &SpotifyMusicCard{
			Name:   data[0].Name,
			Images: data[0].Album.Images[0].URL,
			Url:    string(data[0].URI),
			Sgener: data[0].Album.Name,
		}
		context = utils.ToString(map[string]interface{}{
			"card_music": card,
			"result":     fmt.Sprintf("正在为你播放音乐:%s", data[0].Name),
		})
		log.Debug("播放音乐成功",
			log.Field{Key: "query", Value: query},
			log.Field{Key: "card", Value: card},
		)
	}

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