@orion-agents/orion-code 0.1.0

package/dist/core/checkpoint.js

@@ -0,0 +1,284 @@
+"use strict";
+/**
+ * orion code - File Checkpoints
+ *
+ * Before editing files, create recoverable snapshots so the user can
+ * undo agent changes back to a specific turn.
+ *
+ * Storage: ~/.orion-code/projects/<project-key>/checkpoints/<turnId>/<file>
+ * TTL: 7 days
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CHECKPOINT_TTL_MS = void 0;
+exports.createCheckpoint = createCheckpoint;
+exports.restoreCheckpoint = restoreCheckpoint;
+exports.listCheckpoints = listCheckpoints;
+exports.cleanupCheckpoints = cleanupCheckpoints;
+exports.shouldCreateMultiFileCheckpoint = shouldCreateMultiFileCheckpoint;
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+const config_dir_1 = require("../services/config-dir");
+exports.CHECKPOINT_TTL_MS = 7 * 24 * 60 * 60 * 1000; // 7 days
+function getCheckpointDir(projectPath) {
+    const dir = (0, config_dir_1.getProjectCheckpointsDir)(projectPath);
+    if (!fs.existsSync(dir)) {
+        fs.mkdirSync(dir, { recursive: true, mode: 0o700 });
+    }
+    return dir;
+}
+function getLegacyCheckpointDir(projectPath) {
+    return path.join((0, config_dir_1.getProjectSessionsDir)(projectPath), '_checkpoints');
+}
+function getTurnDir(projectPath, turnId) {
+    const base = getCheckpointDir(projectPath);
+    return path.join(base, `${turnId}`);
+}
+function getExistingTurnDir(projectPath, turnId) {
+    const canonical = path.join((0, config_dir_1.getProjectCheckpointsDir)(projectPath), `${turnId}`);
+    if (fs.existsSync(canonical))
+        return canonical;
+    const legacy = path.join(getLegacyCheckpointDir(projectPath), `${turnId}`);
+    if (fs.existsSync(legacy))
+        return legacy;
+    return canonical;
+}
+function isInside(parentDir, candidatePath) {
+    const relative = path.relative(parentDir, candidatePath);
+    return relative === '' || (!relative.startsWith('..') && !path.isAbsolute(relative));
+}
+function resolveCheckpointTarget(projectPath, filePath) {
+    const projectRoot = path.resolve(projectPath);
+    const absolutePath = path.isAbsolute(filePath)
+        ? path.resolve(filePath)
+        : path.resolve(projectRoot, filePath);
+    if (!isInside(projectRoot, absolutePath))
+        return null;
+    const relativePath = path.relative(projectRoot, absolutePath);
+    if (!relativePath || relativePath.startsWith('..') || path.isAbsolute(relativePath))
+        return null;
+    return { absolutePath, relativePath };
+}
+function resolveCheckpointSourcePath(turnDir, relativePath) {
+    const checkpointPath = path.resolve(turnDir, safeFileName(relativePath));
+    return isInside(path.resolve(turnDir), checkpointPath) ? checkpointPath : null;
+}
+/**
+ * Create a checkpoint for the given files at the current state.
+ * Files are saved individually so they can be restored independently.
+ */
+function createCheckpoint(projectPath, turnId, filePaths) {
+    if (!projectPath || filePaths.length === 0)
+        return null;
+    const dir = getTurnDir(projectPath, turnId);
+    if (fs.existsSync(dir))
+        return null; // Already exists
+    fs.mkdirSync(dir, { recursive: true, mode: 0o700 });
+    const files = [];
+    for (const filePath of filePaths) {
+        try {
+            const target = resolveCheckpointTarget(projectPath, filePath);
+            if (!target)
+                continue;
+            if (!fs.existsSync(target.absolutePath)) {
+                files.push({
+                    path: target.relativePath,
+                    content: '',
+                    sizeBytes: 0,
+                    existed: false,
+                });
+                continue;
+            }
+            const content = fs.readFileSync(target.absolutePath, 'utf8');
+            const checkpointPath = resolveCheckpointSourcePath(dir, target.relativePath);
+            if (!checkpointPath)
+                continue;
+            fs.mkdirSync(path.dirname(checkpointPath), { recursive: true });
+            fs.writeFileSync(checkpointPath, content, { mode: 0o600 });
+            files.push({
+                path: target.relativePath,
+                content: content.slice(0, 200), // Preview only
+                sizeBytes: Buffer.byteLength(content, 'utf8'),
+                existed: true,
+            });
+        }
+        catch {
+            // Skip files that can't be read
+        }
+    }
+    // Write checkpoint metadata
+    const meta = { turnId, createdAt: Date.now(), files };
+    fs.writeFileSync(path.join(dir, '.checkpoint.json'), JSON.stringify(meta, null, 2), { mode: 0o600 });
+    return meta;
+}
+/**
+ * Restore a checkpoint — overwrite current files with checkpointed content.
+ * Returns the list of restored files.
+ */
+function restoreCheckpoint(projectPath, turnId) {
+    if (!projectPath)
+        return { restored: [], error: 'No project path' };
+    const dir = getExistingTurnDir(projectPath, turnId);
+    if (!fs.existsSync(dir)) {
+        return { restored: [], error: `No checkpoint found for turn ${turnId}` };
+    }
+    const metaPath = path.join(dir, '.checkpoint.json');
+    if (!fs.existsSync(metaPath)) {
+        return { restored: [], error: 'Checkpoint metadata missing' };
+    }
+    let meta;
+    try {
+        meta = JSON.parse(fs.readFileSync(metaPath, 'utf8'));
+    }
+    catch {
+        return { restored: [], error: 'Checkpoint metadata corrupted' };
+    }
+    if (!Array.isArray(meta.files)) {
+        return { restored: [], error: 'Checkpoint metadata missing files array' };
+    }
+    const restored = [];
+    for (const file of meta.files) {
+        const target = resolveCheckpointTarget(projectPath, file.path);
+        if (!target) {
+            return { restored, error: `Invalid checkpoint path: ${file.path}` };
+        }
+        const checkpointFile = resolveCheckpointSourcePath(dir, target.relativePath);
+        if (!checkpointFile) {
+            return { restored, error: `Invalid checkpoint source path: ${file.path}` };
+        }
+        try {
+            if (file.existed === false) {
+                if (fs.existsSync(target.absolutePath)) {
+                    const stat = fs.statSync(target.absolutePath);
+                    if (stat.isDirectory()) {
+                        return { restored, error: `Refusing to remove directory from checkpoint restore: ${file.path}` };
+                    }
+                    fs.rmSync(target.absolutePath, { force: true });
+                }
+                restored.push(target.relativePath);
+                continue;
+            }
+            if (!fs.existsSync(checkpointFile))
+                continue;
+            const content = fs.readFileSync(checkpointFile, 'utf8');
+            fs.mkdirSync(path.dirname(target.absolutePath), { recursive: true });
+            fs.writeFileSync(target.absolutePath, content, { mode: 0o600 });
+            restored.push(target.relativePath);
+        }
+        catch {
+            // Skip files that can't be restored
+        }
+    }
+    return { restored };
+}
+/**
+ * List available checkpoints for a project.
+ */
+function listCheckpoints(projectPath) {
+    if (!projectPath)
+        return [];
+    const dirs = [getCheckpointDir(projectPath), getLegacyCheckpointDir(projectPath)];
+    const checkpointsByTurn = new Map();
+    for (const dir of dirs) {
+        if (!fs.existsSync(dir))
+            continue;
+        for (const entry of fs.readdirSync(dir)) {
+            const metaPath = path.join(dir, entry, '.checkpoint.json');
+            if (fs.existsSync(metaPath)) {
+                try {
+                    const checkpoint = JSON.parse(fs.readFileSync(metaPath, 'utf8'));
+                    if (!checkpointsByTurn.has(checkpoint.turnId)) {
+                        checkpointsByTurn.set(checkpoint.turnId, checkpoint);
+                    }
+                }
+                catch {
+                    // Skip corrupted metadata
+                }
+            }
+        }
+    }
+    return Array.from(checkpointsByTurn.values()).sort((a, b) => b.createdAt - a.createdAt);
+}
+/**
+ * Clean up checkpoints older than the TTL (7 days).
+ */
+function cleanupCheckpoints(projectPath) {
+    if (!projectPath)
+        return 0;
+    const dirs = [getCheckpointDir(projectPath), getLegacyCheckpointDir(projectPath)];
+    let cleaned = 0;
+    const now = Date.now();
+    for (const dir of dirs) {
+        if (!fs.existsSync(dir))
+            continue;
+        for (const entry of fs.readdirSync(dir)) {
+            if (entry === '.checkpoint.json')
+                continue; // Skip stray meta files
+            const entryPath = path.join(dir, entry);
+            try {
+                const stat = fs.statSync(entryPath);
+                if (!stat.isDirectory())
+                    continue;
+                // Check the meta file's mtime
+                const metaPath = path.join(entryPath, '.checkpoint.json');
+                if (fs.existsSync(metaPath)) {
+                    const metaStat = fs.statSync(metaPath);
+                    if (now - metaStat.mtimeMs > exports.CHECKPOINT_TTL_MS) {
+                        fs.rmSync(entryPath, { recursive: true, force: true });
+                        cleaned++;
+                    }
+                }
+            }
+            catch {
+                // Skip entries that can't be accessed
+            }
+        }
+    }
+    return cleaned;
+}
+function safeFileName(filePath) {
+    // Replace problematic characters in file names for safe storage
+    return filePath.replace(/[<>:"|?*]/g, '_');
+}
+/**
+ * Returns true when a single turn modifies enough files to warrant an explicit
+ * risky-edit checkpoint with rollback guidance. Wired into createPreToolCheckpoint
+ * in chat-controller.ts, which marks the trace note 'risky_multi_file_checkpoint'
+ * and emits a user-visible rollback hint.
+ */
+function shouldCreateMultiFileCheckpoint(changedFileCount, threshold = 5) {
+    return changedFileCount >= threshold;
+}
+//# sourceMappingURL=checkpoint.js.map

package/dist/core/checkpoint.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"checkpoint.js","sourceRoot":"","sources":["../../src/core/checkpoint.ts"],"names":[],"mappings":";AAAA;;;;;;;;GAQG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA2EH,4CAkDC;AAMD,8CA6DC;AAKD,0CAwBC;AAKD,gDAkCC;AAaD,0EAEC;AAjRD,uCAAyB;AACzB,2CAA6B;AAC7B,uDAAyF;AAE5E,QAAA,iBAAiB,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,SAAS;AAenE,SAAS,gBAAgB,CAAC,WAAmB;IAC3C,MAAM,GAAG,GAAG,IAAA,qCAAwB,EAAC,WAAW,CAAC,CAAC;IAClD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QACxB,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IACtD,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,sBAAsB,CAAC,WAAmB;IACjD,OAAO,IAAI,CAAC,IAAI,CAAC,IAAA,kCAAqB,EAAC,WAAW,CAAC,EAAE,cAAc,CAAC,CAAC;AACvE,CAAC;AAED,SAAS,UAAU,CAAC,WAAmB,EAAE,MAAc;IACrD,MAAM,IAAI,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAC;IAC3C,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,MAAM,EAAE,CAAC,CAAC;AACtC,CAAC;AAED,SAAS,kBAAkB,CAAC,WAAmB,EAAE,MAAc;IAC7D,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,IAAA,qCAAwB,EAAC,WAAW,CAAC,EAAE,GAAG,MAAM,EAAE,CAAC,CAAC;IAChF,IAAI,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC;QAAE,OAAO,SAAS,CAAC;IAC/C,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,sBAAsB,CAAC,WAAW,CAAC,EAAE,GAAG,MAAM,EAAE,CAAC,CAAC;IAC3E,IAAI,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC;QAAE,OAAO,MAAM,CAAC;IACzC,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,QAAQ,CAAC,SAAiB,EAAE,aAAqB;IACxD,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE,aAAa,CAAC,CAAC;IACzD,OAAO,QAAQ,KAAK,EAAE,IAAI,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC;AACvF,CAAC;AAED,SAAS,uBAAuB,CAC9B,WAAmB,EACnB,QAAgB;IAEhB,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IAC9C,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;QAC5C,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC;QACxB,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;IACxC,IAAI,CAAC,QAAQ,CAAC,WAAW,EAAE,YAAY,CAAC;QAAE,OAAO,IAAI,CAAC;IAEtD,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;IAC9D,IAAI,CAAC,YAAY,IAAI,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC;QAAE,OAAO,IAAI,CAAC;IACjG,OAAO,EAAE,YAAY,EAAE,YAAY,EAAE,CAAC;AACxC,CAAC;AAED,SAAS,2BAA2B,CAAC,OAAe,EAAE,YAAoB;IACxE,MAAM,cAAc,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC;IACzE,OAAO,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC;AACjF,CAAC;AAED;;;GAGG;AACH,SAAgB,gBAAgB,CAC9B,WAA+B,EAC/B,MAAc,EACd,SAAmB;IAEnB,IAAI,CAAC,WAAW,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IAExD,MAAM,GAAG,GAAG,UAAU,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;IAC5C,IAAI,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC;QAAE,OAAO,IAAI,CAAC,CAAC,iBAAiB;IAEtD,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IAEpD,MAAM,KAAK,GAAqB,EAAE,CAAC;IACnC,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QACjC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,uBAAuB,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;YAC9D,IAAI,CAAC,MAAM;gBAAE,SAAS;YAEtB,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC;gBACxC,KAAK,CAAC,IAAI,CAAC;oBACT,IAAI,EAAE,MAAM,CAAC,YAAY;oBACzB,OAAO,EAAE,EAAE;oBACX,SAAS,EAAE,CAAC;oBACZ,OAAO,EAAE,KAAK;iBACf,CAAC,CAAC;gBACH,SAAS;YACX,CAAC;YAED,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,MAAM,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;YAC7D,MAAM,cAAc,GAAG,2BAA2B,CAAC,GAAG,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC;YAC7E,IAAI,CAAC,cAAc;gBAAE,SAAS;YAC9B,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAChE,EAAE,CAAC,aAAa,CAAC,cAAc,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;YAE3D,KAAK,CAAC,IAAI,CAAC;gBACT,IAAI,EAAE,MAAM,CAAC,YAAY;gBACzB,OAAO,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,eAAe;gBAC/C,SAAS,EAAE,MAAM,CAAC,UAAU,CAAC,OAAO,EAAE,MAAM,CAAC;gBAC7C,OAAO,EAAE,IAAI;aACd,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,gCAAgC;QAClC,CAAC;IACH,CAAC;IAED,4BAA4B;IAC5B,MAAM,IAAI,GAAe,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,CAAC;IAClE,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,kBAAkB,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IAErG,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;;GAGG;AACH,SAAgB,iBAAiB,CAC/B,WAA+B,EAC/B,MAAc;IAEd,IAAI,CAAC,WAAW;QAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,KAAK,EAAE,iBAAiB,EAAE,CAAC;IAEpE,MAAM,GAAG,GAAG,kBAAkB,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;IACpD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QACxB,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,KAAK,EAAE,gCAAgC,MAAM,EAAE,EAAE,CAAC;IAC3E,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,kBAAkB,CAAC,CAAC;IACpD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC7B,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,KAAK,EAAE,6BAA6B,EAAE,CAAC;IAChE,CAAC;IAED,IAAI,IAAgB,CAAC;IACrB,IAAI,CAAC;QACH,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC;IACvD,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,KAAK,EAAE,+BAA+B,EAAE,CAAC;IAClE,CAAC;IAED,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;QAC/B,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,KAAK,EAAE,yCAAyC,EAAE,CAAC;IAC5E,CAAC;IAED,MAAM,QAAQ,GAAa,EAAE,CAAC;IAC9B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,uBAAuB,CAAC,WAAW,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/D,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,4BAA4B,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;QACtE,CAAC;QACD,MAAM,cAAc,GAAG,2BAA2B,CAAC,GAAG,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC;QAC7E,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,mCAAmC,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;QAC7E,CAAC;QACD,IAAI,CAAC;YACH,IAAI,IAAI,CAAC,OAAO,KAAK,KAAK,EAAE,CAAC;gBAC3B,IAAI,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC;oBACvC,MAAM,IAAI,GAAG,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;oBAC9C,IAAI,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;wBACvB,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,yDAAyD,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;oBACnG,CAAC;oBACD,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,YAAY,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBAClD,CAAC;gBACD,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;gBACnC,SAAS;YACX,CAAC;YAED,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC;gBAAE,SAAS;YAC7C,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;YACxD,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACrE,EAAE,CAAC,aAAa,CAAC,MAAM,CAAC,YAAY,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;YAChE,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QACrC,CAAC;QAAC,MAAM,CAAC;YACP,oCAAoC;QACtC,CAAC;IACH,CAAC;IAED,OAAO,EAAE,QAAQ,EAAE,CAAC;AACtB,CAAC;AAED;;GAEG;AACH,SAAgB,eAAe,CAAC,WAA+B;IAC7D,IAAI,CAAC,WAAW;QAAE,OAAO,EAAE,CAAC;IAE5B,MAAM,IAAI,GAAG,CAAC,gBAAgB,CAAC,WAAW,CAAC,EAAE,sBAAsB,CAAC,WAAW,CAAC,CAAC,CAAC;IAElF,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAsB,CAAC;IACxD,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC;YAAE,SAAS;QAClC,KAAK,MAAM,KAAK,IAAI,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC;YACxC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,EAAE,kBAAkB,CAAC,CAAC;YAC3D,IAAI,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC5B,IAAI,CAAC;oBACH,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAe,CAAC;oBAC/E,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC9C,iBAAiB,CAAC,GAAG,CAAC,UAAU,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;oBACvD,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,0BAA0B;gBAC5B,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;AAC1F,CAAC;AAED;;GAEG;AACH,SAAgB,kBAAkB,CAAC,WAA+B;IAChE,IAAI,CAAC,WAAW;QAAE,OAAO,CAAC,CAAC;IAE3B,MAAM,IAAI,GAAG,CAAC,gBAAgB,CAAC,WAAW,CAAC,EAAE,sBAAsB,CAAC,WAAW,CAAC,CAAC,CAAC;IAElF,IAAI,OAAO,GAAG,CAAC,CAAC;IAChB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAEvB,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC;YAAE,SAAS;QAClC,KAAK,MAAM,KAAK,IAAI,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC;YACxC,IAAI,KAAK,KAAK,kBAAkB;gBAAE,SAAS,CAAC,wBAAwB;YAEpE,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;YACxC,IAAI,CAAC;gBACH,MAAM,IAAI,GAAG,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;gBACpC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE;oBAAE,SAAS;gBAElC,8BAA8B;gBAC9B,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,kBAAkB,CAAC,CAAC;gBAC1D,IAAI,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAC5B,MAAM,QAAQ,GAAG,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;oBACvC,IAAI,GAAG,GAAG,QAAQ,CAAC,OAAO,GAAG,yBAAiB,EAAE,CAAC;wBAC/C,EAAE,CAAC,MAAM,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;wBACvD,OAAO,EAAE,CAAC;oBACZ,CAAC;gBACH,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,sCAAsC;YACxC,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,YAAY,CAAC,QAAgB;IACpC,gEAAgE;IAChE,OAAO,QAAQ,CAAC,OAAO,CAAC,YAAY,EAAE,GAAG,CAAC,CAAC;AAC7C,CAAC;AAED;;;;;GAKG;AACH,SAAgB,+BAA+B,CAAC,gBAAwB,EAAE,SAAS,GAAG,CAAC;IACrF,OAAO,gBAAgB,IAAI,SAAS,CAAC;AACvC,CAAC"}

package/dist/core/cost-tracker.d.ts

@@ -0,0 +1,147 @@
+/**
+ * orion code - 成本追踪模块
+ *
+ * 记录 token 使用量，计算成本，支持预算检查。
+ * 按 Agent、任务、模型、时间维度统计。
+ */
+/** 使用记录 */
+export type CostSource = 'provider' | 'configured' | 'builtin' | 'fallback';
+export interface ModelPricing {
+    /** USD per one million input tokens. */
+    input: number;
+    /** USD per one million output tokens. */
+    output: number;
+    /** USD per one million cached input tokens. Defaults to input pricing. */
+    cachedInput?: number;
+}
+export interface TokenUsage {
+    promptTokens: number;
+    completionTokens: number;
+    cachedPromptTokens?: number;
+    /** Provider-reported billed cost in USD. */
+    costUsd?: number;
+    /** Stable provider request id used to suppress duplicate accounting. */
+    requestId?: string;
+}
+export interface UsageRecord {
+    timestamp: Date;
+    model: string;
+    promptTokens: number;
+    completionTokens: number;
+    totalTokens: number;
+    cachedPromptTokens: number;
+    /** Cost used for accounting, in USD. */
+    costUsd: number;
+    costSource: CostSource;
+    /** @deprecated Use costUsd. Kept for API compatibility. */
+    estimatedCost: number;
+    requestId?: string;
+    requestKind?: string;
+    agentId?: string;
+    taskId?: string;
+}
+/** 统计结果 */
+export interface CostStats {
+    totalPromptTokens: number;
+    totalCompletionTokens: number;
+    totalTokens: number;
+    totalCost: number;
+    providerCost: number;
+    estimatedCost: number;
+    recordCount: number;
+    bySource: Record<CostSource, {
+        cost: number;
+        count: number;
+    }>;
+    byAgent: Record<string, {
+        tokens: number;
+        cost: number;
+        count: number;
+    }>;
+    byTask: Record<string, {
+        tokens: number;
+        cost: number;
+        count: number;
+    }>;
+    byModel: Record<string, {
+        tokens: number;
+        cost: number;
+        count: number;
+    }>;
+}
+/** 时间范围 */
+export interface TimeRange {
+    start: Date;
+    end: Date;
+}
+export interface CostTrackerOptions {
+    pricing?: Record<string, ModelPricing>;
+    defaultPricing?: ModelPricing;
+    onRecord?: (record: UsageRecord) => void;
+}
+export declare class CostTracker {
+    private records;
+    private recordsByRequestId;
+    private budgetLimit;
+    private sessionStartTime;
+    private pricing;
+    private defaultPricing;
+    private onRecord?;
+    constructor(options?: CostTrackerOptions);
+    /**
+     * 记录使用量
+     */
+    record(usage: TokenUsage, meta: {
+        model: string;
+        agentId?: string;
+        taskId?: string;
+        requestKind?: string;
+    }): UsageRecord;
+    /** Configure the durable sink after runtime/session bootstrap. */
+    setRecordSink(onRecord?: (record: UsageRecord) => void): void;
+    /**
+     * 获取统计
+     */
+    getStats(timeRange?: TimeRange): CostStats;
+    /**
+     * 获取本次会话统计
+     */
+    getSessionStats(): CostStats;
+    /**
+     * 检查预算
+     */
+    checkBudget(): {
+        ok: boolean;
+        remaining: number;
+        used: number;
+    };
+    /**
+     * 设置预算限制
+     */
+    setBudget(limit: number): void;
+    /**
+     * 获取预算限制
+     */
+    getBudget(): number | null;
+    /**
+     * 清空记录
+     */
+    clear(): void;
+    /**
+     * 获取所有记录
+     */
+    getRecords(): UsageRecord[];
+    /**
+     * 获取最后一条记录
+     */
+    getLastRecord(): UsageRecord | undefined;
+    /**
+     * 格式化成本显示
+     */
+    formatCost(cost: number): string;
+    /**
+     * 格式化统计摘要
+     */
+    formatSummary(stats: CostStats): string;
+}
+//# sourceMappingURL=cost-tracker.d.ts.map

package/dist/core/cost-tracker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cost-tracker.d.ts","sourceRoot":"","sources":["../../src/core/cost-tracker.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAMH,WAAW;AACX,MAAM,MAAM,UAAU,GAAG,UAAU,GAAG,YAAY,GAAG,SAAS,GAAG,UAAU,CAAC;AAE5E,MAAM,WAAW,YAAY;IAC3B,wCAAwC;IACxC,KAAK,EAAE,MAAM,CAAC;IACd,yCAAyC;IACzC,MAAM,EAAE,MAAM,CAAC;IACf,0EAA0E;IAC1E,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,UAAU;IACzB,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAC;IACzB,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,4CAA4C;IAC5C,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,wEAAwE;IACxE,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,WAAW;IAC1B,SAAS,EAAE,IAAI,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAC;IACzB,WAAW,EAAE,MAAM,CAAC;IACpB,kBAAkB,EAAE,MAAM,CAAC;IAC3B,wCAAwC;IACxC,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,UAAU,CAAC;IACvB,2DAA2D;IAC3D,aAAa,EAAE,MAAM,CAAC;IACtB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,WAAW;AACX,MAAM,WAAW,SAAS;IACxB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,qBAAqB,EAAE,MAAM,CAAC;IAC9B,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IACrB,aAAa,EAAE,MAAM,CAAC;IACtB,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC,UAAU,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC9D,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACzE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACxE,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CAC1E;AAED,WAAW;AACX,MAAM,WAAW,SAAS;IACxB,KAAK,EAAE,IAAI,CAAC;IACZ,GAAG,EAAE,IAAI,CAAC;CACX;AAoDD,MAAM,WAAW,kBAAkB;IACjC,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;IACvC,cAAc,CAAC,EAAE,YAAY,CAAC;IAC9B,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,WAAW,KAAK,IAAI,CAAC;CAC1C;AAMD,qBAAa,WAAW;IACtB,OAAO,CAAC,OAAO,CAAqB;IACpC,OAAO,CAAC,kBAAkB,CAAkC;IAC5D,OAAO,CAAC,WAAW,CAAuB;IAC1C,OAAO,CAAC,gBAAgB,CAAO;IAC/B,OAAO,CAAC,OAAO,CAA+B;IAC9C,OAAO,CAAC,cAAc,CAAe;IACrC,OAAO,CAAC,QAAQ,CAAC,CAAgC;gBAErC,OAAO,GAAE,kBAAuB;IAO5C;;OAEG;IACH,MAAM,CACJ,KAAK,EAAE,UAAU,EACjB,IAAI,EAAE;QACJ,KAAK,EAAE,MAAM,CAAC;QACd,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,GACA,WAAW;IAyDd,kEAAkE;IAClE,aAAa,CAAC,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,WAAW,KAAK,IAAI,GAAG,IAAI;IAI7D;;OAEG;IACH,QAAQ,CAAC,SAAS,CAAC,EAAE,SAAS,GAAG,SAAS;IAsE1C;;OAEG;IACH,eAAe,IAAI,SAAS;IAI5B;;OAEG;IACH,WAAW,IAAI;QAAE,EAAE,EAAE,OAAO,CAAC;QAAC,SAAS,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE;IAU/D;;OAEG;IACH,SAAS,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IAI9B;;OAEG;IACH,SAAS,IAAI,MAAM,GAAG,IAAI;IAI1B;;OAEG;IACH,KAAK,IAAI,IAAI;IAMb;;OAEG;IACH,UAAU,IAAI,WAAW,EAAE;IAI3B;;OAEG;IACH,aAAa,IAAI,WAAW,GAAG,SAAS;IAIxC;;OAEG;IACH,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM;IAUhC;;OAEG;IACH,aAAa,CAAC,KAAK,EAAE,SAAS,GAAG,MAAM;CAexC"}