@ksm0709/context 0.0.26 → 0.0.28

package/README.md

@@ -14,7 +14,7 @@ This plugin automatically registers an MCP (Model Context Protocol) server named
 
 ## OMX Support
 
-This plugin supports OMX (OpenCode Managed eXtension). See [[docs/omx-setup.md]] for setup instructions.
+This plugin supports OMX (OpenCode Managed eXtension). See [docs/omx-setup.md](docs/omx-setup.md) for setup instructions.
 
 ## Installation
 

package/dist/cli/index.js

@@ -27,7 +27,7 @@ function resolveContextDir(projectDir) {
 // package.json
 var package_default = {
   name: "@ksm0709/context",
-  version: "0.0.26",
+  version: "0.0.28",
   author: {
     name: "TaehoKang",
     email: "ksm07091@gmail.com"
@@ -69,7 +69,7 @@ var package_default = {
     "@opencode-ai/plugin": ">=1.0.0"
   },
   dependencies: {
-    "@ksm0709/context": "^0.0.22",
+    "@ksm0709/context": "^0.0.27",
     "@modelcontextprotocol/sdk": "^1.27.1",
     "jsonc-parser": "^3.0.0"
   },
@@ -577,16 +577,88 @@ function updateConfigPaths(contextDir) {
 }
 
 // src/cli/commands/install.ts
-import { join as join5, resolve as resolve3, dirname } from "path";
-import { existsSync as existsSync5, mkdirSync as mkdirSync2, copyFileSync } from "fs";
-import { fileURLToPath } from "url";
+import { join as join6, resolve as resolve4, dirname as dirname2 } from "path";
+import { existsSync as existsSync6, mkdirSync as mkdirSync3, copyFileSync } from "fs";
+import { fileURLToPath as fileURLToPath2 } from "url";
 import { createRequire } from "module";
+
+// src/omx/registry.ts
+import { fileURLToPath } from "url";
+import { dirname, join as join5, resolve as resolve3 } from "path";
+import { existsSync as existsSync5, readFileSync as readFileSync3, writeFileSync as writeFileSync3, mkdirSync as mkdirSync2 } from "fs";
+import { homedir as homedir2 } from "os";
+function resolveMcpPath() {
+  const currentFile = fileURLToPath(import.meta.url);
+  const currentDir = dirname(currentFile);
+  const distMcpPath = resolve3(currentDir, "..", "mcp.js");
+  if (existsSync5(distMcpPath)) {
+    return distMcpPath;
+  }
+  const srcMcpPath = resolve3(currentDir, "..", "mcp.ts");
+  if (existsSync5(srcMcpPath)) {
+    return srcMcpPath;
+  }
+  return distMcpPath;
+}
+function getRegistryPaths() {
+  return [
+    join5(homedir2(), ".omx", "mcp-registry.json"),
+    join5(homedir2(), ".omc", "mcp-registry.json")
+  ];
+}
+function ensureMcpRegistered(sdkLog) {
+  const registryPaths = getRegistryPaths();
+  let targetPath = registryPaths[0];
+  for (const p of registryPaths) {
+    if (existsSync5(p)) {
+      targetPath = p;
+      break;
+    }
+  }
+  let registry = {};
+  if (existsSync5(targetPath)) {
+    try {
+      const content = readFileSync3(targetPath, "utf-8");
+      registry = JSON.parse(content);
+    } catch (e) {
+      if (sdkLog) {
+        sdkLog(`[ERROR] Failed to parse MCP registry at ${targetPath}: ${e instanceof Error ? e.message : String(e)}`);
+      }
+      registry = {};
+    }
+  }
+  const mcpPath = resolveMcpPath();
+  const expectedConfig = {
+    command: "bun",
+    args: [mcpPath]
+  };
+  const currentConfig = registry["context-mcp"];
+  if (!currentConfig || currentConfig.command !== expectedConfig.command || !Array.isArray(currentConfig.args) || currentConfig.args[0] !== expectedConfig.args[0]) {
+    registry["context-mcp"] = expectedConfig;
+    try {
+      mkdirSync2(dirname(targetPath), { recursive: true });
+      writeFileSync3(targetPath, JSON.stringify(registry, null, 2), "utf-8");
+      if (sdkLog) {
+        sdkLog(`[INFO] Registered context-mcp in ${targetPath}`);
+      }
+      return true;
+    } catch (e) {
+      if (sdkLog) {
+        sdkLog(`[ERROR] Failed to write MCP registry to ${targetPath}: ${e instanceof Error ? e.message : String(e)}`);
+      }
+      return false;
+    }
+  }
+  return false;
+}
+
+// src/cli/commands/install.ts
 function resolveOmxSource() {
   try {
-    const cliDir = dirname(fileURLToPath(import.meta.url));
-    const pkgRoot = resolve3(cliDir, "..", "..");
-    const source = join5(pkgRoot, "dist", "omx", "index.mjs");
-    if (existsSync5(source))
+    const cliDir = dirname2(fileURLToPath2(import.meta.url));
+    const pkgRoot = resolve4(cliDir, "..", "..");
+    const source = join6(pkgRoot, "dist", "omx", "index.mjs");
+    if (existsSync6(source))
       return source;
   } catch {}
   try {
@@ -597,17 +669,21 @@ function resolveOmxSource() {
   }
 }
 function installOmx(projectDir, sourcePath) {
-  if (!existsSync5(sourcePath)) {
+  if (!existsSync6(sourcePath)) {
     process.stderr.write(`Could not find OMX plugin source file: ${sourcePath}
 `);
     process.exit(1);
     return;
   }
-  const targetDir = join5(projectDir, ".omx", "hooks");
-  mkdirSync2(targetDir, { recursive: true });
-  copyFileSync(sourcePath, join5(targetDir, "context.mjs"));
+  const targetDir = join6(projectDir, ".omx", "hooks");
+  mkdirSync3(targetDir, { recursive: true });
+  copyFileSync(sourcePath, join6(targetDir, "context.mjs"));
   process.stdout.write(`Installed context plugin to .omx/hooks/context.mjs
 `);
+  if (ensureMcpRegistered()) {
+    process.stdout.write(`Successfully registered context-mcp in ~/.omx/mcp-registry.json
+`);
+  }
 }
 function runInstall(args) {
   const [subcommand] = args;

package/dist/index.js

@@ -25,7 +25,7 @@ import { join as join2 } from "path";
 // package.json
 var package_default = {
   name: "@ksm0709/context",
-  version: "0.0.26",
+  version: "0.0.28",
   author: {
     name: "TaehoKang",
     email: "ksm07091@gmail.com"
@@ -67,7 +67,7 @@ var package_default = {
     "@opencode-ai/plugin": ">=1.0.0"
   },
   dependencies: {
-    "@ksm0709/context": "^0.0.22",
+    "@ksm0709/context": "^0.0.27",
     "@modelcontextprotocol/sdk": "^1.27.1",
     "jsonc-parser": "^3.0.0"
   },

package/dist/mcp.js

@@ -33087,6 +33087,24 @@ ${tags.map((t) => `  - ${t}`).join(`
       };
     }
   });
+  const listToolsMethod = ListToolsRequestSchema.shape.method.value;
+  const originalListToolsHandler = server.server._requestHandlers?.get(listToolsMethod);
+  if (originalListToolsHandler) {
+    server.server.setRequestHandler(ListToolsRequestSchema, async (request, extra) => {
+      const response = await originalListToolsHandler(request, extra);
+      if (response.tools) {
+        response.tools = response.tools.map((tool) => {
+          const newTool = { ...tool };
+          delete newTool.execution;
+          if (newTool.inputSchema && newTool.inputSchema.$schema) {
+            delete newTool.inputSchema.$schema;
+          }
+          return newTool;
+        });
+      }
+      return response;
+    });
+  }
   const transport = new StdioServerTransport;
   server.connect(transport);
   return server;

package/dist/omx/index.mjs

@@ -1,6 +1,6 @@
 // src/omx/index.ts
-import { existsSync as existsSync4, readFileSync as readFileSync4, unlinkSync } from "node:fs";
-import { join as join4 } from "node:path";
+import { existsSync as existsSync5, readFileSync as readFileSync5, unlinkSync } from "node:fs";
+import { join as join5 } from "node:path";
 
 // src/constants.ts
 var DEFAULTS = {
@@ -95,7 +95,7 @@ import { join as join3 } from "node:path";
 // package.json
 var package_default = {
   name: "@ksm0709/context",
-  version: "0.0.26",
+  version: "0.0.28",
   author: {
     name: "TaehoKang",
     email: "ksm07091@gmail.com"
@@ -137,7 +137,7 @@ var package_default = {
     "@opencode-ai/plugin": ">=1.0.0"
   },
   dependencies: {
-    "@ksm0709/context": "^0.0.22",
+    "@ksm0709/context": "^0.0.27",
     "@modelcontextprotocol/sdk": "^1.27.1",
     "jsonc-parser": "^3.0.0"
   },
@@ -545,6 +545,76 @@ function injectIntoAgentsMd(agentsMdPath, content) {
   writeFileAtomically(agentsMdPath, nextContent);
 }
 
+// src/omx/registry.ts
+import { fileURLToPath } from "node:url";
+import { dirname as dirname2, join as join4, resolve } from "node:path";
+import { existsSync as existsSync4, readFileSync as readFileSync4, writeFileSync as writeFileSync3, mkdirSync as mkdirSync3 } from "node:fs";
+import { homedir } from "node:os";
+function resolveMcpPath() {
+  const currentFile = fileURLToPath(import.meta.url);
+  const currentDir = dirname2(currentFile);
+  const distMcpPath = resolve(currentDir, "..", "mcp.js");
+  if (existsSync4(distMcpPath)) {
+    return distMcpPath;
+  }
+  const srcMcpPath = resolve(currentDir, "..", "mcp.ts");
+  if (existsSync4(srcMcpPath)) {
+    return srcMcpPath;
+  }
+  return distMcpPath;
+}
+function getRegistryPaths() {
+  return [
+    join4(homedir(), ".omx", "mcp-registry.json"),
+    join4(homedir(), ".omc", "mcp-registry.json")
+  ];
+}
+function ensureMcpRegistered(sdkLog) {
+  const registryPaths = getRegistryPaths();
+  let targetPath = registryPaths[0];
+  for (const p of registryPaths) {
+    if (existsSync4(p)) {
+      targetPath = p;
+      break;
+    }
+  }
+  let registry = {};
+  if (existsSync4(targetPath)) {
+    try {
+      const content = readFileSync4(targetPath, "utf-8");
+      registry = JSON.parse(content);
+    } catch (e) {
+      if (sdkLog) {
+        sdkLog(`[ERROR] Failed to parse MCP registry at ${targetPath}: ${e instanceof Error ? e.message : String(e)}`);
+      }
+      registry = {};
+    }
+  }
+  const mcpPath = resolveMcpPath();
+  const expectedConfig = {
+    command: "bun",
+    args: [mcpPath]
+  };
+  const currentConfig = registry["context-mcp"];
+  if (!currentConfig || currentConfig.command !== expectedConfig.command || !Array.isArray(currentConfig.args) || currentConfig.args[0] !== expectedConfig.args[0]) {
+    registry["context-mcp"] = expectedConfig;
+    try {
+      mkdirSync3(dirname2(targetPath), { recursive: true });
+      writeFileSync3(targetPath, JSON.stringify(registry, null, 2), "utf-8");
+      if (sdkLog) {
+        sdkLog(`[INFO] Registered context-mcp in ${targetPath}`);
+      }
+      return true;
+    } catch (e) {
+      if (sdkLog) {
+        sdkLog(`[ERROR] Failed to write MCP registry to ${targetPath}: ${e instanceof Error ? e.message : String(e)}`);
+      }
+      return false;
+    }
+  }
+  return false;
+}
+
 // src/omx/tmux-submit.ts
 import { spawnSync } from "node:child_process";
 function runTmux(args) {
@@ -561,7 +631,7 @@ function runTmux(args) {
   return { ok: true };
 }
 function sleep(ms) {
-  return new Promise((resolve) => globalThis.setTimeout(resolve, ms));
+  return new Promise((resolve2) => globalThis.setTimeout(resolve2, ms));
 }
 async function sendTmuxSubmitSequence(target, attempts = 3) {
   const totalAttempts = Math.max(1, Math.floor(attempts));
@@ -660,8 +730,21 @@ var STATIC_KNOWLEDGE_CONTEXT = `## Knowledge Context
 async function onSessionStart(event, sdk) {
   const projectDir = resolveProjectDir(event);
   scaffoldIfNeeded(projectDir);
-  injectIntoAgentsMd(join4(projectDir, "AGENTS.md"), STATIC_KNOWLEDGE_CONTEXT);
+  injectIntoAgentsMd(join5(projectDir, "AGENTS.md"), STATIC_KNOWLEDGE_CONTEXT);
   await sdk.log.info(`Injected context into AGENTS.md for ${projectDir}`);
+  const wasRegistered = ensureMcpRegistered(sdk.log.info);
+  if (wasRegistered) {
+    const warningMsg = "Context MCP was just added to your OMX registry. You must stop this session, run 'omx setup', and restart to use MCP tools.";
+    await sdk.log.info(warningMsg);
+    if (typeof sdk.tmux?.sendKeys === "function") {
+      const sessionName = typeof event.context?.session_name === "string" && event.context.session_name.trim().length > 0 ? event.context.session_name.trim() : undefined;
+      await sdk.tmux.sendKeys({
+        sessionName,
+        text: `# WARNING: ${warningMsg}`,
+        submit: true
+      });
+    }
+  }
 }
 async function onTurnComplete(event, sdk) {
   const projectDir = resolveProjectDir(event);
@@ -687,9 +770,9 @@ async function onTurnComplete(event, sdk) {
   }
   const followupScopeKey = resolveFollowupScopeKey(event);
   let pendingFollowupScopes = typeof sdk.state?.read === "function" ? await sdk.state.read(TURN_END_PENDING_SKIP_KEY, {}) ?? {} : {};
-  const workCompleteFile = join4(projectDir, DEFAULTS.workCompleteFile);
-  if (existsSync4(workCompleteFile)) {
-    const content = readFileSync4(workCompleteFile, "utf-8");
+  const workCompleteFile = join5(projectDir, DEFAULTS.workCompleteFile);
+  if (existsSync5(workCompleteFile)) {
+    const content = readFileSync5(workCompleteFile, "utf-8");
     const { sessionId: fileSessionId, turnId: fileTurnId } = parseWorkComplete(content);
     const currentScopeId = event.session_id ?? event.thread_id ?? "";
     if (!fileSessionId || fileSessionId === currentScopeId) {
@@ -755,7 +838,7 @@ async function onTurnComplete(event, sdk) {
 ${turnEnd}
 </system-reminder>`;
   const sessionName = typeof event.context?.session_name === "string" && event.context.session_name.trim().length > 0 ? event.context.session_name.trim() : undefined;
-  await new Promise((resolve) => globalThis.setTimeout(resolve, 500));
+  await new Promise((resolve2) => globalThis.setTimeout(resolve2, 500));
   if (typeof sdk.tmux?.sendKeys === "function") {}
   const result = await sdk.tmux.sendKeys({
     sessionName,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ksm0709/context",
-  "version": "0.0.26",
+  "version": "0.0.28",
   "author": {
     "name": "TaehoKang",
     "email": "ksm07091@gmail.com"
@@ -42,7 +42,7 @@
     "@opencode-ai/plugin": ">=1.0.0"
   },
   "dependencies": {
-    "@ksm0709/context": "^0.0.22",
+    "@ksm0709/context": "^0.0.27",
     "@modelcontextprotocol/sdk": "^1.27.1",
     "jsonc-parser": "^3.0.0"
   },