@textcortex/zenocode 0.1.2 → 0.1.3

package/scripts/build-branded-opencode.test.mjs

@@ -0,0 +1,25 @@
+import assert from "node:assert/strict";
+import test from "node:test";
+import {
+  buildWrapperBinMap,
+  mapBrandedBinaryPackageName,
+} from "./build-branded-opencode.mjs";
+
+test("mapBrandedBinaryPackageName scopes runtime binaries under zenocode", () => {
+  assert.equal(
+    mapBrandedBinaryPackageName("opencode-darwin-arm64", "@textcortex/zenocode-ai"),
+    "@textcortex/zenocode-darwin-arm64",
+  );
+  assert.equal(
+    mapBrandedBinaryPackageName("opencode-linux-x64-baseline-musl", "@textcortex/zenocode-ai"),
+    "@textcortex/zenocode-linux-x64-baseline-musl",
+  );
+});
+
+test("buildWrapperBinMap includes package, opencode, and zenocode entrypoints", () => {
+  assert.deepEqual(buildWrapperBinMap("@textcortex/zenocode-ai", "zenocode"), {
+    "zenocode-ai": "./bin/opencode",
+    opencode: "./bin/opencode",
+    zenocode: "./bin/opencode",
+  });
+});

package/scripts/{run-codecortex.mjs → run-zenocode.mjs}

@@ -7,7 +7,10 @@ import path from "node:path";
 import process from "node:process";
 import readline from "node:readline/promises";
 import { fileURLToPath } from "node:url";
-import { padBinaryReplacement, patchOpenCodeVersionFooterText } from "./branding-patch.mjs";
+import {
+  patchZenocodeBinaryText,
+  zenocodeLogo,
+} from "./branding-patch.mjs";
 
 const currentFilePath = realpathSync(fileURLToPath(import.meta.url));
 const __dirname = path.dirname(currentFilePath);
@@ -66,19 +69,22 @@ const credentialFiles = [
   path.join(os.homedir(), ".credentials.json"),
   ...devCredentialFiles,
 ].filter((value, index, values) => values.indexOf(value) === index);
-const opencodeLogoSnippet = `var logo = {
-  left: ["                   ", "\\u2588\\u2580\\u2580\\u2588 \\u2588\\u2580\\u2580\\u2588 \\u2588\\u2580\\u2580\\u2588 \\u2588\\u2580\\u2580\\u2584", "\\u2588__\\u2588 \\u2588__\\u2588 \\u2588^^^ \\u2588__\\u2588", "\\u2580\\u2580\\u2580\\u2580 \\u2588\\u2580\\u2580\\u2580 \\u2580\\u2580\\u2580\\u2580 \\u2580~~\\u2580"],
-  right: ["             \\u2584     ", "\\u2588\\u2580\\u2580\\u2580 \\u2588\\u2580\\u2580\\u2588 \\u2588\\u2580\\u2580\\u2588 \\u2588\\u2580\\u2580\\u2588", "\\u2588___ \\u2588__\\u2588 \\u2588__\\u2588 \\u2588^^^", "\\u2580\\u2580\\u2580\\u2580 \\u2580\\u2580\\u2580\\u2580 \\u2580\\u2580\\u2580\\u2580 \\u2580\\u2580\\u2580\\u2580"]
-};
-var marks = "_^~";`;
-const zenocodeTextLogoSnippetCore = `var logo = {
-  left: ["                   ", "   Zenocode        ", "   Zenocode        ", "                   "],
-  right: ["                   ", "                   ", "                   ", "                   "]
-};
-var marks = "_^~";`;
-const zenocodeBanner = `
-   Z E N O C O D E
-`;
+
+function _decodeEscapedLogoLine(line) {
+  return line.replace(/\\u([0-9a-fA-F]{4})/g, (_, codePoint) =>
+    String.fromCharCode(Number.parseInt(codePoint, 16)),
+  );
+}
+
+export function buildZenocodeBanner() {
+  const bannerLines = zenocodeLogo.left
+    .map((leftLine, index) => {
+      const rightLine = zenocodeLogo.right[index] || "";
+      return `${_decodeEscapedLogoLine(leftLine)} ${_decodeEscapedLogoLine(rightLine)}`.trimEnd();
+    })
+    .filter((line) => line.trim());
+  return `\n${bannerLines.join("\n")}\n`;
+}
 const privateDirectoryMode = 0o700;
 const privateFileMode = 0o600;
 
