chore: minor implmentation

Author: MathurAditya724

package.json

@@ -3,7 +3,7 @@
   "version": "0.0.1",
   "scripts": {
     "commit": "czg",
-    "setup": "is-ci || husky",
+    "prepare": "is-ci || husky",
     "format": "biome check --write ."
   },
   "devDependencies": {

packages/core/package.json

@@ -26,7 +26,11 @@
   },
   "peerDependencies": {
     "@modelcontextprotocol/sdk": "^1.7.0",
-    "pino": "^9.6.0"
+    "@standard-community/standard-json": "^0.1.0",
+    "@valibot/to-json-schema": "^1.0.0-rc.0",
+    "arktype": "^2.0.4",
+    "pino": "^9.6.0",
+    "zod-to-json-schema": "^3.24.1"
   },
   "devDependencies": {
     "@standard-schema/spec": "^1.0.0",

packages/core/src/handlers/initializeHandler.ts

@@ -2,6 +2,7 @@ import {
   InitializeRequestSchema,
   LATEST_PROTOCOL_VERSION,
   SUPPORTED_PROTOCOL_VERSIONS,
+  type ServerCapabilities,
 } from "@modelcontextprotocol/sdk/types.js";
 import type { HandlerFn } from "../types";
 
@@ -14,6 +15,12 @@ export const initializeHandler: HandlerFn = (ctx, request, config) => {
   const validatedData = InitializeRequestSchema.parse(request);
   const requestedVersion = validatedData.params.protocolVersion;
 
+  const capabilities: ServerCapabilities = {};
+
+  if (config.tools) {
+    capabilities.tools = {};
+  }
+
   return {
     result: {
       protocolVersion: SUPPORTED_PROTOCOL_VERSIONS.includes(requestedVersion)
@@ -23,9 +30,7 @@ export const initializeHandler: HandlerFn = (ctx, request, config) => {
         name: config.name,
         version: config.version,
       },
-      capabilities: {
-        tools: {},
-      },
+      capabilities,
     },
   };
 };

packages/core/src/handlers/messageHandler.ts

@@ -5,20 +5,29 @@ import {
 import type { HandlerFn } from "../types";
 import { initializeHandler } from "./initializeHandler";
 import { methodNotFoundHandler } from "./notFound";
+import { callToolsHandler, listToolsHandler } from "./tools";
 
 const handlersMap: Record<string, HandlerFn | undefined> = {
   initialize: initializeHandler,
+  "tools/call": callToolsHandler,
+  "tools/list": listToolsHandler,
 };
 
 export const messageHandler: HandlerFn = async (ctx, request, config) => {
   const _request = RequestSchema.parse(request);
 
-  const handler = handlersMap[_request.method] ?? methodNotFoundHandler;
+  const method = _request.method;
+
+  if ((method === "tools/list" || method === "tools/call") && !config.tools) {
+    return;
+  }
+
+  const handler = handlersMap[method] ?? methodNotFoundHandler;
 
   const response = await handler(ctx, request, config);
 
   ctx.logger.info({
-    msg: `Response for ${_request.method}`,
+    msg: `Response for ${method}`,
     response,
   });
 

packages/core/src/handlers/tools/callToolsHandler.ts

@@ -0,0 +1,3 @@
+import type { HandlerFn } from "@/types";
+
+export const callToolsHandler: HandlerFn = () => {};

packages/core/src/handlers/tools/index.ts

@@ -0,0 +1,2 @@
+export * from "./listToolsHandler";
+export * from "./callToolsHandler";

packages/core/src/handlers/tools/listToolsHandler.ts

@@ -0,0 +1,34 @@
+import type { HandlerFn } from "@/types";
+import { ListToolsRequestSchema } from "@modelcontextprotocol/sdk/types.js";
+import { toJsonSchema } from "@standard-community/standard-json";
+
+const EMPTY_OBJECT_JSON_SCHEMA = {
+  type: "object" as const,
+};
+
+export const listToolsHandler: HandlerFn = async (ctx, request, config) => {
+  ctx.logger.info({
+    msg: "Listing tools",
+    request,
+  });
+
+  if (!config.tools) {
+    return;
+  }
+
+  const validatedData = ListToolsRequestSchema.parse(request);
+
+  return {
+    result: {
+      tools: await Promise.all(
+        Object.entries(config.tools).map(async ([name, tool]) => ({
+          name,
+          description: tool.description,
+          inputSchema: tool.schema
+            ? await toJsonSchema(tool.schema)
+            : EMPTY_OBJECT_JSON_SCHEMA,
+        })),
+      ),
+    },
+  };
+};

packages/core/tsconfig.json

@@ -24,7 +24,11 @@
     "forceConsistentCasingInFileNames": true,
     "isolatedModules": true,
     "verbatimModuleSyntax": true,
-    "skipLibCheck": true
+    "skipLibCheck": true,
+    "baseUrl": ".",
+    "paths": {
+      "@/*": ["./src/*"]
+    }
   },
   "include": ["./src/**/*"]
 }