mstro-app 0.1.47

package/dist/server/services/websocket/file-utils.js

@@ -0,0 +1,272 @@
+// Copyright (c) 2025-present Mstro, Inc. All rights reserved.
+// Licensed under the MIT License. See LICENSE file for details.
+/**
+ * File Utilities
+ *
+ * File type detection, gitignore parsing, and directory scanning utilities.
+ */
+import { existsSync, readdirSync, readFileSync, statSync } from 'node:fs';
+import { extname, join, relative, sep } from 'node:path';
+// Directories always excluded from autocomplete scanning
+const ALWAYS_IGNORE_DIRS = new Set([
+    'node_modules', 'dist', 'dist-ssr', '.git', '__pycache__',
+    '.next', '.turbo', 'build', '.cache', 'coverage',
+    '.parcel-cache', '.nuxt', '.output', '.vercel', '.netlify',
+    '.nyc_output', 'venv', '.venv'
+]);
+// File type categorization for icons
+export const FILE_TYPE_MAP = {
+    // JavaScript/TypeScript
+    '.js': 'javascript',
+    '.jsx': 'javascript',
+    '.ts': 'typescript',
+    '.tsx': 'typescript',
+    '.mjs': 'javascript',
+    '.cjs': 'javascript',
+    // Web
+    '.html': 'html',
+    '.htm': 'html',
+    '.css': 'css',
+    '.scss': 'css',
+    '.sass': 'css',
+    '.less': 'css',
+    // Data
+    '.json': 'json',
+    '.yaml': 'yaml',
+    '.yml': 'yaml',
+    '.xml': 'xml',
+    '.toml': 'config',
+    // Config
+    '.env': 'config',
+    '.gitignore': 'config',
+    '.eslintrc': 'config',
+    '.prettierrc': 'config',
+    // Documentation
+    '.md': 'markdown',
+    '.mdx': 'markdown',
+    '.txt': 'text',
+    '.rst': 'text',
+    // Images
+    '.png': 'image',
+    '.jpg': 'image',
+    '.jpeg': 'image',
+    '.gif': 'image',
+    '.svg': 'image',
+    '.webp': 'image',
+    '.ico': 'image',
+    // Other languages
+    '.py': 'python',
+    '.rb': 'ruby',
+    '.go': 'go',
+    '.rs': 'rust',
+    '.java': 'java',
+    '.kt': 'kotlin',
+    '.swift': 'swift',
+    '.c': 'c',
+    '.cpp': 'cpp',
+    '.h': 'c',
+    '.hpp': 'cpp',
+    '.cs': 'csharp',
+    '.php': 'php',
+    '.sh': 'shell',
+    '.bash': 'shell',
+    '.zsh': 'shell',
+    '.fish': 'shell',
+    // Build/Package
+    '.lock': 'lock',
+    '.log': 'log',
+};
+// Directory cache
+export const directoryCache = new Map();
+export const CACHE_TTL_MS = 5000; // 5 second cache
+/**
+ * Get file type from extension
+ */
+export function getFileType(filePath) {
+    const ext = extname(filePath).toLowerCase();
+    return FILE_TYPE_MAP[ext] || 'file';
+}
+/**
+ * Parse .gitignore patterns
+ */
+export function parseGitignore(workingDir) {
+    const gitignorePath = join(workingDir, '.gitignore');
+    if (!existsSync(gitignorePath)) {
+        return [];
+    }
+    try {
+        const content = readFileSync(gitignorePath, 'utf-8');
+        const patterns = content
+            .split('\n')
+            .map(line => line.trim())
+            .filter(line => line && !line.startsWith('#') && !line.startsWith('!'));
+        return patterns;
+    }
+    catch (error) {
+        console.error('[FileUtils] Error parsing .gitignore:', error);
+        return [];
+    }
+}
+/**
+ * Check if path matches gitignore pattern
+ * Follows gitignore semantics:
+ * - Patterns without slashes match anywhere in the path (e.g., "node_modules" matches "foo/node_modules/bar")
+ * - Patterns with slashes are anchored to the root (e.g., "src/foo" only matches "src/foo")
+ * - Patterns ending with / match directories
+ */
+function matchesSinglePattern(normalizedPath, pathSegments, pattern) {
+    const normalizedPattern = pattern.replace(/\\/g, '/');
+    const isDirectoryPattern = normalizedPattern.endsWith('/');
+    const cleanPattern = isDirectoryPattern ? normalizedPattern.slice(0, -1) : normalizedPattern;
+    if (cleanPattern.includes('/')) {
+        return normalizedPath === cleanPattern || normalizedPath.startsWith(`${cleanPattern}/`);
+    }
+    if (pathSegments.includes(cleanPattern)) {
+        return true;
+    }
+    if (cleanPattern.includes('*')) {
+        const regexPattern = cleanPattern
+            .replace(/\./g, '\\.')
+            .replace(/\*\*/g, '.*')
+            .replace(/\*/g, '[^/]*');
+        return new RegExp(`(^|/)${regexPattern}($|/)`).test(normalizedPath);
+    }
+    return false;
+}
+export function isIgnored(path, patterns) {
+    const normalizedPath = path.replace(/\\/g, '/');
+    const pathSegments = normalizedPath.split('/');
+    return patterns.some(pattern => matchesSinglePattern(normalizedPath, pathSegments, pattern));
+}
+/**
+ * Recursively scan directory with depth limit for performance
+ */
+function shouldScanEntry(entry, relativePath, gitignorePatterns) {
+    if (entry.name.startsWith('.'))
+        return false;
+    if (entry.isDirectory() && ALWAYS_IGNORE_DIRS.has(entry.name))
+        return false;
+    if (gitignorePatterns.length > 0 && isIgnored(relativePath, gitignorePatterns))
+        return false;
+    return true;
+}
+export function scanDirectoryRecursiveWithDepth(dir, baseDir, gitignorePatterns, results = [], maxResults = 1000, maxDepth = 10, currentDepth = 0) {
+    try {
+        if (results.length >= maxResults || currentDepth >= maxDepth) {
+            return results;
+        }
+        const entries = readdirSync(dir, { withFileTypes: true });
+        for (const entry of entries) {
+            if (results.length >= maxResults)
+                break;
+            const fullPath = join(dir, entry.name);
+            const relativePath = relative(baseDir, fullPath);
+            if (!shouldScanEntry(entry, relativePath, gitignorePatterns))
+                continue;
+            results.push({ relativePath, isDirectory: entry.isDirectory() });
+            if (entry.isDirectory() && results.length < maxResults && currentDepth + 1 < maxDepth) {
+                scanDirectoryRecursiveWithDepth(fullPath, baseDir, gitignorePatterns, results, maxResults, maxDepth, currentDepth + 1);
+            }
+        }
+    }
+    catch {
+        // Skip directories we can't read
+    }
+    return results;
+}
+/**
+ * Check if a path is in a safe location for reading (outside working dir)
+ */
+export function isPathInSafeLocation(fullPath) {
+    const homeDir = process.env.HOME || process.env.USERPROFILE || '';
+    const safePatterns = [
+        join(homeDir, 'Desktop'),
+        join(homeDir, 'Downloads'),
+        join(homeDir, 'Pictures'),
+        join(homeDir, 'Documents'),
+        '/var/folders/', // macOS temp directories for screenshots
+        '/tmp/',
+        process.env.TMPDIR || '/tmp'
+    ];
+    return safePatterns.some(safePath => fullPath.startsWith(safePath));
+}
+/**
+ * Check if a file is an image based on extension
+ */
+export function isImageFile(filePath) {
+    const imageExtensions = ['.png', '.jpg', '.jpeg', '.gif', '.webp', '.bmp', '.tiff', '.ico', '.svg'];
+    const ext = filePath.toLowerCase().split('.').pop();
+    return ext ? imageExtensions.includes(`.${ext}`) : false;
+}
+function readDirectoryContent(fullPath, filePath, fileName) {
+    try {
+        const entries = readdirSync(fullPath, { withFileTypes: true });
+        const listing = entries
+            .filter(e => !e.name.startsWith('.'))
+            .sort((a, b) => {
+            if (a.isDirectory() && !b.isDirectory())
+                return -1;
+            if (!a.isDirectory() && b.isDirectory())
+                return 1;
+            return a.name.localeCompare(b.name);
+        })
+            .map(e => e.isDirectory() ? `${e.name}/` : e.name)
+            .join('\n');
+        return { path: filePath, fileName, content: listing, size: listing.length };
+    }
+    catch {
+        return { path: filePath, fileName, content: '', error: 'Cannot read directory' };
+    }
+}
+function readImageContent(fullPath, filePath, fileName, stats) {
+    const buffer = readFileSync(fullPath);
+    const ext = fullPath.toLowerCase().split('.').pop() || 'png';
+    const mimeType = ext === 'svg' ? 'image/svg+xml' : ext === 'jpg' ? 'image/jpeg' : `image/${ext}`;
+    return {
+        path: filePath, fileName,
+        content: buffer.toString('base64'),
+        size: stats.size, modifiedAt: stats.mtime.toISOString(),
+        isImage: true, mimeType
+    };
+}
+function readTextContent(fullPath, filePath, fileName, stats) {
+    return {
+        path: filePath, fileName,
+        content: readFileSync(fullPath, 'utf-8'),
+        size: stats.size, modifiedAt: stats.mtime.toISOString()
+    };
+}
+/**
+ * Read file content for context injection
+ */
+export function readFileContent(filePath, workingDir) {
+    try {
+        const fullPath = filePath.startsWith('/') ? filePath : join(workingDir, filePath);
+        const fileName = fullPath.split(sep).pop() || filePath;
+        const normalizedPath = join(fullPath);
+        const isInWorkingDir = normalizedPath.startsWith(join(workingDir));
+        if (!isInWorkingDir && !isPathInSafeLocation(normalizedPath)) {
+            return { path: filePath, fileName, content: '', error: 'Access denied: path outside allowed locations' };
+        }
+        if (!existsSync(fullPath)) {
+            return { path: filePath, fileName, content: '', error: 'File not found' };
+        }
+        const stats = statSync(fullPath);
+        if (stats.isDirectory()) {
+            return readDirectoryContent(fullPath, filePath, fileName);
+        }
+        const isImage = isImageFile(fullPath);
+        const MAX_FILE_SIZE = isImage ? 10 * 1024 * 1024 : 1024 * 1024;
+        if (stats.size > MAX_FILE_SIZE) {
+            return { path: filePath, fileName, content: '', size: stats.size, error: `File too large (${Math.round(stats.size / 1024)}KB). Maximum is ${isImage ? '10MB' : '1MB'}.` };
+        }
+        return isImage
+            ? readImageContent(fullPath, filePath, fileName, stats)
+            : readTextContent(fullPath, filePath, fileName, stats);
+    }
+    catch (error) {
+        console.error('[FileUtils] Error reading file:', error);
+        return { path: filePath, fileName: filePath.split(sep).pop() || filePath, content: '', error: error.message || 'Failed to read file' };
+    }
+}
+//# sourceMappingURL=file-utils.js.map

