@textcortex/zenocode 0.1.2 → 0.1.4

package/scripts/run-zenocode.test.mjs

@@ -0,0 +1,224 @@
+import assert from "node:assert/strict";
+import fs from "node:fs/promises";
+import os from "node:os";
+import path from "node:path";
+import { spawn } from "node:child_process";
+import test from "node:test";
+import {
+  buildPaddedLogoSnippet,
+  patchLogoSnippetText,
+} from "./branding-patch.mjs";
+import {
+  buildOpenCodeConfig,
+  canRecoverRuntimeSessionFromTranscript,
+  buildZenocodeBanner,
+  chooseDefaults,
+  runRuntimeWithSessionRecovery,
+  writePrivateJsonFile,
+} from "./run-zenocode.mjs";
+
+test("chooseDefaults prefers kimi k2.5 thinking for Zenocode", () => {
+  const defaults = chooseDefaults({
+    "glm-5": {},
+    "kimi-k2-5-thinking": {},
+    "gpt-5-2": {},
+  });
+
+  assert.deepEqual(defaults, {
+    model: "kimi-k2-5-thinking",
+    smallModel: "kimi-k2-5-thinking",
+  });
+});
+
+test("buildOpenCodeConfig includes an openai stub for fallback runtime auth plugins", () => {
+  const config = buildOpenCodeConfig({
+    baseUrl: "http://127.0.0.1:8080",
+    providerID: "textcortex",
+    model: "kimi-k2-5-thinking",
+    smallModel: "kimi-k2-5-thinking",
+  });
+
+  assert.deepEqual(config.enabled_providers, ["textcortex"]);
+  assert.equal(config.model, "textcortex/kimi-k2-5-thinking");
+  assert.ok(config.provider.openai);
+  assert.deepEqual(config.provider.openai.models, {});
+});
+
+test("buildZenocodeBanner renders block logo art instead of plain text", () => {
+  const banner = buildZenocodeBanner();
+
+  assert.match(banner, /▀▀█/);
+  assert.match(banner, /█▀▀/);
+  assert.match(banner, /█▄▄/);
+  assert.doesNotMatch(banner, /\bZenocode\b/);
+});
+
+test("patchLogoSnippetText upgrades previously branded Zenocode logos", () => {
+  const legacyTextLogoSnippet = `var logo = {
+  left: ["                   ", "   Zenocode        ", "   Zenocode        ", "                   "],
+  right: ["                   ", "                   ", "                   ", "                   "]
+};
+var marks = "_^~";`;
+
+  const result = patchLogoSnippetText(buildPaddedLogoSnippet(legacyTextLogoSnippet));
+
+  assert.equal(result.patched, true);
+  assert.match(result.text, /\\u2580\\u2580\\u2588/);
+  assert.match(result.text, /\\u2588\\u2580\\u2580/);
+  assert.doesNotMatch(result.text, /\\\\u2580/);
+  assert.doesNotMatch(result.text, /Zenocode/);
+});
+
+test("patchLogoSnippetText upgrades double-escaped Zenocode glyph logos", () => {
+  const brokenGlyphLogoSnippet = String.raw`var logo = {
+  left: ["                   "," \\u2580\\u2580\\u2588 \\u2588\\u2580\\u2580 \\u2588\\u2584 \\u2588 \\u2588\\u2580\\u2588"," \\u2588\\u2584\\u2584 \\u2588\\u2588\\u2584 \\u2588 \\u2580\\u2588 \\u2588\\u2584\\u2588","                   "],
+  right: ["                   "," \\u2588\\u2580\\u2580 \\u2588\\u2580\\u2588 \\u2588\\u2580\\u2584 \\u2588\\u2580\\u2580"," \\u2588\\u2584\\u2584 \\u2588\\u2584\\u2588 \\u2588\\u2584\\u2580 \\u2588\\u2588\\u2584","                   "]
+};
+var marks = "_^~";`;
+
+  const result = patchLogoSnippetText(buildPaddedLogoSnippet(brokenGlyphLogoSnippet));
+
+  assert.equal(result.patched, true);
+  assert.match(result.text, /\\u2580\\u2580\\u2588/);
+  assert.match(result.text, /\\u2588\\u2580\\u2580/);
+  assert.doesNotMatch(result.text, /\\\\u2580/);
+});
+
+test("writePrivateJsonFile stores credentials with private permissions", async (t) => {
+  const tempDir = await fs.mkdtemp(path.join(os.tmpdir(), "zenocode-runtime-"));
+  const credentialsPath = path.join(tempDir, "runtime", "credentials.json");
+
+  t.after(async () => {
+    await fs.rm(tempDir, { recursive: true, force: true });
+  });
+
+  await writePrivateJsonFile(credentialsPath, { access_token: "secret-token" });
+
+  const saved = JSON.parse(await fs.readFile(credentialsPath, "utf-8"));
+  assert.equal(saved.access_token, "secret-token");
+
+  if (process.platform !== "win32") {
+    const dirMode = (await fs.stat(path.dirname(credentialsPath))).mode & 0o777;
+    const fileMode = (await fs.stat(credentialsPath)).mode & 0o777;
+
+    assert.equal(dirMode, 0o700);
+    assert.equal(fileMode, 0o600);
+  }
+});
+
+test("canRecoverRuntimeSessionFromTranscript detects expired session output", () => {
+  const transcript = [
+    "\u001b[2J\u001b[H",
+    "cool, is there anything about zenocode?",
+    "",
+    'Unauthorized: {"detail":"Token has expired"}',
+  ].join("\n");
+
+  assert.equal(canRecoverRuntimeSessionFromTranscript(transcript), true);
+});
+
+test("runRuntimeWithSessionRecovery reruns login after runtime session expiry", async () => {
+  const events = [];
+  let launchCount = 0;
+
+  const result = await runRuntimeWithSessionRecovery({
+    args: ["run"],
+    baseUrl: "http://127.0.0.1:8080",
+    token: "token-1",
+    childOptions: {
+      cwd: process.cwd(),
+      env: {},
+    },
+    canAutoLoginRuntime: true,
+    runLogin: async (baseUrl, loginArgs) => {
+      events.push(["login", baseUrl, loginArgs]);
+    },
+    resolveTokenFn: async () => {
+      events.push(["resolve-token"]);
+      return "token-2";
+    },
+    resolveStoredBaseUrlFn: async () => {
+      events.push(["resolve-base-url"]);
+      return "https://api.textcortex.com";
+    },
+    prepareRuntimeFn: async (baseUrl, token) => {
+      events.push(["prepare", baseUrl, token]);
+      return "kimi-k2-5-thinking";
+    },
+    launchRuntimeFn: async ({ childOptions }) => {
+      launchCount += 1;
+      events.push(["launch", launchCount, childOptions.env.TEXTCORTEX_API_KEY]);
+      if (launchCount === 1) {
+        return { expiredSession: true };
+      }
+      return { code: 0, signal: null, expiredSession: false };
+    },
+  });
+
+  assert.deepEqual(events, [
+    ["launch", 1, "token-1"],
+    ["login", "http://127.0.0.1:8080", []],
+    ["resolve-token"],
+    ["resolve-base-url"],
+    ["prepare", "https://api.textcortex.com", "token-2"],
+    ["launch", 2, "token-2"],
+  ]);
+  assert.deepEqual(result, { code: 0, signal: null, expiredSession: false });
+});
+
+test("logout removes legacy credential fallbacks as well as Zenocode credentials", async (t) => {
+  const tempDir = await fs.mkdtemp(path.join(os.tmpdir(), "zenocode-logout-"));
+  const zenocodeHome = path.join(tempDir, ".zenocode");
+  const codecortexHome = path.join(tempDir, ".codecortex");
+  const repoCredentialsPath = path.join(tempDir, ".credentials.json");
+  const runtimeCredentialsPath = path.join(zenocodeHome, "credentials.json");
+  const legacyRuntimeCredentialsPath = path.join(codecortexHome, "credentials.json");
+  const scriptPath = new URL("./run-zenocode.mjs", import.meta.url);
+
+  t.after(async () => {
+    await fs.rm(tempDir, { recursive: true, force: true });
+  });
+
+  await fs.mkdir(path.dirname(runtimeCredentialsPath), { recursive: true });
+  await fs.mkdir(path.dirname(legacyRuntimeCredentialsPath), { recursive: true });
+  await fs.writeFile(runtimeCredentialsPath, JSON.stringify({ access_token: "current" }));
+  await fs.writeFile(
+    legacyRuntimeCredentialsPath,
+    JSON.stringify({ access_token: "legacy" }),
+  );
+  await fs.writeFile(repoCredentialsPath, JSON.stringify({ access_token: "repo" }));
+
+  const result = await new Promise((resolve, reject) => {
+    const child = spawn(
+      process.execPath,
+      [scriptPath.pathname, "logout"],
+      {
+        cwd: tempDir,
+        env: {
+          ...process.env,
+          ZENOCODE_HOME: zenocodeHome,
+          CODECORTEX_HOME: codecortexHome,
+        },
+        stdio: ["ignore", "pipe", "pipe"],
+      },
+    );
+    let stdout = "";
+    let stderr = "";
+    child.stdout.on("data", (chunk) => {
+      stdout += String(chunk);
+    });
+    child.stderr.on("data", (chunk) => {
+      stderr += String(chunk);
+    });
+    child.on("error", reject);
+    child.on("exit", (code, signal) => resolve({ code, signal, stdout, stderr }));
+  });
+
+  assert.equal(result.code, 0);
+  await assert.rejects(fs.stat(runtimeCredentialsPath), { code: "ENOENT" });
+  await assert.rejects(fs.stat(legacyRuntimeCredentialsPath), { code: "ENOENT" });
+  assert.deepEqual(
+    JSON.parse(await fs.readFile(repoCredentialsPath, "utf-8")),
+    { access_token: "repo" },
+  );
+});