@@ -503,16 +509,28 @@ async function runLoginCommand(baseUrl, args) {
 }
 
 async function runLogoutCommand() {
-  try {
-    await fs.unlink(runtimeCredentialsPath);
-    console.log("Zenocode credentials removed.");
-  } catch (error) {
-    if (error?.code === "ENOENT") {
-      console.log("No local Zenocode credentials found.");
-      return;
+  let removedAnyCredentials = false;
+  for (const credentialsPath of [
+    runtimeCredentialsPath,
+    legacyRuntimeCredentialsPath,
+  ].filter((value, index, values) => values.indexOf(value) === index)) {
+    try {
+      await fs.unlink(credentialsPath);
+      removedAnyCredentials = true;
+    } catch (error) {
+      if (error?.code === "ENOENT") {
+        continue;
+      }
+      throw error;
     }
-    throw error;
   }
+
+  if (removedAnyCredentials) {
+    console.log("Zenocode credentials removed.");
+    return;
+  }
+
+  console.log("No local Zenocode credentials found.");
 }
 
 async function runChild(command, args, options) {
@@ -527,8 +545,130 @@ async function runChild(command, args, options) {
   });
 }
 
-async function runBrandedBinary(args, options) {
-  const result = await runChild(opencodeBinaryPath, args, options);
+function _buildScriptCommand(runtimeBinaryPath, args, transcriptPath) {
+  if (process.platform === "darwin") {
+    return {
+      command: "script",
+      args: ["-q", "-F", "-t", "0", transcriptPath, runtimeBinaryPath, ...args],
+    };
+  }
+  if (process.platform === "linux") {
+    const shellCommand = [runtimeBinaryPath, ...args]
+      .map((value) => `'${String(value).replaceAll("'", `'\\''`)}'`)
+      .join(" ");
+    return {
+      command: "script",
+      args: ["-q", "-f", "-c", shellCommand, transcriptPath],
+    };
+  }
+  return null;
+}
+
+async function _readTranscriptUpdate(transcriptPath, offset) {
+  let handle;
+  try {
+    handle = await fs.open(transcriptPath, "r");
+  } catch (error) {
+    if (error?.code === "ENOENT") {
+      return { offset, text: "" };
+    }
+    throw error;
+  }
+
+  try {
+    const stats = await handle.stat();
+    if (stats.size <= offset) {
+      return { offset, text: "" };
+    }
+
+    const length = stats.size - offset;
+    const buffer = Buffer.alloc(length);
+    await handle.read(buffer, 0, length, offset);
+    return { offset: stats.size, text: buffer.toString("utf-8") };
+  } finally {
+    await handle.close();
+  }
+}
+
+async function _terminateChildProcess(child, exitPromise) {
+  if (child.exitCode !== null || child.signalCode !== null) {
+    return exitPromise;
+  }
+
+  child.kill("SIGTERM");
+  const exited = await Promise.race([exitPromise, sleep(1_500).then(() => null)]);
+  if (exited) {
+    return exited;
+  }
+
+  child.kill("SIGKILL");
+  return exitPromise;
+}
+
+async function _runRuntimeWithTranscriptMonitor(runtimeBinaryPath, args, options) {
+  const transcriptDir = await fs.mkdtemp(path.join(os.tmpdir(), "zenocode-runtime-"));
+  const transcriptPath = path.join(transcriptDir, "session.log");
+  const command = _buildScriptCommand(runtimeBinaryPath, args, transcriptPath);
+  if (!command) {
+    const result = await runChild(runtimeBinaryPath, args, { ...options, stdio: "inherit" });
+    return { ...result, expiredSession: false };
+  }
+
+  try {
+    const child = spawn(command.command, command.args, {
+      ...options,
+      stdio: "inherit",
+    });
+    const exitPromise = new Promise((resolve, reject) => {
+      child.on("error", reject);
+      child.on("exit", (code, signal) => resolve({ code, signal, expiredSession: false }));
+    });
+    let transcriptOffset = 0;
+    let transcriptTail = "";
+
+    while (true) {
+      const outcome = await Promise.race([
+        exitPromise.then((result) => ({ type: "exit", result })),
+        sleep(250).then(() => ({ type: "poll" })),
+      ]);
+
+      const update = await _readTranscriptUpdate(transcriptPath, transcriptOffset);
+      transcriptOffset = update.offset;
+      if (update.text) {
+        transcriptTail = _appendTranscriptTail(transcriptTail, update.text);
+      }
+
+      if (canRecoverRuntimeSessionFromTranscript(transcriptTail)) {
+        await _terminateChildProcess(child, exitPromise);
+        return { expiredSession: true };
+      }
+
+      if (outcome.type === "exit") {
+        return canRecoverRuntimeSessionFromTranscript(transcriptTail)
+          ? { expiredSession: true }
+          : outcome.result;
+      }
+    }
+  } catch (error) {
+    if (error?.code === "ENOENT") {
+      const result = await runChild(runtimeBinaryPath, args, { ...options, stdio: "inherit" });
+      return { ...result, expiredSession: false };
+    }
+    throw error;
+  } finally {
+    await fs.rm(transcriptDir, { recursive: true, force: true });
+  }
+}
+
+async function runRuntimeBinary(runtimeBinaryPath, args, options, monitorRuntimeSession) {
+  if (!monitorRuntimeSession) {
+    const result = await runChild(runtimeBinaryPath, args, { ...options, stdio: "inherit" });
+    return { ...result, expiredSession: false };
+  }
+  return _runRuntimeWithTranscriptMonitor(runtimeBinaryPath, args, options);
+}
+
+function exitWithChildResult(result) {
   if (result.signal) {
     process.kill(process.pid, result.signal);
     return;
@@ -553,14 +693,6 @@ function shouldPatchOpencodeRuntimePackage(packageName) {
   );
 }
 
-function _buildPaddedLogoSnippet(coreSnippet) {
-  return padBinaryReplacement(opencodeLogoSnippet, coreSnippet);
-}
-
-function _buildZenocodeLogoSnippet() {
-  return _buildPaddedLogoSnippet(zenocodeTextLogoSnippetCore);
-}
-
 async function _pathExists(pathToCheck) {
   try {
     await fs.access(pathToCheck);
@@ -688,26 +820,6 @@ async function _adHocSignBinary(binaryPath) {
   }
 }
 
-function _patchLogoSnippetText(text) {
-  const replacement = _buildZenocodeLogoSnippet();
-  if (!replacement || text.includes(replacement)) {
-    return { patched: false, text };
-  }
-
-  const patchTargets = [opencodeLogoSnippet].filter(Boolean);
-  let logoOffset = -1;
-  for (const target of patchTargets) {
-    logoOffset = text.indexOf(target);
-    if (logoOffset !== -1) break;
-  }
-  if (logoOffset === -1) {
-    return { patched: false, text };
-  }
-
-  const nextText = `${text.slice(0, logoOffset)}${replacement}${text.slice(logoOffset + replacement.length)}`;
-  return { patched: true, text: nextText };
-}
-
 async function _patchOpencodeBinaryBranding(binaryPath) {
   let buffer;
   try {
@@ -718,19 +830,9 @@ async function _patchOpencodeBinaryBranding(binaryPath) {
 
   const originalLength = buffer.length;
   let text = buffer.toString("latin1");
-  let patched = false;
-
-  const logoPatch = _patchLogoSnippetText(text);
-  if (logoPatch.patched) {
-    text = logoPatch.text;
-    patched = true;
-  }
-
-  const footerPatch = patchOpenCodeVersionFooterText(text);
-  if (footerPatch.patched) {
-    text = footerPatch.text;
-    patched = true;
-  }
+  const patch = patchZenocodeBinaryText(text);
+  text = patch.text;
+  const patched = patch.patched;
 
   if (!patched) {
     return false;
@@ -859,6 +961,88 @@ async function runPackageLauncher(packageName, args, options) {
   );
 }
 
+async function resolvePinnedRuntimeBinary(packageName, options) {
+  const runners = [
+    { command: _runnerCommand("pnpm"), args: ["dlx", packageName] },
+    { command: _runnerCommand("npx"), args: ["--yes", packageName] },
+  ];
+  const missingRunners = [];
+
+  for (const runner of runners) {
+    try {
+      let pinnedRuntimePath = await _ensurePatchedOpencodeDlxBinaries(packageName, runner, options);
+      if (!pinnedRuntimePath && shouldPatchOpencodeRuntimePackage(packageName)) {
+        const existingPinnedPath = _runtimeBrandedBinaryPath();
+        if (await _pathExists(existingPinnedPath)) {
+          pinnedRuntimePath = existingPinnedPath;
+        }
+      }
+
+      if (pinnedRuntimePath) {
+        return pinnedRuntimePath;
+      }
+    } catch (error) {
+      if (error?.code === "ENOENT") {
+        missingRunners.push(runner.command);
+        continue;
+      }
+      throw error;
+    }
+  }
+
+  if (missingRunners.length === runners.length) {
+    throw new Error(
+      `No package runner found (${missingRunners.join(", ")}). Install pnpm or npm, or set ZENOCODE_OPENCODE_BIN_PATH/CODECORTEX_OPENCODE_BIN_PATH.`,
+    );
+  }
+
+  return null;
+}
+
+export async function runRuntimeWithSessionRecovery({
+  args,
+  baseUrl,
+  token,
+  childOptions,
+  canAutoLoginRuntime,
+  runLogin = runLoginCommand,
+  resolveTokenFn = resolveToken,
+  resolveStoredBaseUrlFn = resolveStoredBaseUrl,
+  prepareRuntimeFn = prepareRuntime,
+  launchRuntimeFn,
+}) {
+  let activeBaseUrl = baseUrl;
+  let activeToken = token;
+
+  while (true) {
+    const result = await launchRuntimeFn({
+      args,
+      childOptions: {
+        ...childOptions,
+        env: {
+          ...(childOptions.env || {}),
+          TEXTCORTEX_API_KEY: activeToken,
+        },
+      },
+      token: activeToken,
+    });
+
+    if (!result.expiredSession) {
+      return result;
+    }
+
+    if (!canAutoLoginRuntime) {
+      return result;
+    }
+
+    console.log("Zenocode session expired. Starting login flow...\n");
+    await runLogin(activeBaseUrl, buildAutoLoginArgs());
+    activeToken = await resolveTokenFn();
+    activeBaseUrl = process.env.TEXTCORTEX_BASE_URL || (await resolveStoredBaseUrlFn()) || activeBaseUrl;
+    await prepareRuntimeFn(activeBaseUrl, activeToken);
+  }
+}
+
 async function packageExistsOnNpm(packageName) {
   const controller = new AbortController();
   const timeout = setTimeout(() => controller.abort(), 4_000);
@@ -925,7 +1109,14 @@ function maybeRenderBanner(args) {
   if (!shouldRenderBanner(args)) {
     return;
   }
-  console.log(`${zenocodeBanner}\n`);
+  console.log(`${buildZenocodeBanner()}\n`);
+}
+
+function buildAutoLoginArgs() {
+  return process.env.ZENOCODE_AUTO_LOGIN_NO_BROWSER === "1" ||
+    process.env.CODECORTEX_AUTO_LOGIN_NO_BROWSER === "1"
+    ? ["--no-launch-browser"]
+    : [];
 }
 
 function isMissingTokenError(error) {
@@ -954,6 +1145,45 @@ function canAutoLogin(args) {
   return true;
 }
 
+const ansiSequencePattern =
+  /\u001B(?:\[[0-?]*[ -/]*[@-~]|\][^\u0007]*(?:\u0007|\u001B\\)|[@-Z\\-_])/g;
+
+function _stripTerminalControlCharacters(text) {
+  let normalized = "";
+  for (const char of text) {
+    if (char === "\b" || char === "\u007f") {
+      normalized = normalized.slice(0, -1);
+      continue;
+    }
+    if (char === "\r") {
+      continue;
+    }
+    if (char < " " && char !== "\n" && char !== "\t") {
+      continue;
+    }
+    normalized += char;
+  }
+  return normalized;
+}
+
+export function normalizeRuntimeTranscript(text) {
+  return _stripTerminalControlCharacters(text.replaceAll(ansiSequencePattern, ""));
+}
+
+export function canRecoverRuntimeSessionFromTranscript(text) {
+  const normalized = normalizeRuntimeTranscript(text);
+  return (
+    /Unauthorized:\s*\{\s*"detail"\s*:\s*"Token has expired"\s*\}/i.test(normalized) ||
+    /authentication token may be missing or expired/i.test(normalized) ||
+    /server returned 401 after successful authentication/i.test(normalized)
+  );
+}
+
+function _appendTranscriptTail(tail, nextChunk, maxLength = 32_768) {
+  const combined = `${tail}${normalizeRuntimeTranscript(nextChunk)}`;
+  return combined.length > maxLength ? combined.slice(-maxLength) : combined;
+}
+
 function shouldAttemptAutoLogin(error, args) {
   if (!isMissingTokenError(error)) return false;
   return canAutoLogin(args);
@@ -969,12 +1199,7 @@ async function resolveTokenWithAutoLogin(baseUrl, args) {
     }
 
     console.log("No local Zenocode credentials found. Starting login flow...\n");
-    const loginArgs =
-      process.env.ZENOCODE_AUTO_LOGIN_NO_BROWSER === "1" ||
-      process.env.CODECORTEX_AUTO_LOGIN_NO_BROWSER === "1"
-        ? ["--no-launch-browser"]
-        : [];
-    await runLoginCommand(baseUrl, loginArgs);
+    await runLoginCommand(baseUrl, buildAutoLoginArgs());
     const token = await resolveToken();
     const persistedBaseUrl = process.env.TEXTCORTEX_BASE_URL || (await resolveStoredBaseUrl()) || baseUrl;
     return { token, baseUrl: persistedBaseUrl };
@@ -994,12 +1219,7 @@ async function prepareRuntimeWithAutoLogin(baseUrl, token, args) {
     }
 
     console.log("Zenocode session expired. Starting login flow...\n");
-    const loginArgs =
-      process.env.ZENOCODE_AUTO_LOGIN_NO_BROWSER === "1" ||
-      process.env.CODECORTEX_AUTO_LOGIN_NO_BROWSER === "1"
-        ? ["--no-launch-browser"]
-        : [];
-    await runLoginCommand(baseUrl, loginArgs);
+    await runLoginCommand(baseUrl, buildAutoLoginArgs());
     const refreshedToken = await resolveToken();
     const refreshedBaseUrl = process.env.TEXTCORTEX_BASE_URL || (await resolveStoredBaseUrl()) || baseUrl;
     const model = await prepareRuntime(refreshedBaseUrl, refreshedToken);
@@ -1047,7 +1267,6 @@ async function main() {
 
   const childOptions = {
     cwd: process.cwd(),
-    stdio: "inherit",
     env: {
       ...process.env,
       OPENCODE_MODELS_PATH: modelsPath,
@@ -1055,13 +1274,37 @@ async function main() {
       TEXTCORTEX_API_KEY: token,
     },
   };
+  const monitorRuntimeSession = canAutoLogin(passthrough);
 
   if (opencodeBinaryPath) {
-    await runBrandedBinary(passthrough, childOptions);
+    const result = await runRuntimeWithSessionRecovery({
+      args: passthrough,
+      baseUrl: runtime.baseUrl,
+      token,
+      childOptions,
+      canAutoLoginRuntime: monitorRuntimeSession,
+      launchRuntimeFn: ({ args, childOptions }) =>
+        runRuntimeBinary(opencodeBinaryPath, args, childOptions, monitorRuntimeSession),
+    });
+    exitWithChildResult(result);
     return;
   }
 
   const launchPackage = await resolveLaunchPackage();
+  const pinnedRuntimePath = await resolvePinnedRuntimeBinary(launchPackage, childOptions);
+  if (pinnedRuntimePath) {
+    const result = await runRuntimeWithSessionRecovery({
+      args: passthrough,
+      baseUrl: runtime.baseUrl,
+      token,
+      childOptions,
+      canAutoLoginRuntime: monitorRuntimeSession,
+      launchRuntimeFn: ({ args, childOptions }) =>
+        runRuntimeBinary(pinnedRuntimePath, args, childOptions, monitorRuntimeSession),
+    });
+    exitWithChildResult(result);
+    return;
+  }
   await runPackageLauncher(launchPackage, passthrough, childOptions);
 }