silverbullet-ai-mcp 0.0.0

package/dist/index.js

@@ -0,0 +1,173 @@
+#!/usr/bin/env node
+import { createServer as createHttpServer } from "node:http";
+import { Server } from "@modelcontextprotocol/sdk/server/index.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
+import { CallToolRequestSchema, ListToolsRequestSchema, } from "@modelcontextprotocol/sdk/types.js";
+const SB_URL = process.env.SB_URL;
+if (!SB_URL) {
+    console.error("SB_URL environment variable is required");
+    process.exit(1);
+}
+const SB_AUTH_TOKEN = process.env.SB_AUTH_TOKEN;
+const MCP_PORT = process.env.MCP_PORT;
+const MCP_TOKEN = process.env.MCP_TOKEN;
+const EXCLUDED_TOOLS = new Set(["ask_user", "navigate"]);
+// Space Lua has no json.decode, so pass arguments as Lua literals.
+// TODO: export this from sb or sbai?
+function toLuaLiteral(value) {
+    if (value === null || value === undefined)
+        return "nil";
+    if (typeof value === "string") {
+        const escaped = value
+            .replace(/\\/g, "\\\\")
+            .replace(/"/g, '\\"')
+            .replace(/\n/g, "\\n")
+            .replace(/\r/g, "\\r")
+            .replace(/\t/g, "\\t");
+        return `"${escaped}"`;
+    }
+    if (typeof value === "number") {
+        return Number.isFinite(value) ? String(value) : "nil";
+    }
+    if (typeof value === "boolean")
+        return value ? "true" : "false";
+    if (Array.isArray(value))
+        return `{${value.map(toLuaLiteral).join(", ")}}`;
+    if (typeof value === "object") {
+        const pairs = Object.entries(value)
+            .map(([k, v]) => `[${toLuaLiteral(k)}]=${toLuaLiteral(v)}`)
+            .join(", ");
+        return `{${pairs}}`;
+    }
+    return "nil";
+}
+async function evalLuaScript(script) {
+    const headers = {
+        "Content-Type": "text/plain",
+        "X-Timeout": "60",
+    };
+    if (SB_AUTH_TOKEN)
+        headers["Authorization"] = `Bearer ${SB_AUTH_TOKEN}`;
+    const res = await fetch(`${SB_URL}/.runtime/lua_script`, {
+        method: "POST",
+        headers,
+        body: script,
+    });
+    const text = await res.text();
+    let json = {};
+    try {
+        json = JSON.parse(text);
+    }
+    catch {
+        // invalid json or not json (e.g. "Unauthorized" from SB)
+    }
+    if (!res.ok || json.error) {
+        throw new Error(json.error ??
+            `SilverBullet returned HTTP ${res.status} ${res.statusText}`);
+    }
+    return json.result;
+}
+// Current tool set, refreshed at startup and via the refresh_tools tool.
+let toolList = [];
+async function refreshTools() {
+    const tools = (await evalLuaScript(`return system.invokeFunction("silverbullet-ai.listTools")`));
+    toolList = tools.filter((t) => t.source !== "mcp" && !EXCLUDED_TOOLS.has(t.name));
+    return toolList.length;
+}
+const REFRESH_TOOL = {
+    name: "refresh_tools",
+    description: "Reload the list of available tools from SilverBullet. Use this if tools were added or changed since this server started.",
+    inputSchema: { type: "object", properties: {} },
+    annotations: { readOnlyHint: true },
+};
+const errText = (e) => e instanceof Error ? e.message : String(e);
+const ok = (text) => ({
+    content: [{ type: "text", text }],
+    isError: false,
+});
+const err = (text) => ({
+    content: [{ type: "text", text }],
+    isError: true,
+});
+function buildServer() {
+    const server = new Server({ name: "silverbullet-ai-mcp", version: "0.0.0" }, { capabilities: { tools: { listChanged: true } } });
+    server.setRequestHandler(ListToolsRequestSchema, () => ({
+        tools: [
+            REFRESH_TOOL,
+            ...toolList.map((t) => ({
+                name: t.name,
+                description: t.description,
+                inputSchema: t.parameters,
+                annotations: { readOnlyHint: t.readOnly === true },
+            })),
+        ],
+    }));
+    server.setRequestHandler(CallToolRequestSchema, async (request) => {
+        const { name, arguments: args } = request.params;
+        if (name === REFRESH_TOOL.name) {
+            try {
+                const count = await refreshTools();
+                await server.sendToolListChanged();
+                return ok(`Reloaded ${count} tools.`);
+            }
+            catch (e) {
+                return err(errText(e));
+            }
+        }
+        if (!toolList.some((t) => t.name === name)) {
+            return err(`Unknown tool: ${name}`);
+        }
+        try {
+            const r = (await evalLuaScript(`return system.invokeFunction("silverbullet-ai.callTool", ${toLuaLiteral(name)}, ${toLuaLiteral(args ?? {})})`));
+            return r.success
+                ? ok(r.result ?? "")
+                : err(r.error ?? "Tool call failed");
+        }
+        catch (e) {
+            return err(errText(e));
+        }
+    });
+    return server;
+}
+try {
+    await refreshTools();
+}
+catch (e) {
+    console.error(`Failed to load tools from SilverBullet: ${errText(e)}`);
+    process.exit(1);
+}
+console.error(`silverbullet-ai-mcp: exposing ${toolList.length} tools: ${toolList
+    .map((t) => t.name)
+    .join(", ")}`);
+if (MCP_PORT) {
+    const httpServer = createHttpServer(async (req, res) => {
+        if (MCP_TOKEN && req.headers.authorization !== `Bearer ${MCP_TOKEN}`) {
+            res.writeHead(401).end();
+            return;
+        }
+        const server = buildServer();
+        const transport = new StreamableHTTPServerTransport({
+            sessionIdGenerator: undefined,
+        });
+        res.on("close", () => {
+            transport.close();
+            server.close();
+        });
+        try {
+            await server.connect(transport);
+            await transport.handleRequest(req, res);
+        }
+        catch (e) {
+            console.error("Error handling request:", e);
+            if (!res.headersSent)
+                res.writeHead(500).end();
+        }
+    });
+    httpServer.listen(Number(MCP_PORT), () => {
+        console.error(`silverbullet-ai-mcp listening on port ${MCP_PORT}`);
+    });
+}
+else {
+    await buildServer().connect(new StdioServerTransport());
+}

package/package.json

@@ -0,0 +1,27 @@
+{
+  "name": "silverbullet-ai-mcp",
+  "version": "0.0.0",
+  "description": "MCP server bridge exposing silverbullet-ai tools from a SilverBullet server",
+  "license": "AGPL-3.0-only",
+  "type": "module",
+  "bin": {
+    "silverbullet-ai-mcp": "dist/index.js"
+  },
+  "files": [
+    "dist"
+  ],
+  "engines": {
+    "node": ">=20"
+  },
+  "scripts": {
+    "build": "tsc",
+    "prepublishOnly": "npm run build"
+  },
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.29.0"
+  },
+  "devDependencies": {
+    "@types/node": "^24.0.0",
+    "typescript": "^5.7.0"
+  }
+}