trigger.dev 4.0.0-v4-beta.28 → 4.0.1

package/dist/esm/mcp/tools/tasks.d.ts

@@ -0,0 +1,200 @@
+import { ToolMeta } from "../types.js";
+export declare const getCurrentWorker: {
+    name: string;
+    title: string;
+    description: string;
+    inputSchema: {
+        projectRef: import("zod").ZodOptional<import("zod").ZodString>;
+        configPath: import("zod").ZodOptional<import("zod").ZodString>;
+        environment: import("zod").ZodDefault<import("zod").ZodEnum<["dev", "staging", "prod", "preview"]>>;
+        branch: import("zod").ZodOptional<import("zod").ZodString>;
+    };
+    handler: (input: unknown, extra: ToolMeta) => Promise<{
+        [x: string]: unknown;
+        content: ({
+            [x: string]: unknown;
+            type: "text";
+            text: string;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+        } | {
+            [x: string]: unknown;
+            type: "image";
+            data: string;
+            mimeType: string;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+        } | {
+            [x: string]: unknown;
+            type: "audio";
+            data: string;
+            mimeType: string;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+        } | {
+            [x: string]: unknown;
+            type: "resource_link";
+            name: string;
+            uri: string;
+            description?: string | undefined;
+            title?: string | undefined;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+            mimeType?: string | undefined;
+        } | {
+            [x: string]: unknown;
+            type: "resource";
+            resource: {
+                [x: string]: unknown;
+                text: string;
+                uri: string;
+                _meta?: {
+                    [x: string]: unknown;
+                } | undefined;
+                mimeType?: string | undefined;
+            } | {
+                [x: string]: unknown;
+                uri: string;
+                blob: string;
+                _meta?: {
+                    [x: string]: unknown;
+                } | undefined;
+                mimeType?: string | undefined;
+            };
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+        })[];
+        isError?: boolean | undefined;
+        _meta?: {
+            [x: string]: unknown;
+        } | undefined;
+        structuredContent?: {
+            [x: string]: unknown;
+        } | undefined;
+    }>;
+};
+export declare const triggerTaskTool: {
+    name: string;
+    title: string;
+    description: string;
+    inputSchema: {
+        projectRef: import("zod").ZodOptional<import("zod").ZodString>;
+        configPath: import("zod").ZodOptional<import("zod").ZodString>;
+        environment: import("zod").ZodDefault<import("zod").ZodEnum<["dev", "staging", "prod", "preview"]>>;
+        branch: import("zod").ZodOptional<import("zod").ZodString>;
+    } & {
+        taskId: import("zod").ZodString;
+        payload: import("zod").ZodAny;
+        options: import("zod").ZodOptional<import("zod").ZodObject<{
+            queue: import("zod").ZodOptional<import("zod").ZodObject<{
+                name: import("zod").ZodString;
+            }, "strip", import("zod").ZodTypeAny, {
+                name: string;
+            }, {
+                name: string;
+            }>>;
+            delay: import("zod").ZodOptional<import("zod").ZodUnion<[import("zod").ZodString, import("zod").ZodDate]>>;
+            idempotencyKey: import("zod").ZodOptional<import("zod").ZodString>;
+            machine: import("zod").ZodOptional<import("zod").ZodEnum<["micro", "small-1x", "small-2x", "medium-1x", "medium-2x", "large-1x", "large-2x"]>>;
+            maxAttempts: import("zod").ZodOptional<import("zod").ZodNumber>;
+            maxDuration: import("zod").ZodOptional<import("zod").ZodNumber>;
+            tags: import("zod").ZodOptional<import("zod").ZodArray<import("zod").ZodString, "many">>;
+            ttl: import("zod").ZodDefault<import("zod").ZodUnion<[import("zod").ZodString, import("zod").ZodNumber]>>;
+        }, "strip", import("zod").ZodTypeAny, {
+            ttl: string | number;
+            tags?: string[] | undefined;
+            idempotencyKey?: string | undefined;
+            maxAttempts?: number | undefined;
+            maxDuration?: number | undefined;
+            queue?: {
+                name: string;
+            } | undefined;
+            machine?: "micro" | "small-1x" | "small-2x" | "medium-1x" | "medium-2x" | "large-1x" | "large-2x" | undefined;
+            delay?: string | Date | undefined;
+        }, {
+            tags?: string[] | undefined;
+            idempotencyKey?: string | undefined;
+            maxAttempts?: number | undefined;
+            maxDuration?: number | undefined;
+            queue?: {
+                name: string;
+            } | undefined;
+            machine?: "micro" | "small-1x" | "small-2x" | "medium-1x" | "medium-2x" | "large-1x" | "large-2x" | undefined;
+            delay?: string | Date | undefined;
+            ttl?: string | number | undefined;
+        }>>;
+    };
+    handler: (input: unknown, extra: ToolMeta) => Promise<{
+        [x: string]: unknown;
+        content: ({
+            [x: string]: unknown;
+            type: "text";
+            text: string;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+        } | {
+            [x: string]: unknown;
+            type: "image";
+            data: string;
+            mimeType: string;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+        } | {
+            [x: string]: unknown;
+            type: "audio";
+            data: string;
+            mimeType: string;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+        } | {
+            [x: string]: unknown;
+            type: "resource_link";
+            name: string;
+            uri: string;
+            description?: string | undefined;
+            title?: string | undefined;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+            mimeType?: string | undefined;
+        } | {
+            [x: string]: unknown;
+            type: "resource";
+            resource: {
+                [x: string]: unknown;
+                text: string;
+                uri: string;
+                _meta?: {
+                    [x: string]: unknown;
+                } | undefined;
+                mimeType?: string | undefined;
+            } | {
+                [x: string]: unknown;
+                uri: string;
+                blob: string;
+                _meta?: {
+                    [x: string]: unknown;
+                } | undefined;
+                mimeType?: string | undefined;
+            };
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+        })[];
+        isError?: boolean | undefined;
+        _meta?: {
+            [x: string]: unknown;
+        } | undefined;
+        structuredContent?: {
+            [x: string]: unknown;
+        } | undefined;
+    }>;
+};

