codex-lens 0.1.1 → 0.1.3

package/build.js

@@ -1,4 +1,5 @@
-import { build } from 'esbuild';
+import { build as esbuildBuild } from 'esbuild';
+import { build as viteBuild } from 'vite';
 import { rmSync, existsSync, mkdirSync, cpSync } from 'fs';
 import { resolve, dirname } from 'path';
 import { fileURLToPath } from 'url';
@@ -18,6 +19,8 @@ async function buildAll() {
     'src/proxy.js',
     'src/aggregator.js',
     'src/watcher.js',
+    'src/pty-manager.js',
+    'src/snapshot-manager.js',
     'src/lib/sse-parser.js',
     'src/lib/diff-builder.js',
     'src/lib/log-manager.js',
@@ -26,7 +29,7 @@ async function buildAll() {
 
   for (const file of backendFiles) {
     try {
-      await build({
+      await esbuildBuild({
         entryPoints: [resolve(__dirname, file)],
         outfile: resolve(__dirname, file.replace('src/', 'dist/')),
         bundle: false,
@@ -43,21 +46,11 @@ async function buildAll() {
     }
   }
 
-  await build({
-    entryPoints: [resolve(__dirname, 'src/main.jsx')],
-    outfile: resolve(__dirname, 'dist/main.js'),
-    bundle: true,
-    platform: 'browser',
-    target: 'es2020',
-    format: 'esm',
-    jsx: 'automatic',
-    loader: {
-      '.js': 'jsx',
-      '.jsx': 'jsx',
-    },
-    external: [],
+  console.log('Building frontend with Vite...');
+  await viteBuild({
+    configFile: resolve(__dirname, 'vite.config.js'),
   });
-  console.log('Built: src/main.jsx -> dist/main.js');
+  console.log('Built: frontend -> dist/public/');
 
   if (existsSync(resolve(__dirname, 'public'))) {
     cpSync(resolve(__dirname, 'public'), resolve(outDir, 'public'), { recursive: true });

package/dist/pty-manager.js

@@ -0,0 +1,182 @@
+#!/usr/bin/env node
+import { fileURLToPath } from "node:url";
+import { join, dirname } from "node:path";
+import { platform, arch } from "node:os";
+import { chmodSync, statSync } from "node:fs";
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+let ptyProcess = null;
+let dataListeners = [];
+let exitListeners = [];
+let lastExitCode = null;
+let outputBuffer = "";
+let lastPtyCols = 120;
+let lastPtyRows = 30;
+const MAX_BUFFER = 2e5;
+let batchBuffer = "";
+let batchScheduled = false;
+async function getPty() {
+  const ptyMod = await import("node-pty");
+  return ptyMod.default || ptyMod;
+}
+function findSafeSliceStart(buf, rawStart) {
+  const scanLimit = Math.min(rawStart + 64, buf.length);
+  let i = rawStart;
+  while (i < scanLimit) {
+    const ch = buf.charCodeAt(i);
+    if (ch === 27) {
+      let j = i + 1;
+      while (j < scanLimit && !(buf.charCodeAt(j) >= 64 && buf.charCodeAt(j) <= 126 && j > i + 1)) {
+        j++;
+      }
+      if (j < scanLimit) {
+        return j + 1;
+      }
+      i = j;
+      continue;
+    }
+    if (ch >= 32 && ch <= 63) {
+      i++;
+      continue;
+    }
+    break;
+  }
+  return i < buf.length ? i : rawStart;
+}
+function flushBatch() {
+  batchScheduled = false;
+  if (!batchBuffer) return;
+  const chunk = batchBuffer;
+  batchBuffer = "";
+  for (const cb of dataListeners) {
+    try {
+      cb(chunk);
+    } catch {
+    }
+  }
+}
+function fixSpawnHelperPermissions() {
+  try {
+    const os = platform();
+    const cpu = arch();
+    const helperPath = join(__dirname, "node_modules", "node-pty", "prebuilds", `${os}-${cpu}`, "spawn-helper");
+    const stat = statSync(helperPath);
+    if (!(stat.mode & 73)) {
+      chmodSync(helperPath, stat.mode | 493);
+    }
+  } catch {
+  }
+}
+async function spawnCodex(codexBinary, projectRoot, proxyPort) {
+  if (ptyProcess) {
+    killPty();
+  }
+  const pty = await getPty();
+  fixSpawnHelperPermissions();
+  const shell = platform() === "win32" ? "powershell.exe" : "bash";
+  const args = platform() === "win32" ? ["-NoExit", "-Command", `Set-Location "${projectRoot}"; & "${codexBinary}"`] : [];
+  const env = { ...process.env };
+  env.OPENAI_BASE_URL = `http://127.0.0.1:${proxyPort}`;
+  if (platform() === "win32") {
+    env.WINPTY = "1";
+  }
+  lastExitCode = null;
+  outputBuffer = "";
+  ptyProcess = pty.spawn(shell, args, {
+    name: "xterm-256color",
+    cols: lastPtyCols,
+    rows: lastPtyRows,
+    cwd: projectRoot,
+    env,
+    useConpty: false
+  });
+  ptyProcess.onData((data) => {
+    outputBuffer += data;
+    if (outputBuffer.length > MAX_BUFFER) {
+      const rawStart = outputBuffer.length - MAX_BUFFER;
+      const safeStart = findSafeSliceStart(outputBuffer, rawStart);
+      outputBuffer = outputBuffer.slice(safeStart);
+    }
+    batchBuffer += data;
+    if (!batchScheduled) {
+      batchScheduled = true;
+      setImmediate(flushBatch);
+    }
+  });
+  ptyProcess.onExit(({ exitCode }) => {
+    flushBatch();
+    lastExitCode = exitCode;
+    ptyProcess = null;
+    for (const cb of exitListeners) {
+      try {
+        cb(exitCode);
+      } catch {
+      }
+    }
+  });
+  return ptyProcess;
+}
+function writeToPty(data) {
+  if (ptyProcess) {
+    ptyProcess.write(data);
+    return true;
+  }
+  return false;
+}
+function resizePty(cols, rows) {
+  lastPtyCols = cols;
+  lastPtyRows = rows;
+  if (ptyProcess) {
+    try {
+      ptyProcess.resize(cols, rows);
+    } catch {
+    }
+  }
+}
+function killPty() {
+  if (ptyProcess) {
+    flushBatch();
+    batchBuffer = "";
+    batchScheduled = false;
+    try {
+      ptyProcess.kill();
+    } catch {
+    }
+    ptyProcess = null;
+  }
+}
+function onPtyData(cb) {
+  dataListeners.push(cb);
+  return () => {
+    dataListeners = dataListeners.filter((l) => l !== cb);
+  };
+}
+function onPtyExit(cb) {
+  exitListeners.push(cb);
+  return () => {
+    exitListeners = exitListeners.filter((l) => l !== cb);
+  };
+}
+function getPtyPid() {
+  return ptyProcess ? ptyProcess.pid : null;
+}
+function getPtyState() {
+  return {
+    running: !!ptyProcess,
+    exitCode: lastExitCode
+  };
+}
+function getOutputBuffer() {
+  return outputBuffer;
+}
+export {
+  getOutputBuffer,
+  getPtyPid,
+  getPtyState,
+  killPty,
+  onPtyData,
+  onPtyExit,
+  resizePty,
+  spawnCodex,
+  writeToPty
+};