opencode-heartbeat-approval 0.2.1 → 0.3.1

package/dist/index.js

@@ -10,6 +10,11 @@ var __export = (target, all) => {
     });
 };
 
+// src/plugin.ts
+import { readFile } from "fs/promises";
+import { join } from "path";
+import { homedir } from "os";
+
 // node_modules/zod/v4/classic/external.js
 var exports_external = {};
 __export(exports_external, {
@@ -12331,23 +12336,26 @@ function tool(input) {
 }
 tool.schema = exports_external;
 // src/plugin.ts
-var DEFAULT_RUNNER_URL = "http://127.0.0.1:3210";
 var POLL_INTERVAL_MS = 1e4;
-function getRunnerUrl(override) {
-  return override ?? process.env.HEARTBEAT_RUNNER_URL ?? DEFAULT_RUNNER_URL;
-}
-function buildHeaders() {
-  const headers = { "Content-Type": "application/json" };
-  const token = process.env.HEARTBEAT_AUTH_TOKEN;
-  if (token) {
-    headers["Authorization"] = `Bearer ${token}`;
+var JSON_HEADERS = { "Content-Type": "application/json" };
+var SESSION_MAP_PATH = join(homedir(), ".config", "opencode", "heartbeat", "session-map.json");
+async function lookupSessionMapping(sessionId) {
+  try {
+    const text = await readFile(SESSION_MAP_PATH, "utf-8");
+    const mapping = JSON.parse(text);
+    const entry = mapping[sessionId];
+    if (entry && typeof entry.port === "number" && entry.port > 0 && entry.port < 65536) {
+      return `http://127.0.0.1:${entry.port}`;
+    }
+    return null;
+  } catch {
+    return null;
   }
-  return headers;
 }
 async function createGate(runnerUrl, params) {
   const resp = await fetch(`${runnerUrl}/api/approval`, {
     method: "POST",
-    headers: buildHeaders(),
+    headers: JSON_HEADERS,
     body: JSON.stringify(params)
   });
   if (resp.status === 409) {
@@ -12362,7 +12370,7 @@ async function createGate(runnerUrl, params) {
 async function pollGate(runnerUrl) {
   const resp = await fetch(`${runnerUrl}/api/approval`, {
     method: "GET",
-    headers: buildHeaders()
+    headers: JSON_HEADERS
   });
   if (!resp.ok) {
     const text = await resp.text().catch(() => "");
@@ -12398,8 +12406,7 @@ Returns: { status: "approved" | "rejected" | "expired" | "unavailable" | "cancel
           prompt: tool.schema.string().describe("What you need from the human \u2014 be specific and concise"),
           type: tool.schema.string().optional().describe("Type of request: 'approval' (need permission) or 'assistance' (need human help). Default: 'approval'"),
           artifacts: tool.schema.string().optional().describe("Comma-separated list of relevant file paths or artifact names"),
-          deadline_hours: tool.schema.number().optional().describe("Hours before the request expires (default: 24, max: 168)"),
-          runner_url: tool.schema.string().optional().describe("URL of this role's heartbeat runner (e.g., http://127.0.0.1:3211). Required for multi-role setups.")
+          deadline_hours: tool.schema.number().optional().describe("Hours before the request expires (default: 24, max: 168)")
         },
         async execute(args, ctx) {
           const artifacts = args.artifacts ? args.artifacts.split(",").map((s) => s.trim()).filter(Boolean) : undefined;
@@ -12413,7 +12420,21 @@ Returns: { status: "approved" | "rejected" | "expired" | "unavailable" | "cancel
             });
           }
           const requestType = args.type === "assistance" ? "assistance" : "approval";
-          const resolvedUrl = getRunnerUrl(args.runner_url);
+          const envPort = process.env.APPROVAL_PORT;
+          let resolvedUrl = null;
+          if (envPort && /^\d+$/.test(envPort)) {
+            resolvedUrl = `http://127.0.0.1:${envPort}`;
+          } else {
+            resolvedUrl = await lookupSessionMapping(ctx.sessionID);
+          }
+          if (!resolvedUrl) {
+            return JSON.stringify({
+              status: "unavailable",
+              type: requestType,
+              error: `No approval server found for session ${ctx.sessionID}. Ensure heartbeat runner is running for this role.`,
+              approval_id: null
+            });
+          }
           let gateResult;
           try {
             gateResult = await createGate(resolvedUrl, {

package/dist/plugin.js

@@ -10,6 +10,11 @@ var __export = (target, all) => {
     });
 };
 
+// src/plugin.ts
+import { readFile } from "fs/promises";
+import { join } from "path";
+import { homedir } from "os";
+
 // node_modules/zod/v4/classic/external.js
 var exports_external = {};
 __export(exports_external, {
@@ -12331,23 +12336,26 @@ function tool(input) {
 }
 tool.schema = exports_external;
 // src/plugin.ts
-var DEFAULT_RUNNER_URL = "http://127.0.0.1:3210";
 var POLL_INTERVAL_MS = 1e4;
-function getRunnerUrl(override) {
-  return override ?? process.env.HEARTBEAT_RUNNER_URL ?? DEFAULT_RUNNER_URL;
-}
-function buildHeaders() {
-  const headers = { "Content-Type": "application/json" };
-  const token = process.env.HEARTBEAT_AUTH_TOKEN;
-  if (token) {
-    headers["Authorization"] = `Bearer ${token}`;
+var JSON_HEADERS = { "Content-Type": "application/json" };
+var SESSION_MAP_PATH = join(homedir(), ".config", "opencode", "heartbeat", "session-map.json");
+async function lookupSessionMapping(sessionId) {
+  try {
+    const text = await readFile(SESSION_MAP_PATH, "utf-8");
+    const mapping = JSON.parse(text);
+    const entry = mapping[sessionId];
+    if (entry && typeof entry.port === "number" && entry.port > 0 && entry.port < 65536) {
+      return `http://127.0.0.1:${entry.port}`;
+    }
+    return null;
+  } catch {
+    return null;
   }
-  return headers;
 }
 async function createGate(runnerUrl, params) {
   const resp = await fetch(`${runnerUrl}/api/approval`, {
     method: "POST",
-    headers: buildHeaders(),
+    headers: JSON_HEADERS,
     body: JSON.stringify(params)
   });
   if (resp.status === 409) {
@@ -12362,7 +12370,7 @@ async function createGate(runnerUrl, params) {
 async function pollGate(runnerUrl) {
   const resp = await fetch(`${runnerUrl}/api/approval`, {
     method: "GET",
-    headers: buildHeaders()
+    headers: JSON_HEADERS
   });
   if (!resp.ok) {
     const text = await resp.text().catch(() => "");
@@ -12398,8 +12406,7 @@ Returns: { status: "approved" | "rejected" | "expired" | "unavailable" | "cancel
           prompt: tool.schema.string().describe("What you need from the human \u2014 be specific and concise"),
           type: tool.schema.string().optional().describe("Type of request: 'approval' (need permission) or 'assistance' (need human help). Default: 'approval'"),
           artifacts: tool.schema.string().optional().describe("Comma-separated list of relevant file paths or artifact names"),
-          deadline_hours: tool.schema.number().optional().describe("Hours before the request expires (default: 24, max: 168)"),
-          runner_url: tool.schema.string().optional().describe("URL of this role's heartbeat runner (e.g., http://127.0.0.1:3211). Required for multi-role setups.")
+          deadline_hours: tool.schema.number().optional().describe("Hours before the request expires (default: 24, max: 168)")
         },
         async execute(args, ctx) {
           const artifacts = args.artifacts ? args.artifacts.split(",").map((s) => s.trim()).filter(Boolean) : undefined;
@@ -12413,7 +12420,21 @@ Returns: { status: "approved" | "rejected" | "expired" | "unavailable" | "cancel
             });
           }
           const requestType = args.type === "assistance" ? "assistance" : "approval";
-          const resolvedUrl = getRunnerUrl(args.runner_url);
+          const envPort = process.env.APPROVAL_PORT;
+          let resolvedUrl = null;
+          if (envPort && /^\d+$/.test(envPort)) {
+            resolvedUrl = `http://127.0.0.1:${envPort}`;
+          } else {
+            resolvedUrl = await lookupSessionMapping(ctx.sessionID);
+          }
+          if (!resolvedUrl) {
+            return JSON.stringify({
+              status: "unavailable",
+              type: requestType,
+              error: `No approval server found for session ${ctx.sessionID}. Ensure heartbeat runner is running for this role.`,
+              approval_id: null
+            });
+          }
           let gateResult;
           try {
             gateResult = await createGate(resolvedUrl, {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-heartbeat-approval",
-  "version": "0.2.1",
+  "version": "0.3.1",
   "description": "OpenCode plugin providing request_human_input MCP tool for Heartbeat pipeline (approval + assistance)",
   "type": "module",
   "main": "dist/index.js",

package/src/plugin.ts

@@ -1,8 +1,11 @@
+import { readFile } from "node:fs/promises";
+import { join } from "node:path";
+import { homedir } from "node:os";
 import { tool } from "@opencode-ai/plugin";
 import type { PluginInput } from "@opencode-ai/plugin";
 
-const DEFAULT_RUNNER_URL = "http://127.0.0.1:3210";
 const POLL_INTERVAL_MS = 10_000;
+const JSON_HEADERS: Record<string, string> = { "Content-Type": "application/json" };
 
 interface CreateResponse {
   approval_id: string;
@@ -18,25 +21,28 @@ interface PollResponse {
   type?: string;
 }
 
-function getRunnerUrl(override?: string): string {
-  return override ?? process.env.HEARTBEAT_RUNNER_URL ?? DEFAULT_RUNNER_URL;
-}
-
-function buildHeaders(): Record<string, string> {
-  const headers: Record<string, string> = { "Content-Type": "application/json" };
-  const token = process.env.HEARTBEAT_AUTH_TOKEN;
-  if (token) {
-    headers["Authorization"] = `Bearer ${token}`;
-  }
-  return headers;
-}
-
 interface ConflictResponse {
   error: string;
   message: string;
   existing_approval_id: string;
 }
 
+const SESSION_MAP_PATH = join(homedir(), ".config", "opencode", "heartbeat", "session-map.json");
+
+async function lookupSessionMapping(sessionId: string): Promise<string | null> {
+  try {
+    const text = await readFile(SESSION_MAP_PATH, "utf-8");
+    const mapping = JSON.parse(text) as Record<string, { port: number; role: string; created_at: string }>;
+    const entry = mapping[sessionId];
+    if (entry && typeof entry.port === "number" && entry.port > 0 && entry.port < 65536) {
+      return `http://127.0.0.1:${entry.port}`;
+    }
+    return null;
+  } catch {
+    return null;
+  }
+}
+
 async function createGate(runnerUrl: string, params: {
   prompt: string;
   artifacts?: string[];
@@ -46,7 +52,7 @@ async function createGate(runnerUrl: string, params: {
 }): Promise<CreateResponse | ConflictResponse> {
   const resp = await fetch(`${runnerUrl}/api/approval`, {
     method: "POST",
-    headers: buildHeaders(),
+    headers: JSON_HEADERS,
     body: JSON.stringify(params),
   });
   if (resp.status === 409) {
@@ -62,7 +68,7 @@ async function createGate(runnerUrl: string, params: {
 async function pollGate(runnerUrl: string): Promise<PollResponse> {
   const resp = await fetch(`${runnerUrl}/api/approval`, {
     method: "GET",
-    headers: buildHeaders(),
+    headers: JSON_HEADERS,
   });
   if (!resp.ok) {
     const text = await resp.text().catch(() => "");
@@ -106,10 +112,6 @@ Returns: { status: "approved" | "rejected" | "expired" | "unavailable" | "cancel
             .number()
             .optional()
             .describe("Hours before the request expires (default: 24, max: 168)"),
-          runner_url: tool.schema
-            .string()
-            .optional()
-            .describe("URL of this role's heartbeat runner (e.g., http://127.0.0.1:3211). Required for multi-role setups."),
         },
         async execute(args, ctx) {
           const artifacts = args.artifacts
@@ -125,7 +127,22 @@ Returns: { status: "approved" | "rejected" | "expired" | "unavailable" | "cancel
             });
           }
           const requestType = (args.type === "assistance" ? "assistance" : "approval") as "approval" | "assistance";
-          const resolvedUrl = getRunnerUrl(args.runner_url);
+
+          const envPort = process.env.APPROVAL_PORT;
+          let resolvedUrl: string | null = null;
+          if (envPort && /^\d+$/.test(envPort)) {
+            resolvedUrl = `http://127.0.0.1:${envPort}`;
+          } else {
+            resolvedUrl = await lookupSessionMapping(ctx.sessionID);
+          }
+          if (!resolvedUrl) {
+            return JSON.stringify({
+              status: "unavailable",
+              type: requestType,
+              error: `No approval server found for session ${ctx.sessionID}. Ensure heartbeat runner is running for this role.`,
+              approval_id: null,
+            });
+          }
 
           let gateResult: CreateResponse | ConflictResponse;
           try {