@starlove/orbit 1.0.7 → 1.1.0

package/src/mcp-server.js

@@ -5,6 +5,46 @@
 // "active board" state.
 
 import { createOrbitClient } from "./mcp/orbit-client.js";
+import { existsSync, mkdirSync } from "node:fs";
+import { dirname, join, resolve } from "node:path";
+import {
+  archivedTicketsForBoard,
+  archiveTicket,
+  checkpointTicket,
+  claimNext,
+  completeTicket,
+  createBoardEntry,
+  createComment,
+  createTicket,
+  deleteTicket,
+  exportBoard,
+  getBoardContext,
+  getContextPack,
+  readComments,
+  readTicket,
+  getTicketBlockers,
+  getTicketRelations,
+  localAgentActor,
+  restoreTicket,
+  searchTickets,
+  updateBoard,
+  updateTicket
+} from "./core/board.js";
+import { closeAllConnections, createBoardSchema, openConnection } from "./core/db.js";
+import {
+  getBoardByRegistryId,
+  getBoardBySlug,
+  getBoardByRepoPath,
+  insertBoard,
+  listBoards,
+  openBoardDb,
+  syncRegistryFromBoardDb,
+  touchBoardActive
+} from "./core/registry.js";
+import { scheduleAutomaticBoardBackup } from "./core/backups.js";
+import { provisionRepoBoard } from "./core/provision-repo-board.js";
+import { seedIfEmpty } from "./core/seed.js";
+import { now, normalizePath } from "./core/util.js";
 
 const SERVER_INFO = {
   name: "minimal-agent-board",
@@ -12,6 +52,133 @@ const SERVER_INFO = {
 };
 
 const orbitClient = await createOrbitClient();
+// Per-process session board. Set on init by walking up from cwd, mutable via
+// `board_set_active`. NEVER read by anything outside this file — every
+// downstream call receives an explicit ctx.
+let sessionBoardRow = null;
+
+function setSessionBoard(row) {
+  sessionBoardRow = row;
+  if (row) touchBoardActive(row.id);
+}
+
+function getSessionBoard() {
+  return sessionBoardRow;
+}
+
+/**
+ * Walk up from `startDir` toward the filesystem root, returning the first
+ * directory that satisfies `predicate`. Returns null if none found.
+ */
+function walkUp(startDir, predicate) {
+  let dir = startDir;
+  while (true) {
+    if (predicate(dir)) return dir;
+    const parent = dirname(dir);
+    if (parent === dir) return null;
+    dir = parent;
+  }
+}
+
+/**
+ * Initialise the MCP session board. Resolution order:
+ *   1. Walk up from PROJECT_ROOT (cwd unless `orbit mcp --cwd` or env overrides it)
+ *      for an existing `.orbit/board.db` — finds the board even when the agent
+ *      is launched from a subdirectory or MCP config uses an absolute command.
+ *   2. Walk up from PROJECT_ROOT for a `.git/` directory — auto-creates a board at
+ *      the git root on first launch.
+ *   3. Fall back to PROJECT_ROOT itself — supports non-git projects and global installs.
+ */
+function initMcpSessionBoard() {
+  const start = normalizePath(process.env.PROJECT_ROOT ? resolve(process.env.PROJECT_ROOT) : process.cwd());
+
+  // 1. Prefer the registry row for PROJECT_ROOT (or one of its ancestors). New
+  // Orbit boards live in central DATA_DIR storage, so there may be no in-repo
+  // .orbit/board.db to discover.
+  const registeredRoot = walkUp(start, (dir) => Boolean(getBoardByRepoPath(normalizePath(dir))));
+  if (registeredRoot) {
+    setSessionBoard(getBoardByRepoPath(normalizePath(registeredRoot)));
+    return;
+  }
+
+  // 2. Walk up for an existing legacy board db.
+  const boardRoot = walkUp(start, (dir) => existsSync(join(dir, ".orbit", "board.db")));
+  if (boardRoot) {
+    const root = normalizePath(boardRoot);
+    const existing = getBoardByRepoPath(root);
+    if (existing) { setSessionBoard(existing); return; }
+    // db exists on disk but isn't in the registry yet — register it.
+    const dbPath = join(boardRoot, ".orbit", "board.db");
+    const db = openConnection(dbPath);
+    createBoardSchema(db);
+    const seeded = seedIfEmpty(db, root);
+    if (seeded) {
+      const t = now();
+      insertBoard({ id: seeded.id, slug: seeded.slug, name: seeded.name, repo_path: root, db_path: dbPath, repo_url: seeded.repo_url || "", default_branch: seeded.default_branch || "main", last_active_at: t, created_at: t, updated_at: t });
+    }
+    const fresh = getBoardByRepoPath(root);
+    if (fresh) { syncRegistryFromBoardDb(fresh, db); setSessionBoard(fresh); }
+    return;
+  }
+
+  // 3. Walk up for a git root; fall back to cwd for non-git projects.
+  const gitRoot = walkUp(start, (dir) => existsSync(join(dir, ".git")));
+  const result = provisionRepoBoard(normalizePath(gitRoot || start));
+  if (result.registryRow) setSessionBoard(result.registryRow);
+}
+
+function createBoardAtRoot(root) {
+  const dbPath = join(root, ".orbit", "board.db");
+  mkdirSync(dirname(dbPath), { recursive: true });
+  const db = openConnection(dbPath);
+  createBoardSchema(db);
+  const seeded = seedIfEmpty(db, root);
+  if (seeded) {
+    const t = now();
+    insertBoard({ id: seeded.id, slug: seeded.slug, name: seeded.name, repo_path: root, db_path: dbPath, repo_url: seeded.repo_url || "", default_branch: seeded.default_branch || "main", last_active_at: t, created_at: t, updated_at: t });
+  }
+  const fresh = getBoardByRepoPath(root);
+  if (fresh) { syncRegistryFromBoardDb(fresh, db); setSessionBoard(fresh); }
+}
+
+initMcpSessionBoard();
+closeAllConnections();
+
+/** Build a ctx for the session board (or a board the tool argument selected). */
+function ctxFor(boardRow, actor) {
+  if (!boardRow) throw rpcError(-32602, "No board in session; pass board_id or board_slug, or call board_set_active first.");
+  return { actor, board: boardRow, db: openBoardDb(boardRow) };
+}
+
+function sessionCtx(actor) {
+  return ctxFor(getSessionBoard(), actor);
+}
+
+function withAutomaticBackup(ctx, result) {
+  scheduleAutomaticBoardBackup(ctx.board, ctx.db);
+  return result;
+}
+
+/** Resolve a registry row from explicit args (no session fallback). */
+function resolveExplicitBoard(args) {
+  if (args.board_id) {
+    const row = getBoardByRegistryId(args.board_id);
+    if (!row) throw rpcError(-32004, "Board id not found.");
+    return row;
+  }
+  if (args.board_slug || args.board) {
+    const row = getBoardBySlug(args.board_slug || args.board);
+    if (!row) throw rpcError(-32004, "Board slug not found.");
+    return row;
+  }
+  return null;
+}
+
+/** Resolve a board for a tool that accepts board_id/board_slug AND can fall
+ *  back to the session board. */
+function resolveBoardOrSession(args) {
+  return resolveExplicitBoard(args) || getSessionBoard();
+}
 
 const TOOL_DEFS = [
   {
@@ -75,7 +242,7 @@ const TOOL_DEFS = [
   {
     name: "board_get_ticket_context",
     description:
-      "Read the full context pack for a ticket, including board agent_instructions (project-level agent context), relations, blockers, comments, and child cards. Heavy — prefer board_read_ticket when you only need title/description/comments.",
+      "Read the default agent context pack for a ticket, including board agent_instructions (project-level agent context), relations, blockers, implementation fields, and child cards. Ticket comments are intentionally omitted; use board_read_comments for explicit comment retrieval.",
     inputSchema: {
       type: "object",
       properties: {
@@ -90,7 +257,7 @@ const TOOL_DEFS = [
   {
     name: "board_read_ticket",
     description:
-      "Look up a ticket by ticket_id, number, or exact title (case-insensitive) on the session's active board, and return its title, description, labels, state, and full comment thread plus the board manual (agent_instructions + journal + deployment paths).",
+      "Look up a ticket by ticket_id, number, or exact title (case-insensitive) on the session's active board, and return its title, description, labels, state, and the board manual (agent_instructions + journal + deployment paths). Ticket comments are intentionally omitted; use board_read_comments for explicit comment retrieval.",
     inputSchema: {
       type: "object",
       properties: {
@@ -332,7 +499,9 @@ const TOOL_DEFS = [
       type: "object",
       properties: {
         board_id: { type: "string" },
-        board_slug: { type: "string" }
+        board_slug: { type: "string" },
+        include_attachments: { type: "boolean", description: "Embed attached image bytes in the JSON snapshot." },
+        include_images: { type: "boolean", description: "Alias for include_attachments." }
       },
       additionalProperties: false
     },
@@ -341,7 +510,7 @@ const TOOL_DEFS = [
   {
     name: "board_update_settings",
     description:
-      "PATCH-equivalent for board settings: name, repo_url, system_path, default_branch, agent_instructions (project-level agent context), and project_notes (Notes For You). Same fields as PATCH /api/boards/:id.",
+      "PATCH-equivalent for board settings: name (display rename only; slug/canonical URLs stay unchanged), repo_url, system_path, default_branch, agent_instructions (project-level agent context), and project_notes (Notes For You).",
     inputSchema: {
       type: "object",
       properties: {
@@ -364,13 +533,21 @@ const tools = new Map(TOOL_DEFS.map((tool) => [tool.name, tool]));
 
 let inputBuffer = Buffer.alloc(0);
 let expectedBodyLength = -1;
+let shuttingDown = false;
+
+function shutdown(code = 0) {
+  if (shuttingDown) return;
+  shuttingDown = true;
+  orbitClient.close?.();
+  process.exit(code);
+}
 
 process.stdin.on("data", (chunk) => {
   inputBuffer = Buffer.concat([inputBuffer, chunk]);
   drainInput();
 });
 
-process.stdin.on("end", () => process.exit(0));
+process.stdin.on("end", () => shutdown(0));
 process.stdin.resume();
 
 function drainInput() {
@@ -449,6 +626,8 @@ async function handleMessage(body, transport = "content-length") {
     }, transport);
   } catch (error) {
     writeMessage(errorResponse(message.id, error.code || -32000, error.message || "Internal error."), transport);
+  } finally {
+    orbitClient.close?.();
   }
 }
 
@@ -544,6 +723,11 @@ process.on("unhandledRejection", (error) => {
   writeMessage(errorResponse(null, -32000, error?.message || "Unhandled rejection."));
 });
 
+process.once("SIGINT", () => shutdown(0));
+process.once("SIGTERM", () => shutdown(0));
+process.once("exit", () => orbitClient.close?.());
+
 if (process.env.MAB_MCP_STDERR_LOG === "1") {
-  console.error(`Starscape Orbit MCP ready. Mode: ${orbitClient.mode}. Target: ${typeof orbitClient.sessionLabel === "function" ? orbitClient.sessionLabel() : orbitClient.sessionLabel}`);
+  const target = typeof orbitClient.sessionLabel === "function" ? orbitClient.sessionLabel() : orbitClient.sessionLabel;
+  console.error(`Starscape Orbit MCP ready. Mode: ${orbitClient.mode}. Target: ${target}`);
 }

package/src/server.js

@@ -8,6 +8,7 @@ import {
   startSSEStream,
   startupSummary
 } from "./core/board.js";
+import { closeAllConnections } from "./core/db.js";
 
 const server = createServer(async (req, res) => {
   try {
@@ -46,3 +47,22 @@ server.listen(boardRuntime.port, () => {
   console.log(`Boards (${summary.boardCount}):`);
   for (const path of summary.boardPaths) console.log(`  - ${path}`);
 });
+
+let shuttingDown = false;
+
+function shutdown() {
+  if (shuttingDown) return;
+  shuttingDown = true;
+  server.close(() => {
+    closeAllConnections();
+    process.exit(0);
+  });
+  setTimeout(() => {
+    closeAllConnections();
+    process.exit(0);
+  }, 1500).unref();
+}
+
+process.once("SIGINT", shutdown);
+process.once("SIGTERM", shutdown);
+process.once("exit", () => closeAllConnections());