using System.Collections.Generic;
using LinkToolAddin.host.mcp;

namespace LinkToolAddin.host;

public class McpServerList
{
    private Dictionary<string,McpServer> servers = new Dictionary<string, McpServer>();

    public McpServerList()
    {
        servers.Add("gaode",new SseMcpServer
        {
            Name = "gaode",
            Type = "sse",
            Description = "高德地图API",
            IsActive = true,
            BaseUrl = "https://mcp.amap.com/sse?key=ed418512c94ade8f83d42c37b77d2bb2",
            Headers = new Dictionary<string, string>()
            {
                {"Content-Type","application/json"}
            }
        });
        servers.Add("ArcGisPro", new InnerMcpServer
        {
            Name = "ArcGisPro",
            Type = "inner",
            Description = "可以调用arcgis的地理处理工具或执行python代码等",
            IsActive = true
        });
        servers.Add("KnowledgeBase", new InnerMcpServer
        {
            Name = "KnowledgeBase",
            Type = "inner",
            Description = "可以调用进行查询知识库，获取相关参考信息。",
            IsActive = true
        });
        servers.Add("filesystem", new StdioMcpServer()
        {
            Name = "filesystem",
            Type = "stdio",
            Command = "npxh",
            Args = new List<string>()
            {
                "-y",
                "@modelcontextprotocol/server-filesystem",
                "D:\\01_Project\\20250305_LinkTool\\20250420_AiDemoProject\\TestData"
            }
        });
        servers.Add("fetch", new StdioMcpServer()
        {
            Name = "fetch",
            Type = "stdio",
            Command = "uvx",
            Args = new List<string>()
            {
                "mcp-server-fetch"
            }
        });
        servers.Add("bing-search", new StdioMcpServer()
        {
            Name = "bing-search",
            Type = "stdio",
            Command = "npx",
            Args = new List<string>()
            {
                "bing-cn-mcp"
            }
        });
        servers.Add("mcp-python-interpreter", new StdioMcpServer()
        {
            Name = "mcp-python-interpreter",
            Type = "stdio",
            Command = "uvx",
            Args = new List<string>()
            {
                "--native-tls",
                "mcp-python-interpreter",
                "--dir",
                "D:\\01_Project\\20250305_LinkTool\\20250420_AiDemoProject\\TestData",
                "--python-path",
                "C:\\Program Files\\ArcGIS\\Pro\\bin\\Python\\envs\\custom\\python.exe"
            }
        });
    }
    
    public McpServer GetServer(string name)
    {
        if (servers.ContainsKey(name))
        {
            return servers[name];
        }
        else
        {
            return null;
        }
    }
    
    public List<McpServer> GetAllServers()
    {
        List<McpServer> serverList = new List<McpServer>();
        foreach (var server in servers)
        {
            serverList.Add(server.Value);
        }
        return serverList;
    }
}