ucode-agent 1.7.0 → 1.8.0

package/src/ui/theme.js

@@ -276,3 +276,47 @@ export function planLine(items) {
   });
   return `  ${sky(`plan ${done}/${list.length}`)}  ${parts.join(dim('  ·  '))}`;
 }
+
+/**
+ * The background ucode paints behind itself.
+ *
+ * A terminal's own background is whatever the person set it to years ago:
+ * white, solarized, a photograph. The interface was drawn for a dark one, and
+ * on a light terminal the dim greys it relies on turn to near-invisible smoke.
+ * So ucode paints its own ground for as long as it is running, and the
+ * alternate screen gives it back untouched on exit.
+ *
+ * Near-black rather than black: a true #000 against a bright room is a hole,
+ * and the box edges lose their softness. This is the shade a code editor
+ * settles on for the same reason.
+ */
+export const BACKGROUND = process.env.UCODE_BG || '#131316';
+
+const rgb = (hex) => {
+  const m = /^#?([0-9a-f]{2})([0-9a-f]{2})([0-9a-f]{2})$/i.exec(hex.trim());
+  return m ? [parseInt(m[1], 16), parseInt(m[2], 16), parseInt(m[3], 16)] : [19, 19, 22];
+};
+
+/** Turn the background on. Everything painted after this sits on it. */
+export const BG_ON = (() => {
+  if (process.env.NO_COLOR || process.env.UCODE_BG === 'off') return '';
+  const [r, g, b] = rgb(BACKGROUND);
+  return `\x1b[48;2;${r};${g};${b}m`;
+})();
+
+/** Hand the terminal its own colours back. */
+export const BG_OFF = BG_ON ? '\x1b[0m' : '';
+
+/**
+ * Keep the background on across a line that resets it.
+ *
+ * chalk closes a foreground with 39 and a background with 49, and 49 means
+ * "the terminal's default" — which is exactly the colour being painted over.
+ * A diff line, which sets its own background, would therefore punch a hole
+ * through to the terminal's ground for the rest of the line. Re-asserting the
+ * background after every reset closes those holes.
+ */
+export function onBackground(text) {
+  if (!BG_ON) return text;
+  return BG_ON + String(text).replace(/\x1b\[(?:0|49)m/g, (m) => m + BG_ON);
+}

package/ucode.js

@@ -1,124 +1,132 @@
-#!/usr/bin/env node
-/**
- * ucode.js — the command.
- *
- * Parses the arguments, builds an Agent, and gets out of the way. Everything
- * of substance is under src/.
- */
-
-import path from 'node:path';
-import process from 'node:process';
-import { readFile } from 'node:fs/promises';
-import { realpathSync } from 'node:fs';
-import { pathToFileURL, fileURLToPath } from 'node:url';
-
-import { Agent } from './src/core/loop.js';
-import { setModel, modelName, MODELS, DEFAULT_MODEL, ENV_FILE } from './src/core/provider.js';
-import { VERSION } from './src/core/version.js';
-import { Plain } from './src/ui/plain.js';
-import { blue, dim, sky } from './src/ui/theme.js';
-
-function parseArgs(argv) {
-  const args = { debug: false, model: null, cwd: process.cwd(), help: false, plan: false, version: false };
-
-  for (let i = 0; i < argv.length; i++) {
-    const a = argv[i];
-    if (a === '--debug') args.debug = true;
-    else if (a === '--plan') args.plan = true;
-    else if (a === '--model' || a === '-m') args.model = argv[++i];
-    else if (a === '--cwd' || a === '-C') args.cwd = path.resolve(argv[++i]);
-    else if (a === '--help' || a === '-h') args.help = true;
-    else if (a === '--version' || a === '-v') args.version = true;
-  }
-
-  return args;
-}
-
-function usage() {
-  const entries = Object.entries(MODELS);
-  const width = Math.max(...entries.map(([, m]) => m.name.length));
-  const models = entries
-    .map(([id, m]) => `      ${m.name.padEnd(width)}   ${dim(id)}`)
-    .join('\n');
-
-  process.stdout.write(
-    `\n  ${blue('ucode')} — a terminal coding agent\n\n` +
-    '  ucode [options]\n\n' +
-    `    -m, --model <id>   which model to use (default: ${modelName(DEFAULT_MODEL)})\n` +
-    '    -C, --cwd <dir>    work in another directory\n' +
-    '        --plan         start in plan mode: read and research, change nothing\n' +
-    '        --debug        print stack traces when something breaks\n' +
-    '    -v, --version      print the version\n' +
-    '    -h, --help         this message\n' +
-    '    doctor             check that everything ucode needs is working\n\n' +
-    `  ${sky('Models')}\n${models}\n\n` +
-    `  Needs UCODE_API_KEY in the environment or in ${ENV_FILE}\n` +
-    '  Free keys: https://openrouter.ai/keys\n\n'
-  );
-}
-
-async function main() {
-  const args = parseArgs(process.argv.slice(2));
-
-  if (args.help) return usage();
-
-  if (process.argv[2] === 'doctor') {
-    const { runDoctor } = await import('./src/core/doctor.js');
-    process.stdout.write(`${(await runDoctor()).join('\n')}\n`);
-    return;
-  }
-
-  if (args.version) {
-    process.stdout.write(`${VERSION}\n`);
-    return;
-  }
-
-  if (args.model) {
-    try {
-      setModel(args.model);
-    } catch (err) {
-      new Plain({ cwd: args.cwd }).error(err);
-      process.exitCode = 1;
-      return;
-    }
-  }
-
-  const agent = new Agent({ cwd: args.cwd, debug: args.debug });
-  if (args.plan) agent.ui.mode = 'plan';
-
-  try {
-    await agent.start();
-  } catch (err) {
-    agent.ui.error(err, { debug: args.debug });
-    await agent.persist().catch(() => {});
-    agent.ui.close();
-    process.exitCode = 1;
-  }
-}
-
-/**
- * Was this file launched, or imported?
- *
- * Importing it — which another front end would do to reuse Agent — must not
- * open a terminal session. The obvious check is comparing `import.meta.url`
- * with argv[1], and it is wrong: after `npm link`, argv[1] arrives as the path
- * through the symlink in the global node_modules while `import.meta.url` has
- * already been resolved to the real file. The two never match, so the command
- * starts, matches nothing, and exits successfully having done absolutely
- * nothing — which is a great deal harder to diagnose than a crash.
- *
- * Comparing real paths is what actually answers the question.
- */
-function launchedDirectly() {
-  const entry = process.argv[1];
-  if (!entry) return false;
-
-  const self = fileURLToPath(import.meta.url);
-  try {
-    return realpathSync(entry) === realpathSync(self);
-  } catch {
-    return pathToFileURL(entry).href === import.meta.url;
-  }
-}
-
-if (launchedDirectly()) main();
+#!/usr/bin/env node
+/**
+ * ucode.js — the command.
+ *
+ * Parses the arguments, builds an Agent, and gets out of the way. Everything
+ * of substance is under src/.
+ */
+
+import path from 'node:path';
+import process from 'node:process';
+import { readFile } from 'node:fs/promises';
+import { realpathSync } from 'node:fs';
+import { pathToFileURL, fileURLToPath } from 'node:url';
+
+import { Agent } from './src/core/loop.js';
+import { setModel, modelName, MODELS, DEFAULT_MODEL, ENV_FILE } from './src/core/provider.js';
+import { VERSION } from './src/core/version.js';
+import { Plain } from './src/ui/plain.js';
+import { blue, dim, sky } from './src/ui/theme.js';
+
+function parseArgs(argv) {
+  const args = { debug: false, model: null, cwd: process.cwd(), help: false, plan: false, version: false };
+
+  for (let i = 0; i < argv.length; i++) {
+    const a = argv[i];
+    if (a === '--debug') args.debug = true;
+    else if (a === '--plan') args.plan = true;
+    else if (a === '--model' || a === '-m') args.model = argv[++i];
+    else if (a === '--cwd' || a === '-C') args.cwd = path.resolve(argv[++i]);
+    else if (a === '--help' || a === '-h') args.help = true;
+    else if (a === '--version' || a === '-v') args.version = true;
+  }
+
+  return args;
+}
+
+function usage() {
+  const entries = Object.entries(MODELS);
+  const width = Math.max(...entries.map(([, m]) => m.name.length));
+  const models = entries
+    .map(([id, m]) => `      ${m.name.padEnd(width)}   ${dim(id)}`)
+    .join('\n');
+
+  process.stdout.write(
+    `\n  ${blue('ucode')} — a terminal coding agent\n\n` +
+    '  ucode [options]\n\n' +
+    `    -m, --model <id>   which model to use (default: ${modelName(DEFAULT_MODEL)})\n` +
+    '    -C, --cwd <dir>    work in another directory\n' +
+    '        --plan         start in plan mode: read and research, change nothing\n' +
+    '        --debug        print stack traces when something breaks\n' +
+    '    -v, --version      print the version\n' +
+    '    -h, --help         this message\n' +
+    '    doctor             check that everything ucode needs is working\n' +
+    '    login <key>        save your key for every folder on this machine\n\n' +
+    `  ${sky('Models')}\n${models}\n\n` +
+    `  Needs UCODE_API_KEY in the environment or in ${ENV_FILE}\n` +
+    '  Free keys: https://openrouter.ai/keys\n\n'
+  );
+}
+
+async function main() {
+  const args = parseArgs(process.argv.slice(2));
+
+  if (args.help) return usage();
+
+  if (process.argv[2] === 'login') {
+    const { saveKey } = await import('./src/core/login.js');
+    process.stdout.write(`${await saveKey(process.argv[3])}
+`);
+    return;
+  }
+
+  if (process.argv[2] === 'doctor') {
+    const { runDoctor } = await import('./src/core/doctor.js');
+    process.stdout.write(`${(await runDoctor()).join('\n')}\n`);
+    return;
+  }
+
+  if (args.version) {
+    process.stdout.write(`${VERSION}\n`);
+    return;
+  }
+
+  if (args.model) {
+    try {
+      setModel(args.model);
+    } catch (err) {
+      new Plain({ cwd: args.cwd }).error(err);
+      process.exitCode = 1;
+      return;
+    }
+  }
+
+  const agent = new Agent({ cwd: args.cwd, debug: args.debug });
+  if (args.plan) agent.ui.mode = 'plan';
+
+  try {
+    await agent.start();
+  } catch (err) {
+    agent.ui.error(err, { debug: args.debug });
+    await agent.persist().catch(() => {});
+    agent.ui.close();
+    process.exitCode = 1;
+  }
+}
+
+/**
+ * Was this file launched, or imported?
+ *
+ * Importing it — which another front end would do to reuse Agent — must not
+ * open a terminal session. The obvious check is comparing `import.meta.url`
+ * with argv[1], and it is wrong: after `npm link`, argv[1] arrives as the path
+ * through the symlink in the global node_modules while `import.meta.url` has
+ * already been resolved to the real file. The two never match, so the command
+ * starts, matches nothing, and exits successfully having done absolutely
+ * nothing — which is a great deal harder to diagnose than a crash.
+ *
+ * Comparing real paths is what actually answers the question.
+ */
+function launchedDirectly() {
+  const entry = process.argv[1];
+  if (!entry) return false;
+
+  const self = fileURLToPath(import.meta.url);
+  try {
+    return realpathSync(entry) === realpathSync(self);
+  } catch {
+    return pathToFileURL(entry).href === import.meta.url;
+  }
+}
+
+if (launchedDirectly()) main();