import { Client } from '@modelcontextprotocol/sdk/client/index.js';

import { MCP_SERVER_TYPE, SSEClientTransportOptions } from './types';

export interface IMCPServer {
  isStarted(): boolean;
  start(): Promise<void>;
  getServerName(): string;
  callTool(toolName: string, toolCallId: string, arg_string: string): ReturnType<Client['callTool']>;
  getTools(): ReturnType<Client['listTools']>;
  getClient(): Client | null;
  update(command: string, args?: string[], env?: { [key: string]: string }): void;
  stop(): void;
}

export interface MCPServerManager {
  callTool(
    serverName: string,
    toolName: string,
    toolCallId: string,
    arg_string: string,
  ): ReturnType<Client['callTool']>;
  removeServer(name: string): void;
  addOrUpdateServer(description: MCPServerDescription): void;
  // invoke in node.js only
  addOrUpdateServerDirectly(server: any): void;
  initBuiltinServer(builtinMCPServer: any, enabled: boolean): void;
  getTools(serverName: string): ReturnType<Client['listTools']>;
  getServerNames(): Promise<string[]>;
  startServer(serverName: string): Promise<void>;
  stopServer(serverName: string): Promise<void>;
  getStartedServers(): Promise<string[]>;
  registerTools(serverName: string): Promise<void>;
  addExternalMCPServers(servers: MCPServerDescription[]): void;
  getServers(): Map<string, IMCPServer>;
  getServerByName(name: string): IMCPServer | undefined;
}

export type MCPTool = Awaited<ReturnType<MCPServerManager['getTools']>>['tools'][number];

export type MCPToolParameter = Awaited<ReturnType<MCPServerManager['getTools']>>['tools'][number]['inputSchema'];

export interface BaseMCPServerDescription {
  /**
   * The unique name of the MCP server.
   */
  name: string;
  /**
   * Whether to enable the MCP server.
   */
  enabled?: boolean;
}

export interface StdioMCPServerDescription extends BaseMCPServerDescription {
  /**
   * The unique name of the MCP server.
   */
  name: string;

  /**
   * The command to execute the MCP server.
   */
  command: string;

  /**
   * An array of arguments to pass to the command.
   */
  args?: string[];

  /**
   * Optional environment variables to set when starting the server.
   */
  env?: { [key: string]: string };
}

export interface SSEMCPServerDescription extends BaseMCPServerDescription {
  /**
   * The host of the MCP server.
   */
  url: string;
  transportOptions?: SSEClientTransportOptions;
}

export interface BuiltinMCPServerDescription {
  /**
   * The unique name of the MCP server.
   */
  name: string;
  /**
   * Whether to enable the MCP server.
   */
  enabled?: boolean;
}

export type MCPServerDescription =
  | ({ type: MCP_SERVER_TYPE.STDIO } & StdioMCPServerDescription)
  | ({ type: MCP_SERVER_TYPE.SSE } & SSEMCPServerDescription)
  | ({ type: MCP_SERVER_TYPE.BUILTIN } & BuiltinMCPServerDescription);

export const MCPServerManager = Symbol('MCPServerManager');
export const MCPServerManagerPath = 'ServicesMCPServerManager';

export const MCPServersEnabledKey = 'mcp_servers_enabled';