package/dist/esm/mcp/tools/tasks.js

@@ -0,0 +1,117 @@
+import { toolsMetadata } from "../config.js";
+import { CommonProjectsInput, TriggerTaskInput } from "../schemas.js";
+import { respondWithError, toolHandler } from "../utils.js";
+export const getCurrentWorker = {
+    name: toolsMetadata.get_current_worker.name,
+    title: toolsMetadata.get_current_worker.title,
+    description: toolsMetadata.get_current_worker.description,
+    inputSchema: CommonProjectsInput.shape,
+    handler: toolHandler(CommonProjectsInput.shape, async (input, { ctx }) => {
+        ctx.logger?.log("calling get_current_worker", { input });
+        if (ctx.options.devOnly && input.environment !== "dev") {
+            return respondWithError(`This MCP server is only available for the dev environment. You tried to access the ${input.environment} environment. Remove the --dev-only flag to access other environments.`);
+        }
+        const projectRef = await ctx.getProjectRef({
+            projectRef: input.projectRef,
+            cwd: input.configPath,
+        });
+        const cliApiClient = await ctx.getCliApiClient(input.branch);
+        const workerResult = await cliApiClient.getWorkerByTag(projectRef, input.environment, "current");
+        if (!workerResult.success) {
+            return respondWithError(workerResult.error);
+        }
+        const { worker, urls } = workerResult.data;
+        const contents = [
+            `Current worker for ${input.environment} is ${worker.version} using ${worker.sdkVersion} of the SDK.`,
+        ];
+        if (worker.tasks.length > 0) {
+            contents.push(`The worker has ${worker.tasks.length} tasks registered:`);
+            for (const task of worker.tasks) {
+                if (task.payloadSchema) {
+                    contents.push(`- ${task.slug} in ${task.filePath} (payload schema: ${JSON.stringify(task.payloadSchema)})`);
+                }
+                else {
+                    contents.push(`- ${task.slug} in ${task.filePath}`);
+                }
+            }
+        }
+        else {
+            contents.push(`The worker has no tasks registered.`);
+        }
+        contents.push(`\n`);
+        contents.push(`URLs:`);
+        contents.push(`- Runs: ${urls.runs}`);
+        contents.push(`\n`);
+        contents.push(`You can use the list_runs tool with the version ${worker.version} to get the list of runs for this worker.`);
+        if (typeof worker.sdkVersion === "string" &&
+            typeof worker.cliVersion === "string" &&
+            worker.sdkVersion !== worker.cliVersion) {
+            contents.push(`WARNING: The SDK version (${worker.sdkVersion}) is different from the CLI version (${worker.cliVersion}). This might cause issues with the task execution. Make sure to pin the CLI and the SDK versions to ${worker.sdkVersion}.`);
+        }
+        return {
+            content: [{ type: "text", text: contents.join("\n") }],
+        };
+    }),
+};
+export const triggerTaskTool = {
+    name: toolsMetadata.trigger_task.name,
+    title: toolsMetadata.trigger_task.title,
+    description: toolsMetadata.trigger_task.description,
+    inputSchema: TriggerTaskInput.shape,
+    handler: toolHandler(TriggerTaskInput.shape, async (input, { ctx }) => {
+        ctx.logger?.log("calling trigger_task", { input });
+        if (ctx.options.devOnly && input.environment !== "dev") {
+            return respondWithError(`This MCP server is only available for the dev environment. You tried to access the ${input.environment} environment. Remove the --dev-only flag to access other environments.`);
+        }
+        const projectRef = await ctx.getProjectRef({
+            projectRef: input.projectRef,
+            cwd: input.configPath,
+        });
+        const apiClient = await ctx.getApiClient({
+            projectRef,
+            environment: input.environment,
+            scopes: ["write:tasks"],
+            branch: input.branch,
+        });
+        ctx.logger?.log("triggering task", { input });
+        let payload = input.payload;
+        if (typeof payload === "string") {
+            try {
+                payload = JSON.parse(payload);
+            }
+            catch {
+                ctx.logger?.log("payload is not a valid JSON string, using as is", { payload });
+            }
+        }
+        const result = await apiClient.triggerTask(input.taskId, {
+            payload,
+            options: input.options,
+        });
+        const taskRunUrl = await ctx.getDashboardUrl(`/projects/v3/${projectRef}/runs/${result.id}`);
+        const contents = [
+            `Task ${input.taskId} triggered and run with ID created: ${result.id}.`,
+            `View the run in the dashboard: ${taskRunUrl}`,
+            `You can also use the get_run_details tool to get the details of the run.`,
+        ];
+        if (input.environment === "dev") {
+            const cliApiClient = await ctx.getCliApiClient(input.branch);
+            const devStatus = await cliApiClient.getDevStatus(projectRef);
+            const isConnected = devStatus.success ? devStatus.data.isConnected : false;
+            const connectionMessage = isConnected
+                ? undefined
+                : "The dev CLI is not connected to this project, because it is not currently running. Make sure to run the dev command to execute triggered tasks.";
+            if (connectionMessage) {
+                contents.push(connectionMessage);
+            }
+        }
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: contents.join("\n"),
+                },
+            ],
+        };
+    }),
+};
+//# sourceMappingURL=tasks.js.map

