ineedcodes 0.2.0 → 1.0.1

package/agent.mjs

@@ -1,244 +0,0 @@
-#!/usr/bin/env node
-// ineed — minimal, self-contained terminal AI agent. No dependencies.
-
-import * as fs from 'node:fs';
-import * as os from 'node:os';
-import * as path from 'node:path';
-import { spawn } from 'node:child_process';
-import * as readline from 'node:readline';
-
-const VERSION = '0.2.0';
-const CONFIG_DIR = path.join(os.homedir(), '.ineedcodes');
-const CONFIG_FILE = path.join(CONFIG_DIR, 'config.json');
-const A = (c, t) => process.stdout.isTTY ? `\x1b[${c}m${t}\x1b[0m` : t;
-const bold = t => A('1', t), dim = t => A('2', t), green = t => A('32', t);
-const red = t => A('31', t), yellow = t => A('33', t), cyan = t => A('36', t), gray = t => A('90', t);
-const trunc = (s, n = 120) => { const o = s.replaceAll('\n', ' '); return o.length > n ? o.slice(0, n - 1) + '…' : o; };
-
-/* ── config ─────────────────────────────────────────────── */
-function loadConfig() {
-  try { return JSON.parse(fs.readFileSync(CONFIG_FILE, 'utf8')); } catch { return null; }
-}
-function saveConfig(cfg) {
-  fs.mkdirSync(CONFIG_DIR, { recursive: true });
-  fs.writeFileSync(CONFIG_FILE, JSON.stringify(cfg, null, 2));
-}
-
-/* ── provider API (OpenAI-compatible) ───────────────────── */
-async function chat(cfg, messages, tools) {
-  const body = { model: cfg.model, messages, temperature: 0.2 };
-  if (tools?.length) body.tools = tools.map(t => ({ type: 'function', function: { name: t.name, description: t.description, parameters: t.parameters } }));
-  const res = await fetch(cfg.baseUrl.replace(/\/+$/, '') + '/chat/completions', {
-    method: 'POST',
-    headers: { 'content-type': 'application/json', authorization: `Bearer ${cfg.apiKey}` },
-    body: JSON.stringify(body)
-  });
-  if (!res.ok) throw new Error(`HTTP ${res.status}: ${(await res.text()).slice(0, 200)}`);
-  return (await res.json()).choices?.[0]?.message ?? { role: 'assistant', content: '', tool_calls: [] };
-}
-
-/* ── tools ──────────────────────────────────────────────── */
-const TOOLS = [
-  { name: 'read_file', description: 'Read a text file (UTF-8).', parameters: { type: 'object', properties: { path: { type: 'string' } }, required: ['path'] } },
-  { name: 'write_file', description: 'Create or overwrite a file.', parameters: { type: 'object', properties: { path: { type: 'string' }, content: { type: 'string' } }, required: ['path', 'content'] } },
-  { name: 'list_files', description: 'List files in a directory (recursive, skips node_modules/.git/dist).', parameters: { type: 'object', properties: { path: { type: 'string' } }, required: ['path'] } },
-  { name: 'shell', description: 'Run a shell command and capture stdout, stderr, exit code.', parameters: { type: 'object', properties: { command: { type: 'string' } }, required: ['command'] } }
-];
-
-function runTool(name, input, cwd) {
-  try {
-    if (name === 'read_file') {
-      const p = path.resolve(cwd, input.path);
-      if (path.basename(p).startsWith('.env') || /(^|\/)\.ssh\//.test(p)) return { ok: false, output: 'Refused: looks like a secret file.' };
-      return { ok: true, output: fs.readFileSync(p, 'utf8').slice(0, 60_000) };
-    }
-    if (name === 'write_file') {
-      const p = path.resolve(cwd, input.path);
-      fs.mkdirSync(path.dirname(p), { recursive: true });
-      fs.writeFileSync(p, input.content);
-      return { ok: true, output: `Wrote ${p} (${input.content.length} bytes).` };
-    }
-    if (name === 'list_files') {
-      const out = [];
-      const skip = new Set(['node_modules', '.git', 'dist', 'build', '.next']);
-      (function walk(d, depth) {
-        if (out.length > 400 || depth > 4) return;
-        for (const e of fs.readdirSync(d, { withFileTypes: true })) {
-          if (skip.has(e.name)) continue;
-          out.push((e.isDirectory() ? e.name + '/' : e.name));
-          if (e.isDirectory()) walk(path.join(d, e.name), depth + 1);
-        }
-      })(path.resolve(cwd, input.path || '.'), 0);
-      return { ok: true, output: out.join('\n') };
-    }
-    if (name === 'shell') {
-      return { ok: true, output: 'RUNNING' }; // replaced by async shellRun
-    }
-    return { ok: false, output: `Unknown tool: ${name}` };
-  } catch (err) {
-    return { ok: false, output: `Error: ${err.message}` };
-  }
-}
-
-function shellRun(command, cwd) {
-  return new Promise(resolve => {
-    const child = spawn(command, { cwd, shell: true, env: { ...process.env, NO_COLOR: '1' } });
-    let out = '';
-    const timer = setTimeout(() => { try { child.kill('SIGKILL'); } catch {} out += '\n[timeout after 120s]'; }, 120_000);
-    child.stdout.on('data', c => { out += c.toString(); if (out.length > 100_000) child.kill('SIGKILL'); });
-    child.stderr.on('data', c => { out += c.toString(); });
-    child.on('close', code => { clearTimeout(timer); resolve({ ok: code === 0, output: `exit code: ${code}\n${out.slice(0, 20_000)}` }); });
-    child.on('error', err => { clearTimeout(timer); resolve({ ok: false, output: `spawn error: ${err.message}` }); });
-  });
-}
-
-/* ── agent loop ─────────────────────────────────────────── */
-const SYSTEM = `You are ineed, an autonomous terminal agent on the user's machine.
-Rules:
-- Use the tools to do real work. Never invent command output; success claims need evidence.
-- Prefer targeted edits over full rewrites. Work only inside the current folder.
-- Never push to remotes or delete data without being asked.
-- When the objective is met, reply with the final result: what changed, what you ran, and the evidence.`;
-
-async function runObjective(cfg, objective, cwd, history) {
-  const messages = [
-    { role: 'system', content: `${SYSTEM}\nWorking directory: ${cwd}` },
-    ...history,
-    { role: 'user', content: objective }
-  ];
-  let answer = '';
-
-  for (let step = 0; step < 30; step++) {
-    const msg = await chat(cfg, messages, TOOLS);
-    messages.push(msg);
-
-    if (msg.content) { console.log('  ' + dim(trunc(msg.content, 200))); answer = msg.content; }
-    const calls = msg.tool_calls ?? [];
-    if (calls.length === 0) return { answer, history: [{ role: 'user', content: objective }, { role: 'assistant', content: answer }] };
-
-    for (const call of calls) {
-      const input = JSON.parse(call.function?.arguments || '{}');
-      console.log('  ' + cyan('▸ ' + call.function?.name) + ' ' + gray(trunc(call.function?.arguments || '', 90)));
-      let result;
-      if (call.function?.name === 'shell') result = await shellRun(input.command, cwd);
-      else result = runTool(call.function?.name, input, cwd);
-      console.log('  ' + (result.ok ? green('◂ ok') : red('◂ fail')) + ' ' + gray(trunc(result.output, 100)));
-      messages.push({ role: 'tool', tool_call_id: call.id, content: result.output.slice(0, 20_000) });
-    }
-  }
-  return { answer: answer || 'Stopped: too many steps.', history: [{ role: 'user', content: objective }, { role: 'assistant', content: answer }] };
-}
-
-/* ── setup wizard ───────────────────────────────────────── */
-const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
-let askQueue = [];
-let askPending = null;
-let wizardMode = true;
-let replHandler = null;
-rl.on('line', l => {
-  const line = l.trim();
-  if (askPending) { const r = askPending; askPending = null; r(line); return; }
-  if (wizardMode) { askQueue.push(line); return; }
-  if (replHandler) replHandler(line);
-});
-rl.on('close', () => { if (askPending) { const r = askPending; askPending = null; r(''); } });
-const ask = q => new Promise(res => {
-  process.stdout.write(q + ' ');
-  if (askQueue.length > 0) res(askQueue.shift());
-  else askPending = res;
-});
-
-async function wizard() {
-  console.log('\n' + bold('Welcome to ineed!') + ' Let me connect to your AI provider. One time only.\n');
-  const baseUrl = await ask(cyan('1. API base URL (e.g. https://api.openai.com/v1):'));
-  const apiKey = await ask(cyan('2. API key:'));
-  if (!baseUrl || !apiKey) { console.log(red('Both are required. Run `ineed` again.')); process.exit(1); }
-  console.log(dim('\nChecking…'));
-  let models = [];
-  try {
-    const res = await fetch(baseUrl.replace(/\/+$/, '') + '/models', { headers: { authorization: `Bearer ${apiKey}` } });
-    if (res.ok) models = (await res.json()).data?.map(m => m.id).filter(Boolean) ?? [];
-  } catch {}
-  if (models.length > 0) console.log(green(`✓ Connected. ${models.length} models found.`));
-  else console.log(yellow('⚠ Could not list models (some servers hide them). Type a model id from the provider docs.'));
-  const model = await ask(cyan(`3. Model id ${models.length ? '(e.g. ' + (models.find(m => /gpt-4o-mini|fast|flash/i.test(m)) ?? models[0]) + ')' : ''}:`));
-  if (!model) { console.log(red('Model is required. Run `ineed` again.')); process.exit(1); }
-
-  console.log(dim('\nSending a test message…'));
-  try {
-    const m = await chat({ baseUrl, apiKey, model }, [{ role: 'user', content: 'Reply with exactly: OK' }]);
-    console.log(green('✓ Works!') + dim(` model replied: ${trunc(m.content ?? '', 40)}`));
-  } catch (err) {
-    console.log(yellow(`⚠ Test failed: ${err.message}`));
-    const keep = await ask(yellow('Save anyway? [y/N]:'));
-    if (!/^y/i.test(keep)) { console.log(dim('Not saved. Run `ineed` to retry.')); process.exit(1); }
-  }
-  saveConfig({ baseUrl, apiKey, model });
-  console.log(green('\nSaved to ' + CONFIG_FILE));
-  console.log('All set! Type your goal below.\n');
-}
-
-/* ── main ───────────────────────────────────────────────── */
-const args = process.argv.slice(2);
-if (args[0] === '--version' || args[0] === '-v') { console.log(`ineed ${VERSION}`); process.exit(0); }
-if (args[0] === '--help' || args[0] === '-h') {
-  console.log(`
-${bold('ineed')} v${VERSION} — your terminal, now autonomous
-
-  ${cyan('ineed')}                     interactive session here
-  ${cyan('ineed "do something"')}     one-shot task
-  ${cyan('ineed --reset')}            forget saved config and run the wizard again
-`);
-  process.exit(0);
-}
-
-const cwd = process.cwd();
-let cfg = loadConfig();
-if (args[0] === '--reset') { try { fs.unlinkSync(CONFIG_FILE); } catch {} cfg = null; }
-
-if (!cfg) { await wizard(); cfg = loadConfig(); }
-
-if (args.length) {
-  const { answer } = await runObjective(cfg, args.join(' '), cwd, []);
-  console.log('\n' + bold(green('Done')) + (answer ? '\n  ' + trunc(answer, 500) : ''));
-  process.exit(0);
-}
-
-console.log(bold(green('ineed')) + dim(` v${VERSION} · ${cfg.model} · ${cwd}`));
-console.log(dim('Say what you want. /help for commands. Ctrl+C twice to exit.\n'));
-
-// main REPL reuses the shared rl above
-rl.setPrompt('\n' + bold(green('ineed')) + dim(' ❯ '));
-let lastAnswer = '';
-const history = [];
-
-const handleLine = async input => {
-  if (!input) { rl.prompt(); return; }
-  if (input === '/exit' || input === '/quit' || input === 'exit') { console.log(dim('Goodbye.')); process.exit(0); }
-  if (input === '/help') {
-    console.log('  ' + cyan('/model') + '   show or change model   ' + cyan('/reset') + '  redo setup   ' + cyan('/exit') + '  quit');
-    rl.prompt(); return;
-  }
-  if (input.startsWith('/model')) {
-    console.log('Model: ' + bold(cfg.model) + dim(' — change with `ineed --reset` or edit ' + CONFIG_FILE));
-    rl.prompt(); return;
-  }
-  if (input === '/reset') { try { fs.unlinkSync(CONFIG_FILE); } catch {} console.log(dim('Config cleared. Exiting: run `ineed` to set up again.')); process.exit(0); }
-
-  try {
-    const { answer, history: h } = await runObjective(cfg, input, cwd, history);
-    history.push(...h);
-    lastAnswer = answer;
-    console.log('\n' + bold(green('Done')) + (answer ? '\n  ' + answer.split('\n').slice(0, 12).join('\n  ') : ''));
-  } catch (err) {
-    console.log(red('\nError: ' + err.message));
-  }
-  rl.prompt();
-};
-
-wizardMode = false;
-replHandler = handleLine;
-const backlog = askQueue.splice(0);
-for (const line of backlog) await handleLine(line);
-rl.prompt();
-rl.on('SIGINT', () => { console.log(dim('\nGoodbye.')); process.exit(0); });