initialized mcp support

app/src/modules/mcp/$object.ts

@@ -0,0 +1,136 @@
+import * as s from "jsonv-ts";
+import { type Resource, Tool, type ToolAnnotation, type ToolHandlerCtx } from "jsonv-ts/mcp";
+import { excludePropertyTypes, rescursiveClean } from "./utils";
+import { autoFormatString, getPath } from "bknd/utils";
+import type { App } from "App";
+import type { ModuleBuildContext } from "modules";
+
+export interface McpToolOptions {
+   title?: string;
+   description?: string;
+   annotations?: ToolAnnotation;
+   tools?: Tool<any, any, any>[];
+   resources?: Resource<any, any, any, any>[];
+}
+
+export interface ObjectToolSchemaOptions extends s.IObjectOptions {
+   mcp?: McpToolOptions;
+}
+
+type AppToolContext = {
+   app: App;
+   ctx: () => ModuleBuildContext;
+};
+type AppToolHandlerCtx = ToolHandlerCtx<AppToolContext>;
+
+export class ObjectToolSchema<
+   const P extends s.TProperties = s.TProperties,
+   const O extends s.IObjectOptions = s.IObjectOptions,
+> extends s.ObjectSchema<P, O> {
+   public readonly mcp: McpToolOptions;
+   private cleanSchema: s.ObjectSchema<P, O>;
+
+   constructor(
+      public name: string,
+      properties: P,
+      options?: ObjectToolSchemaOptions,
+   ) {
+      const { mcp, ...rest } = options || {};
+
+      super(properties, rest as any);
+      this.name = name;
+      this.mcp = mcp || {};
+      this.cleanSchema = this.getCleanSchema();
+   }
+
+   private getMcpOptions(action: "get" | "update") {
+      const { tools, resources, ...rest } = this.mcp;
+      const label = (text?: string) =>
+         text && [autoFormatString(action), text].filter(Boolean).join(" ");
+      return {
+         title: label(this.title ?? this.mcp.title),
+         description: label(this.description ?? this.mcp.description),
+         annotations: {
+            destructiveHint: true,
+            idempotentHint: true,
+            ...rest.annotations,
+         },
+      };
+   }
+
+   private getCleanSchema() {
+      const props = excludePropertyTypes(this as any, [ObjectToolSchema]);
+      const schema = s.strictObject(props);
+      return rescursiveClean(schema, {
+         removeRequired: true,
+         removeDefault: false,
+      }) as s.ObjectSchema<P, O>;
+   }
+
+   private toolGet(node: s.Node<ObjectToolSchema>) {
+      return new Tool(
+         [this.name, "get"].join("_"),
+         {
+            ...this.getMcpOptions("get"),
+            inputSchema: s.strictObject({
+               path: s
+                  .string({
+                     pattern: /^[a-zA-Z0-9_.]{0,}$/,
+                     description: "(optional) path to the property to get, e.g. `key.subkey`",
+                  })
+                  .optional(),
+               include_secrets: s.boolean({ default: false }).optional(),
+            }),
+         },
+         async (params, ctx: AppToolHandlerCtx) => {
+            const configs = ctx.context.app.toJSON(params.include_secrets);
+            const config = getPath(configs, node.instancePath);
+            const value = getPath(config, params.path ?? []);
+            return ctx.json({
+               path: params.path ?? "",
+               value: value ?? null,
+            });
+         },
+      );
+   }
+
+   private toolUpdate(node: s.Node<ObjectToolSchema>) {
+      const schema = this.cleanSchema;
+      return new Tool(
+         [this.name, "update"].join("_"),
+         {
+            ...this.getMcpOptions("update"),
+            inputSchema: s.strictObject({
+               full: s.boolean({ default: false }).optional(),
+               value: s
+                  .strictObject(schema.properties as any)
+                  .partial() as unknown as s.ObjectSchema<P, O>,
+            }),
+         },
+         async (params, ctx: AppToolHandlerCtx) => {
+            return ctx.json(params);
+         },
+      );
+   }
+
+   getTools(node: s.Node<ObjectToolSchema>): Tool<any, any, any>[] {
+      const { tools = [] } = this.mcp;
+      return [this.toolGet(node), this.toolUpdate(node), ...tools];
+   }
+
+   override toJSON(): s.JSONSchemaDefinition {
+      const { toJSON, "~standard": _, mcp, cleanSchema, name, ...rest } = this;
+      return JSON.parse(JSON.stringify(rest));
+   }
+}
+
+export const $object = <
+   const P extends s.TProperties = s.TProperties,
+   const O extends ObjectToolSchemaOptions = ObjectToolSchemaOptions,
+>(
+   name: string,
+   properties: P,
+   options?: s.StrictOptions<ObjectToolSchemaOptions, O>,
+): ObjectToolSchema<P, O> & O => {
+   return new ObjectToolSchema(name, properties, options) as any;
+};

app/src/modules/mcp/index.ts

@@ -0,0 +1 @@
+export * from "./$object";

app/src/modules/mcp/utils.spec.ts

@@ -0,0 +1,36 @@
+import { describe, it, expect } from "bun:test";
+import { excludePropertyTypes, rescursiveClean } from "./utils";
+import { s } from "../../core/utils/schema";
+
+describe("rescursiveOptional", () => {
+   it("should make all properties optional", () => {
+      const schema = s.strictObject({
+         a: s.string(),
+         b: s.number(),
+         nested: s.strictObject({
+            c: s.string().optional(),
+            d: s.number(),
+            nested2: s.record(s.string()),
+         }),
+      });
+
+      //console.log(schema.toJSON());
+      console.log(
+         rescursiveClean(schema, {
+            removeRequired: true,
+            removeDefault: true,
+         }).toJSON(),
+      );
+      /* const result = rescursiveOptional(schema);
+      expect(result.properties.a.optional).toBe(true); */
+   });
+
+   it("should exclude properties", () => {
+      const schema = s.strictObject({
+         a: s.string(),
+         b: s.number(),
+      });
+
+      console.log(excludePropertyTypes(schema, [s.StringSchema]));
+   });
+});

app/src/modules/mcp/utils.ts

@@ -0,0 +1,51 @@
+import * as s from "jsonv-ts";
+import { isPlainObject, transformObject } from "bknd/utils";
+
+export function rescursiveClean(
+   input: s.Schema,
+   opts?: {
+      removeRequired?: boolean;
+      removeDefault?: boolean;
+   },
+): s.Schema {
+   const json = input.toJSON();
+
+   const removeRequired = (obj: any) => {
+      if (isPlainObject(obj)) {
+         if ("required" in obj && opts?.removeRequired) {
+            obj.required = undefined;
+         }
+
+         if ("default" in obj && opts?.removeDefault) {
+            obj.default = undefined;
+         }
+
+         if ("properties" in obj && isPlainObject(obj.properties)) {
+            for (const key in obj.properties) {
+               obj.properties[key] = removeRequired(obj.properties[key]);
+            }
+         }
+      }
+
+      return obj;
+   };
+
+   removeRequired(json);
+   return s.fromSchema(json);
+}
+
+export function excludePropertyTypes(
+   input: s.ObjectSchema<any, any>,
+   props: (new (...args: any[]) => s.Schema)[],
+): s.TProperties {
+   const properties = { ...input.properties };
+
+   return transformObject(properties, (value, key) => {
+      for (const prop of props) {
+         if (value instanceof prop) {
+            return undefined;
+         }
+      }
+      return value;
+   });
+}