@gitset-dev/cli 1.2.0 → 2.0.0

package/lib/cli-config.js

@@ -0,0 +1,207 @@
+'use strict';
+
+/**
+ * `gitset config` — manage local BYOAI provider credentials.
+ * Keys are stored only on this machine (~/.gitset/config.json, chmod 600)
+ * and are sent directly to the chosen AI provider, never to Gitset.
+ */
+const config = require('./config');
+const { SUPPORTED, isForbiddenModel } = require('./ai');
+const theme = require('../src/utils/theme');
+const { selectOption } = require('../src/utils/ui');
+const { runQuickSetup } = require('./setup-wizard');
+
+const ok = (s) => console.log(`${theme.success('✓')} ${s}`);
+const err = (s) => console.error(`${theme.error('✗')} ${s}`);
+
+function flag(argv, name) {
+  const i = argv.indexOf(name);
+  return i !== -1 && i + 1 < argv.length && !argv[i + 1].startsWith('--') ? argv[i + 1] : null;
+}
+const has = (argv, name) => argv.includes(name);
+
+const PROVIDERS = [...SUPPORTED.filter((p) => p !== 'mock'), 'custom'];
+
+function usage() {
+  console.log(`gitset config — BYOAI provider setup
+
+  ${theme.accent('gitset config')}                interactive setup (recommended)
+  gitset config set <provider> --key <api-key> [--model <m>] [--base-url <url>] [--default]
+  gitset config list
+  gitset config remove <provider>
+  gitset config theme <dark|light>
+  gitset config path
+
+Providers: ${PROVIDERS.join(', ')}
+Env override: ANTHROPIC_API_KEY, OPENAI_API_KEY, GEMINI_API_KEY, OPENROUTER_API_KEY, DEEPSEEK_API_KEY`);
+}
+
+function printList() {
+  const rows = config.list();
+  if (rows.length === 0) {
+    console.log('No providers configured yet.');
+    return;
+  }
+  for (const r of rows) {
+    const def = r.isDefault ? theme.success(' (default)') : '';
+    const key = r.keyLast4 ? `••••${r.keyLast4}` : theme.warn('no key');
+    console.log(`${theme.bold(r.provider)}${def}\n  key: ${key}${r.model ? `  model: ${theme.accent(r.model)}` : ''}${r.baseUrl ? `  base: ${r.baseUrl}` : ''}`);
+  }
+}
+
+// Bare `gitset config` — an always-interactive control panel, not a one-shot
+// printout. Nothing set up yet? Straight to the add-provider wizard. Already
+// configured? A real menu (add / remove / default / theme), looping until
+// the user picks "Done" — never a dead end that just prints and exits.
+async function runInteractiveHub() {
+  for (;;) {
+    const rows = config.list();
+    if (rows.length === 0) {
+      return await runQuickSetup();
+    }
+
+    console.log();
+    console.log(theme.bold('Configured providers:'));
+    printList();
+
+    let choice;
+    try {
+      choice = await selectOption('What do you want to do?', [
+        { label: 'Add another provider', value: 'add' },
+        { label: 'Set the default provider', value: 'default' },
+        { label: 'Remove a provider', value: 'remove' },
+        { label: `Switch theme (currently ${theme.mode()})`, value: 'theme' },
+        { label: 'Done', value: 'done' },
+      ]);
+    } catch {
+      return true; // e.g. Ctrl+D — leave quietly, nothing pending to save
+    }
+
+    if (choice === 'done') return true;
+
+    if (choice === 'add') { await runQuickSetup(); continue; }
+
+    if (choice === 'default') {
+      const provider = await selectOption(
+        'Make which provider the default?',
+        rows.map((r) => ({ label: r.provider + (r.isDefault ? '  (current default)' : ''), value: r.provider })),
+      );
+      const row = rows.find((r) => r.provider === provider);
+      config.setProvider(provider, { model: row.model || undefined, baseUrl: row.baseUrl || undefined, makeDefault: true });
+      ok(`${provider} is now the default.`);
+      continue;
+    }
+
+    if (choice === 'remove') {
+      const provider = await selectOption(
+        'Remove which provider?',
+        rows.map((r) => ({ label: r.provider + (r.isDefault ? '  (default)' : ''), value: r.provider })),
+      );
+      config.removeProvider(provider);
+      ok(`Removed ${provider}.`);
+      continue;
+    }
+
+    if (choice === 'theme') {
+      const mode = await selectOption('Pick a theme:', [
+        { label: 'Dark terminal', value: 'dark' },
+        { label: 'Light terminal', value: 'light' },
+      ]);
+      config.setTheme(mode);
+      ok(`Theme set to ${mode}.`);
+      continue;
+    }
+  }
+}
+
+async function runConfigCommand(argv) {
+  const sub = argv[0];
+
+  if (!sub) {
+    if (!process.stdin.isTTY) {
+      if (config.list().length === 0) {
+        err('No AI provider configured. In a non-interactive shell, set one directly:');
+        console.log('  gitset config set anthropic --key <api-key> --default');
+        return 1;
+      }
+      printList();
+      return 0;
+    }
+    const ranOk = await runInteractiveHub();
+    return ranOk ? 0 : 1;
+  }
+
+  if (sub === 'help' || sub === '--help') { usage(); return 0; }
+
+  if (sub === 'path') { console.log(config.FILE); return 0; }
+
+  if (sub === 'list') { printList(); return 0; }
+
+  if (sub === 'theme') {
+    const value = argv[1];
+    if (value !== 'dark' && value !== 'light') {
+      err('Usage: gitset config theme <dark|light>');
+      return 1;
+    }
+    config.setTheme(value);
+    ok(`Theme set to ${value}.`);
+    return 0;
+  }
+
+  if (sub === 'set') {
+    const provider = argv[1] && !argv[1].startsWith('--') ? argv[1].toLowerCase() : null;
+
+    // `gitset config set` with no provider named — same wizard as bare
+    // `gitset config`, just reachable from the more discoverable verb.
+    if (!provider) {
+      if (!process.stdin.isTTY) {
+        err('Usage: gitset config set <provider> --key <api-key> [--model m] [--default]');
+        return 1;
+      }
+      const didSetUp = await runQuickSetup();
+      return didSetUp ? 0 : 1;
+    }
+
+    if (!PROVIDERS.includes(provider)) {
+      err(`Unknown provider "${provider}". One of: ${PROVIDERS.join(', ')}`);
+      return 1;
+    }
+    const apiKey = flag(argv, '--key');
+    const model = flag(argv, '--model');
+    const baseUrl = flag(argv, '--base-url');
+    if (!apiKey && !model && !baseUrl) {
+      err('Nothing to set. Provide at least --key (or --model / --base-url).');
+      return 1;
+    }
+    if (isForbiddenModel(model)) {
+      err(`Model "${model}" can't be used with Gitset — choose a different model.`);
+      return 1;
+    }
+    if (provider === 'custom' && !baseUrl && !config.list().find((x) => x.provider === 'custom')) {
+      err('Provider "custom" requires --base-url (an OpenAI-compatible endpoint).');
+      return 1;
+    }
+    config.setProvider(provider, {
+      apiKey: apiKey || undefined,
+      model: model || undefined,
+      baseUrl: baseUrl || undefined,
+      makeDefault: has(argv, '--default'),
+    });
+    ok(`Saved ${provider}${has(argv, '--default') ? ' (default)' : ''}. Stored locally at ${config.FILE} (chmod 600).`);
+    return 0;
+  }
+
+  if (sub === 'remove') {
+    const provider = argv[1] ? argv[1].toLowerCase() : null;
+    if (!provider) { err('Usage: gitset config remove <provider>'); return 1; }
+    if (config.removeProvider(provider)) { ok(`Removed ${provider}.`); return 0; }
+    err(`Provider "${provider}" was not configured.`);
+    return 1;
+  }
+
+  err(`Unknown subcommand "${sub}".`);
+  usage();
+  return 1;
+}
+
+module.exports = { runConfigCommand };

