ineedcodes 1.0.1 → 1.1.0

package/src/ui.js

@@ -1,6 +1,6 @@
 // ui.js: terminal helpers. No dependencies, respects NO_COLOR and non-TTY.
 
-export const VERSION = '1.0.0';
+export const VERSION = '1.1.0';
 
 const USE_COLOR = process.stdout.isTTY && !process.env.NO_COLOR;
 const wrap = (code, t) => USE_COLOR ? `\x1b[${code}m${t}\x1b[0m` : String(t);
@@ -18,17 +18,80 @@ export const trunc = (s, n = 120) => {
   return o.length > n ? o.slice(0, n - 1) + '...' : o;
 };
 
-export const LOGO = [
-  '####  ##   ',
-  ' ##   ##   ',
-  ' ##   ##   ',
-  ' ##   ##   ',
-  ' ####  ### '
-].join('\n');
+const plain = s => String(s).replace(/\x1b\[[0-9;]*m/g, '');
+
+// Rounded box around lines. Width follows the longest line, capped to the terminal.
+export function box(lines, colorFn = t => t) {
+  const cols = process.stdout.columns || 80;
+  const inner = Math.min(cols - 4, Math.max(10, ...lines.map(l => plain(l).length)) + 2);
+  const top = colorFn('╭' + '─'.repeat(inner) + '╮');
+  const bot = colorFn('╰' + '─'.repeat(inner) + '╯');
+  const mid = lines.map(l => colorFn('│') + ' ' + l + ' '.repeat(Math.max(0, inner - plain(l).length - 1)) + colorFn('│'));
+  return [top, ...mid, bot].join('\n');
+}
+
+const SPIN_FRAMES = ['⠋', '⠙', '⠹', '⠸', '⠼', '⠴', '⠦', '⠧', '⠇', '⠏'];
+
+// Spinner for TTYs only. In pipes and tests it becomes a no-op.
+export function startSpinner(text = 'thinking') {
+  if (!process.stdout.isTTY || process.env.NO_COLOR) return { stop: () => {} };
+  let i = 0;
+  let stopped = false;
+  const line = () => `\r${cyan(SPIN_FRAMES[i++ % SPIN_FRAMES.length])} ${dim(text + '...')}  `;
+  process.stdout.write(line());
+  const iv = setInterval(() => process.stdout.write(line()), 90);
+  return {
+    stop() {
+      if (stopped) return;
+      stopped = true;
+      clearInterval(iv);
+      process.stdout.write('\r' + ' '.repeat(plain(text).length + 20) + '\r');
+    }
+  };
+}
+
+export const RULE = () => dim('─'.repeat(Math.min(process.stdout.columns || 80, 64)));
+
+// user bubble: quoted, colored, compact
+export const userBubble = text => {
+  const w = Math.min(process.stdout.columns || 80, 60);
+  const wrapped = [];
+  for (const raw of String(text).split('\n')) {
+    let line = raw;
+    while (line.length > w - 4) {
+      wrapped.push('  ' + yellow('│ ') + line.slice(0, w - 4));
+      line = line.slice(w - 4);
+    }
+    wrapped.push('  ' + yellow('│ ') + line);
+  }
+  return wrapped.join('\n') + '\n  ' + yellow('╰' + '─'.repeat(w - 4) + '╯');
+};
+
+// ── raw screen plumbing for the full-screen chat layout ──
+export const screen = {
+  enter() { process.stdout.write('\x1b[?1049h\x1b[?25l\x1b[H\x1b[2J'); },
+  exit() { process.stdout.write('\x1b[r\x1b[?25h\x1b[?1049l'); },
+  region(top, bot) { process.stdout.write(`\x1b[${top};${bot}r`); },
+  resetRegion() { process.stdout.write('\x1b[r'); },
+  at(row, col = 1) { process.stdout.write(`\x1b[${row};${col}H`); },
+  clearLine() { process.stdout.write('\x1b[2K'); }
+};
 
 export const BANNER = () =>
   bold(green('ineed')) + dim(` v${VERSION}`) + dim(' · your terminal, now autonomous');
 
+// Big startup logo, ANSI-shadow style. Six fixed-width lines.
+const LOGO_LINES = [
+  '██╗ ███╗   ██╗  ███████╗  ███████╗  ██████╗ ',
+  '██║ ████╗  ██║  ██╔════╝  ██╔════╝  ██╔══██╗',
+  '██║ ██╔██╗ ██║  █████╗    █████╗    ██║  ██║',
+  '██║ ██║╚██╗██║  ██╔══╝    ██╔══╝    ██║  ██║',
+  '██║ ██║ ╚████║  ███████╗  ███████╗  ██████╔╝',
+  '╚═╝ ╚═╝  ╚═══╝  ╚══════╝  ╚══════╝  ╚═════╝ '
+];
+export const LOGO = LOGO_LINES.join('\n');
+export const logo = () => LOGO_LINES.map(l => green(bold(l))).join('\n');
+
 // Ask a question and await one line. `secret` hides typed characters.
 // onLine: receiver for lines typed when no question is pending (REPL dispatch).
 export function makeInput(rl, onLine) {

package/src/wizard.js

@@ -2,7 +2,7 @@
 
 import { fetchModels, chat } from './provider.js';
 import { saveConfig } from './config.js';
-import { bold, dim, red, yellow, green, cyan, trunc } from './ui.js';
+import { bold, dim, red, yellow, green, cyan, trunc, logo, BANNER, startSpinner } from './ui.js';
 
 export async function testConnection(cfg, signal) {
   // proves reachability + auth + a working chat in one shot
@@ -15,7 +15,10 @@ export async function wizard(ask, { fromCommand = false } = {}) {
   const abortIfEnded = () => { if (inputEnded()) throw Object.assign(new Error('Setup aborted: input ended.'), { aborted: true }); };
 
   console.log('');
-  console.log(bold('Welcome to ineed') + dim(' - let\'s connect you to an AI provider. You only do this once.'));
+  console.log(logo());
+  console.log('');
+  console.log(BANNER());
+  console.log(bold('Welcome!') + dim(' Let\'s connect you to an AI provider. You only do this once.'));
   console.log(dim('Any OpenAI-compatible API works: OpenAI, OmniRoute, LM Studio, Ollama, vLLM, and more.'));
   console.log('');
 
@@ -39,9 +42,11 @@ export async function wizard(ask, { fromCommand = false } = {}) {
   const probe = { baseUrl, apiKey, model: 'x' };
   console.log(dim('\n   Checking connection...'));
   let models = [];
+  const connSpin = startSpinner('connecting');
   try {
     models = await fetchModels(probe);
   } catch {}
+  connSpin.stop();
   if (models.length > 0) {
     console.log(green(`   Connected. ${models.length} models available.`));
   } else {
@@ -65,11 +70,15 @@ export async function wizard(ask, { fromCommand = false } = {}) {
   console.log(dim(`   Testing ${model}...`));
   let saved = false;
   while (!saved) {
+    let testSpin = null;
     try {
+      testSpin = startSpinner('testing ' + model);
       const reply = await testConnection({ baseUrl, apiKey, model });
+      testSpin.stop();
       console.log(green('   Works.') + dim(` Replied: ${trunc(reply, 40)}`));
       saved = true;
     } catch (err) {
+      testSpin?.stop();
       console.log(red('   Test failed: ' + err.message));
       const choice = await ask('   [r]etry key, [m]odel, [l]ist models, [b]ase url, or [s]ave anyway? ');
       if (choice === '') abortIfEnded();