@gonzih/cc-tg 0.2.13 → 0.2.15

package/dist/bot.d.ts

@@ -43,3 +43,4 @@ export declare class CcTgBot {
     private killSession;
     stop(): void;
 }
+export declare function splitMessage(text: string, maxLen?: number): string[];

package/dist/bot.js

@@ -3,7 +3,7 @@
  * One ClaudeProcess per chat_id — sessions are isolated per user.
  */
 import TelegramBot from "node-telegram-bot-api";
-import { existsSync, createWriteStream, mkdirSync, statSync } from "fs";
+import { existsSync, createWriteStream, mkdirSync, statSync, readdirSync } from "fs";
 import { resolve, basename, join } from "path";
 import os from "os";
 import { execSync } from "child_process";
@@ -358,10 +358,45 @@ export class CcTgBot {
             }
             else if (name === "Bash") {
                 const cmd = input.command ?? "";
-                // yt-dlp / ffmpeg -o "path"
-                const oFlag = cmd.match(/-o\s+["']?([^\s"']+\.[\w]{1,10})["']?/);
-                if (oFlag)
-                    session.writtenFiles.add(resolve(cwd ?? process.cwd(), oFlag[1]));
+                if (/\byt-dlp\b|\bffmpeg\b/.test(cmd)) {
+                    // Scan output dir for recently modified media files (template paths like /tmp/%(title)s.%(ext)s
+                    // make the actual filename unknowable at tracking time)
+                    const oFlagMatch = cmd.match(/-o\s+["']?([^\s"']+)/);
+                    let scanDir = "/tmp/";
+                    if (oFlagMatch) {
+                        const oPath = oFlagMatch[1].replace(/["'].*$/, "");
+                        const dirEnd = oPath.lastIndexOf("/");
+                        if (dirEnd > 0)
+                            scanDir = oPath.slice(0, dirEnd + 1);
+                    }
+                    const MEDIA_EXTS = new Set([".mp3", ".mp4", ".wav", ".ogg", ".flac", ".webm", ".m4a", ".aac"]);
+                    const nowMs = Date.now();
+                    try {
+                        for (const entry of readdirSync(scanDir)) {
+                            const dotIdx = entry.lastIndexOf(".");
+                            if (dotIdx < 0)
+                                continue;
+                            const ext = entry.slice(dotIdx).toLowerCase();
+                            if (!MEDIA_EXTS.has(ext))
+                                continue;
+                            const full = join(scanDir, entry);
+                            try {
+                                if (nowMs - statSync(full).mtimeMs <= 90_000) {
+                                    console.log(`[claude:files] tracked yt-dlp/ffmpeg output: ${full}`);
+                                    session.writtenFiles.add(full);
+                                }
+                            }
+                            catch { /* skip unreadable entries */ }
+                        }
+                    }
+                    catch { /* scanDir doesn't exist or unreadable */ }
+                }
+                else {
+                    // Other bash commands: try to extract output path from -o flag
+                    const oFlag = cmd.match(/-o\s+["']?([^\s"']+\.[\w]{1,10})["']?/);
+                    if (oFlag)
+                        session.writtenFiles.add(resolve(cwd ?? process.cwd(), oFlag[1]));
+                }
                 // mv source dest — track dest
                 const mvMatch = cmd.match(/\bmv\s+\S+\s+["']?([^\s"']+)["']?$/);
                 if (mvMatch)
@@ -387,7 +422,7 @@ export class CcTgBot {
             /credential/i, /secret/i, /password/i, /passwd/i, /\.env/i,
             /api[_-]?key/i, /token/i, /private[_-]?key/i, /id_rsa/i,
             /\.pem$/i, /\.key$/i, /\.pfx$/i, /\.p12$/i,
-            /gmail/i, /oauth/i, /auth/i,
+            /gmail/i, /oauth/i, /\bauth\b/i,
         ];
         return sensitivePatterns.some((p) => p.test(name));
     }
@@ -395,11 +430,15 @@ export class CcTgBot {
         // Extract file path candidates from result text
         // Match: /absolute/path/file.ext or relative like ./foo/bar.csv or just foo.pdf
         const pathPattern = /(?:^|[\s`'"(])(\/?[\w.\-/]+\.[\w]{1,10})(?:[\s`'")\n]|$)/gm;
+        const quotedPattern = /"([^"]+\.[a-zA-Z0-9]{1,10})"|'([^']+\.[a-zA-Z0-9]{1,10})'/g;
         const candidates = new Set();
         let match;
         while ((match = pathPattern.exec(resultText)) !== null) {
             candidates.add(match[1]);
         }
+        while ((match = quotedPattern.exec(resultText)) !== null) {
+            candidates.add(match[1] ?? match[2]);
+        }
         const safeDirs = ["/tmp/", "/var/folders/", os.homedir() + "/Downloads/"];
         const isSafeDir = (p) => safeDirs.some(d => p.startsWith(d)) || p.startsWith(this.opts.cwd ?? process.cwd());
         const toUpload = [];
@@ -741,7 +780,7 @@ function downloadToFile(url, destPath) {
         }).on("error", reject);
     });
 }
-function splitMessage(text, maxLen = 4096) {
+export function splitMessage(text, maxLen = 4096) {
     if (text.length <= maxLen)
         return [text];
     const chunks = [];

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gonzih/cc-tg",
-  "version": "0.2.13",
+  "version": "0.2.15",
   "description": "Claude Code Telegram bot — chat with Claude Code via Telegram",
   "type": "module",
   "bin": {
@@ -9,7 +9,10 @@
   "scripts": {
     "build": "tsc",
     "start": "node dist/index.js",
-    "dev": "node --loader ts-node/esm src/index.ts"
+    "dev": "node --loader ts-node/esm src/index.ts",
+    "test": "vitest run",
+    "test:watch": "vitest",
+    "test:coverage": "vitest run --coverage"
   },
   "files": [
     "dist/"
@@ -20,7 +23,9 @@
   "devDependencies": {
     "@types/node": "^22.0.0",
     "@types/node-telegram-bot-api": "^0.64.0",
-    "typescript": "^5.5.0"
+    "@vitest/coverage-v8": "^4.1.0",
+    "typescript": "^5.5.0",
+    "vitest": "^4.1.0"
   },
   "repository": {
     "type": "git",