@thegitai/cli 1.0.0-preview.5 → 1.0.0-preview.51

package/dist/src/permissions.js

@@ -0,0 +1,288 @@
+import { getUnquotedShellText } from './agent-mode.js';
+export const PERMISSION_BUCKETS = ['create', 'edit', 'delete', 'run', 'browse'];
+export function createSessionGrants() {
+    return { buckets: [], commandPrefixes: [], origins: [] };
+}
+export function bucketActionLabel(bucket) {
+    if (bucket === 'create')
+        return 'create files';
+    if (bucket === 'edit')
+        return 'edit files';
+    if (bucket === 'delete')
+        return 'delete files';
+    if (bucket === 'browse')
+        return 'act on any website';
+    return 'run commands';
+}
+export function originForUrl(url) {
+    const raw = String(url ?? '').trim();
+    if (!raw)
+        return null;
+    try {
+        const parsed = new URL(/^[a-z][a-z0-9+.-]*:/i.test(raw) ? raw : `https://${raw}`);
+        if (parsed.protocol === 'about:' || parsed.protocol === 'chrome:')
+            return null;
+        return parsed.origin;
+    }
+    catch {
+        return null;
+    }
+}
+export function isOriginGranted(grants, origin) {
+    if (!grants || !origin)
+        return false;
+    return grants.origins?.includes(origin) === true;
+}
+export function grantOrigin(grants, origin) {
+    const normalized = String(origin ?? '').trim();
+    if (!normalized)
+        return;
+    const origins = grants.origins ??= [];
+    if (!origins.includes(normalized))
+        origins.push(normalized);
+}
+const NO_PREFIX_GRANT_BINARIES = new Set([
+    'rm',
+    'rmdir',
+    'mv',
+    'dd',
+    'mkfs',
+    'shred',
+    'sudo',
+    'doas',
+    'su',
+    'chmod',
+    'chown',
+    'chgrp',
+    'kill',
+    'pkill',
+    'killall',
+    'shutdown',
+    'reboot',
+    'curl',
+    'wget',
+    'ssh',
+    'scp',
+    'nc',
+    'eval',
+    'exec',
+    'source',
+]);
+const INTERPRETER_OR_WRAPPER_BINARIES = new Set([
+    'bash',
+    'sh',
+    'zsh',
+    'fish',
+    'dash',
+    'ksh',
+    'csh',
+    'tcsh',
+    'python',
+    'python2',
+    'python3',
+    'node',
+    'deno',
+    'bun',
+    'ruby',
+    'perl',
+    'php',
+    'lua',
+    'osascript',
+    'awk',
+    'gawk',
+    'sed',
+    'env',
+    'timeout',
+    'xargs',
+    'nohup',
+    'setsid',
+    'watch',
+    'script',
+    'nice',
+    'ionice',
+    'stdbuf',
+    'time',
+    'sudo',
+    'doas',
+    'su',
+    'find',
+    'make',
+]);
+const SCRIPT_RUNNER_VERBS = new Set(['run', 'exec', 'run-script', 'x', 'dlx']);
+const VERB_PATTERN = /^[A-Za-z][A-Za-z0-9_:-]*$/;
+const BINARY_PATTERN = /^[A-Za-z0-9][A-Za-z0-9._+-]*$/;
+const SHELL_CONTROL_OPERATOR_PATTERN = /[&|;<>\n]/;
+const COMMAND_SUBSTITUTION_PATTERN = /`|\$\(/;
+function normalizeCommand(command) {
+    return String(command ?? '').trim().replace(/\s+/g, ' ');
+}
+function parseCommand(command) {
+    const raw = String(command ?? '');
+    if (!raw.trim())
+        return null;
+    if (COMMAND_SUBSTITUTION_PATTERN.test(raw))
+        return null;
+    if (SHELL_CONTROL_OPERATOR_PATTERN.test(getUnquotedShellText(raw)))
+        return null;
+    const all = normalizeCommand(raw).match(/\S+/g) ?? [];
+    const tokens = [];
+    for (const token of all) {
+        if (!tokens.length && /^[A-Za-z_][A-Za-z0-9_]*=/.test(token))
+            continue;
+        if (!tokens.length && token === 'command')
+            continue;
+        tokens.push(token);
+    }
+    if (!tokens.length)
+        return null;
+    const binary = (tokens[0].split('/').pop() ?? '').trim();
+    if (!binary)
+        return null;
+    tokens[0] = binary;
+    return { canonical: tokens.join(' '), tokens };
+}
+function grantPrefixFromTokens(tokens) {
+    const binary = tokens[0];
+    if (!BINARY_PATTERN.test(binary))
+        return null;
+    if (NO_PREFIX_GRANT_BINARIES.has(binary))
+        return null;
+    if (INTERPRETER_OR_WRAPPER_BINARIES.has(binary))
+        return null;
+    const verb = tokens[1];
+    if (!verb || !VERB_PATTERN.test(verb))
+        return null;
+    if (SCRIPT_RUNNER_VERBS.has(verb)) {
+        const script = tokens[2];
+        if (!script || !VERB_PATTERN.test(script))
+            return null;
+        return `${binary} ${verb} ${script}`;
+    }
+    return `${binary} ${verb}`;
+}
+export function commandGrantPrefix(command) {
+    const parsed = parseCommand(command);
+    return parsed ? grantPrefixFromTokens(parsed.tokens) : null;
+}
+export function isBucketGranted(grants, bucket) {
+    return Boolean(grants?.buckets.includes(bucket));
+}
+export function isCommandGranted(grants, command) {
+    if (!grants)
+        return false;
+    if (grants.buckets.includes('run'))
+        return true;
+    if (!grants.commandPrefixes.length)
+        return false;
+    const parsed = parseCommand(command);
+    if (!parsed)
+        return false;
+    if (grantPrefixFromTokens(parsed.tokens) === null)
+        return false;
+    return grants.commandPrefixes.some((prefix) => parsed.canonical === prefix || parsed.canonical.startsWith(`${prefix} `));
+}
+export function grantBucket(grants, bucket) {
+    if (!grants.buckets.includes(bucket))
+        grants.buckets.push(bucket);
+}
+export function grantCommandPrefix(grants, prefix) {
+    const normalized = normalizeCommand(prefix);
+    if (!normalized)
+        return;
+    if (!grants.commandPrefixes.includes(normalized)) {
+        grants.commandPrefixes.push(normalized);
+    }
+}
+export function buildPermissionOptions(bucket, command, origin) {
+    const options = [
+        { label: 'Approve once', decision: { kind: 'once' } },
+    ];
+    const prefix = bucket === 'run' && command ? commandGrantPrefix(command) : null;
+    if (prefix) {
+        options.push({
+            label: `Always allow: ${prefix}`,
+            decision: { kind: 'always-prefix', prefix },
+        });
+    }
+    if (bucket === 'browse' && origin) {
+        options.push({
+            label: `Allow this site for the session: ${origin}`,
+            decision: { kind: 'always-origin', origin },
+        });
+    }
+    options.push({
+        label: `Always allow: ${bucketActionLabel(bucket)}`,
+        decision: { kind: 'always-bucket' },
+    });
+    options.push({ label: 'Deny', decision: { kind: 'deny' } });
+    return options;
+}
+function declinedSubject(bucket) {
+    if (bucket === 'run') {
+        return { noun: 'command', effect: 'Nothing was executed' };
+    }
+    if (bucket === 'create') {
+        return { noun: 'file creation', effect: 'Nothing was created' };
+    }
+    if (bucket === 'delete') {
+        return { noun: 'file deletion', effect: 'Nothing was deleted' };
+    }
+    if (bucket === 'browse') {
+        return { noun: 'browser action', effect: 'The page was not touched' };
+    }
+    return { noun: 'edit', effect: 'Nothing was changed' };
+}
+function declinedResponse(bucket, toolName, extra) {
+    const { noun, effect } = declinedSubject(bucket);
+    return {
+        ok: false,
+        skipped: true,
+        ...extra,
+        failureCategory: 'user_declined',
+        failureDetails: {
+            category: 'user_declined',
+            tool: toolName,
+            action: 'Respect the real user’s decision. Do not retry the same or an equivalent action; reconsider the approach or ask one specific question if needed.',
+        },
+        error: `The real user rejected this proposed ${noun}. ${effect}; this was not a tool failure or an automated system skip.`,
+    };
+}
+export async function ensurePermission(context, request, toolName, extra = {}) {
+    if (context.autoYes)
+        return null;
+    if (isBucketGranted(context.grants, request.bucket))
+        return null;
+    if (request.bucket === 'run' &&
+        request.command &&
+        isCommandGranted(context.grants, request.command)) {
+        return null;
+    }
+    if (request.bucket === 'browse' &&
+        isOriginGranted(context.grants, request.origin ?? null)) {
+        return null;
+    }
+    if (!context.requestPermission) {
+        return {
+            ok: false,
+            ...extra,
+            error: `requestPermission is required when autoYes is false (${toolName})`,
+        };
+    }
+    const options = buildPermissionOptions(request.bucket, request.command, request.origin);
+    const decision = await context.requestPermission({ ...request, options });
+    if (decision.kind === 'deny') {
+        return declinedResponse(request.bucket, toolName, extra);
+    }
+    if (context.grants) {
+        if (decision.kind === 'always-bucket') {
+            grantBucket(context.grants, request.bucket);
+        }
+        else if (decision.kind === 'always-prefix') {
+            grantCommandPrefix(context.grants, decision.prefix);
+        }
+        else if (decision.kind === 'always-origin') {
+            grantOrigin(context.grants, decision.origin);
+        }
+    }
+    return null;
+}

package/dist/src/preferences.js

@@ -0,0 +1,39 @@
+import { existsSync, mkdirSync, readFileSync, renameSync, rmSync, writeFileSync } from 'node:fs';
+import path from 'node:path';
+import { getClientStateDir } from './client-state.js';
+export const DEFAULT_PREFERENCES = {
+    notifications: true,
+};
+export function getPreferencesPath(env = process.env) {
+    return path.join(getClientStateDir(env), 'prefs.json');
+}
+export function readPreferences(env = process.env) {
+    const filePath = getPreferencesPath(env);
+    if (!existsSync(filePath))
+        return { ...DEFAULT_PREFERENCES };
+    try {
+        const parsed = JSON.parse(readFileSync(filePath, 'utf8'));
+        return {
+            notifications: parsed.notifications === false ? false : DEFAULT_PREFERENCES.notifications,
+        };
+    }
+    catch {
+        return { ...DEFAULT_PREFERENCES };
+    }
+}
+export function writePreferences(preferences, env = process.env) {
+    const filePath = getPreferencesPath(env);
+    mkdirSync(path.dirname(filePath), { recursive: true, mode: 0o700 });
+    const tempPath = `${filePath}.${process.pid}.tmp`;
+    try {
+        writeFileSync(tempPath, `${JSON.stringify(preferences, null, 2)}\n`, {
+            encoding: 'utf8',
+            mode: 0o600,
+        });
+        renameSync(tempPath, filePath);
+    }
+    catch (error) {
+        rmSync(tempPath, { force: true });
+        throw error;
+    }
+}

package/dist/src/quota-notice.js

@@ -0,0 +1,35 @@
+import { isQuotaExceededError, ServerApiError } from './api/http.js';
+import { USAGE_WINDOWS, formatUtcMinute, usageWindowLabel } from './usage.js';
+export function buildQuotaNotice(error) {
+    if (!(error instanceof ServerApiError) || !isQuotaExceededError(error)) {
+        return null;
+    }
+    const label = usageWindowLabel(String(error.details.window ?? ''));
+    const inSentence = label.toLowerCase();
+    const resetsAt = error.details.resetsAt ?? null;
+    const lines = [];
+    if (label) {
+        const reset = resetsAt ? formatUtcMinute(resetsAt) : '';
+        lines.push(reset && reset !== 'unknown'
+            ? `You've used your ${inSentence} limit. It resets at ${reset}.`
+            : `You've used your ${inSentence} limit.`);
+    }
+    else {
+        lines.push(firstMessageLine(error));
+    }
+    lines.push(label
+        ? `Run /usage to see your ${otherWindowNames(label)} windows.`
+        : 'Run /usage to see your remaining usage.');
+    return { title: 'Usage limit reached', body: lines.join('\n') };
+}
+function otherWindowNames(usedLabel) {
+    const names = USAGE_WINDOWS.map(([name]) => name)
+        .filter((name) => name !== usedLabel)
+        .map((name) => name.toLowerCase());
+    return names.length > 1
+        ? `${names.slice(0, -1).join(', ')} and ${names[names.length - 1]}`
+        : (names[0] ?? '');
+}
+function firstMessageLine(error) {
+    return String(error.message ?? '').split('\n')[0]?.trim() || 'Usage limit reached.';
+}

