dub 0.64.6 → 0.65.0

package/dist/esm/mcp-server/tools.js

@@ -1,78 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-import { isAsyncIterable, isBinaryData, valueToBase64 } from "./shared.js";
-// Optional function to assist with formatting tool results
-export async function formatResult(value, init) {
-    if (typeof value === "undefined") {
-        return { content: [] };
-    }
-    const { response } = init;
-    const contentType = response?.headers.get("content-type") ?? "";
-    let content = [];
-    if (contentType.search(/\bjson\b/g)) {
-        content = [{ type: "text", text: JSON.stringify(value) }];
-    }
-    else if (contentType.startsWith("text/event-stream")
-        && isAsyncIterable(value)) {
-        content = await consumeSSE(value);
-    }
-    else if (contentType.startsWith("text/") && typeof value === "string") {
-        content = [{ type: "text", text: value }];
-    }
-    else if (isBinaryData(value) && contentType.startsWith("image/")) {
-        const data = await valueToBase64(value);
-        content = data == null
-            ? []
-            : [{ type: "image", data, mimeType: contentType }];
-    }
-    else {
-        return {
-            content: [{
-                    type: "text",
-                    text: `Unsupported content type: "${contentType}"`,
-                }],
-            isError: true,
-        };
-    }
-    return { content };
-}
-async function consumeSSE(value) {
-    const content = [];
-    for await (const chunk of value) {
-        if (typeof chunk === "string") {
-            content.push({ type: "text", text: chunk });
-        }
-        else {
-            content.push({ type: "text", text: JSON.stringify(chunk) });
-        }
-    }
-    return content;
-}
-export function createRegisterTool(logger, server, sdk, allowedScopes, allowedTools) {
-    return (tool) => {
-        if (allowedTools && !allowedTools.has(tool.name)) {
-            return;
-        }
-        const scopes = tool.scopes ?? [];
-        if (allowedScopes.size > 0 && scopes.length === 0) {
-            return;
-        }
-        if (allowedScopes.size > 0
-            && !scopes.every((s) => allowedScopes.has(s))) {
-            return;
-        }
-        if (tool.args) {
-            server.tool(tool.name, tool.description, tool.args, async (args, ctx) => {
-                return tool.tool(sdk, args, ctx);
-            });
-        }
-        else {
-            server.tool(tool.name, tool.description, async (ctx) => {
-                return tool.tool(sdk, ctx);
-            });
-        }
-        logger.debug("Registered tool", { name: tool.name });
-    };
-}
-//# sourceMappingURL=tools.js.map

package/dist/esm/mcp-server/tools.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"tools.js","sourceRoot":"","sources":["../../../src/mcp-server/tools.ts"],"names":[],"mappings":"AAAA;;GAEG;AASH,OAAO,EAAE,eAAe,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAyB3E,2DAA2D;AAC3D,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,KAAc,EACd,IAAyC;IAEzC,IAAI,OAAO,KAAK,KAAK,WAAW,EAAE,CAAC;QACjC,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;IACzB,CAAC;IAED,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC;IAC1B,MAAM,WAAW,GAAG,QAAQ,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;IAChE,IAAI,OAAO,GAA8B,EAAE,CAAC;IAE5C,IAAI,WAAW,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC;QACpC,OAAO,GAAG,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAC5D,CAAC;SAAM,IACL,WAAW,CAAC,UAAU,CAAC,mBAAmB,CAAC;WACxC,eAAe,CAAC,KAAK,CAAC,EACzB,CAAC;QACD,OAAO,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,CAAC;IACpC,CAAC;SAAM,IAAI,WAAW,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACxE,OAAO,GAAG,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IAC5C,CAAC;SAAM,IAAI,YAAY,CAAC,KAAK,CAAC,IAAI,WAAW,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QACnE,MAAM,IAAI,GAAG,MAAM,aAAa,CAAC,KAAK,CAAC,CAAC;QACxC,OAAO,GAAG,IAAI,IAAI,IAAI;YACpB,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC;IACvD,CAAC;SAAM,CAAC;QACN,OAAO;YACL,OAAO,EAAE,CAAC;oBACR,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,8BAA8B,WAAW,GAAG;iBACnD,CAAC;YACF,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;IAED,OAAO,EAAE,OAAO,EAAE,CAAC;AACrB,CAAC;AAED,KAAK,UAAU,UAAU,CACvB,KAA6B;IAE7B,MAAM,OAAO,GAA8B,EAAE,CAAC;IAE9C,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,KAAK,EAAE,CAAC;QAChC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QAC9C,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,MAAqB,EACrB,MAAiB,EACjB,GAAY,EACZ,aAA4B,EAC5B,YAA0B;IAE1B,OAAO,CAAoC,IAAuB,EAAQ,EAAE;QAC1E,IAAI,YAAY,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YACjD,OAAO;QACT,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC;QACjC,IAAI,aAAa,CAAC,IAAI,GAAG,CAAC,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAClD,OAAO;QACT,CAAC;QAED,IACE,aAAa,CAAC,IAAI,GAAG,CAAC;eACnB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAW,EAAE,EAAE,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EACvD,CAAC;YACD,OAAO;QACT,CAAC;QAED,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YACd,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;gBACtE,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,EAAE,GAAG,CAAC,CAAC;YACnC,CAAC,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,WAAW,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;gBACrD,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YAC7B,CAAC,CAAC,CAAC;QACL,CAAC;QAED,MAAM,CAAC,KAAK,CAAC,iBAAiB,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IACvD,CAAC,CAAC;AACJ,CAAC"}

