primocode 11.0.0 → 11.1.0

package/lib/cli/headless.js

@@ -0,0 +1,102 @@
+/**
+ * headless.js — `primocode -p "pedido"`: sem tela, para script e CI.
+ *
+ *   primocode -p "arruma o teste"                 texto no stdout
+ *   primocode -p "..." --output-format json       um JSON no fim: {type, is_error, result, duration_ms}
+ *   git diff | primocode -p "revisa isso"         o stdin entra no pedido
+ *   primocode -p "..." --permission-mode plan     só lê (ver lib/permission/perigo.js)
+ *
+ * Mesma ideia do `claude -p` e do `codex exec` (flags documentadas deles; o
+ * código é nosso). O pedido passa pelo MESMO caminho do digitado (sendPrompt).
+ * Código de saída: 0 ok, 1 erro.
+ */
+'use strict';
+
+const path = require('path');
+const { stripAnsi } = require('../tui/ui');
+
+const FORMATOS = ['text', 'json'];
+
+/** Tira -p/--print, --output-format e --permission-mode dos argumentos. */
+function analisar(argv) {
+    const r = { ativo: false, formato: 'text', permissao: null, resto: [], erro: null };
+    for (let i = 0; i < argv.length; i++) {
+        const a = argv[i];
+        const [chave, valor] = a.startsWith('--') && a.includes('=') ? [a.slice(0, a.indexOf('=')), a.slice(a.indexOf('=') + 1)] : [a, null];
+        if (a === '-p' || a === '--print') r.ativo = true;
+        else if (chave === '--output-format') {
+            const v = valor !== null ? valor : argv[++i];
+            if (!FORMATOS.includes(v)) r.erro = `--output-format aceita ${FORMATOS.join(' ou ')}`;
+            else r.formato = v;
+        } else if (chave === '--permission-mode') r.permissao = valor !== null ? valor : argv[++i];
+        else r.resto.push(a);
+    }
+    return r;
+}
+
+/** O pedido: os argumentos soltos, mais o stdin (que vem depois, como contexto). */
+function montarPedido(palavras, stdin) {
+    const pedido = palavras.join(' ').trim();
+    const extra = String(stdin || '').trim();
+    if (pedido && extra) return `${pedido}\n\n${extra}`;
+    return pedido || extra;
+}
+
+function lerStdin() {
+    if (process.stdin.isTTY) return Promise.resolve('');
+    return new Promise((resolve) => {
+        let s = '';
+        process.stdin.setEncoding('utf8');
+        process.stdin.on('data', (d) => { s += d; });
+        process.stdin.on('end', () => resolve(s));
+        process.stdin.on('error', () => resolve(s));
+    });
+}
+
+/** Roda `fn` guardando tudo o que ela escreve. `calar` esconde da tela. */
+async function capturar(fn, { calar = false } = {}) {
+    const log = console.log;
+    const escrever = process.stdout.write.bind(process.stdout);
+    let texto = '';
+    const pega = (s) => { texto += stripAnsi(s); };
+    console.log = (...a) => { pega(a.join(' ') + '\n'); if (!calar) log(...a); };
+    process.stdout.write = (s, ...resto) => { pega(typeof s === 'string' ? s : Buffer.from(s).toString()); return calar ? true : escrever(s, ...resto); };
+    try { await fn(); } finally { console.log = log; process.stdout.write = escrever; }
+    return texto.replace(/\n{3,}/g, '\n\n').trim();
+}
+
+/**
+ * A pasta do pedido no modo script: a de onde o comando rodou. Antes o `-p`
+ * ignorava a pasta e trabalhava no último projeto da sessão (currentProject):
+ * `cd meu-app && primocode -p "grave o PLANO.md"` gravava em outro lugar.
+ * null = segue o padrão: na casa, na raiz do disco e no ~/primocode (lá o
+ * PrimoCode escolhe ou cria o projeto), e no app, que manda a pasta dele.
+ */
+function pastaDoScript({ cwd, casa, raiz, app = false } = {}) {
+    if (app || !cwd) return null;
+    // realpath: no macOS a pasta temporária é /var e /private/var ao mesmo tempo
+    const real = (p) => { try { return require('fs').realpathSync(p); } catch { return path.resolve(p); } };
+    const r = real(cwd);
+    if (r === path.parse(r).root) return null;
+    if (casa && r === real(casa)) return null;
+    if (raiz && r === real(raiz)) return null;
+    return r;
+}
+
+function resultadoJson({ texto, erro, inicio, permissao, conversa, markdown }) {
+    return JSON.stringify({ type: 'result', is_error: !!erro, result: erro ? String(erro) : texto, duration_ms: Date.now() - inicio, permission_mode: permissao || 'auto', conversa: conversa || null, markdown: erro ? null : (markdown || null) });
+}
+
+/* A última resposta do agente como ele escreveu (markdown cru), sem o que o terminal desenha em volta. */
+function ultimaResposta(historico, desde = 0) {
+    for (let i = (historico || []).length - 1; i >= desde; i--) {
+        const m = historico[i];
+        if (!m || m.role !== 'assistant') continue;
+        const t = typeof m.content === 'string' ? m.content
+            : Array.isArray(m.content) ? m.content.map((p) => (typeof p === 'string' ? p : (p && p.type === 'text' && p.text) || '')).join('\n') : '';
+        if (t.trim()) return t.trim();
+    }
+    return null;
+}
+
+module.exports = { analisar, montarPedido, lerStdin, capturar, resultadoJson, ultimaResposta, pastaDoScript, FORMATOS };