package/dist/src/session-safety.js

@@ -4,7 +4,6 @@ import path from 'node:path';
 import { ARTIFACT_IGNORE_DIRS, normalizeProjectRelativePath, shouldIgnoreArtifactPath, } from './artifact-policy.js';
 import { hashBytes, readFileEditSnapshot, storedContentBuffer, } from './edit-journal.js';
 import { deleteProjectFile, resolveProjectPath, writeProjectFile, writeProjectFileBuffer, } from './patcher.js';
-import { removeIndexFile, upsertIndexFile } from './project-index.js';
 const MAX_CHECKPOINTS = 20;
 const MAX_SESSION_EDITS = 500;
 const MAX_READ_RECORDS = 200;
@@ -672,7 +671,6 @@ export async function restoreCheckpointFiles(args) {
     const restored = [];
     const applied = [];
     let changed = false;
-    const syncPolicy = args.projectIndex.initialized;
     for (const snapshot of targets) {
         const before = readFileEditSnapshot(args.rootDir, snapshot.filePath);
         try {
@@ -680,15 +678,11 @@ export async function restoreCheckpointFiles(args) {
                 const result = writeStoredProjectFile(args.rootDir, snapshot.filePath, snapshot.content ?? '', snapshot.contentEncoding);
                 if (result.changed)
                     changed = true;
-                if (syncPolicy)
-                    await upsertIndexFile(args.projectIndex, snapshot.filePath);
             }
             else {
                 const result = deleteProjectFile(args.rootDir, snapshot.filePath);
                 if (result.deleted)
                     changed = true;
-                if (syncPolicy)
-                    await removeIndexFile(args.projectIndex, snapshot.filePath);
             }
             const after = readFileEditSnapshot(args.rootDir, snapshot.filePath);
             applied.push({ snapshot, before, after });
@@ -710,15 +704,9 @@ export async function restoreCheckpointFiles(args) {
                             throw new Error('previous file content is unavailable');
                         }
                         writeStoredProjectFile(args.rootDir, item.snapshot.filePath, item.before.content, item.before.contentEncoding);
-                        if (syncPolicy) {
-                            await upsertIndexFile(args.projectIndex, item.snapshot.filePath);
-                        }
                     }
                     else {
                         deleteProjectFile(args.rootDir, item.snapshot.filePath);
-                        if (syncPolicy) {
-                            await removeIndexFile(args.projectIndex, item.snapshot.filePath);
-                        }
                     }
                     rolledBack.push({ filePath: item.snapshot.filePath });
                 }

package/dist/src/session-store.js

@@ -1,12 +1,16 @@
+import { spawnSync } from 'node:child_process';
 import { createHash } from 'node:crypto';
 import { existsSync, mkdirSync, readFileSync, readdirSync, renameSync, rmSync, writeFileSync, } from 'node:fs';
 import path from 'node:path';
 import { getClientStateDir } from './client-state.js';
+import { findStoredImageByContent, pruneSessionImages, readSessionImage, sweepOrphanSessionImages, } from './core/session-image-store.js';
+import { sweepGeneratedImages } from './tools/save-generated-image.js';
 import { normalizeAssistantEditJournal } from './edit-journal.js';
+import { createSessionGrants } from './permissions.js';
 import { cloneSessionSafetyState, createSessionSafetyState, mergeLocalSessionSafetyState, normalizeSessionSafetyState, } from './session-safety.js';
-import { truncate } from './utils.js';
+import { singleLinePreview } from './utils.js';
 const SESSION_STORE_VERSION = 1;
-const MAX_RECENT_SESSIONS = 5;
+export const MAX_RECENT_SESSIONS = 10;
 function cloneJson(value) {
     return JSON.parse(JSON.stringify(value ?? null));
 }
@@ -77,6 +81,45 @@ function listSessionFiles(rootDir, env = process.env) {
         .filter((name) => name.endsWith('.json'))
         .map((name) => path.join(dir, name));
 }
+function normalizeBranch(value) {
+    const text = String(value ?? '')
+        .replace(/[\r\n\t]/g, ' ')
+        .trim();
+    return text ? text.slice(0, 120) : null;
+}
+function dehydrateHistoryImages(history) {
+    return history.map((entry) => ({
+        ...entry,
+        parts: (entry.parts ?? []).map((part) => {
+            if (!part?.inlineData?.data)
+                return part;
+            const cachePath = part.imageCachePath ?? findStoredImageByContent(part.inlineData.data);
+            if (!cachePath)
+                return part;
+            return {
+                imageRef: { cachePath, mimeType: part.inlineData.mimeType },
+                text: '[image stored in this session]',
+            };
+        }),
+    }));
+}
+function rehydrateHistoryImages(history) {
+    return history.map((entry) => ({
+        ...entry,
+        parts: (entry.parts ?? []).map((part) => {
+            const ref = part?.imageRef;
+            if (!ref?.cachePath)
+                return part;
+            const stored = readSessionImage(String(ref.cachePath));
+            if (!stored)
+                return { text: part.text ?? '[image no longer available]' };
+            return {
+                inlineData: { mimeType: stored.mimeType, data: stored.base64Data },
+                imageCachePath: ref.cachePath,
+            };
+        }),
+    }));
+}
 function normalizeHistory(value) {
     if (!Array.isArray(value))
         return [];
@@ -107,6 +150,7 @@ function normalizeSnapshot(raw, rootDir) {
         createdAt: normalizeIsoDate(raw.createdAt),
         updatedAt: normalizeIsoDate(raw.updatedAt),
         modelId,
+        branch: normalizeBranch(raw.branch),
         history: cloneJson(normalizeHistory(raw.history)),
         clientState: sanitizeClientState(raw.clientState),
         serverState: sanitizeOpaqueState(raw.serverState),
@@ -145,14 +189,58 @@ function assertSessionNameAvailable(rootDir, name, sessionId, env = process.env)
         throw new Error(`Session name "${name}" is already used by ${duplicate.id}. Use --session "${name}" to resume it or choose a different name.`);
     }
 }
+function listAllSessionIds(env = process.env) {
+    const ids = new Set();
+    const projectsDir = path.join(getSessionBaseDir(env), 'sessions', 'projects');
+    if (!existsSync(projectsDir))
+        return ids;
+    try {
+        for (const project of readdirSync(projectsDir)) {
+            const dir = path.join(projectsDir, project);
+            try {
+                for (const file of readdirSync(dir)) {
+                    if (file.endsWith('.json'))
+                        ids.add(path.basename(file, '.json'));
+                }
+            }
+            catch {
+            }
+        }
+    }
+    catch {
+        return ids;
+    }
+    return ids;
+}
 export function pruneSavedSessions(rootDir, env = process.env) {
     const snapshots = loadAllSnapshots(rootDir, env);
     const keep = new Set(snapshots.slice(0, MAX_RECENT_SESSIONS).map((snapshot) => snapshot.id));
     for (const snapshot of snapshots.slice(MAX_RECENT_SESSIONS)) {
         if (!keep.has(snapshot.id)) {
             rmSync(getSessionPath(rootDir, snapshot.id, env), { force: true });
+            pruneSessionImages(snapshot.id, env);
         }
     }
+    sweepOrphanSessionImages({
+        activeSessionIds: listAllSessionIds(env),
+        env,
+    });
+    sweepGeneratedImages({ env });
+}
+export function readGitBranch(rootDir) {
+    try {
+        const result = spawnSync('git', ['rev-parse', '--abbrev-ref', 'HEAD'], {
+            cwd: rootDir,
+            encoding: 'utf8',
+            timeout: 1500,
+            stdio: ['ignore', 'pipe', 'ignore'],
+        });
+        const branch = result.status === 0 ? String(result.stdout ?? '').trim() : '';
+        return branch && branch !== 'HEAD' ? branch : null;
+    }
+    catch {
+        return null;
+    }
 }
 export function snapshotFromSession(session) {
     const now = new Date().toISOString();
@@ -168,6 +256,7 @@ export function snapshotFromSession(session) {
         createdAt,
         updatedAt: now,
         modelId: session.modelId,
+        branch: readGitBranch(session.rootDir),
         history: cloneJson(session.history),
         clientState: {
             editCounter: session.clientState.editCounter,
@@ -187,6 +276,7 @@ export function saveSessionState(session, env = process.env) {
 }
 export function applySessionSnapshot(session, snapshot, options = {}) {
     const currentAgentMode = session.agentMode;
+    const previousSessionId = session.sessionId;
     session.sessionId = snapshot.id;
     session.sessionName = snapshot.name;
     session.sessionCreatedAt = snapshot.createdAt;
@@ -196,6 +286,9 @@ export function applySessionSnapshot(session, snapshot, options = {}) {
     session.serverState = sanitizeOpaqueState(snapshot.serverState);
     session.agentMode = options.preserveAgentMode ? currentAgentMode : 'default';
     session.autoYes = session.agentMode === 'auto-accept';
+    if (previousSessionId !== session.sessionId) {
+        session.grants = createSessionGrants();
+    }
     session.turnState = {
         id: null,
         historyStartIndex: session.history.length,
@@ -210,6 +303,9 @@ export function applySessionSnapshot(session, snapshot, options = {}) {
         safety: mergeLocalSessionSafetyState(session.clientState.safety, snapshot.clientState.safety),
     };
 }
+export function sessionHasUserMessage(session) {
+    return session.history.some((entry) => Boolean(userPromptText(entry)));
+}
 function extractMarkedSection(text, marker) {
     const index = text.indexOf(marker);
     if (index === -1)
@@ -218,23 +314,27 @@ function extractMarkedSection(text, marker) {
     const end = after.indexOf('\n\n');
     return (end === -1 ? after : after.slice(0, end)).trim();
 }
-function extractLastUserMessage(history) {
-    for (let i = history.length - 1; i >= 0; i--) {
-        const entry = history[i];
-        if (!entry || entry.role !== 'user')
-            continue;
-        const text = (entry.parts ?? [])
-            .map((part) => (typeof part?.text === 'string' ? part.text : ''))
-            .filter(Boolean)
-            .join('\n')
-            .trim();
-        if (!text)
-            continue;
-        const request = extractMarkedSection(text, 'Current user request:') ||
-            extractMarkedSection(text, 'User request:');
-        const message = extractMarkedSection(text, 'Current user message:') ||
-            extractMarkedSection(text, 'User message:');
-        return truncate(request || message || text, 120);
+export function userPromptText(entry) {
+    if (!entry || entry.role !== 'user' || entry.kind !== 'turnStart')
+        return '';
+    const text = (entry.parts ?? [])
+        .map((part) => (typeof part?.text === 'string' ? part.text : ''))
+        .filter(Boolean)
+        .join('\n')
+        .trim();
+    if (!text)
+        return '';
+    const request = extractMarkedSection(text, 'Current user request:') ||
+        extractMarkedSection(text, 'User request:');
+    const message = extractMarkedSection(text, 'Current user message:') ||
+        extractMarkedSection(text, 'User message:');
+    return (request || message || text).trim();
+}
+function extractFirstUserPrompt(history) {
+    for (const entry of history) {
+        const text = userPromptText(entry);
+        if (text)
+            return singleLinePreview(text, 120);
     }
     return '';
 }
@@ -247,8 +347,9 @@ function metadataFromSnapshot(snapshot) {
         updatedAt: snapshot.updatedAt,
         modelId: snapshot.modelId,
         messageCount: snapshot.history.length,
-        lastUserMessage: extractLastUserMessage(snapshot.history),
+        lastUserMessage: extractFirstUserPrompt(snapshot.history),
         summaryPreview: '',
+        branch: snapshot.branch ?? null,
     };
 }
 export function listSessionMetadata(rootDir, env = process.env) {