chore: minor changes

Author: MathurAditya724

examples/mcp-sdk/src/index.ts

@@ -3,10 +3,9 @@ import { Hono } from "hono";
 import {
   muppet,
   describeTool,
-  // staticResource,
-  // dynamicResource,
   describePrompt,
   mValidator,
+  registerResources,
 } from "muppet";
 import z from "zod";
 import pino from "pino";
@@ -38,35 +37,18 @@ app.post(
   },
 );
 
-// Define static resources
-// app.post(
-//   "/static/*",
-//   staticResource({
-//     name: "Static Resource",
-//     description: "A static resource",
-//     resource: {
-//       path: "E:/dev/muppet/muppet/examples/mcp-sdk/dist/static",
-//     },
-//   }),
-// );
-
-// // Define Dynamic resources
-// app.post(
-//   "/dynamic/*",
-//   dynamicResource({
-//     name: "Dynamic Resource",
-//     description: "A dynamic resource",
-//   }),
-//   (c) => {
-//     return c.json([
-//       {
-//         uri: "file:///logs/app.log",
-//         name: "Application Logs",
-//         mimeType: "text/plain",
-//       },
-//     ]);
-//   },
-// );
+app.post(
+  "/documents",
+  registerResources((c) => {
+    return c.json([
+      {
+        uri: "https://lorem.ipsum",
+        name: "Todo list",
+        mimeType: "text/plain",
+      },
+    ]);
+  }),
+);
 
 // Define a simple prompt
 app.post(
@@ -102,4 +84,30 @@ muppet(app, {
       "/Users/adityamathur/dev/muppet-dev/muppet/examples/mcp-sdk/dist/main.log",
     ),
   },
+  resources: {
+    https: () => {
+      return [
+        {
+          uri: "task1",
+          text: "Task 1",
+        },
+        {
+          uri: "task2",
+          text: "Task 2",
+        },
+        {
+          uri: "task3",
+          text: "Task 3",
+        },
+        {
+          uri: "task4",
+          text: "Task 4",
+        },
+        {
+          uri: "task5",
+          text: "Task 5",
+        },
+      ];
+    },
+  },
 });

packages/core/src/muppet.ts

@@ -19,6 +19,7 @@ import type {
   DescribeOptions,
   MuppetConfiguration,
   PromptConfiguration,
+  Resource,
   ServerConfiguration,
   ToolHandlerResponse,
 } from "./types";
@@ -407,20 +408,40 @@ function createResourceApp<
   });
 
   app.post("/list", async (c) => {
-    const resources = await Promise.all(
+    const responses = await Promise.all(
       Object.values(config.resources ?? {}).map(async ({ path }) => {
         const res = await hono.request(path, {
           method: "POST",
           headers: c.req.header(),
         });
 
-        return res.json();
+        return res.json() as Promise<Resource[]>;
       }),
     );
 
+    const resources = responses.flat(2).map((resource: Resource) => {
+      const schema = {
+        name: resource.name,
+        description: resource.description,
+        mimeType: resource.mimeType,
+      };
+
+      if (resource.type === "template") {
+        return {
+          ...schema,
+          uriTemplate: resource.uri,
+        };
+      }
+
+      return {
+        ...schema,
+        uri: resource.uri,
+      };
+    });
+
     return c.json({
       result: {
-        resources: resources.flat(),
+        resources,
       },
     });
   });

packages/core/src/resources.ts

@@ -1,14 +1,15 @@
 import type { Env, Handler, Input } from "hono";
 import type { BlankEnv, BlankInput, TypedResponse } from "hono/types";
 import { McpPrimitives, uniqueSymbol } from "./utils";
+import type { Resource } from "./types";
 
 export function registerResources<
   E extends Env = BlankEnv,
   P extends string = string,
   I extends Input = BlankInput,
 >(
-  handler: Handler<E, P, I, TypedResponse<"json", 200>>,
-): Handler<E, P, I, TypedResponse<"json", 200>> {
+  handler: Handler<E, P, I, TypedResponse<Resource[], 200>>,
+): Handler<E, P, I, TypedResponse<Resource[], 200>> {
   return Object.assign(handler, {
     [uniqueSymbol]: {
       type: McpPrimitives.RESOURCES,

packages/core/src/types.ts

@@ -44,6 +44,35 @@ export type ResourceConfiguration = Record<
   }
 >;
 
+export type Resource =
+  | {
+      type?: "direct";
+      uri: string;
+      name: string;
+      description?: string;
+      mimeType?: string;
+    }
+  | {
+      type: "template";
+      uri: string;
+      name: string;
+      description?: string;
+      mimeType?: string;
+    };
+
+export type ResourceResponse = {
+  uri: string;
+  mimeType?: string;
+  /**
+   * For text resources
+   */
+  text?: string;
+  /**
+   * For binary resources (base64 encoded)
+   */
+  blob?: string;
+};
+
 export type ConceptConfiguration = Record<
   string,
   | (DescribeOptions & {
@@ -72,19 +101,8 @@ export type MuppetConfiguration = {
   events?: Emitter<AvailableEvents>;
   resources?: Record<
     string,
-    (uri: string) => Promisify<
-      {
-        uri: string;
-        mimeType: string;
-        /**
-         * For text resources
-         */
-        text?: string;
-        /**
-         * For binary resources (base64 encoded)
-         */
-        blob?: string;
-      }[]
-    >
+    (
+      uri: string,
+    ) => Promisify<ResourceResponse[] | { contents: ResourceResponse[] }>
   >;
 };