@unkey/api/src/mcp-server/server.ts

Developer-friendly & type-safe Typescript SDK specifically catered to leverage *@unkey/api* API.

/*
 * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
 */

import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { UnkeyCore } from "../core.js";
import { SDKOptions } from "../lib/config.js";
import type { ConsoleLogger } from "./console-logger.js";
import { createRegisterPrompt } from "./prompts.js";
import {
  createRegisterResource,
  createRegisterResourceTemplate,
} from "./resources.js";
import { MCPScope } from "./scopes.js";
import { createRegisterTool } from "./tools.js";
import { tool$apisCreateApi } from "./tools/apisCreateApi.js";
import { tool$identitiesCreateIdentity } from "./tools/identitiesCreateIdentity.js";
import { tool$livenessLiveness } from "./tools/livenessLiveness.js";
import { tool$ratelimitDeleteOverride } from "./tools/ratelimitDeleteOverride.js";
import { tool$ratelimitGetOverride } from "./tools/ratelimitGetOverride.js";
import { tool$ratelimitLimit } from "./tools/ratelimitLimit.js";
import { tool$ratelimitListOverrides } from "./tools/ratelimitListOverrides.js";
import { tool$ratelimitSetOverride } from "./tools/ratelimitSetOverride.js";

export function createMCPServer(deps: {
  logger: ConsoleLogger;
  allowedTools?: string[] | undefined;
  scopes?: MCPScope[] | undefined;
  serverURL?: string | undefined;
  rootKey?: SDKOptions["rootKey"] | undefined;
  serverIdx?: SDKOptions["serverIdx"] | undefined;
}) {
  const server = new McpServer({
    name: "Unkey",
    version: "2.0.0-alpha.7",
  });

  const client = new UnkeyCore({
    rootKey: deps.rootKey,
    serverURL: deps.serverURL,
    serverIdx: deps.serverIdx,
  });

  const scopes = new Set(deps.scopes);

  const allowedTools = deps.allowedTools && new Set(deps.allowedTools);
  const tool = createRegisterTool(
    deps.logger,
    server,
    client,
    scopes,
    allowedTools,
  );
  const resource = createRegisterResource(deps.logger, server, client, scopes);
  const resourceTemplate = createRegisterResourceTemplate(
    deps.logger,
    server,
    client,
    scopes,
  );
  const prompt = createRegisterPrompt(deps.logger, server, client, scopes);
  const register = { tool, resource, resourceTemplate, prompt };
  void register; // suppress unused warnings

  tool(tool$ratelimitLimit);
  tool(tool$ratelimitSetOverride);
  tool(tool$ratelimitGetOverride);
  tool(tool$ratelimitListOverrides);
  tool(tool$ratelimitDeleteOverride);
  tool(tool$identitiesCreateIdentity);
  tool(tool$apisCreateApi);
  tool(tool$livenessLiveness);

  return server;
}