@agentbean/daemon 0.1.8 → 0.1.9

package/dist/post-process.js

@@ -1,10 +1,23 @@
-import { readdirSync, statSync, existsSync } from 'node:fs';
+import { readdirSync, realpathSync, statSync, existsSync } from 'node:fs';
 import { isAbsolute, join, resolve } from 'node:path';
 import { homedir } from 'node:os';
 import { logger } from './log.js';
 const CODE_BLOCK_RE = /```python\n([\s\S]*?)```/g;
 const CODEX_IMG_DIR = join(homedir(), '.codex', 'generated_images');
 const OUTPUT_FILE_EXT_RE = /\.(png|jpe?g|gif|webp|svg|pdf|txt|csv|json|md|mp4|mov|zip)$/i;
+const IGNORED_OUTPUT_DIRS = new Set([
+    '.git',
+    '.hg',
+    '.svn',
+    '.cache',
+    '.next',
+    '.nuxt',
+    '.turbo',
+    'node_modules',
+    'vendor',
+]);
+const MAX_OUTPUT_FILES_PER_ROOT = 2000;
+const OUTPUT_DIR_ENV = 'AGENT_BEAN_OUTPUT_DIRS';
 export function listAllFiles(dir, maxDepth = 10, depth = 0) {
     if (!existsSync(dir) || depth > maxDepth)
         return [];
@@ -13,10 +26,15 @@ export function listAllFiles(dir, maxDepth = 10, depth = 0) {
         const full = join(dir, entry.name);
         if (entry.isSymbolicLink())
             continue;
-        if (entry.isDirectory())
+        if (entry.isDirectory()) {
+            if (IGNORED_OUTPUT_DIRS.has(entry.name))
+                continue;
             results.push(...listAllFiles(full, maxDepth, depth + 1));
+        }
         else
             results.push(full);
+        if (results.length >= MAX_OUTPUT_FILES_PER_ROOT)
+            break;
     }
     return results;
 }
@@ -59,41 +77,85 @@ function extractMentionedFiles(reply, workspace, dispatchStart) {
     }
     return files;
 }
-export async function postProcess(reply, workspace, kind, dispatchStart) {
-    const outputFiles = new Set();
-    // Codex native image detection
-    if (kind === 'codex') {
-        const allCodexFiles = listAllFiles(CODEX_IMG_DIR);
-        for (const f of allCodexFiles) {
+function outputDirsFromEnv() {
+    const raw = process.env[OUTPUT_DIR_ENV];
+    if (!raw?.trim())
+        return [];
+    return raw
+        .split(/[,:;]/)
+        .map((item) => item.trim())
+        .filter(Boolean);
+}
+function canonicalPath(path) {
+    try {
+        return realpathSync(path);
+    }
+    catch {
+        return path;
+    }
+}
+function resolveOutputRoots(workspace, outputDirs = []) {
+    const roots = new Set();
+    if (workspace)
+        roots.add(resolve(workspace));
+    for (const raw of [...outputDirs, ...outputDirsFromEnv()]) {
+        const expanded = raw.replace(/^~(?=$|\/)/, homedir());
+        const root = isAbsolute(expanded)
+            ? expanded
+            : workspace
+                ? resolve(workspace, expanded)
+                : resolve(expanded);
+        roots.add(root);
+    }
+    return [...roots].filter((root) => {
+        try {
+            return statSync(root).isDirectory();
+        }
+        catch {
+            return false;
+        }
+    });
+}
+function collectRecentOutputFiles(roots, dispatchStart) {
+    const files = new Set();
+    for (const root of roots) {
+        for (const filePath of listAllFiles(root, 8)) {
+            if (!OUTPUT_FILE_EXT_RE.test(filePath))
+                continue;
             try {
-                const st = statSync(f);
-                if (st.mtimeMs > dispatchStart) {
-                    outputFiles.add(f);
+                const st = statSync(filePath);
+                if (st.isFile() && st.mtimeMs > dispatchStart) {
+                    files.add(canonicalPath(filePath));
                 }
             }
             catch { }
         }
     }
-    for (const filePath of extractMentionedFiles(reply, workspace, dispatchStart)) {
-        outputFiles.add(filePath);
-    }
-    // Detect files created during this dispatch
-    if (workspace) {
-        for (const entry of readdirSync(workspace, { withFileTypes: true })) {
-            if (entry.isDirectory())
-                continue;
-            const f = join(workspace, entry.name);
-            if (entry.name.startsWith('.agentbean-exec-'))
+    return [...files];
+}
+export async function postProcess(reply, workspace, kind, dispatchStart, options = {}) {
+    const outputFiles = new Set();
+    // Codex native image detection
+    if (kind === 'codex') {
+        const allCodexFiles = listAllFiles(CODEX_IMG_DIR, 4);
+        for (const f of allCodexFiles) {
+            if (!OUTPUT_FILE_EXT_RE.test(f))
                 continue;
             try {
                 const st = statSync(f);
                 if (st.mtimeMs > dispatchStart) {
-                    outputFiles.add(f);
+                    outputFiles.add(canonicalPath(f));
                 }
             }
             catch { }
         }
     }
+    for (const filePath of extractMentionedFiles(reply, workspace, dispatchStart)) {
+        outputFiles.add(canonicalPath(filePath));
+    }
+    for (const filePath of collectRecentOutputFiles(resolveOutputRoots(workspace, options.outputDirs), dispatchStart)) {
+        outputFiles.add(filePath);
+    }
     // Extract code blocks for logging but do NOT auto-execute (security)
     if (workspace) {
         const codeBlocks = [];

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@agentbean/daemon",
   "private": false,
-  "version": "0.1.8",
+  "version": "0.1.9",
   "type": "module",
   "main": "./dist/index.js",
   "bin": {