import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
import { roots } from "../server/roots.js"

// Tool configuration
const name = "get-roots-list";
const config = {
  title: "Get Roots List Tool",
  description:
    "Lists the current MCP roots provided by the client. Demonstrates the roots protocol capability even though this server doesn't access files.",
  inputSchema: {},
};

/**
 * Registers the 'get-roots-list' tool with the given MCP server.
 *
 * If the client does not support the roots capability, the tool is not registered.
 *
 * The registered tool interacts with the MCP roots capability, which enables the server to access
 * information about the client's workspace directories or file system roots.
 *
 * When supported, the server automatically retrieves and formats the current list of roots from the
 * client upon connection and whenever the client sends a `roots/list_changed` notification.
 *
 * Therefore, this tool displays the roots that the server currently knows about for the connected
 * client. If for some reason the server never got the initial roots list, the tool will request the
 * list from the client again.
 *
 * @param {McpServer} server - The server instance interacting with the MCP client
 */
export const registerGetRootsListTool = (server: McpServer) => {
  const clientSupportsRoots =
    server.server.getClientCapabilities()?.roots?.listChanged;
  if (!clientSupportsRoots) {
    server.registerTool(
      name,
      config,
      async (args, extra): Promise<CallToolResult> => {
        const currentRoots = roots?.has(extra.sessionId)
          ? roots.get(extra.sessionId)
          : (await server.server.listRoots()).roots
        if (currentRoots && currentRoots.length === 0) {
          return {
            content: [
              {
                type: "text",
                text:
                  "The client supports roots but no roots are currently configured.\n\n" +
                  "This could mean:\n" +
                  "1. The client hasn't provided any roots yet\n" +
                  "2. The client provided an empty roots list\n" +
                  "3. The roots configuration is still being loaded",
              },
            ],
          };
        }

        const rootsList = currentRoots ?  currentRoots
          .map((root, index) => {
            return `${index + 1}. ${root.name || "Unnamed Root"}\n   URI: ${
              root.uri
            }`;
          })
          .join("\n\n") : 'No roots found'

        return {
          content: [
            {
              type: "text",
              text:
                `Current MCP Roots (${currentRoots!.length} total):\n\n${rootsList}\n\n` +
                "Note: This server demonstrates the roots protocol capability but doesn't actually access files. " +
                "The roots are provided by the MCP client and can be used by servers that need file system access.",
            },
          ],
        };
      }
    );
  }
};