ocean-brain 0.1.6 → 0.2.1

package/dist/index.js

@@ -5,10 +5,21 @@ import { execSync } from "child_process";
 import fs from "fs";
 import os from "os";
 import path from "path";
-import { fileURLToPath } from "url";
+import { fileURLToPath, pathToFileURL } from "url";
 import { Command } from "commander";
 var __dirname = path.dirname(fileURLToPath(import.meta.url));
 var serverRoot = path.resolve(__dirname, "..", "server");
+function findBin(name, fromDir) {
+  let dir = path.resolve(fromDir, "..");
+  while (true) {
+    const bin = path.resolve(dir, "node_modules", ".bin", name);
+    if (fs.existsSync(bin)) return bin;
+    const parent = path.dirname(dir);
+    if (parent === dir) break;
+    dir = parent;
+  }
+  throw new Error(`Could not find "${name}" binary. Make sure it is installed.`);
+}
 var pkg = JSON.parse(
   fs.readFileSync(path.resolve(__dirname, "..", "package.json"), "utf-8")
 );
@@ -28,7 +39,7 @@ program.command("serve", { isDefault: true }).description("Start the Ocean Brain
   process.env.PORT = process.env.PORT || opts.port;
   process.env.HOST = process.env.HOST || opts.host;
   const schemaPath = path.resolve(serverRoot, "prisma/schema.prisma");
-  const prisma = path.resolve(__dirname, "..", "node_modules", ".bin", "prisma");
+  const prisma = findBin("prisma", __dirname);
   execSync(`"${prisma}" generate --schema="${schemaPath}"`, {
     stdio: "inherit",
     env: { ...process.env }
@@ -37,6 +48,11 @@ program.command("serve", { isDefault: true }).description("Start the Ocean Brain
     stdio: "inherit",
     env: { ...process.env }
   });
-  await import(path.resolve(serverRoot, "dist/main.js"));
+  const serverEntry = pathToFileURL(path.resolve(serverRoot, "dist/main.js")).href;
+  await import(serverEntry);
+});
+program.command("mcp").description("Start MCP server for AI integration").option("-s, --server <url>", "Ocean Brain server URL", "http://localhost:6683").action(async (opts) => {
+  const { startMcpServer } = await import("./mcp.js");
+  await startMcpServer(opts.server);
 });
 program.parse();

package/dist/mcp.js

@@ -0,0 +1,194 @@
+#!/usr/bin/env node
+
+// src/mcp.ts
+import fs from "fs";
+import path from "path";
+import { fileURLToPath } from "url";
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { z } from "zod";
+var __dirname = path.dirname(fileURLToPath(import.meta.url));
+var pkg = JSON.parse(
+  fs.readFileSync(path.resolve(__dirname, "..", "package.json"), "utf-8")
+);
+async function graphql(serverUrl, query, variables) {
+  const response = await fetch(`${serverUrl}/graphql`, {
+    method: "POST",
+    headers: { "Content-Type": "application/json" },
+    body: JSON.stringify({ query, variables })
+  });
+  if (!response.ok) {
+    throw new Error(`GraphQL request failed: ${response.status} ${response.statusText}`);
+  }
+  const result = await response.json();
+  if (result.errors?.length) {
+    throw new Error(`GraphQL error: ${result.errors[0].message}`);
+  }
+  return result.data;
+}
+async function startMcpServer(serverUrl) {
+  const server = new McpServer({
+    name: "ocean-brain",
+    version: pkg.version
+  });
+  server.tool(
+    "search_notes",
+    "Search notes by keyword. Returns a list of matching notes with title and tags. Use read_note to get full content of a specific note.",
+    {
+      query: z.string().describe("Search keyword"),
+      limit: z.number().optional().default(10).describe("Max results (default: 10)")
+    },
+    async ({ query, limit }) => {
+      const data = await graphql(serverUrl, `
+                query ($searchFilter: SearchFilterInput, $pagination: PaginationInput) {
+                    allNotes(searchFilter: $searchFilter, pagination: $pagination) {
+                        totalCount
+                        notes {
+                            id
+                            title
+                            updatedAt
+                            tags { id name }
+                            contentAsMarkdown
+                        }
+                    }
+                }
+            `, {
+        searchFilter: { query, sortBy: "updatedAt", sortOrder: "desc" },
+        pagination: { limit, offset: 0 }
+      });
+      const result = data?.allNotes;
+      const notes = result.notes.map((note) => ({
+        id: note.id,
+        title: note.title,
+        updatedAt: note.updatedAt,
+        tags: note.tags.map((t) => t.name),
+        preview: note.contentAsMarkdown.slice(0, 100)
+      }));
+      return {
+        content: [{
+          type: "text",
+          text: JSON.stringify({ totalCount: result.totalCount, notes }, null, 2)
+        }]
+      };
+    }
+  );
+  server.tool(
+    "read_note",
+    "Read content of a note by ID. Returns truncated by default (1000 chars). Set maxLength to 0 for full content only when necessary.",
+    {
+      id: z.string().describe("Note ID"),
+      maxLength: z.number().optional().default(1e3).describe("Max content length in characters. 0 for full content. (default: 1000)")
+    },
+    async ({ id, maxLength }) => {
+      const data = await graphql(serverUrl, `
+                query ($id: ID!) {
+                    note(id: $id) {
+                        id
+                        title
+                        contentAsMarkdown
+                        createdAt
+                        updatedAt
+                        tags { id name }
+                    }
+                }
+            `, { id });
+      const note = data?.note;
+      let markdown = note.contentAsMarkdown;
+      const totalLength = markdown.length;
+      let truncated = false;
+      if (maxLength > 0 && markdown.length > maxLength) {
+        markdown = markdown.slice(0, maxLength);
+        truncated = true;
+      }
+      const output = [
+        `# ${note.title}`,
+        "",
+        `Tags: ${note.tags.map((t) => t.name).join(", ") || "(none)"}`,
+        `Created: ${note.createdAt}`,
+        `Updated: ${note.updatedAt}`,
+        ...truncated ? [`Content: ${totalLength} chars (showing first ${maxLength})`] : [],
+        "",
+        markdown,
+        ...truncated ? ["\n... (truncated, use maxLength: 0 to read full content)"] : []
+      ].join("\n");
+      return {
+        content: [{
+          type: "text",
+          text: output
+        }]
+      };
+    }
+  );
+  server.tool(
+    "list_tags",
+    "List all tags with their note counts.",
+    {},
+    async () => {
+      const data = await graphql(serverUrl, `
+                query ($searchFilter: SearchFilterInput, $pagination: PaginationInput) {
+                    allTags(searchFilter: $searchFilter, pagination: $pagination) {
+                        totalCount
+                        tags {
+                            id
+                            name
+                            referenceCount
+                        }
+                    }
+                }
+            `, {
+        searchFilter: { query: "" },
+        pagination: { limit: 100, offset: 0 }
+      });
+      const result = data?.allTags;
+      return {
+        content: [{
+          type: "text",
+          text: JSON.stringify(result, null, 2)
+        }]
+      };
+    }
+  );
+  server.tool(
+    "list_recent_notes",
+    "List recently updated notes. Returns titles and tags only. Use read_note to get content of a specific note.",
+    {
+      limit: z.number().optional().default(10).describe("Max results (default: 10)")
+    },
+    async ({ limit }) => {
+      const data = await graphql(serverUrl, `
+                query ($searchFilter: SearchFilterInput, $pagination: PaginationInput) {
+                    allNotes(searchFilter: $searchFilter, pagination: $pagination) {
+                        totalCount
+                        notes {
+                            id
+                            title
+                            updatedAt
+                            tags { id name }
+                        }
+                    }
+                }
+            `, {
+        searchFilter: { query: "", sortBy: "updatedAt", sortOrder: "desc" },
+        pagination: { limit, offset: 0 }
+      });
+      const result = data?.allNotes;
+      const notes = result.notes.map((note) => ({
+        id: note.id,
+        title: note.title,
+        updatedAt: note.updatedAt,
+        tags: note.tags.map((t) => t.name)
+      }));
+      return {
+        content: [{
+          type: "text",
+          text: JSON.stringify({ totalCount: result.totalCount, notes }, null, 2)
+        }]
+      };
+    }
+  );
+  const transport = new StdioServerTransport();
+  await server.connect(transport);
+}
+export {
+  startMcpServer
+};