package/src/mcp-server/cli/start/command.ts

@@ -1,94 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { buildCommand } from "@stricli/core";
-import { numberParser } from "@stricli/core";
-import * as z from "zod";
-import { consoleLoggerLevels } from "../../console-logger.js";
-
-export const startCommand = buildCommand({
-  loader: async () => {
-    const { main } = await import("./impl.js");
-    return main;
-  },
-  parameters: {
-    flags: {
-      transport: {
-        kind: "enum",
-        brief: "The transport to use for communicating with the server",
-        default: "stdio",
-        values: ["stdio", "sse"],
-      },
-      port: {
-        kind: "parsed",
-        brief: "The port to use when the SSE transport is enabled",
-        default: "2718",
-        parse: (val: string) =>
-          z.coerce.number().int().gte(0).lt(65536).parse(val),
-      },
-      tool: {
-        kind: "parsed",
-        brief: "Specify tools to mount on the server",
-        optional: true,
-        variadic: true,
-        parse: (value) => {
-          return z.string().parse(value);
-        },
-      },
-      "api-token": {
-        kind: "parsed",
-        brief: "Sets the token auth field for the API",
-        optional: true,
-        parse: (value) => {
-          return z.string().parse(value);
-        },
-      },
-      "server-url": {
-        kind: "parsed",
-        brief: "Overrides the default server URL used by the SDK",
-        optional: true,
-        parse: (value) => new URL(value).toString(),
-      },
-      "server-index": {
-        kind: "parsed",
-        brief: "Selects a predefined server used by the SDK",
-        optional: true,
-        parse: numberParser,
-      },
-      "log-level": {
-        kind: "enum",
-        brief: "The log level to use for the server",
-        default: "info",
-        values: consoleLoggerLevels,
-      },
-      env: {
-        kind: "parsed",
-        brief: "Environment variables made available to the server",
-        optional: true,
-        variadic: true,
-        parse: (val: string) => {
-          const sepIdx = val.indexOf("=");
-          if (sepIdx === -1) {
-            throw new Error("Invalid environment variable format");
-          }
-
-          const key = val.slice(0, sepIdx);
-          const value = val.slice(sepIdx + 1);
-
-          return [
-            z.string().nonempty({
-              message: "Environment variable key must be a non-empty string",
-            }).parse(key),
-            z.string().nonempty({
-              message: "Environment variable value must be a non-empty string",
-            }).parse(value),
-          ] satisfies [string, string];
-        },
-      },
-    },
-  },
-  docs: {
-    brief: "Run the Model Context Protocol server",
-  },
-});

package/src/mcp-server/cli/start/impl.ts