package/dist/esm/mcp/tools/tasks.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tasks.js","sourceRoot":"","sources":["../../../../src/mcp/tools/tasks.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAC7C,OAAO,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AAEtE,OAAO,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE5D,MAAM,CAAC,MAAM,gBAAgB,GAAG;IAC9B,IAAI,EAAE,aAAa,CAAC,kBAAkB,CAAC,IAAI;IAC3C,KAAK,EAAE,aAAa,CAAC,kBAAkB,CAAC,KAAK;IAC7C,WAAW,EAAE,aAAa,CAAC,kBAAkB,CAAC,WAAW;IACzD,WAAW,EAAE,mBAAmB,CAAC,KAAK;IACtC,OAAO,EAAE,WAAW,CAAC,mBAAmB,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE;QACvE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,4BAA4B,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAEzD,IAAI,GAAG,CAAC,OAAO,CAAC,OAAO,IAAI,KAAK,CAAC,WAAW,KAAK,KAAK,EAAE,CAAC;YACvD,OAAO,gBAAgB,CACrB,sFAAsF,KAAK,CAAC,WAAW,wEAAwE,CAChL,CAAC;QACJ,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC;YACzC,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,GAAG,EAAE,KAAK,CAAC,UAAU;SACtB,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,MAAM,GAAG,CAAC,eAAe,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAE7D,MAAM,YAAY,GAAG,MAAM,YAAY,CAAC,cAAc,CACpD,UAAU,EACV,KAAK,CAAC,WAAW,EACjB,SAAS,CACV,CAAC;QAEF,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;YAC1B,OAAO,gBAAgB,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;QAC9C,CAAC;QAED,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,YAAY,CAAC,IAAI,CAAC;QAE3C,MAAM,QAAQ,GAAG;YACf,sBAAsB,KAAK,CAAC,WAAW,OAAO,MAAM,CAAC,OAAO,UAAU,MAAM,CAAC,UAAU,cAAc;SACtG,CAAC;QAEF,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,QAAQ,CAAC,IAAI,CAAC,kBAAkB,MAAM,CAAC,KAAK,CAAC,MAAM,oBAAoB,CAAC,CAAC;YAEzE,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBAChC,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;oBACvB,QAAQ,CAAC,IAAI,CACX,KAAK,IAAI,CAAC,IAAI,OAAO,IAAI,CAAC,QAAQ,qBAAqB,IAAI,CAAC,SAAS,CACnE,IAAI,CAAC,aAAa,CACnB,GAAG,CACL,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,QAAQ,CAAC,IAAI,CAAC,KAAK,IAAI,CAAC,IAAI,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;gBACtD,CAAC;YACH,CAAC;QACH,CAAC;aAAM,CAAC;YACN,QAAQ,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAC;QACvD,CAAC;QAED,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpB,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACvB,QAAQ,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QACtC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpB,QAAQ,CAAC,IAAI,CACX,mDAAmD,MAAM,CAAC,OAAO,2CAA2C,CAC7G,CAAC;QAEF,IACE,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ;YACrC,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ;YACrC,MAAM,CAAC,UAAU,KAAK,MAAM,CAAC,UAAU,EACvC,CAAC;YACD,QAAQ,CAAC,IAAI,CACX,6BAA6B,MAAM,CAAC,UAAU,wCAAwC,MAAM,CAAC,UAAU,wGAAwG,MAAM,CAAC,UAAU,GAAG,CACpO,CAAC;QACJ,CAAC;QAED,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;SACvD,CAAC;IACJ,CAAC,CAAC;CACH,CAAC;AAEF,MAAM,CAAC,MAAM,eAAe,GAAG;IAC7B,IAAI,EAAE,aAAa,CAAC,YAAY,CAAC,IAAI;IACrC,KAAK,EAAE,aAAa,CAAC,YAAY,CAAC,KAAK;IACvC,WAAW,EAAE,aAAa,CAAC,YAAY,CAAC,WAAW;IACnD,WAAW,EAAE,gBAAgB,CAAC,KAAK;IACnC,OAAO,EAAE,WAAW,CAAC,gBAAgB,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE;QACpE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,sBAAsB,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAEnD,IAAI,GAAG,CAAC,OAAO,CAAC,OAAO,IAAI,KAAK,CAAC,WAAW,KAAK,KAAK,EAAE,CAAC;YACvD,OAAO,gBAAgB,CACrB,sFAAsF,KAAK,CAAC,WAAW,wEAAwE,CAChL,CAAC;QACJ,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC;YACzC,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,GAAG,EAAE,KAAK,CAAC,UAAU;SACtB,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,MAAM,GAAG,CAAC,YAAY,CAAC;YACvC,UAAU;YACV,WAAW,EAAE,KAAK,CAAC,WAAW;YAC9B,MAAM,EAAE,CAAC,aAAa,CAAC;YACvB,MAAM,EAAE,KAAK,CAAC,MAAM;SACrB,CAAC,CAAC;QAEH,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,iBAAiB,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAE9C,IAAI,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;QAE5B,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YAChC,IAAI,CAAC;gBACH,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAChC,CAAC;YAAC,MAAM,CAAC;gBACP,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,iDAAiD,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAClF,CAAC;QACH,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,EAAE;YACvD,OAAO;YACP,OAAO,EAAE,KAAK,CAAC,OAAO;SACvB,CAAC,CAAC;QAEH,MAAM,UAAU,GAAG,MAAM,GAAG,CAAC,eAAe,CAAC,gBAAgB,UAAU,SAAS,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC;QAE7F,MAAM,QAAQ,GAAG;YACf,QAAQ,KAAK,CAAC,MAAM,uCAAuC,MAAM,CAAC,EAAE,GAAG;YACvE,kCAAkC,UAAU,EAAE;YAC9C,0EAA0E;SAC3E,CAAC;QAEF,IAAI,KAAK,CAAC,WAAW,KAAK,KAAK,EAAE,CAAC;YAChC,MAAM,YAAY,GAAG,MAAM,GAAG,CAAC,eAAe,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAC7D,MAAM,SAAS,GAAG,MAAM,YAAY,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;YAC9D,MAAM,WAAW,GAAG,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,KAAK,CAAC;YAC3E,MAAM,iBAAiB,GAAG,WAAW;gBACnC,CAAC,CAAC,SAAS;gBACX,CAAC,CAAC,iJAAiJ,CAAC;YAEtJ,IAAI,iBAAiB,EAAE,CAAC;gBACtB,QAAQ,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACnC,CAAC;QACH,CAAC;QAED,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC;iBAC1B;aACF;SACF,CAAC;IACJ,CAAC,CAAC;CACH,CAAC"}

