Merge pull request #36 from SugiKent/feature/tool-args

add enabledTools args

Author: suekou

README.md

@@ -64,6 +64,15 @@ or
 - `NOTION_API_TOKEN` (required): Your Notion API integration token.
 - `NOTION_MARKDOWN_CONVERSION`: Set to "true" to enable experimental Markdown conversion. This can significantly reduce token consumption when viewing content, but may cause issues when trying to edit page content.
 
+## Command Line Arguments
+
+- `--enabledTools`: Comma-separated list of tools to enable (e.g. "notion_retrieve_page,notion_query_database"). When specified, only the listed tools will be available. If not specified, all tools are enabled.
+
+Read-only tools example (copy-paste friendly):
+```bash
+node build/index.js --enabledTools=notion_retrieve_block,notion_retrieve_block_children,notion_retrieve_page,notion_query_database,notion_retrieve_database,notion_search,notion_list_all_users,notion_retrieve_user,notion_retrieve_bot_user,notion_retrieve_comments
+```
+
 ## Advanced Configuration
 
 ### Markdown Conversion

notion/package-lock.json

@@ -28,6 +28,8 @@
   },
   "devDependencies": {
     "@types/node": "^20.11.24",
-    "typescript": "^5.3.3"
+    "@types/yargs": "^17.0.33",
+    "typescript": "^5.3.3",
+    "yargs": "^17.7.2"
   }
 }

notion/package.json

@@ -1,11 +1,28 @@
 import { expect, test, describe, vi, beforeEach } from "vitest";
 import { NotionClientWrapper } from "./index.js";
 import { PageResponse } from "./types/index.js";
+import { filterTools } from "./index.js";
 
 vi.mock("./markdown/index.js", () => ({
   convertToMarkdown: vi.fn().mockReturnValue("# Test"),
 }));
 
+// Mock tool list
+const mockInputSchema = { type: "object" as const }
+const mockTools = [
+    {
+        name: "notion_retrieve_block",
+        inputSchema: mockInputSchema
+    },
+    {
+        name: "notion_retrieve_page",
+        inputSchema: mockInputSchema
+    },
+    {
+        name: "notion_query_database",
+        inputSchema: mockInputSchema
+    }
+];
 global.fetch = vi.fn();
 
 describe("NotionClientWrapper", () => {
@@ -162,4 +179,26 @@ describe("NotionClientWrapper", () => {
 
     expect(convertToMarkdown).toHaveBeenCalledWith(response);
   });
+
+  describe("filterTools", () => {
+    test("should return all tools when no filter specified", () => {
+      const result = filterTools(mockTools);
+      expect(result).toEqual(mockTools);
+    });
+
+    test("should filter tools based on enabledTools", () => {
+      const enabledToolsSet = new Set(["notion_retrieve_block", "notion_query_database"]);
+      const result = filterTools(mockTools, enabledToolsSet);
+      expect(result).toEqual([
+        { name: "notion_retrieve_block", inputSchema: mockInputSchema },
+        { name: "notion_query_database", inputSchema: mockInputSchema }
+      ]);
+    });
+
+    test("should return empty array when no tools match", () => {
+      const enabledToolsSet = new Set(["non_existent_tool"]);
+      const result = filterTools(mockTools, enabledToolsSet);
+      expect(result).toEqual([]);
+    });
+  });
 });