package/dist/server/services/websocket/file-utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"file-utils.js","sourceRoot":"","sources":["../../../../server/services/websocket/file-utils.ts"],"names":[],"mappings":"AAAA,8DAA8D;AAC9D,gEAAgE;AAEhE;;;;GAIG;AAEH,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,SAAS,CAAC;AAC1E,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,EAAE,MAAM,WAAW,CAAC;AAGzD,yDAAyD;AACzD,MAAM,kBAAkB,GAAG,IAAI,GAAG,CAAC;IACjC,cAAc,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,aAAa;IACzD,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,QAAQ,EAAE,UAAU;IAChD,eAAe,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,UAAU;IAC1D,aAAa,EAAE,MAAM,EAAE,OAAO;CAC/B,CAAC,CAAC;AAEH,qCAAqC;AACrC,MAAM,CAAC,MAAM,aAAa,GAA2B;IACnD,wBAAwB;IACxB,KAAK,EAAE,YAAY;IACnB,MAAM,EAAE,YAAY;IACpB,KAAK,EAAE,YAAY;IACnB,MAAM,EAAE,YAAY;IACpB,MAAM,EAAE,YAAY;IACpB,MAAM,EAAE,YAAY;IACpB,MAAM;IACN,OAAO,EAAE,MAAM;IACf,MAAM,EAAE,MAAM;IACd,MAAM,EAAE,KAAK;IACb,OAAO,EAAE,KAAK;IACd,OAAO,EAAE,KAAK;IACd,OAAO,EAAE,KAAK;IACd,OAAO;IACP,OAAO,EAAE,MAAM;IACf,OAAO,EAAE,MAAM;IACf,MAAM,EAAE,MAAM;IACd,MAAM,EAAE,KAAK;IACb,OAAO,EAAE,QAAQ;IACjB,SAAS;IACT,MAAM,EAAE,QAAQ;IAChB,YAAY,EAAE,QAAQ;IACtB,WAAW,EAAE,QAAQ;IACrB,aAAa,EAAE,QAAQ;IACvB,gBAAgB;IAChB,KAAK,EAAE,UAAU;IACjB,MAAM,EAAE,UAAU;IAClB,MAAM,EAAE,MAAM;IACd,MAAM,EAAE,MAAM;IACd,SAAS;IACT,MAAM,EAAE,OAAO;IACf,MAAM,EAAE,OAAO;IACf,OAAO,EAAE,OAAO;IAChB,MAAM,EAAE,OAAO;IACf,MAAM,EAAE,OAAO;IACf,OAAO,EAAE,OAAO;IAChB,MAAM,EAAE,OAAO;IACf,kBAAkB;IAClB,KAAK,EAAE,QAAQ;IACf,KAAK,EAAE,MAAM;IACb,KAAK,EAAE,IAAI;IACX,KAAK,EAAE,MAAM;IACb,OAAO,EAAE,MAAM;IACf,KAAK,EAAE,QAAQ;IACf,QAAQ,EAAE,OAAO;IACjB,IAAI,EAAE,GAAG;IACT,MAAM,EAAE,KAAK;IACb,IAAI,EAAE,GAAG;IACT,MAAM,EAAE,KAAK;IACb,KAAK,EAAE,QAAQ;IACf,MAAM,EAAE,KAAK;IACb,KAAK,EAAE,OAAO;IACd,OAAO,EAAE,OAAO;IAChB,MAAM,EAAE,OAAO;IACf,OAAO,EAAE,OAAO;IAChB,gBAAgB;IAChB,OAAO,EAAE,MAAM;IACf,MAAM,EAAE,KAAK;CACd,CAAC;AAEF,kBAAkB;AAClB,MAAM,CAAC,MAAM,cAAc,GAAG,IAAI,GAAG,EAAsB,CAAC;AAC5D,MAAM,CAAC,MAAM,YAAY,GAAG,IAAI,CAAC,CAAC,iBAAiB;AAEnD;;GAEG;AACH,MAAM,UAAU,WAAW,CAAC,QAAgB;IAC1C,MAAM,GAAG,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;IAC5C,OAAO,aAAa,CAAC,GAAG,CAAC,IAAI,MAAM,CAAC;AACtC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,cAAc,CAAC,UAAkB;IAC/C,MAAM,aAAa,GAAG,IAAI,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;IACrD,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,CAAC;QAC/B,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,YAAY,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;QACrD,MAAM,QAAQ,GAAG,OAAO;aACrB,KAAK,CAAC,IAAI,CAAC;aACX,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;aACxB,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;QAC1E,OAAO,QAAQ,CAAC;IAClB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,uCAAuC,EAAE,KAAK,CAAC,CAAC;QAC9D,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED;;;;;;GAMG;AACH,SAAS,oBAAoB,CAAC,cAAsB,EAAE,YAAsB,EAAE,OAAe;IAC3F,MAAM,iBAAiB,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;IACtD,MAAM,kBAAkB,GAAG,iBAAiB,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;IAC3D,MAAM,YAAY,GAAG,kBAAkB,CAAC,CAAC,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC;IAE7F,IAAI,YAAY,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QAC/B,OAAO,cAAc,KAAK,YAAY,IAAI,cAAc,CAAC,UAAU,CAAC,GAAG,YAAY,GAAG,CAAC,CAAC;IAC1F,CAAC;IAED,IAAI,YAAY,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;QACxC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,YAAY,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QAC/B,MAAM,YAAY,GAAG,YAAY;aAC9B,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC;aACrB,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC;aACtB,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAC3B,OAAO,IAAI,MAAM,CAAC,QAAQ,YAAY,OAAO,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACtE,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,SAAS,CAAC,IAAY,EAAE,QAAkB;IACxD,MAAM,cAAc,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;IAChD,MAAM,YAAY,GAAG,cAAc,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAC/C,OAAO,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,oBAAoB,CAAC,cAAc,EAAE,YAAY,EAAE,OAAO,CAAC,CAAC,CAAC;AAC/F,CAAC;AAED;;GAEG;AACH,SAAS,eAAe,CAAC,KAAmD,EAAE,YAAoB,EAAE,iBAA2B;IAC7H,IAAI,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;QAAE,OAAO,KAAK,CAAC;IAC7C,IAAI,KAAK,CAAC,WAAW,EAAE,IAAI,kBAAkB,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC;QAAE,OAAO,KAAK,CAAC;IAC5E,IAAI,iBAAiB,CAAC,MAAM,GAAG,CAAC,IAAI,SAAS,CAAC,YAAY,EAAE,iBAAiB,CAAC;QAAE,OAAO,KAAK,CAAC;IAC7F,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,+BAA+B,CAC7C,GAAW,EACX,OAAe,EACf,iBAA2B,EAC3B,UAAiE,EAAE,EACnE,aAAqB,IAAI,EACzB,WAAmB,EAAE,EACrB,eAAuB,CAAC;IAExB,IAAI,CAAC;QACH,IAAI,OAAO,CAAC,MAAM,IAAI,UAAU,IAAI,YAAY,IAAI,QAAQ,EAAE,CAAC;YAC7D,OAAO,OAAO,CAAC;QACjB,CAAC;QAED,MAAM,OAAO,GAAG,WAAW,CAAC,GAAG,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;QAE1D,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,IAAI,OAAO,CAAC,MAAM,IAAI,UAAU;gBAAE,MAAM;YAExC,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;YACvC,MAAM,YAAY,GAAG,QAAQ,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAEjD,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,YAAY,EAAE,iBAAiB,CAAC;gBAAE,SAAS;YAEvE,OAAO,CAAC,IAAI,CAAC,EAAE,YAAY,EAAE,WAAW,EAAE,KAAK,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;YAEjE,IAAI,KAAK,CAAC,WAAW,EAAE,IAAI,OAAO,CAAC,MAAM,GAAG,UAAU,IAAI,YAAY,GAAG,CAAC,GAAG,QAAQ,EAAE,CAAC;gBACtF,+BAA+B,CAAC,QAAQ,EAAE,OAAO,EAAE,iBAAiB,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,YAAY,GAAG,CAAC,CAAC,CAAC;YACzH,CAAC;QACH,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,iCAAiC;IACnC,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB,CAAC,QAAgB;IACnD,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,IAAI,OAAO,CAAC,GAAG,CAAC,WAAW,IAAI,EAAE,CAAC;IAClE,MAAM,YAAY,GAAG;QACnB,IAAI,CAAC,OAAO,EAAE,SAAS,CAAC;QACxB,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC;QAC1B,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC;QACzB,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC;QAC1B,eAAe,EAAE,yCAAyC;QAC1D,OAAO;QACP,OAAO,CAAC,GAAG,CAAC,MAAM,IAAI,MAAM;KAC7B,CAAC;IACF,OAAO,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC;AACtE,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,WAAW,CAAC,QAAgB;IAC1C,MAAM,eAAe,GAAG,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;IACpG,MAAM,GAAG,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC;IACpD,OAAO,GAAG,CAAC,CAAC,CAAC,eAAe,CAAC,QAAQ,CAAC,IAAI,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;AAC3D,CAAC;AAID,SAAS,oBAAoB,CAAC,QAAgB,EAAE,QAAgB,EAAE,QAAgB;IAChF,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,WAAW,CAAC,QAAQ,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;QAC/D,MAAM,OAAO,GAAG,OAAO;aACpB,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;aACpC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACb,IAAI,CAAC,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC,CAAC,WAAW,EAAE;gBAAE,OAAO,CAAC,CAAC,CAAC;YACnD,IAAI,CAAC,CAAC,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC,WAAW,EAAE;gBAAE,OAAO,CAAC,CAAC;YAClD,OAAO,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACtC,CAAC,CAAC;aACD,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;aACjD,IAAI,CAAC,IAAI,CAAC,CAAC;QACd,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC;IAC9E,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,uBAAuB,EAAE,CAAC;IACnF,CAAC;AACH,CAAC;AAED,SAAS,gBAAgB,CAAC,QAAgB,EAAE,QAAgB,EAAE,QAAgB,EAAE,KAAoC;IAClH,MAAM,MAAM,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IACtC,MAAM,GAAG,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,KAAK,CAAC;IAC7D,MAAM,QAAQ,GAAG,GAAG,KAAK,KAAK,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,GAAG,KAAK,KAAK,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS,GAAG,EAAE,CAAC;IACjG,OAAO;QACL,IAAI,EAAE,QAAQ,EAAE,QAAQ;QACxB,OAAO,EAAE,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAClC,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,UAAU,EAAE,KAAK,CAAC,KAAK,CAAC,WAAW,EAAE;QACvD,OAAO,EAAE,IAAI,EAAE,QAAQ;KACxB,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,QAAgB,EAAE,QAAgB,EAAE,QAAgB,EAAE,KAAoC;IACjH,OAAO;QACL,IAAI,EAAE,QAAQ,EAAE,QAAQ;QACxB,OAAO,EAAE,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC;QACxC,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,UAAU,EAAE,KAAK,CAAC,KAAK,CAAC,WAAW,EAAE;KACxD,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,eAAe,CAAC,QAAgB,EAAE,UAAkB;IAClE,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAClF,MAAM,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,QAAQ,CAAC;QAEvD,MAAM,cAAc,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtC,MAAM,cAAc,GAAG,cAAc,CAAC,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;QACnE,IAAI,CAAC,cAAc,IAAI,CAAC,oBAAoB,CAAC,cAAc,CAAC,EAAE,CAAC;YAC7D,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,+CAA+C,EAAE,CAAC;QAC3G,CAAC;QAED,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC1B,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,gBAAgB,EAAE,CAAC;QAC5E,CAAC;QAED,MAAM,KAAK,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACjC,IAAI,KAAK,CAAC,WAAW,EAAE,EAAE,CAAC;YACxB,OAAO,oBAAoB,CAAC,QAAQ,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAC5D,CAAC;QAED,MAAM,OAAO,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;QACtC,MAAM,aAAa,GAAG,OAAO,CAAC,CAAC,CAAC,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,GAAG,IAAI,CAAC;QAC/D,IAAI,KAAK,CAAC,IAAI,GAAG,aAAa,EAAE,CAAC;YAC/B,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,mBAAmB,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC,mBAAmB,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,GAAG,EAAE,CAAC;QAC5K,CAAC;QAED,OAAO,OAAO;YACZ,CAAC,CAAC,gBAAgB,CAAC,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,CAAC;YACvD,CAAC,CAAC,eAAe,CAAC,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;IAC3D,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QACpB,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAC;QACxD,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,IAAI,qBAAqB,EAAE,CAAC;IACzI,CAAC;AACH,CAAC"}