package/package.json

@@ -1,33 +1,43 @@
 {
-  "name": "ocean-brain",
-  "version": "0.1.6",
-  "type": "module",
-  "bin": {
-    "ocean-brain": "./dist/index.js"
-  },
-  "files": [
-    "dist",
-    "server/dist",
-    "server/prisma/schema.prisma",
-    "server/prisma/migrations",
-    "server/client/dist"
-  ],
-  "dependencies": {
-    "@graphql-tools/schema": "^10.0.31",
-    "@graphql-tools/utils": "^11.0.0",
-    "@prisma/client": "^6.19.2",
-    "commander": "^14.0.3",
-    "express": "^5.2.1",
-    "express-session": "^1.19.0",
-    "express-winston": "^4.2.0",
-    "graphql-http": "^1.22.4",
-    "prisma": "^6.19.2",
-    "sqlite3": "^5.1.7",
-    "winston": "^3.19.0"
-  },
-  "scripts": {
-    "dev": "tsx src/index.ts",
-    "build": "tsup",
-    "type-check": "tsc --noEmit"
-  }
-}
+    "name": "ocean-brain",
+    "version": "0.2.1",
+    "type": "module",
+    "repository": {
+        "type": "git",
+        "url": "https://github.com/baealex/ocean-brain"
+    },
+    "homepage": "https://github.com/baealex/ocean-brain",
+    "bugs": {
+        "url": "https://github.com/baealex/ocean-brain/issues"
+    },
+    "bin": {
+        "ocean-brain": "dist/index.js"
+    },
+    "files": [
+        "dist",
+        "server/dist",
+        "server/prisma/schema.prisma",
+        "server/prisma/migrations",
+        "server/client/dist"
+    ],
+    "scripts": {
+        "dev": "tsx src/index.ts",
+        "build": "tsup",
+        "type-check": "tsc --noEmit"
+    },
+    "dependencies": {
+        "@graphql-tools/schema": "^10.0.31",
+        "@graphql-tools/utils": "^11.0.0",
+        "@modelcontextprotocol/sdk": "^1.27.1",
+        "@prisma/client": "^6.19.2",
+        "commander": "^14.0.3",
+        "express": "^5.2.1",
+        "express-session": "^1.19.0",
+        "express-winston": "^4.2.0",
+        "graphql-http": "^1.22.4",
+        "prisma": "^6.19.2",
+        "sqlite3": "^5.1.7",
+        "winston": "^3.19.0",
+        "zod": "^3.25.0"
+    }
+}