package/dist/esm/mcp/tools.d.ts

@@ -0,0 +1,2 @@
+import { McpContext } from "./context.js";
+export declare function registerTools(context: McpContext): void;

package/dist/esm/mcp/tools.js

@@ -0,0 +1,39 @@
+import { deployTool, listDeploysTool } from "./tools/deploys.js";
+import { searchDocsTool } from "./tools/docs.js";
+import { createProjectInOrgTool, initializeProjectTool, listOrgsTool, listProjectsTool, } from "./tools/orgs.js";
+import { listPreviewBranchesTool } from "./tools/previewBranches.js";
+import { cancelRunTool, getRunDetailsTool, listRunsTool } from "./tools/runs.js";
+import { getCurrentWorker, triggerTaskTool } from "./tools/tasks.js";
+import { respondWithError } from "./utils.js";
+export function registerTools(context) {
+    const tools = [
+        searchDocsTool,
+        listOrgsTool,
+        listProjectsTool,
+        createProjectInOrgTool,
+        initializeProjectTool,
+        getCurrentWorker,
+        triggerTaskTool,
+        listRunsTool,
+        getRunDetailsTool,
+        cancelRunTool,
+        deployTool,
+        listDeploysTool,
+        listPreviewBranchesTool,
+    ];
+    for (const tool of tools) {
+        context.server.registerTool(tool.name, {
+            annotations: { title: tool.title },
+            description: tool.description,
+            inputSchema: tool.inputSchema,
+        }, async (input, extra) => {
+            try {
+                return tool.handler(input, { ...extra, ctx: context });
+            }
+            catch (error) {
+                return respondWithError(error);
+            }
+        });
+    }
+}
+//# sourceMappingURL=tools.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"tools.js","sourceRoot":"","sources":["../../../src/mcp/tools.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACjE,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,EACL,sBAAsB,EACtB,qBAAqB,EACrB,YAAY,EACZ,gBAAgB,GACjB,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AACrE,OAAO,EAAE,aAAa,EAAE,iBAAiB,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AACjF,OAAO,EAAE,gBAAgB,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AACrE,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAE9C,MAAM,UAAU,aAAa,CAAC,OAAmB;IAC/C,MAAM,KAAK,GAAG;QACZ,cAAc;QACd,YAAY;QACZ,gBAAgB;QAChB,sBAAsB;QACtB,qBAAqB;QACrB,gBAAgB;QAChB,eAAe;QACf,YAAY;QACZ,iBAAiB;QACjB,aAAa;QACb,UAAU;QACV,eAAe;QACf,uBAAuB;KACxB,CAAC;IAEF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,OAAO,CAAC,MAAM,CAAC,YAAY,CACzB,IAAI,CAAC,IAAI,EACT;YACE,WAAW,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE;YAClC,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,WAAW,EAAE,IAAI,CAAC,WAAW;SAC9B,EACD,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;YACrB,IAAI,CAAC;gBACH,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,GAAG,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,CAAC;YACzD,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,gBAAgB,CAAC,KAAK,CAAC,CAAC;YACjC,CAAC;QACH,CAAC,CACF,CAAC;IACJ,CAAC;AACH,CAAC"}

