@decocms/runtime 0.0.1-testing-beta.1

package/src/mastra.ts

@@ -0,0 +1,666 @@
+/* oxlint-disable no-explicit-any */
+/* oxlint-disable ban-types */
+import { HttpServerTransport } from "@deco/mcp/http";
+import {
+  createTool as mastraCreateTool,
+  Tool,
+  type ToolAction,
+  type ToolExecutionContext,
+} from "@mastra/core";
+import { RuntimeContext } from "@mastra/core/di";
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { z } from "zod";
+import { zodToJsonSchema } from "zod-to-json-schema";
+import type { DefaultEnv } from "./index.ts";
+import {
+  ResourceCreateInputSchema,
+  ResourceCreateOutputSchema,
+  ResourceDeleteInputSchema,
+  ResourceDeleteOutputSchema,
+  ResourceSearchInputSchema,
+  ResourceSearchOutputSchema,
+  ResourcesListOutputSchema,
+  ResourcesReadInputSchema,
+  ResourcesReadOutputSchema,
+  ResourceUpdateInputSchema,
+  ResourceUpdateOutputSchema,
+} from "./resources.ts";
+import { createStateValidationTool, State } from "./state.ts";
+import { ViewsListOutputSchema } from "./views.ts";
+
+export const createRuntimeContext = (prev?: RuntimeContext<AppContext>) => {
+  const runtimeContext = new RuntimeContext<AppContext>();
+  const store = State.getStore();
+  if (!store) {
+    if (prev) {
+      return prev;
+    }
+    throw new Error("Missing context, did you forget to call State.bind?");
+  }
+  const { env, ctx, req } = store;
+  runtimeContext.set("env", env);
+  runtimeContext.set("ctx", ctx);
+  runtimeContext.set("req", req);
+  return runtimeContext;
+};
+
+/**
+ * creates a private tool that always ensure for athentication before being executed
+ */
+export function createPrivateTool<
+  TSchemaIn extends z.ZodSchema = z.ZodSchema,
+  TSchemaOut extends z.ZodSchema | undefined = undefined,
+  TSuspendSchema extends z.ZodSchema = z.ZodSchema,
+  TResumeSchema extends z.ZodSchema = z.ZodSchema,
+  TContext extends
+    ToolExecutionContext<TSchemaIn> = ToolExecutionContext<TSchemaIn>,
+  TExecute extends ToolAction<
+    TSchemaIn,
+    TSchemaOut,
+    any,
+    any,
+    TContext
+  >["execute"] = ToolAction<
+    TSchemaIn,
+    TSchemaOut,
+    any,
+    any,
+    TContext
+  >["execute"],
+>(
+  opts: ToolAction<
+    TSchemaIn,
+    TSchemaOut,
+    TSuspendSchema,
+    TResumeSchema,
+    TContext
+  > & {
+    execute?: TExecute;
+  },
+): [TSchemaIn, TSchemaOut, TSuspendSchema, TResumeSchema, TExecute] extends [
+  z.ZodSchema,
+  z.ZodSchema,
+  z.ZodSchema,
+  z.ZodSchema,
+  Function,
+]
+  ? Tool<TSchemaIn, TSchemaOut, TSuspendSchema, TResumeSchema, TContext> & {
+      inputSchema: TSchemaIn;
+      outputSchema: TSchemaOut;
+      execute: (context: TContext) => Promise<any>;
+    }
+  : Tool<TSchemaIn, TSchemaOut, TSuspendSchema, TResumeSchema, TContext> {
+  const execute = opts.execute;
+  if (typeof execute === "function") {
+    opts.execute = ((input, options) => {
+      const env = input.runtimeContext.get("env") as DefaultEnv;
+      if (env) {
+        env.DECO_REQUEST_CONTEXT.ensureAuthenticated();
+      }
+      return execute(input, options);
+    }) as TExecute;
+  }
+  return createTool(opts);
+}
+
+export interface StreamableTool<TSchemaIn extends z.ZodSchema = z.ZodSchema> {
+  id: string;
+  inputSchema: TSchemaIn;
+  streamable?: true;
+  description?: string;
+  execute: (input: ToolExecutionContext<TSchemaIn>) => Promise<Response>;
+}
+
+export function createStreamableTool<
+  TSchemaIn extends z.ZodSchema = z.ZodSchema,
+>(streamableTool: StreamableTool<TSchemaIn>): StreamableTool<TSchemaIn> {
+  return {
+    ...streamableTool,
+    execute: (input: ToolExecutionContext<TSchemaIn>) => {
+      return streamableTool.execute({
+        ...input,
+        runtimeContext: createRuntimeContext(input.runtimeContext),
+      });
+    },
+  };
+}
+
+export function createTool<
+  TSchemaIn extends z.ZodSchema | undefined = undefined,
+  TSchemaOut extends z.ZodSchema | undefined = undefined,
+  TSuspendSchema extends z.ZodSchema = z.ZodSchema,
+  TResumeSchema extends z.ZodSchema = z.ZodSchema,
+  TContext extends ToolExecutionContext<
+    TSchemaIn,
+    TSuspendSchema,
+    TResumeSchema
+  > = ToolExecutionContext<TSchemaIn, TSuspendSchema, TResumeSchema>,
+  TExecute extends ToolAction<
+    TSchemaIn,
+    TSchemaOut,
+    TSuspendSchema,
+    TResumeSchema,
+    TContext
+  >["execute"] = ToolAction<
+    TSchemaIn,
+    TSchemaOut,
+    TSuspendSchema,
+    TResumeSchema,
+    TContext
+  >["execute"],
+>(
+  opts: ToolAction<
+    TSchemaIn,
+    TSchemaOut,
+    TSuspendSchema,
+    TResumeSchema,
+    TContext
+  > & {
+    execute?: TExecute;
+  },
+): [TSchemaIn, TSchemaOut, TSuspendSchema, TResumeSchema, TExecute] extends [
+  z.ZodSchema,
+  z.ZodSchema,
+  z.ZodSchema,
+  z.ZodSchema,
+  Function,
+]
+  ? Tool<TSchemaIn, TSchemaOut, TSuspendSchema, TResumeSchema, TContext> & {
+      inputSchema: TSchemaIn;
+      outputSchema: TSchemaOut;
+      execute: (context: TContext) => Promise<any>;
+    }
+  : Tool<TSchemaIn, TSchemaOut, TSuspendSchema, TResumeSchema, TContext> {
+  // @ts-expect-error - TSchemaIn is not a ZodType
+  return mastraCreateTool({
+    ...opts,
+    execute:
+      typeof opts?.execute === "function"
+        ? (((input) => {
+            return opts.execute!({
+              ...input,
+              runtimeContext: createRuntimeContext(input.runtimeContext),
+            });
+          }) as TExecute)
+        : opts.execute,
+  });
+}
+
+export type ExecWithContext<TF extends (...args: any[]) => any> = (
+  input: Omit<Parameters<TF>[0], "runtimeContext"> & {
+    runtimeContext: RuntimeContext<AppContext>;
+  },
+) => ReturnType<TF>;
+
+export interface ViewExport {
+  title: string;
+  icon: string;
+  url: string;
+  tools?: string[];
+  rules?: string[];
+  installBehavior?: "none" | "open" | "autoPin";
+}
+
+export type Resources<Env = any, TSchema extends z.ZodTypeAny = never> = Array<
+  (env: Env & DefaultEnv<TSchema>) => {
+    name: string;
+    icon: string;
+    title: string;
+    description?: string;
+    tools: {
+      read: (args: { uri: string }) => Promise<unknown>;
+      search: (args: {
+        term: string;
+        cursor?: string;
+        limit?: number;
+      }) => Promise<unknown>;
+      create?: (
+        args: z.infer<typeof ResourceCreateInputSchema>,
+      ) => Promise<unknown>;
+      update?: (
+        args: z.infer<typeof ResourceUpdateInputSchema>,
+      ) => Promise<unknown>;
+      delete?: (
+        args: z.infer<typeof ResourceDeleteInputSchema>,
+      ) => Promise<unknown>;
+    };
+    views?: {
+      list?: { url?: string; tools?: string[]; rules?: string[] };
+      detail?: {
+        url?: string;
+        mimeTypePattern?: string;
+        resourceName?: string;
+        tools?: string[];
+        rules?: string[];
+      };
+    };
+  }
+>;
+export interface Integration {
+  id: string;
+  appId: string;
+}
+export type CreatedTool =
+  | ReturnType<typeof createTool>
+  | ReturnType<typeof createStreamableTool>;
+export function isStreamableTool(tool: CreatedTool): tool is StreamableTool {
+  return tool && "streamable" in tool && tool.streamable === true;
+}
+export interface CreateMCPServerOptions<
+  Env = any,
+  TSchema extends z.ZodTypeAny = never,
+> {
+  before?: (env: Env & DefaultEnv<TSchema>) => Promise<void> | void;
+  oauth?: {
+    state?: TSchema;
+    scopes?: string[];
+  };
+  views?: (
+    env: Env & DefaultEnv<TSchema>,
+  ) => Promise<ViewExport[]> | ViewExport[];
+  resources?: Resources<Env, TSchema>;
+  tools?:
+    | Array<
+        (
+          env: Env & DefaultEnv<TSchema>,
+        ) =>
+          | Promise<CreatedTool>
+          | CreatedTool
+          | CreatedTool[]
+          | Promise<CreatedTool[]>
+      >
+    | ((
+        env: Env & DefaultEnv<TSchema>,
+      ) => CreatedTool[] | Promise<CreatedTool[]>);
+}
+
+export type Fetch<TEnv = any> = (
+  req: Request,
+  env: TEnv,
+  ctx: ExecutionContext,
+) => Promise<Response> | Response;
+
+export interface AppContext<TEnv = any> {
+  env: TEnv;
+  ctx: { waitUntil: (promise: Promise<any>) => void };
+  req?: Request;
+}
+
+const decoChatOAuthToolsFor = <TSchema extends z.ZodTypeAny = never>({
+  state: schema,
+  scopes,
+}: CreateMCPServerOptions<any, TSchema>["oauth"] = {}) => {
+  const jsonSchema = schema
+    ? zodToJsonSchema(schema)
+    : { type: "object", properties: {} };
+  return [
+    createTool({
+      id: "DECO_CHAT_OAUTH_START",
+      description: "OAuth for Deco Chat",
+      inputSchema: z.object({
+        returnUrl: z.string(),
+      }),
+      outputSchema: z.object({
+        stateSchema: z.any(),
+        scopes: z.array(z.string()).optional(),
+      }),
+      execute: () => {
+        return Promise.resolve({
+          stateSchema: jsonSchema,
+          scopes,
+        });
+      },
+    }),
+  ];
+};
+
+type CallTool = (opts: {
+  toolCallId: string;
+  toolCallInput: any;
+}) => Promise<any>;
+
+export type MCPServer<TEnv = any, TSchema extends z.ZodTypeAny = never> = {
+  fetch: Fetch<TEnv & DefaultEnv<TSchema>>;
+  callTool: CallTool;
+};
+
+export const createMCPServer = <
+  TEnv = any,
+  TSchema extends z.ZodTypeAny = never,
+>(
+  options: CreateMCPServerOptions<TEnv, TSchema>,
+): MCPServer<TEnv, TSchema> => {
+  const createServer = async (bindings: TEnv & DefaultEnv<TSchema>) => {
+    await options.before?.(bindings);
+
+    const server = new McpServer(
+      { name: "@deco/mcp-api", version: "1.0.0" },
+      { capabilities: { tools: {} } },
+    );
+
+    // Resolve resources first; build resource tools; append later
+    const resolvedResources = await Promise.all(
+      options.resources?.map((r) => r(bindings)) ?? [],
+    );
+    const readHandlers = new Map<
+      string,
+      (a: { uri: string }) => Promise<any>
+    >();
+    const searchHandlers = new Map<
+      string,
+      (a: { term: string; cursor?: string; limit?: number }) => Promise<any>
+    >();
+    const createHandlers = new Map<string, (a: any) => Promise<any>>();
+    const updateHandlers = new Map<string, (a: any) => Promise<any>>();
+    const deleteHandlers = new Map<string, (a: any) => Promise<any>>();
+    for (const r of resolvedResources) {
+      if (r?.tools?.read) readHandlers.set(r.name, r.tools.read);
+      if (r?.tools?.search) searchHandlers.set(r.name, r.tools.search);
+      if (r?.tools?.create) createHandlers.set(r.name, r.tools.create);
+      if (r?.tools?.update) updateHandlers.set(r.name, r.tools.update);
+      if (r?.tools?.delete) deleteHandlers.set(r.name, r.tools.delete);
+    }
+    const resourceTools: ReturnType<typeof createTool>[] = [];
+    if (resolvedResources.length > 0) {
+      resourceTools.push(
+        createTool({
+          id: "DECO_CHAT_RESOURCES_READ",
+          description: "Read a resource by uri (name + uri)",
+          inputSchema: ResourcesReadInputSchema,
+          outputSchema: ResourcesReadOutputSchema,
+          execute: (input) => {
+            // @ts-expect-error - input.name is not a string
+            const fn = readHandlers.get(input.name);
+            if (!fn) {
+              // @ts-expect-error - input.name is not a string
+              throw new Error(`READ not implemented for ${input.name}`);
+            }
+            // @ts-expect-error - input.name is not a string
+            return fn({ uri: input.uri });
+          },
+        }),
+      );
+      resourceTools.push(
+        createTool({
+          id: "DECO_CHAT_RESOURCES_SEARCH",
+          description: "Search resources (name + term)",
+          inputSchema: ResourceSearchInputSchema,
+          outputSchema: ResourceSearchOutputSchema,
+          execute: (input) => {
+            // @ts-expect-error - input.name is not a string
+            const fn = searchHandlers.get(input.name);
+            if (!fn) {
+              // @ts-expect-error - input.name is not a string
+              throw new Error(`SEARCH not implemented for ${input.name}`);
+            }
+            // @ts-expect-error - input.name is not a string
+            const { term, cursor, limit } = input;
+            return fn({ term, cursor, limit });
+          },
+        }),
+      );
+      resourceTools.push(
+        createTool({
+          id: "DECO_CHAT_RESOURCES_CREATE",
+          description: "Create a resource (name + content)",
+          inputSchema: ResourceCreateInputSchema,
+          outputSchema: ResourceCreateOutputSchema,
+          execute: (input) => {
+            // @ts-expect-error - input.name is not a string
+            const fn = createHandlers.get(input.name);
+            if (!fn) {
+              // @ts-expect-error - input.name is not a string
+              throw new Error(`CREATE not implemented for ${input.name}`);
+            }
+            return fn(input);
+          },
+        }),
+      );
+      resourceTools.push(
+        createTool({
+          id: "DECO_CHAT_RESOURCES_UPDATE",
+          description: "Update a resource (name + uri)",
+          inputSchema: ResourceUpdateInputSchema,
+          outputSchema: ResourceUpdateOutputSchema,
+          execute: (input) => {
+            // @ts-expect-error - input.name is not a string
+            const fn = updateHandlers.get(input.name);
+            if (!fn) {
+              // @ts-expect-error - input.name is not a string
+              throw new Error(`UPDATE not implemented for ${input.name}`);
+            }
+            return fn(input);
+          },
+        }),
+      );
+      resourceTools.push(
+        createTool({
+          id: "DECO_CHAT_RESOURCES_DELETE",
+          description: "Delete a resource (name + uri)",
+          inputSchema: ResourceDeleteInputSchema,
+          outputSchema: ResourceDeleteOutputSchema,
+          execute: (input) => {
+            // @ts-expect-error - input.name is not a string
+            const fn = deleteHandlers.get(input.name);
+            if (!fn) {
+              // @ts-expect-error - input.name is not a string
+              throw new Error(`DELETE not implemented for ${input.name}`);
+            }
+            return fn(input);
+          },
+        }),
+      );
+      resourceTools.push(
+        createTool({
+          id: "DECO_CHAT_RESOURCES_LIST",
+          description: "List resource types",
+          inputSchema: z.object({}),
+          outputSchema: ResourcesListOutputSchema,
+          execute: () =>
+            Promise.resolve({
+              resources: resolvedResources.map((r) => ({
+                name: r.name,
+                icon: r.icon,
+                title: r.title,
+                description: r.description ?? "",
+                hasCreate: Boolean(createHandlers.get(r.name)),
+                hasUpdate: Boolean(updateHandlers.get(r.name)),
+                hasDelete: Boolean(deleteHandlers.get(r.name)),
+              })),
+            }),
+        }),
+      );
+    }
+
+    const toolsFn =
+      typeof options.tools === "function"
+        ? options.tools
+        : async (bindings: TEnv & DefaultEnv<TSchema>) => {
+            if (typeof options.tools === "function") {
+              return await options.tools(bindings);
+            }
+            return await Promise.all(
+              options.tools?.flatMap(async (tool) => {
+                const toolResult = tool(bindings);
+                const awaited = await toolResult;
+                if (Array.isArray(awaited)) {
+                  return awaited;
+                }
+                return [awaited];
+              }) ?? [],
+            ).then((t) => t.flat());
+          };
+    const tools = await toolsFn(bindings);
+
+    tools.push(...decoChatOAuthToolsFor<TSchema>(options.oauth));
+    tools.push(createStateValidationTool(options.oauth?.state));
+
+    tools.push(
+      createTool({
+        id: `DECO_CHAT_VIEWS_LIST`,
+        description: "List views exposed by this MCP",
+        inputSchema: z.any(),
+        outputSchema: ViewsListOutputSchema,
+        execute: async () => {
+          const base = ((await options.views?.(bindings)) ?? []).map((v) => ({
+            id: undefined,
+            // Stable machine name for routing: UPPERCASE + underscores
+            name: v.title.toUpperCase().replace(/[^A-Z0-9]/g, "_"),
+            title: v.title,
+            description: undefined,
+            icon: v.icon,
+            url: v.url,
+            tools: v.tools ?? [],
+            rules: v.rules ?? [],
+            installBehavior: v.installBehavior ?? "none",
+          }));
+          const resourceViews = resolvedResources
+            .map((r) => {
+              const listUrl =
+                r.views?.list?.url ??
+                `internal://resource/list?name=${encodeURIComponent(r.name)}`;
+
+              // Default CRUD tool ids for resources
+              const defaultListTools: string[] = (() => {
+                const base = [
+                  "DECO_CHAT_RESOURCES_LIST",
+                  "DECO_CHAT_RESOURCES_READ",
+                  "DECO_CHAT_RESOURCES_SEARCH",
+                ];
+                const canCreate = Boolean(createHandlers.get(r.name));
+                const canUpdate = Boolean(updateHandlers.get(r.name));
+                const canDelete = Boolean(deleteHandlers.get(r.name));
+                if (canCreate) base.push("DECO_CHAT_RESOURCES_CREATE");
+                if (canUpdate) base.push("DECO_CHAT_RESOURCES_UPDATE");
+                if (canDelete) base.push("DECO_CHAT_RESOURCES_DELETE");
+                return base;
+              })();
+
+              const defaultListRules: string[] = [
+                `You are viewing the ${
+                  r.title ?? r.name
+                } resources list. Resources are changeable via Resource tools (DECO_CHAT_RESOURCES_*). Use the appropriate tools to read, search, create, update, or delete items; do not fabricate data.`,
+              ];
+
+              const list = [
+                {
+                  name: `${r.name.toUpperCase()}_LIST`,
+                  title: `${r.name} List`,
+                  description: r.description,
+                  icon: r.icon,
+                  url: listUrl,
+                  tools: r.views?.list?.tools ?? defaultListTools,
+                  rules: r.views?.list?.rules ?? defaultListRules,
+                },
+              ];
+              const detailUrl =
+                r.views?.detail?.url ??
+                `internal://resource/detail?name=${encodeURIComponent(r.name)}`;
+              const detail = [
+                {
+                  name: `${r.name.toUpperCase()}_DETAIL`,
+                  title: `${r.name} Detail`,
+                  description: r.description,
+                  icon: r.icon,
+                  url: detailUrl,
+                  mimeTypePattern: r.views?.detail?.mimeTypePattern,
+                  resourceName: r.views?.detail?.resourceName ?? r.name,
+                  tools: r.views?.detail?.tools ?? [],
+                  rules: r.views?.detail?.rules ?? [],
+                },
+              ];
+              return [...list, ...detail];
+            })
+            .flat();
+
+          return { views: [...base, ...resourceViews] };
+        },
+      }),
+    );
+
+    for (const tool of tools) {
+      server.registerTool(
+        tool.id,
+        {
+          _meta: {
+            streamable: isStreamableTool(tool),
+          },
+          description: tool.description,
+          inputSchema:
+            tool.inputSchema && "shape" in tool.inputSchema
+              ? (tool.inputSchema.shape as z.ZodRawShape)
+              : z.object({}).shape,
+          outputSchema: isStreamableTool(tool)
+            ? z.object({ bytes: z.record(z.string(), z.number()) }).shape
+            : tool.outputSchema &&
+                typeof tool.outputSchema === "object" &&
+                "shape" in tool.outputSchema
+              ? (tool.outputSchema.shape as z.ZodRawShape)
+              : z.object({}).shape,
+        },
+        async (args) => {
+          let result = await tool.execute!({
+            context: args,
+            runId: crypto.randomUUID(),
+            runtimeContext: createRuntimeContext(),
+          });
+
+          if (isStreamableTool(tool) && result instanceof Response) {
+            result = { bytes: await result.bytes() };
+          }
+          return {
+            structuredContent: result,
+            content: [
+              {
+                type: "text",
+                text: JSON.stringify(result),
+              },
+            ],
+          };
+        },
+      );
+    }
+
+    return { server, tools };
+  };
+
+  const fetch = async (
+    req: Request,
+    env: TEnv & DefaultEnv<TSchema>,
+    _ctx: ExecutionContext,
+  ) => {
+    const { server } = await createServer(env);
+    const transport = new HttpServerTransport();
+
+    await server.connect(transport);
+
+    return await transport.handleMessage(req);
+  };
+
+  const callTool: CallTool = async ({ toolCallId, toolCallInput }) => {
+    const currentState = State.getStore();
+    if (!currentState) {
+      throw new Error("Missing state, did you forget to call State.bind?");
+    }
+    const env = currentState?.env;
+    const { tools } = await createServer(env);
+    const tool = tools.find((t) => t.id === toolCallId);
+    const execute = tool?.execute;
+    if (!execute) {
+      throw new Error(
+        `Tool ${toolCallId} not found or does not have an execute function`,
+      );
+    }
+
+    return execute({
+      context: toolCallInput,
+      runId: crypto.randomUUID(),
+      runtimeContext: createRuntimeContext(),
+    });
+  };
+
+  return {
+    fetch,
+    callTool,
+  };
+};