package services

import (
	"os"
	"path/filepath"
	"runtime"
	"strings"
)

// McpMarketItem MCP 市场条目
type McpMarketItem struct {
	Key         string   `json:"key"`
	Name        string   `json:"name"`
	Description string   `json:"description"`
	Provider    string   `json:"provider"`
	Transport   string   `json:"transport"` // stdio/http/sse
	Command     string   `json:"command"`
	Args        []string `json:"args"`
	Env         []string `json:"env"`
	URL         string   `json:"url"`
	Tags        []string `json:"tags"`
	BuiltIn     bool     `json:"built_in"` // 是否为项目内置（演示）服务
}

// demoMcpPath 解析内置演示 MCP 服务可执行文件路径
// 优先级：1) 服务器工作目录下 bin/demo-mcp 2) go run ./cmd/demo-mcp 兜底
func demoMcpPath() (string, []string) {
	ext := ""
	if runtime.GOOS == "windows" {
		ext = ".exe"
	}
	candidates := []string{
		filepath.Join("bin", "demo-mcp"+ext),
		filepath.Join("..", "bin", "demo-mcp"+ext),
	}
	if wd, err := os.Getwd(); err == nil {
		for _, c := range candidates {
			p := filepath.Join(wd, c)
			if info, err := os.Stat(p); err == nil && !info.IsDir() {
				return p, nil
			}
		}
	}
	// 兜底：go run ./cmd/demo-mcp（需要 Go 工具链）
	return "go", []string{"run", "./cmd/demo-mcp"}
}

// GetMcpMarket 返回 MCP 市场服务列表
func GetMcpMarket() []McpMarketItem {
	cmd, args := demoMcpPath()
	list := []McpMarketItem{
		{
			Key:         "demo",
			Name:        "内置演示 MCP",
			Description: "项目自带演示服务：提供当前时间、计算器、模拟天气、回声等工具，用于验证 MCP 全链路。",
			Provider:    "本项目",
			Transport:   "stdio",
			Command:     cmd,
			Args:        args,
			Tags:        []string{"演示", "零配置"},
			BuiltIn:     true,
		},
		{
			Key:         "tianyancha",
			Name:        "天眼查 MCP",
			Description: "企业工商信息查询、股权穿透、司法风险等数据服务。",
			Provider:    "天眼查",
			Transport:   "http",
			URL:         "https://mcp.tianyancha.com/mcp",
			Tags:        []string{"企业信息"},
		},
		{
			Key:         "qcc",
			Name:        "企查查 MCP",
			Description: "企业信用信息、工商资料、经营风险等数据查询服务。",
			Provider:    "企查查",
			Transport:   "http",
			URL:         "https://mcp.qcc.com/mcp",
			Tags:        []string{"企业信息"},
		},
		{
			Key:         "qianzhan",
			Name:        "前瞻 MCP",
			Description: "前瞻产业研究院行业数据、研究报告等。",
			Provider:    "前瞻",
			Transport:   "http",
			URL:         "https://mcp.qianzhan.com/mcp",
			Tags:        []string{"行业数据"},
		},
		{
			Key:         "itjuzi",
			Name:        "IT桔子 MCP",
			Description: "创业公司数据、投融资事件、行业洞察等。",
			Provider:    "IT桔子",
			Transport:   "http",
			URL:         "https://mcp.itjuzi.com/mcp",
			Tags:        []string{"创投数据"},
		},
		{
			Key:         "boss",
			Name:        "BOSS直聘 MCP",
			Description: "BOSS直聘企业招聘、职位等数据服务。",
			Provider:    "BOSS直聘",
			Transport:   "http",
			URL:         "https://mcp.zhipin.com/mcp",
			Tags:        []string{"招聘"},
		},
	}
	return list
}

// FindMarketItem 按 key 查找市场条目
func FindMarketItem(key string) (McpMarketItem, bool) {
	for _, item := range GetMcpMarket() {
		if strings.EqualFold(item.Key, key) {
			return item, true
		}
	}
	return McpMarketItem{}, false
}