package/dist/esm/mcp/types.d.ts

@@ -0,0 +1,6 @@
+import { RequestHandlerExtra } from "@modelcontextprotocol/sdk/shared/protocol.js";
+import { ServerNotification, ServerRequest } from "@modelcontextprotocol/sdk/types.js";
+import { McpContext } from "./context.js";
+export type ToolMeta = RequestHandlerExtra<ServerRequest, ServerNotification> & {
+    ctx: McpContext;
+};

package/dist/esm/mcp/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/esm/mcp/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/mcp/types.ts"],"names":[],"mappings":""}

package/dist/esm/mcp/utils.d.ts

@@ -0,0 +1,89 @@
+import type { CallToolResult, ServerNotification } from "@modelcontextprotocol/sdk/types.js";
+import { z } from "zod";
+import { ToolMeta } from "./types.js";
+export declare function respondWithError(error: unknown): CallToolResult;
+export type ToolHandlerMeta = ToolMeta & {
+    createProgressTracker: (total: number) => ProgressTracker;
+};
+export declare function toolHandler<TInputShape extends z.ZodRawShape>(shape: TInputShape, handler: (input: z.output<z.ZodObject<TInputShape>>, meta: ToolHandlerMeta) => Promise<CallToolResult>): (input: unknown, extra: ToolMeta) => Promise<{
+    [x: string]: unknown;
+    content: ({
+        [x: string]: unknown;
+        type: "text";
+        text: string;
+        _meta?: {
+            [x: string]: unknown;
+        } | undefined;
+    } | {
+        [x: string]: unknown;
+        type: "image";
+        data: string;
+        mimeType: string;
+        _meta?: {
+            [x: string]: unknown;
+        } | undefined;
+    } | {
+        [x: string]: unknown;
+        type: "audio";
+        data: string;
+        mimeType: string;
+        _meta?: {
+            [x: string]: unknown;
+        } | undefined;
+    } | {
+        [x: string]: unknown;
+        type: "resource_link";
+        name: string;
+        uri: string;
+        description?: string | undefined;
+        title?: string | undefined;
+        _meta?: {
+            [x: string]: unknown;
+        } | undefined;
+        mimeType?: string | undefined;
+    } | {
+        [x: string]: unknown;
+        type: "resource";
+        resource: {
+            [x: string]: unknown;
+            text: string;
+            uri: string;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+            mimeType?: string | undefined;
+        } | {
+            [x: string]: unknown;
+            uri: string;
+            blob: string;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+            mimeType?: string | undefined;
+        };
+        _meta?: {
+            [x: string]: unknown;
+        } | undefined;
+    })[];
+    isError?: boolean | undefined;
+    _meta?: {
+        [x: string]: unknown;
+    } | undefined;
+    structuredContent?: {
+        [x: string]: unknown;
+    } | undefined;
+}>;
+declare class ProgressTracker {
+    #private;
+    private progress;
+    private progressToken;
+    private total;
+    private message;
+    private sendNotification;
+    constructor(total: number, sendNotification: (notification: ServerNotification) => Promise<void>, progressToken?: string | number);
+    updateProgress(progress: number, message?: string): Promise<void>;
+    incrementProgress(increment: number, message?: string): Promise<void>;
+    complete(message?: string): Promise<void>;
+    getProgress(): number;
+}
+export {};