@@ -1,130 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
-import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
-import express from "express";
-import { SDKOptions } from "../../../lib/config.js";
-import { LocalContext } from "../../cli.js";
-import {
-  ConsoleLoggerLevel,
-  createConsoleLogger,
-} from "../../console-logger.js";
-import { createMCPServer } from "../../server.js";
-
-interface StartCommandFlags {
-  readonly transport: "stdio" | "sse";
-  readonly port: number;
-  readonly tool?: string[];
-  readonly "api-token"?: string | undefined;
-  readonly "server-url"?: string;
-  readonly "server-index"?: SDKOptions["serverIdx"];
-  readonly "log-level": ConsoleLoggerLevel;
-  readonly env?: [string, string][];
-}
-
-export async function main(this: LocalContext, flags: StartCommandFlags) {
-  flags.env?.forEach(([key, value]) => {
-    process.env[key] = value;
-  });
-
-  switch (flags.transport) {
-    case "stdio":
-      await startStdio(flags);
-      break;
-    case "sse":
-      await startSSE(flags);
-      break;
-    default:
-      throw new Error(`Invalid transport: ${flags.transport}`);
-  }
-}
-
-async function startStdio(flags: StartCommandFlags) {
-  const logger = createConsoleLogger(flags["log-level"]);
-  const transport = new StdioServerTransport();
-  const server = createMCPServer({
-    logger,
-    allowedTools: flags.tool,
-    ...{ token: flags["api-token"] ?? "" },
-    serverURL: flags["server-url"],
-    serverIdx: flags["server-index"],
-  });
-  await server.connect(transport);
-
-  const abort = async () => {
-    await server.close();
-    process.exit(0);
-  };
-  process.on("SIGTERM", abort);
-  process.on("SIGINT", abort);
-}
-
-async function startSSE(flags: StartCommandFlags) {
-  const logger = createConsoleLogger(flags["log-level"]);
-  const app = express();
-  const mcpServer = createMCPServer({
-    logger,
-    allowedTools: flags.tool,
-    ...{ token: flags["api-token"] ?? "" },
-    serverURL: flags["server-url"],
-    serverIdx: flags["server-index"],
-  });
-  let transport: SSEServerTransport | undefined;
-  const controller = new AbortController();
-
-  app.get("/sse", async (_req, res) => {
-    transport = new SSEServerTransport("/message", res);
-
-    await mcpServer.connect(transport);
-
-    mcpServer.server.onclose = async () => {
-      res.end();
-    };
-  });
-
-  app.post("/message", async (req, res) => {
-    if (!transport) {
-      throw new Error("Server transport not initialized");
-    }
-
-    await transport.handlePostMessage(req, res);
-  });
-
-  const httpServer = app.listen(flags.port, "0.0.0.0", () => {
-    const ha = httpServer.address();
-    const host = typeof ha === "string" ? ha : `${ha?.address}:${ha?.port}`;
-    logger.info("MCP HTTP server started", { host });
-  });
-
-  let closing = false;
-  controller.signal.addEventListener("abort", async () => {
-    if (closing) {
-      logger.info("Received second signal. Forcing shutdown.");
-      process.exit(1);
-    }
-    closing = true;
-
-    logger.info("Shutting down MCP server");
-
-    await mcpServer.close();
-
-    logger.info("Shutting down HTTP server");
-
-    const timer = setTimeout(() => {
-      logger.info("Forcing shutdown");
-      process.exit(1);
-    }, 5000);
-
-    httpServer.close(() => {
-      clearTimeout(timer);
-      logger.info("Graceful shutdown complete");
-      process.exit(0);
-    });
-  });
-
-  const abort = () => controller.abort();
-  process.on("SIGTERM", abort);
-  process.on("SIGINT", abort);
-}

package/src/mcp-server/cli.ts

@@ -1,13 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { CommandContext, StricliProcess } from "@stricli/core";
-
-export interface LocalContext extends CommandContext {
-  readonly process: StricliProcess;
-}
-
-export function buildContext(process: NodeJS.Process): LocalContext {
-  return { process: process as StricliProcess };
-}

package/src/mcp-server/console-logger.ts

@@ -1,71 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-export const consoleLoggerLevels = [
-  "debug",
-  "warning",
-  "info",
-  "error",
-] as const;
-
-export type ConsoleLoggerLevel = (typeof consoleLoggerLevels)[number];
-
-export type ConsoleLogger = {
-  [key in ConsoleLoggerLevel]: (
-    message: string,
-    data?: Record<string, unknown>,
-  ) => void;
-};
-
-export function createConsoleLogger(level: ConsoleLoggerLevel): ConsoleLogger {
-  const min = consoleLoggerLevels.indexOf(level);
-  const noop = () => {};
-
-  const logger: ConsoleLogger = {
-    debug: noop,
-    warning: noop,
-    info: noop,
-    error: noop,
-  };
-
-  return consoleLoggerLevels.reduce((logger, level, i) => {
-    if (i < min) {
-      return logger;
-    }
-
-    logger[level] = log.bind(null, level);
-
-    return logger;
-  }, logger);
-}
-
-function log(
-  level: ConsoleLoggerLevel,
-  message: string,
-  data?: Record<string, unknown>,
-) {
-  let line = "";
-  const allData = [{ msg: message, l: level }, data];
-
-  for (const ctx of allData) {
-    for (const [key, value] of Object.entries(ctx || {})) {
-      if (value == null) {
-        line += ` ${key}=<${value}>`;
-      } else if (typeof value === "function") {
-        line += ` ${key}=<function>`;
-      } else if (typeof value === "symbol") {
-        line += ` ${key}=${value.toString()}`;
-      } else if (typeof value === "string") {
-        const v = value.search(/\s/g) >= 0 ? JSON.stringify(value) : value;
-        line += ` ${key}=${v}`;
-      } else if (typeof value !== "object") {
-        line += ` ${key}=${value}`;
-      } else {
-        line += ` ${key}="${JSON.stringify(value)}"`;
-      }
-    }
-  }
-
-  console.error(line);
-}

package/src/mcp-server/extensions.ts