package/lib/cli-issue.js

@@ -0,0 +1,144 @@
+'use strict';
+
+/**
+ * `gitset issue` — local BYOAI GitHub issue drafting.
+ * Generates a markdown issue from a description (+ optional repo context),
+ * optional interactive refine, optional `gh issue create` (arg-array, no shell).
+ *
+ * Flags: --title <t> --message/-m <desc> --provider --model
+ *        --create (gh issue create) --yes --print --json
+ * Exit: 0 ok · 1 usage · 2 provider error.
+ */
+const { execFileSync } = require('child_process');
+const readline = require('readline');
+const genLocal = require('./generate-local');
+const { suggestLabels } = require('./labels-ai');
+const theme = require('../src/utils/theme');
+
+const out = (s = '') => console.log(s);
+const errln = (s) => console.error(`${theme.error('✗')} ${s}`);
+
+function getFlag(argv, ...names) {
+  for (const n of names) {
+    const i = argv.indexOf(n);
+    if (i !== -1 && argv[i + 1] && !argv[i + 1].startsWith('-')) return argv[i + 1];
+  }
+  return null;
+}
+const has = (argv, ...n) => n.some((x) => argv.includes(x));
+
+function ask(q) {
+  const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+  return new Promise((r) => rl.question(q, (a) => { rl.close(); r(a.trim()); }));
+}
+
+const GENERIC_HEADING = /^(summary|description|overview|context|details|scope|changes|testing|steps\b.*|acceptance\b.*)$/i;
+function deriveTitle(markdown) {
+  const lines = String(markdown).split('\n').map((l) => l.trim());
+  const clean = (l) => l.replace(/^#+\s*/, '').replace(/^\*\*|\*\*$/g, '').trim();
+  for (const l of lines) {
+    if (/^#+\s/.test(l) && !GENERIC_HEADING.test(clean(l))) return clean(l).slice(0, 120);
+  }
+  return null;
+}
+
+async function runIssueCommand(argv) {
+  const json = has(argv, '--json');
+  const printOnly = has(argv, '--print');
+  const nonInteractive = has(argv, '--yes', '-y') || json || printOnly || !process.stdin.isTTY;
+
+  let description = getFlag(argv, '--message', '-m');
+  if (!description) {
+    if (nonInteractive) { errln('Provide an issue description with --message "<text>".'); return 1; }
+    description = await ask('Describe the issue: ');
+  }
+  if (!description) { errln('An issue description is required.'); return 1; }
+
+  let title = getFlag(argv, '--title');
+
+  async function gen(previous, instruction) {
+    if (!json && !printOnly) process.stderr.write(theme.dim('… generating issue\n'));
+    try {
+      return await genLocal.generate({
+        tool: 'issue',
+        ctx: { title: title || description.slice(0, 80), context: description, previous: previous || '', instruction: instruction || '' },
+        provider: getFlag(argv, '--provider'),
+        model: getFlag(argv, '--model'),
+        maxTokens: 2048,
+        interactive: !nonInteractive,
+      });
+    } catch (e) {
+      if (e instanceof genLocal.AIError) { errln(`AI provider error (${e.code}): ${e.message}`); throw { exit: 2 }; }
+      errln(e.message); throw { exit: /gitset config/.test(e.message || '') ? 1 : 2 };
+    }
+  }
+
+  let result;
+  try { result = await gen(); } catch (e) { return e.exit || 2; }
+
+  if (!title) title = deriveTitle(result.text) || description.slice(0, 80);
+  let selectedLabels = [];
+
+  if (json) { out(JSON.stringify({ title, body: result.text, provider: result.provider })); return 0; }
+  if (printOnly) { out(`# ${title}\n\n${result.text}`); return 0; }
+
+  function ensureLabel(l) {
+    try { execFileSync('gh', ['label', 'create', l.name, '--color', l.color || 'ededed', '--description', l.description || ''], { stdio: 'ignore' }); } catch {  }
+  }
+  function create(body) {
+    const args = ['issue', 'create', '--title', title, '--body', body];
+    for (const l of selectedLabels) { ensureLabel(l); args.push('--label', l.name); }
+    try {
+      execFileSync('gh', args, { stdio: 'inherit' });
+      return 0;
+    } catch {
+      errln('gh issue create failed (is `gh` installed & authenticated?).');
+      return 1;
+    }
+  }
+
+  if (nonInteractive) {
+    if (has(argv, '--create')) return create(result.text);
+    out(`# ${title}\n\n${result.text}`);
+    return 0;
+  }
+
+  for (;;) {
+    out();
+    out(theme.bold(`Title: ${title}`));
+    out(theme.accent(result.text.slice(0, 1500) + (result.text.length > 1500 ? '\n… (truncated)' : '')));
+    out();
+    out(theme.dim(`tip: edit ~/.gitset/templates/issue.md (or run \`gitset template edit issue\`) to change the format`));
+    const ch = (await ask(
+      `${theme.key('c', 'reate')}  ${theme.key('l', 'abels')}  ${theme.key('r', 'efine')}  ${theme.key('g', 'regenerate')}  ${theme.key('t', 'itle')}  ${theme.key('p', 'rint')}  ${theme.key('q', 'uit')} > `,
+    )).toLowerCase();
+    if (ch === 'c') return create(result.text);
+    if (ch === 'l') {
+      try {
+        let existing = '';
+        try { existing = JSON.parse(execFileSync('gh', ['label', 'list', '--limit', '100', '--json', 'name'], { encoding: 'utf8', stdio: ['pipe', 'pipe', 'ignore'] })).map((x) => x.name).join(', '); } catch {  }
+        process.stderr.write(theme.dim('… suggesting labels\n'));
+        const sug = await suggestLabels({ title, body: result.text, existing, provider: getFlag(argv, '--provider'), model: getFlag(argv, '--model') });
+        if (sug.length === 0) { out(theme.warn('No label suggestions.')); continue; }
+        out('Suggested: ' + sug.map((l, i) => `[${i + 1}] ${l.name}`).join('  '));
+        const pick = await ask('Apply which? (comma numbers, "all", or blank to skip): ');
+        if (pick.toLowerCase() === 'all') selectedLabels = sug;
+        else { const idx = pick.split(',').map((s) => parseInt(s.trim(), 10) - 1).filter((n) => n >= 0 && n < sug.length); selectedLabels = idx.map((i) => sug[i]); }
+        if (selectedLabels.length) out(theme.success(`✓ labels: ${selectedLabels.map((l) => l.name).join(', ')}`));
+      } catch (e) { errln(e.message); }
+      continue;
+    }
+    if (ch === 'q' || ch === '') { out('Aborted.'); return 0; }
+    if (ch === 't') { const t = await ask('New title: '); if (t) title = t; continue; }
+    if (ch === 'p') { out(`# ${title}\n\n${result.text}`); continue; }
+    if (ch === 'g') { try { result = await gen(); } catch (e) { return e.exit || 2; } continue; }
+    if (ch === 'r') {
+      const instr = await ask('Refinement instruction: ');
+      if (instr) { try { result = await gen(result.text, instr); } catch (e) { return e.exit || 2; } }
+      continue;
+    }
+    out(theme.warn('Unrecognized option.'));
+  }
+}
+
+module.exports = { runIssueCommand };

package/lib/cli-pr.js

@@ -0,0 +1,168 @@
+'use strict';
+
+/**
+ * `gitset pr` — local BYOAI pull-request description from the branch diff.
+ * Generates from `git diff <base>...<head>` + commit log, optional refine,
+ * optional `gh pr create` (arg-array, no shell injection).
+ *
+ * Flags: --base <branch> --title <t> --provider --model
+ *        --create --yes --print --json
+ * Exit: 0 ok · 1 usage/git · 2 provider error.
+ */
+const { execFileSync } = require('child_process');
+const readline = require('readline');
+const genLocal = require('./generate-local');
+const { suggestLabels } = require('./labels-ai');
+const theme = require('../src/utils/theme');
+
+const out = (s = '') => console.log(s);
+const errln = (s) => console.error(`${theme.error('✗')} ${s}`);
+
+function getFlag(argv, ...names) {
+  for (const n of names) {
+    const i = argv.indexOf(n);
+    if (i !== -1 && argv[i + 1] && !argv[i + 1].startsWith('-')) return argv[i + 1];
+  }
+  return null;
+}
+const has = (argv, ...n) => n.some((x) => argv.includes(x));
+function ask(q) {
+  const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+  return new Promise((r) => rl.question(q, (a) => { rl.close(); r(a.trim()); }));
+}
+function git(args) {
+  return execFileSync('git', args, { encoding: 'utf8', maxBuffer: 64 * 1024 * 1024 }).trim();
+}
+
+const GENERIC_HEADING = /^(summary|description|overview|context|details|scope|changes|testing|steps\b.*|acceptance\b.*)$/i;
+function deriveTitle(md) {
+  const lines = String(md).split('\n').map((x) => x.trim());
+  const clean = (l) => l.replace(/^#+\s*/, '').replace(/^(title:|\*\*)\s*/i, '').replace(/\*\*$/, '').trim();
+  for (const l of lines) {
+    if (/^#+\s/.test(l) && !GENERIC_HEADING.test(clean(l))) return clean(l).slice(0, 120);
+  }
+  return null;
+}
+
+function titleFromBranch(branch) {
+  const tail = String(branch || '').split('/').pop() || 'Pull request';
+  const words = tail.replace(/[-_]+/g, ' ').trim();
+  return words ? words.charAt(0).toUpperCase() + words.slice(1) : 'Pull request';
+}
+
+async function runPrCommand(argv) {
+  const json = has(argv, '--json');
+  const printOnly = has(argv, '--print');
+  const nonInteractive = has(argv, '--yes', '-y') || json || printOnly || !process.stdin.isTTY;
+
+  let head;
+  try { head = git(['rev-parse', '--abbrev-ref', 'HEAD']); }
+  catch { errln('Not a git repository.'); return 1; }
+
+  let base = getFlag(argv, '--base');
+  if (!base) {
+    for (const cand of ['main', 'master', 'develop']) {
+      try { git(['rev-parse', '--verify', cand]); base = cand; break; } catch {  }
+    }
+    base = base || 'main';
+  }
+  if (base === head) { errln(`Base and head are both "${head}". Pass --base <branch>.`); return 1; }
+
+  let diff = '';
+  let commits = '';
+  try {
+    diff = git(['diff', `${base}...${head}`, '--no-color']);
+    commits = git(['log', `${base}..${head}`, '--pretty=format:- %h %s (%an)']);
+  } catch {
+    errln(`Could not diff ${base}...${head}. Does base "${base}" exist?`); return 1;
+  }
+  if (!diff.trim()) { errln(`No changes between ${base} and ${head}.`); return 1; }
+
+  let title = getFlag(argv, '--title');
+
+  async function gen(previous, instruction) {
+    if (!json && !printOnly) process.stderr.write(theme.dim(`… generating PR (${base}...${head})\n`));
+    try {
+      return await genLocal.generate({
+        tool: 'pr',
+        ctx: { diff, commits, previous: previous || '', instruction: instruction || '' },
+        provider: getFlag(argv, '--provider'),
+        model: getFlag(argv, '--model'),
+        maxTokens: 4096,
+        interactive: !nonInteractive,
+      });
+    } catch (e) {
+      if (e instanceof genLocal.AIError) { errln(`AI provider error (${e.code}): ${e.message}`); throw { exit: 2 }; }
+      errln(e.message); throw { exit: /gitset config/.test(e.message || '') ? 1 : 2 };
+    }
+  }
+
+  let result;
+  try { result = await gen(); } catch (e) { return e.exit || 2; }
+
+  if (!title) title = deriveTitle(result.text) || titleFromBranch(head);
+  let selectedLabels = [];
+
+  if (json) { out(JSON.stringify({ title, body: result.text, base, head, provider: result.provider })); return 0; }
+  if (printOnly) { out(`# ${title}\n\n${result.text}`); return 0; }
+
+  function ensureLabel(l) {
+    try { execFileSync('gh', ['label', 'create', l.name, '--color', l.color || 'ededed', '--description', l.description || ''], { stdio: 'ignore' }); } catch {  }
+  }
+  function create(body) {
+    const args = ['pr', 'create', '--base', base, '--head', head, '--title', title, '--body', body];
+    for (const l of selectedLabels) { ensureLabel(l); args.push('--label', l.name); }
+    try {
+      execFileSync('gh', args, { stdio: 'inherit' });
+      return 0;
+    } catch {
+      errln('gh pr create failed (is `gh` installed & authenticated, branch pushed?).');
+      return 1;
+    }
+  }
+
+  if (nonInteractive) {
+    if (has(argv, '--create')) return create(result.text);
+    out(`# ${title}\n\n${result.text}`);
+    return 0;
+  }
+
+  for (;;) {
+    out();
+    out(theme.bold(`${title}  ${theme.dim(`(${base} ← ${head})`)}`));
+    out(theme.accent(result.text.slice(0, 1500) + (result.text.length > 1500 ? '\n… (truncated)' : '')));
+    out();
+    out(theme.dim(`tip: edit ~/.gitset/templates/pr.md (or run \`gitset template edit pr\`) to change the format`));
+    const ch = (await ask(
+      `${theme.key('c', 'reate')}  ${theme.key('l', 'abels')}  ${theme.key('r', 'efine')}  ${theme.key('g', 'regenerate')}  ${theme.key('t', 'itle')}  ${theme.key('p', 'rint')}  ${theme.key('q', 'uit')} > `,
+    )).toLowerCase();
+    if (ch === 'c') return create(result.text);
+    if (ch === 'l') {
+      try {
+        let existing = '';
+        try { existing = JSON.parse(execFileSync('gh', ['label', 'list', '--limit', '100', '--json', 'name'], { encoding: 'utf8', stdio: ['pipe', 'pipe', 'ignore'] })).map((x) => x.name).join(', '); } catch {  }
+        process.stderr.write(theme.dim('… suggesting labels\n'));
+        const sug = await suggestLabels({ title, body: result.text, existing, provider: getFlag(argv, '--provider'), model: getFlag(argv, '--model') });
+        if (sug.length === 0) { out(theme.warn('No label suggestions.')); continue; }
+        out('Suggested: ' + sug.map((l, i) => `[${i + 1}] ${l.name}`).join('  '));
+        const pick = await ask('Apply which? (comma numbers, "all", or blank to skip): ');
+        if (pick.toLowerCase() === 'all') selectedLabels = sug;
+        else { const idx = pick.split(',').map((s) => parseInt(s.trim(), 10) - 1).filter((n) => n >= 0 && n < sug.length); selectedLabels = idx.map((i) => sug[i]); }
+        if (selectedLabels.length) out(theme.success(`✓ labels: ${selectedLabels.map((l) => l.name).join(', ')}`));
+      } catch (e) { errln(e.message); }
+      continue;
+    }
+    if (ch === 'q' || ch === '') { out('Aborted.'); return 0; }
+    if (ch === 't') { const t = await ask('New title: '); if (t) title = t; continue; }
+    if (ch === 'p') { out(`# ${title}\n\n${result.text}`); continue; }
+    if (ch === 'g') { try { result = await gen(); } catch (e) { return e.exit || 2; } continue; }
+    if (ch === 'r') {
+      const instr = await ask('Refinement instruction: ');
+      if (instr) { try { result = await gen(result.text, instr); } catch (e) { return e.exit || 2; } }
+      continue;
+    }
+    out(theme.warn('Unrecognized option.'));
+  }
+}
+
+module.exports = { runPrCommand };

package/lib/cli-readme.js

@@ -0,0 +1,153 @@
+'use strict';
+
+/**
+ * `gitset readme` — local BYOAI README generation.
+ * Gathers repo context locally (git-tracked files + key file contents),
+ * generates via the user's own provider, no backend.
+ *
+ * Flags: --provider --model --output <file> --force --yes --print --json
+ * Exit: 0 ok · 1 usage/io · 2 provider error.
+ */
+const fs = require('fs');
+const path = require('path');
+const { execFileSync } = require('child_process');
+const readline = require('readline');
+const genLocal = require('./generate-local');
+const theme = require('../src/utils/theme');
+
+const out = (s = '') => console.log(s);
+const errln = (s) => console.error(`${theme.error('✗')} ${s}`);
+
+function getFlag(argv, n) {
+  const i = argv.indexOf(n);
+  return i !== -1 && argv[i + 1] && !argv[i + 1].startsWith('-') ? argv[i + 1] : null;
+}
+const has = (argv, ...n) => n.some((x) => argv.includes(x));
+
+function ask(q) {
+  const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+  return new Promise((r) => rl.question(q, (a) => { rl.close(); r(a.trim()); }));
+}
+
+const KEY_FILES = [
+  'package.json', 'pyproject.toml', 'Cargo.toml', 'go.mod', 'pom.xml',
+  'composer.json', 'Gemfile', 'requirements.txt', 'tsconfig.json',
+  'src/index.ts', 'src/index.js', 'src/main.ts', 'src/main.py',
+  'index.js', 'main.go', 'main.py', 'app.py',
+];
+
+function gatherContext(cwd) {
+  let files = [];
+  try {
+    files = execFileSync('git', ['ls-files'], { cwd, encoding: 'utf8', maxBuffer: 16 * 1024 * 1024 })
+      .split('\n').map((s) => s.trim()).filter(Boolean);
+  } catch {
+    errln('Not a git repository (run inside your repo).');
+    return null;
+  }
+  const tree = files.slice(0, 400).join('\n');
+  let ctx = `FILE TREE (${files.length} tracked files):\n${tree}\n`;
+
+  let budget = 16000;
+  for (const f of KEY_FILES) {
+    if (budget <= 0) break;
+    if (!files.includes(f)) continue;
+    try {
+      const body = fs.readFileSync(path.join(cwd, f), 'utf8').slice(0, Math.min(budget, 6000));
+      ctx += `\n--- ${f} ---\n${body}\n`;
+      budget -= body.length;
+    } catch {  }
+  }
+  const projectName = (() => {
+    try {
+      if (files.includes('package.json')) {
+        return JSON.parse(fs.readFileSync(path.join(cwd, 'package.json'), 'utf8')).name || path.basename(cwd);
+      }
+    } catch {  }
+    return path.basename(cwd);
+  })();
+  return { projectName, context: ctx };
+}
+
+async function runReadmeCommand(argv) {
+  const cwd = process.cwd();
+  const json = has(argv, '--json');
+  const printOnly = has(argv, '--print');
+  const nonInteractive = has(argv, '--yes', '-y') || json || printOnly || !process.stdin.isTTY;
+  const outFile = path.join(cwd, getFlag(argv, '--output') || 'README.md');
+
+  const gathered = gatherContext(cwd);
+  if (!gathered) return 1;
+
+  let templateText = '';
+  const tmplPath = getFlag(argv, '--template');
+  if (tmplPath) {
+    try { templateText = fs.readFileSync(tmplPath, 'utf8'); }
+    catch { errln(`Template not found: ${tmplPath}`); return 1; }
+  }
+
+  async function gen(previous, instruction) {
+    if (!json && !printOnly) process.stderr.write(theme.dim('… generating README\n'));
+    try {
+      const r = await genLocal.generate({
+        tool: 'readme',
+        ctx: { projectName: gathered.projectName, context: gathered.context, template: templateText, previous: previous || '', instruction: instruction || '' },
+        provider: getFlag(argv, '--provider'),
+        model: getFlag(argv, '--model'),
+        maxTokens: 8192,
+        interactive: !nonInteractive,
+      });
+      return r;
+    } catch (e) {
+      if (e instanceof genLocal.AIError) { errln(`AI provider error (${e.code}): ${e.message}`); throw { exit: 2 }; }
+      errln(e.message); throw { exit: e.message && /gitset config/.test(e.message) ? 1 : 2 };
+    }
+  }
+
+  let result;
+  try { result = await gen(); } catch (e) { return e.exit || 2; }
+
+  if (json) { out(JSON.stringify({ content: result.text, provider: result.provider, model: result.model })); return 0; }
+  if (printOnly) { out(result.text); return 0; }
+
+  function write(content) {
+    if (fs.existsSync(outFile) && !has(argv, '--force') && nonInteractive) {
+      errln(`${path.basename(outFile)} exists. Use --force to overwrite.`);
+      return 1;
+    }
+    fs.writeFileSync(outFile, content.endsWith('\n') ? content : content + '\n');
+    out(`${theme.success('✓')} Wrote ${path.relative(cwd, outFile)} (${content.length} bytes, via ${result.provider})`);
+    return 0;
+  }
+
+  if (nonInteractive) return write(result.text);
+
+  for (;;) {
+    out();
+    out(theme.bold(`README preview (${result.text.length} chars, via ${result.provider}):`));
+    out(theme.dim(result.text.slice(0, 1200) + (result.text.length > 1200 ? '\n… (truncated preview)' : '')));
+    out();
+    if (!tmplPath) out(theme.dim(`tip: edit ~/.gitset/templates/readme.md (or run \`gitset template edit readme\`) to change the format`));
+    const ch = (await ask(
+      `${theme.key('w', 'rite')}  ${theme.key('r', 'efine')}  ${theme.key('g', 'regenerate')}  ${theme.key('p', 'rint')}  ${theme.key('q', 'uit')} > `,
+    )).toLowerCase();
+    if (ch === 'w' || ch === '') {
+      if (fs.existsSync(outFile) && !has(argv, '--force')) {
+        const ow = (await ask(`${path.basename(outFile)} exists. Overwrite? [y/N] `)).toLowerCase();
+        if (ow !== 'y') continue;
+      }
+      return write(result.text);
+    }
+    if (ch === 'q') { out('Aborted.'); return 0; }
+    if (ch === 'p') { out(result.text); continue; }
+    if (ch === 'g') { try { result = await gen(); } catch (e) { return e.exit || 2; } continue; }
+    if (ch === 'r') {
+      const instr = await ask('Refinement instruction: ');
+      if (instr) { try { result = await gen(result.text, instr); } catch (e) { return e.exit || 2; } }
+      continue;
+    }
+    out(theme.warn('Unrecognized option.'));
+  }
+}
+
+module.exports = { runReadmeCommand };