package/dist/esm/mcp/utils.js

@@ -0,0 +1,95 @@
+import { z } from "zod";
+export function respondWithError(error) {
+    return {
+        isError: true,
+        content: [
+            {
+                type: "text",
+                text: JSON.stringify({ error: enumerateError(error) }),
+            },
+        ],
+    };
+}
+function enumerateError(error) {
+    if (!error) {
+        return error;
+    }
+    if (typeof error !== "object") {
+        return error;
+    }
+    const newError = {};
+    const errorProps = ["name", "message"];
+    for (const prop of errorProps) {
+        if (prop in error) {
+            newError[prop] = error[prop];
+        }
+    }
+    return newError;
+}
+export function toolHandler(shape, handler) {
+    return async (input, extra) => {
+        const parsedInput = z.object(shape).safeParse(input);
+        if (!parsedInput.success) {
+            return respondWithError(parsedInput.error);
+        }
+        function createProgressTracker(total) {
+            return new ProgressTracker(total, extra.sendNotification, extra._meta?.progressToken);
+        }
+        return handler(parsedInput.data, { ...extra, createProgressTracker });
+    };
+}
+class ProgressTracker {
+    progress = 0;
+    progressToken;
+    total;
+    message;
+    sendNotification;
+    constructor(total, sendNotification, progressToken) {
+        this.message = "";
+        this.progressToken = progressToken;
+        this.progress = 0;
+        this.total = total;
+        this.sendNotification = sendNotification;
+    }
+    async updateProgress(progress, message) {
+        this.progress = progress;
+        if (message) {
+            this.message = message;
+        }
+        await this.#sendNotification(progress, this.message);
+    }
+    async incrementProgress(increment, message) {
+        this.progress += increment;
+        // make sure the progress is never greater than the total
+        this.progress = Math.min(this.progress, this.total);
+        if (message) {
+            this.message = message;
+        }
+        await this.#sendNotification(this.progress, this.message);
+    }
+    async complete(message) {
+        this.progress = this.total;
+        if (message) {
+            this.message = message;
+        }
+        await this.#sendNotification(this.progress, this.message);
+    }
+    getProgress() {
+        return this.progress;
+    }
+    async #sendNotification(progress, message) {
+        if (!this.progressToken) {
+            return;
+        }
+        await this.sendNotification({
+            method: "notifications/progress",
+            params: {
+                progress,
+                total: this.total,
+                message: this.message,
+                progressToken: this.progressToken,
+            },
+        });
+    }
+}
+//# sourceMappingURL=utils.js.map