package/scripts/run-codecortex.test.mjs

@@ -1,59 +0,0 @@
-import assert from "node:assert/strict";
-import fs from "node:fs/promises";
-import os from "node:os";
-import path from "node:path";
-import test from "node:test";
-import {
-  buildOpenCodeConfig,
-  chooseDefaults,
-  writePrivateJsonFile,
-} from "./run-codecortex.mjs";
-
-test("chooseDefaults prefers kimi k2.5 thinking for Zenocode", () => {
-  const defaults = chooseDefaults({
-    "glm-5": {},
-    "kimi-k2-5-thinking": {},
-    "gpt-5-2": {},
-  });
-
-  assert.deepEqual(defaults, {
-    model: "kimi-k2-5-thinking",
-    smallModel: "kimi-k2-5-thinking",
-  });
-});
-
-test("buildOpenCodeConfig includes an openai stub for fallback runtime auth plugins", () => {
-  const config = buildOpenCodeConfig({
-    baseUrl: "http://127.0.0.1:8080",
-    providerID: "textcortex",
-    model: "kimi-k2-5-thinking",
-    smallModel: "kimi-k2-5-thinking",
-  });
-
-  assert.deepEqual(config.enabled_providers, ["textcortex"]);
-  assert.equal(config.model, "textcortex/kimi-k2-5-thinking");
-  assert.ok(config.provider.openai);
-  assert.deepEqual(config.provider.openai.models, {});
-});
-
-test("writePrivateJsonFile stores credentials with private permissions", async (t) => {
-  const tempDir = await fs.mkdtemp(path.join(os.tmpdir(), "zenocode-runtime-"));
-  const credentialsPath = path.join(tempDir, "runtime", "credentials.json");
-
-  t.after(async () => {
-    await fs.rm(tempDir, { recursive: true, force: true });
-  });
-
-  await writePrivateJsonFile(credentialsPath, { access_token: "secret-token" });
-
-  const saved = JSON.parse(await fs.readFile(credentialsPath, "utf-8"));
-  assert.equal(saved.access_token, "secret-token");
-
-  if (process.platform !== "win32") {
-    const dirMode = (await fs.stat(path.dirname(credentialsPath))).mode & 0o777;
-    const fileMode = (await fs.stat(credentialsPath)).mode & 0o777;
-
-    assert.equal(dirMode, 0o700);
-    assert.equal(fileMode, 0o600);
-  }
-});