@@ -1,17 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { ZodRawShape } from "zod";
-import { PromptArgsRawShape, PromptDefinition } from "./prompts.js";
-import { ResourceDefinition, ResourceTemplateDefinition } from "./resources.js";
-import { ToolDefinition } from "./tools.js";
-
-export type Register = {
-  tool: <A extends ZodRawShape | undefined>(def: ToolDefinition<A>) => void;
-  resource: (def: ResourceDefinition) => void;
-  resourceTemplate: (def: ResourceTemplateDefinition) => void;
-  prompt: <A extends PromptArgsRawShape | undefined>(
-    prompt: PromptDefinition<A>,
-  ) => void;
-};

package/src/mcp-server/mcp-server.ts

@@ -1,26 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { buildApplication, buildRouteMap, run } from "@stricli/core";
-import process from "node:process";
-import { buildContext } from "./cli.js";
-import { startCommand } from "./cli/start/command.js";
-
-const routes = buildRouteMap({
-  routes: {
-    start: startCommand,
-  },
-  docs: {
-    brief: "MCP server CLI",
-  },
-});
-
-export const app = buildApplication(routes, {
-  name: "mcp",
-  versionInfo: {
-    currentVersion: "0.64.6",
-  },
-});
-
-run(app, process.argv.slice(2), buildContext(process));

package/src/mcp-server/prompts.ts

@@ -1,117 +0,0 @@
-/*
- * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
- */
-
-import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-import { RequestHandlerExtra } from "@modelcontextprotocol/sdk/shared/protocol.js";
-import { GetPromptResult } from "@modelcontextprotocol/sdk/types.js";
-import {
-  objectOutputType,
-  ZodOptional,
-  ZodType,
-  ZodTypeAny,
-  ZodTypeDef,
-} from "zod";
-import { DubCore } from "../core.js";
-import { ConsoleLogger } from "./console-logger.js";
-import { MCPScope } from "./scopes.js";
-
-// '@modelcontextprotocol/sdk' currently does not export this type
-export type PromptArgsRawShape = {
-  [k: string]:
-    | ZodType<string, ZodTypeDef, string>
-    | ZodOptional<ZodType<string, ZodTypeDef, string>>;
-};
-
-export type PromptDefinition<
-  Args extends undefined | PromptArgsRawShape = undefined,
-> = Args extends PromptArgsRawShape ? {
-    name: string;
-    description?: string;
-    scopes?: MCPScope[];
-    args: Args;
-    prompt: (
-      client: DubCore,
-      args: objectOutputType<Args, ZodTypeAny>,
-      extra: RequestHandlerExtra,
-    ) => GetPromptResult | Promise<GetPromptResult>;
-  }
-  : {
-    name: string;
-    description?: string;
-    scopes?: MCPScope[];
-    args?: undefined;
-    prompt: (
-      client: DubCore,
-      extra: RequestHandlerExtra,
-    ) => GetPromptResult | Promise<GetPromptResult>;
-  };
-
-// Optional function to assist with formatting prompt results
-export async function formatResult(value: string): Promise<GetPromptResult> {
-  return {
-    messages: [
-      {
-        role: "user",
-        content: {
-          type: "text",
-          text: value,
-        },
-      },
-    ],
-  };
-}
-
-export function createRegisterPrompt(
-  logger: ConsoleLogger,
-  server: McpServer,
-  sdk: DubCore,
-  allowedScopes: Set<MCPScope>,
-): <A extends PromptArgsRawShape | undefined>(
-  prompt: PromptDefinition<A>,
-) => void {
-  return <A extends PromptArgsRawShape | undefined>(
-    prompt: PromptDefinition<A>,
-  ): void => {
-    const scopes = prompt.scopes ?? [];
-    if (allowedScopes.size > 0 && scopes.length === 0) {
-      return;
-    }
-
-    if (
-      allowedScopes.size > 0
-      && !scopes.every((s: MCPScope) => allowedScopes.has(s))
-    ) {
-      return;
-    }
-
-    if (prompt.args) {
-      if (prompt.description) {
-        server.prompt(
-          prompt.name,
-          prompt.description,
-          prompt.args,
-          async (args, ctx) => prompt.prompt(sdk, args, ctx),
-        );
-      } else {
-        server.prompt(
-          prompt.name,
-          prompt.args,
-          async (args, ctx) => prompt.prompt(sdk, args, ctx),
-        );
-      }
-    } else {
-      if (prompt.description) {
-        server.prompt(
-          prompt.name,
-          prompt.description,
-          async (ctx) => prompt.prompt(sdk, ctx),
-        );
-      } else {
-        server.prompt(prompt.name, async (ctx) => prompt.prompt(sdk, ctx));
-      }
-    }
-
-    logger.debug("Registered prompt", { name: prompt.name });
-  };
-}