polarisagi-knowledge-base 0.1.0

package/dist/index.js

@@ -0,0 +1,177 @@
+#!/usr/bin/env node
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+const readline = __importStar(require("readline"));
+const fs = __importStar(require("fs/promises"));
+const path = __importStar(require("path"));
+// Allowed directory validation
+const allowedDir = process.env.POLARIS_KB_ALLOWED_DIR;
+function isPathAllowed(targetPath) {
+    if (!allowedDir) {
+        return true;
+    }
+    const absPath = path.resolve(targetPath);
+    const rootPath = path.resolve(allowedDir);
+    return absPath === rootPath || absPath.startsWith(rootPath + path.sep);
+}
+// MCP JSON-RPC setup
+const rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stdout,
+    terminal: false
+});
+function sendResult(id, result) {
+    const msg = {
+        jsonrpc: "2.0",
+        id,
+        result
+    };
+    console.log(JSON.stringify(msg));
+}
+function sendError(id, code, message) {
+    const msg = {
+        jsonrpc: "2.0",
+        id,
+        error: { code, message }
+    };
+    console.log(JSON.stringify(msg));
+}
+rl.on('line', async (line) => {
+    if (!line.trim())
+        return;
+    try {
+        const req = JSON.parse(line);
+        if (req.jsonrpc !== "2.0" || !req.method || req.id === undefined) {
+            return;
+        }
+        const id = req.id;
+        if (req.method === "initialize") {
+            sendResult(id, {
+                protocolVersion: "2024-11-05",
+                capabilities: { tools: {} },
+                serverInfo: {
+                    name: "polaris-knowledge-base-mcp",
+                    version: "0.1.0"
+                }
+            });
+        }
+        else if (req.method === "tools/list") {
+            sendResult(id, {
+                tools: [
+                    {
+                        name: "list_files",
+                        description: "List files in a given directory path",
+                        inputSchema: {
+                            type: "object",
+                            properties: {
+                                path: {
+                                    type: "string",
+                                    description: "The absolute path to the directory"
+                                }
+                            },
+                            required: ["path"]
+                        }
+                    },
+                    {
+                        name: "read_content",
+                        description: "Read the textual content of a specific file",
+                        inputSchema: {
+                            type: "object",
+                            properties: {
+                                path: {
+                                    type: "string",
+                                    description: "The absolute path to the file"
+                                }
+                            },
+                            required: ["path"]
+                        }
+                    }
+                ]
+            });
+        }
+        else if (req.method === "tools/call") {
+            const params = req.params || {};
+            const args = params.arguments || {};
+            if (params.name === "list_files") {
+                const targetPath = args.path;
+                if (typeof targetPath !== "string") {
+                    sendResult(id, { isError: true, content: [{ type: "text", text: "path is required and must be a string" }] });
+                    return;
+                }
+                if (!isPathAllowed(targetPath)) {
+                    sendResult(id, { isError: true, content: [{ type: "text", text: "path is outside the allowed directory (POLARIS_KB_ALLOWED_DIR)" }] });
+                    return;
+                }
+                try {
+                    const entries = await fs.readdir(targetPath, { withFileTypes: true });
+                    const fileNames = entries.map(e => e.isDirectory() ? `${e.name}/` : e.name);
+                    const result = `Files in ${targetPath}:\n${fileNames.join('\n')}`;
+                    sendResult(id, { content: [{ type: "text", text: result }] });
+                }
+                catch (err) {
+                    sendResult(id, { isError: true, content: [{ type: "text", text: `failed to read dir: ${err.message}` }] });
+                }
+            }
+            else if (params.name === "read_content") {
+                const targetPath = args.path;
+                if (typeof targetPath !== "string") {
+                    sendResult(id, { isError: true, content: [{ type: "text", text: "path is required and must be a string" }] });
+                    return;
+                }
+                if (!isPathAllowed(targetPath)) {
+                    sendResult(id, { isError: true, content: [{ type: "text", text: "path is outside the allowed directory (POLARIS_KB_ALLOWED_DIR)" }] });
+                    return;
+                }
+                try {
+                    const data = await fs.readFile(targetPath, 'utf8');
+                    sendResult(id, { content: [{ type: "text", text: data }] });
+                }
+                catch (err) {
+                    sendResult(id, { isError: true, content: [{ type: "text", text: `failed to read file: ${err.message}` }] });
+                }
+            }
+            else {
+                sendError(id, -32601, `Tool not found: ${params.name}`);
+            }
+        }
+        else {
+            sendError(id, -32601, "Method not found");
+        }
+    }
+    catch (e) {
+        // ignore parse errors or malformed json
+    }
+});

package/package.json

@@ -0,0 +1,21 @@
+{
+  "name": "polarisagi-knowledge-base",
+  "version": "0.1.0",
+  "description": "MCP Plugin for reading local knowledge base files",
+  "files": [
+    "dist"
+  ],
+  "main": "dist/index.js",
+  "bin": {
+    "polarisagi-knowledge-base": "./dist/index.js"
+  },
+  "scripts": {
+    "build": "tsc",
+    "start": "node dist/index.js"
+  },
+  "dependencies": {},
+  "devDependencies": {
+    "@types/node": "^20.0.0",
+    "typescript": "^5.0.0"
+  }
+}