package/dist/esm/mcp/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/mcp/utils.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,MAAM,UAAU,gBAAgB,CAAC,KAAc;IAC7C,OAAO;QACL,OAAO,EAAE,IAAI;QACb,OAAO,EAAE;YACP;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,cAAc,CAAC,KAAK,CAAC,EAAE,CAAC;aACvD;SACF;KACF,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CAAC,KAAc;IACpC,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,QAAQ,GAA4B,EAAE,CAAC;IAE7C,MAAM,UAAU,GAAG,CAAC,MAAM,EAAE,SAAS,CAAU,CAAC;IAEhD,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;QAC9B,IAAI,IAAI,IAAI,KAAK,EAAE,CAAC;YAClB,QAAQ,CAAC,IAAI,CAAC,GAAI,KAAiC,CAAC,IAAI,CAAC,CAAC;QAC5D,CAAC;IACH,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAMD,MAAM,UAAU,WAAW,CACzB,KAAkB,EAClB,OAG4B;IAE5B,OAAO,KAAK,EAAE,KAAc,EAAE,KAAe,EAAE,EAAE;QAC/C,MAAM,WAAW,GAAG,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAErD,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;YACzB,OAAO,gBAAgB,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAC7C,CAAC;QAED,SAAS,qBAAqB,CAAC,KAAa;YAC1C,OAAO,IAAI,eAAe,CAAC,KAAK,EAAE,KAAK,CAAC,gBAAgB,EAAE,KAAK,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;QACxF,CAAC;QAED,OAAO,OAAO,CAAC,WAAW,CAAC,IAAI,EAAE,EAAE,GAAG,KAAK,EAAE,qBAAqB,EAAE,CAAC,CAAC;IACxE,CAAC,CAAC;AACJ,CAAC;AAED,MAAM,eAAe;IACX,QAAQ,GAAW,CAAC,CAAC;IACrB,aAAa,CAA8B;IAC3C,KAAK,CAAS;IACd,OAAO,CAAS;IAChB,gBAAgB,CAAsD;IAE9E,YACE,KAAa,EACb,gBAAqE,EACrE,aAA+B;QAE/B,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;QAClB,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,QAAQ,GAAG,CAAC,CAAC;QAClB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,QAAgB,EAAE,OAAgB;QACrD,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QAEzB,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACzB,CAAC;QAED,MAAM,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IACvD,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,SAAiB,EAAE,OAAgB;QACzD,IAAI,CAAC,QAAQ,IAAI,SAAS,CAAC;QAE3B,yDAAyD;QACzD,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QAEpD,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACzB,CAAC;QAED,MAAM,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,OAAgB;QAC7B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC;QAC3B,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACzB,CAAC;QACD,MAAM,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IAC5D,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,QAAgB,EAAE,OAAe;QACvD,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;YACxB,OAAO;QACT,CAAC;QAED,MAAM,IAAI,CAAC,gBAAgB,CAAC;YAC1B,MAAM,EAAE,wBAAwB;YAChC,MAAM,EAAE;gBACN,QAAQ;gBACR,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,aAAa,EAAE,IAAI,CAAC,aAAa;aAClC;SACF,CAAC,CAAC;IACL,CAAC;CACF"}

package/dist/esm/rules/install.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/esm/rules/install.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=install.js.map

package/dist/esm/rules/install.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"install.js","sourceRoot":"","sources":["../../../src/rules/install.ts"],"names":[],"mappings":""}