package/dist/server/services/websocket/handler.d.ts

@@ -0,0 +1,246 @@
+import type { WSContext } from './types.js';
+export interface UsageReport {
+    tokensUsed: number;
+    sessionId?: string;
+    movementId?: string;
+}
+export type UsageReporter = (report: UsageReport) => void;
+export declare class WebSocketImproviseHandler {
+    private sessions;
+    private connections;
+    private autocompleteService;
+    private frecencyPath;
+    private usageReporter;
+    /** Per-tab selected git directory (tabId -> directory path) */
+    private gitDirectories;
+    constructor();
+    /**
+     * Set the usage reporter callback for sending usage data to platform
+     */
+    setUsageReporter(reporter: UsageReporter): void;
+    /**
+     * Load frecency data from disk
+     */
+    private loadFrecencyData;
+    /**
+     * Save frecency data to disk
+     */
+    private saveFrecencyData;
+    /**
+     * Record a file selection for frecency scoring
+     */
+    recordFileSelection(filePath: string): void;
+    /**
+     * Handle new WebSocket connection
+     */
+    handleConnection(ws: WSContext, _workingDir: string): void;
+    /**
+     * Handle incoming WebSocket message
+     */
+    handleMessage(ws: WSContext, message: string, workingDir: string): Promise<void>;
+    /**
+     * Dispatch a parsed message to the appropriate handler
+     */
+    private dispatchMessage;
+    /**
+     * Handle session-related messages (execute, cancel, history, new, approve, reject)
+     */
+    private handleSessionMessage;
+    /**
+     * Handle history/session listing messages
+     */
+    private handleHistoryMessage;
+    /**
+     * Handle file-related messages (autocomplete, readFile, recordSelection, notifications)
+     */
+    private handleFileMessage;
+    /**
+     * Handle terminal messages
+     */
+    private handleTerminalMessage;
+    /**
+     * Handle file explorer operations with success/error response pattern
+     */
+    private sendFileResult;
+    private handleListDirectory;
+    private handleWriteFile;
+    private handleCreateFile;
+    private handleCreateDirectory;
+    private handleDeleteFile;
+    private handleRenameFile;
+    private handleFileExplorerMessage;
+    /**
+     * Get a session or throw
+     */
+    private requireSession;
+    /**
+     * Set up event listeners for a session
+     */
+    private setupSessionListeners;
+    /**
+     * Resume a historical session for conversation continuity
+     * Falls back to creating a new session if the historical session cannot be found
+     * (e.g., server restarted before the session was saved to disk)
+     */
+    private resumeHistoricalSession;
+    /**
+     * Initialize a new tab with its own session
+     */
+    private initializeTab;
+    /**
+     * Get session for a specific tab
+     */
+    private getSession;
+    /**
+     * Handle connection close
+     */
+    handleClose(ws: WSContext): void;
+    /**
+     * Send message to WebSocket client
+     */
+    private send;
+    /**
+     * Get count of all historical sessions without reading file contents
+     */
+    private getSessionsCount;
+    /**
+     * Get paginated list of historical sessions from disk
+     * Returns minimal metadata - movements are stripped to just userPrompt preview
+     */
+    private getSessionsList;
+    /**
+     * Get a full session by ID (includes all movement data)
+     */
+    private getSessionById;
+    /**
+     * Delete a single session from disk
+     */
+    private deleteSession;
+    /**
+     * Clear all sessions from disk
+     */
+    private clearAllSessions;
+    /**
+     * Search sessions using grep on the history directory
+     * Searches through session file contents for matching text
+     * Returns paginated results with minimal metadata
+     */
+    private movementMatchesQuery;
+    private buildSessionSummary;
+    private searchSessions;
+    /**
+     * Cleanup session
+     */
+    cleanupSession(sessionId: string): void;
+    /**
+     * Clean up stale sessions
+     */
+    cleanupStaleSessions(): void;
+    /**
+     * Generate a notification summary using Claude Haiku
+     * Sends the result as a notificationSummary message
+     */
+    private generateNotificationSummary;
+    /**
+     * Create a fallback summary when AI summarization fails
+     */
+    private createFallbackSummary;
+    /**
+     * Handle git-related messages
+     */
+    private handleGitMessage;
+    /**
+     * Execute a git command and return stdout
+     */
+    private executeGitCommand;
+    /**
+     * Parse git status --porcelain output into structured format
+     */
+    private parseGitStatus;
+    /**
+     * Handle git status request
+     */
+    private handleGitStatus;
+    /**
+     * Handle git stage request
+     */
+    private handleGitStage;
+    /**
+     * Handle git unstage request
+     */
+    private handleGitUnstage;
+    /**
+     * Handle git commit request (with user-provided message)
+     */
+    private handleGitCommit;
+    /**
+     * Handle git commit with AI-generated message
+     * Uses Claude Code to analyze staged changes and generate a commit message
+     */
+    private handleGitCommitWithAI;
+    /**
+     * Extract the actual commit message from Claude's output.
+     * Sometimes Claude outputs reasoning before the actual message, so we need to parse it.
+     */
+    private extractCommitMessage;
+    /**
+     * Handle git push request
+     */
+    private handleGitPush;
+    /**
+     * Handle git log request
+     */
+    private handleGitLog;
+    /** Directories to skip when scanning for git repos */
+    private static readonly SKIP_DIRS;
+    /** Get the current branch name for a git repository */
+    private getRepoBranch;
+    /** Check if a directory name should be skipped when scanning */
+    private shouldSkipDir;
+    /** Recursively scan directories for git repositories */
+    private scanForGitRepos;
+    /**
+     * Discover git repositories in the working directory and subdirectories
+     */
+    private handleGitDiscoverRepos;
+    /**
+     * Set the git directory for operations
+     */
+    private handleGitSetDirectory;
+    /**
+     * Initialize a new terminal session or reconnect to existing one
+     */
+    private handleTerminalInit;
+    /**
+     * Reconnect to an existing terminal session
+     */
+    private handleTerminalReconnect;
+    /**
+     * List all active terminal sessions
+     */
+    private handleTerminalList;
+    /**
+     * Handle terminal input
+     */
+    private handleTerminalInput;
+    /**
+     * Handle terminal resize
+     */
+    private handleTerminalResize;
+    /**
+     * Handle terminal close
+     */
+    private handleTerminalClose;
+    private persistentHandlers;
+    private terminalListenerCleanups;
+    /**
+     * Initialize a persistent (tmux-backed) terminal session
+     * These sessions survive server restarts
+     */
+    private handleTerminalInitPersistent;
+    /**
+     * List all persistent terminal sessions (including those from previous server runs)
+     */
+    private handleTerminalListPersistent;
+}
+//# sourceMappingURL=handler.d.ts.map

