@agentbean/daemon 0.1.7 → 0.1.9

package/dist/post-process.js

@@ -1,9 +1,23 @@
-import { readdirSync, statSync, existsSync } from 'node:fs';
-import { join } from 'node:path';
+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 [];
@@ -12,45 +26,136 @@ 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;
 }
-export async function postProcess(reply, workspace, kind, dispatchStart) {
-    const outputFiles = [];
-    // Codex native image detection
-    if (kind === 'codex') {
-        const allCodexFiles = listAllFiles(CODEX_IMG_DIR);
-        for (const f of allCodexFiles) {
+function normalizeCandidatePath(raw) {
+    const cleaned = raw
+        .trim()
+        .replace(/^file:\/\//, '')
+        .replace(/^["'`<({\[]+/, '')
+        .replace(/["'`>)}\].,;:]+$/, '');
+    if (!cleaned || !OUTPUT_FILE_EXT_RE.test(cleaned))
+        return null;
+    return cleaned.replace(/^~(?=$|\/)/, homedir());
+}
+function extractMentionedFiles(reply, workspace, dispatchStart) {
+    const candidates = new Set();
+    const markdownLinkRe = /!?\[[^\]]*]\(([^)\s]+)\)/g;
+    const plainPathRe = /(?:^|[\s"'`(<])((?:~?\/|\.{1,2}\/)?[\w@%+=:,./-]+\.(?:png|jpe?g|gif|webp|svg|pdf|txt|csv|json|md|mp4|mov|zip))(?:$|[\s"'`)>.,;:])/gim;
+    let match;
+    while ((match = markdownLinkRe.exec(reply)) !== null) {
+        const normalized = normalizeCandidatePath(match[1]);
+        if (normalized)
+            candidates.add(normalized);
+    }
+    while ((match = plainPathRe.exec(reply)) !== null) {
+        const normalized = normalizeCandidatePath(match[1]);
+        if (normalized)
+            candidates.add(normalized);
+    }
+    const files = [];
+    for (const candidate of candidates) {
+        const abs = isAbsolute(candidate) ? candidate : workspace ? resolve(workspace, candidate) : null;
+        if (!abs)
+            continue;
+        try {
+            const st = statSync(abs);
+            if (st.isFile() && st.mtimeMs > dispatchStart)
+                files.push(abs);
+        }
+        catch { }
+    }
+    return files;
+}
+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.push(f);
+                const st = statSync(filePath);
+                if (st.isFile() && st.mtimeMs > dispatchStart) {
+                    files.add(canonicalPath(filePath));
                 }
             }
             catch { }
         }
     }
-    // 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.push(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 = [];
@@ -64,8 +169,9 @@ export async function postProcess(reply, workspace, kind, dispatchStart) {
         }
     }
     let replyText = reply;
-    if (outputFiles.length > 0) {
-        replyText += '\n\n已生成文件:\n' + outputFiles.map((f) => `- ${f}`).join('\n');
+    const files = [...outputFiles];
+    if (files.length > 0) {
+        replyText += '\n\n已生成文件:\n' + files.map((f) => `- ${f}`).join('\n');
     }
-    return { replyText, outputFiles };
+    return { replyText, outputFiles: files };
 }

package/package.json

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