package/dist/server/services/websocket/handler.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"handler.d.ts","sourceRoot":"","sources":["../../../../server/services/websocket/handler.ts"],"names":[],"mappings":"AA2BA,OAAO,KAAK,EAAsK,SAAS,EAAE,MAAM,YAAY,CAAC;AAEhN,MAAM,WAAW,WAAW;IAC1B,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,MAAM,aAAa,GAAG,CAAC,MAAM,EAAE,WAAW,KAAK,IAAI,CAAC;AAE1D,qBAAa,yBAAyB;IACpC,OAAO,CAAC,QAAQ,CAAuD;IACvE,OAAO,CAAC,WAAW,CAAkD;IACrE,OAAO,CAAC,mBAAmB,CAAsB;IACjD,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,aAAa,CAA8B;IACnD,+DAA+D;IAC/D,OAAO,CAAC,cAAc,CAAkC;;IAQxD;;OAEG;IACH,gBAAgB,CAAC,QAAQ,EAAE,aAAa,GAAG,IAAI;IAI/C;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAYxB;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAYxB;;OAEG;IACH,mBAAmB,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAK3C;;OAEG;IACH,gBAAgB,CAAC,EAAE,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,GAAG,IAAI;IAI1D;;OAEG;IACG,aAAa,CACjB,EAAE,EAAE,SAAS,EACb,OAAO,EAAE,MAAM,EACf,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,IAAI,CAAC;IAgBhB;;OAEG;YACW,eAAe;IA4D7B;;OAEG;IACH,OAAO,CAAC,oBAAoB;IA6C5B;;OAEG;IACH,OAAO,CAAC,oBAAoB;IA8B5B;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAoBzB;;OAEG;IACH,OAAO,CAAC,qBAAqB;IA8B7B;;OAEG;IACH,OAAO,CAAC,cAAc;IAOtB,OAAO,CAAC,mBAAmB;IAM3B,OAAO,CAAC,eAAe;IAMvB,OAAO,CAAC,gBAAgB;IAKxB,OAAO,CAAC,qBAAqB;IAK7B,OAAO,CAAC,gBAAgB;IAKxB,OAAO,CAAC,gBAAgB;IAMxB,OAAO,CAAC,yBAAyB;IAYjC;;OAEG;IACH,OAAO,CAAC,cAAc;IAMtB;;OAEG;IACH,OAAO,CAAC,qBAAqB;IA8C7B;;;;OAIG;YACW,uBAAuB;IAyDrC;;OAEG;YACW,aAAa;IAkC3B;;OAEG;IACH,OAAO,CAAC,UAAU;IAUlB;;OAEG;IACH,WAAW,CAAC,EAAE,EAAE,SAAS,GAAG,IAAI;IAIhC;;OAEG;IACH,OAAO,CAAC,IAAI;IAQZ;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAYxB;;;OAGG;IACH,OAAO,CAAC,eAAe;IAqDvB;;OAEG;IACH,OAAO,CAAC,cAAc;IAkCtB;;OAEG;IACH,OAAO,CAAC,aAAa;IA+BrB;;OAEG;IACH,OAAO,CAAC,gBAAgB;IA6BxB;;;;OAIG;IACH,OAAO,CAAC,oBAAoB;IAS5B,OAAO,CAAC,mBAAmB;IAgB3B,OAAO,CAAC,cAAc;IA2CtB;;OAEG;IACH,cAAc,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI;IAIvC;;OAEG;IACH,oBAAoB,IAAI,IAAI;IAG5B;;;OAGG;YACW,2BAA2B;IAgIzC;;OAEG;IACH,OAAO,CAAC,qBAAqB;IAY7B;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAkBxB;;OAEG;IACH,OAAO,CAAC,iBAAiB;IA4BzB;;OAEG;IACH,OAAO,CAAC,cAAc;IAyDtB;;OAEG;YACW,eAAe;IAyC7B;;OAEG;YACW,cAAc;IAgC5B;;OAEG;YACW,gBAAgB;IAoB9B;;OAEG;YACW,eAAe;IA2C7B;;;OAGG;YACW,qBAAqB;IAsInC;;;OAGG;IACH,OAAO,CAAC,oBAAoB;IAsE5B;;OAEG;YACW,aAAa;IAc3B;;OAEG;YACW,YAAY;IA0B1B,sDAAsD;IACtD,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAsC;IAEvE,uDAAuD;YACzC,aAAa;IAK3B,gEAAgE;IAChE,OAAO,CAAC,aAAa;IAIrB,wDAAwD;YAC1C,eAAe;IAwB7B;;OAEG;YACW,sBAAsB;IA2BpC;;OAEG;YACW,qBAAqB;IA0CnC;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAyH1B;;OAEG;IACH,OAAO,CAAC,uBAAuB;IA4F/B;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAe1B;;OAEG;IACH,OAAO,CAAC,mBAAmB;IA6B3B;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAsB5B;;OAEG;IACH,OAAO,CAAC,mBAAmB;IA0B3B,OAAO,CAAC,kBAAkB,CAA+H;IAGzJ,OAAO,CAAC,wBAAwB,CAAsC;IAEtE;;;OAGG;IACH,OAAO,CAAC,4BAA4B;IAmFpC;;OAEG;IACH,OAAO,CAAC,4BAA4B;CAkBrC"}