gitmaps 1.1.0 → 1.1.2

package/app/api/repo/tree/route.ts

@@ -1,104 +1,188 @@
-import { measure } from 'measure-fn';
-import simpleGit from 'simple-git';
-import { readFileSync, existsSync } from 'fs';
-import path from 'path';
-import { validateRepoPath } from '../validate-path';
-
-const BINARY_EXTS = new Set(['png', 'jpg', 'jpeg', 'gif', 'bmp', 'ico', 'svg', 'webp', 'mp3', 'mp4', 'wav', 'ogg', 'avi', 'mov', 'zip', 'tar', 'gz', 'rar', '7z', 'pdf', 'doc', 'docx', 'xls', 'xlsx', 'ppt', 'exe', 'dll', 'so', 'dylib', 'woff', 'woff2', 'ttf', 'eot', 'otf', 'lock']);
-
-export async function POST(req: Request) {
-    return measure('api:repo:tree', async () => {
-        try {
-            const { path: repoPath } = await req.json();
-
-            if (!repoPath) {
-                return new Response('Repository path is required', { status: 400 });
-            }
-
-            const blocked = validateRepoPath(repoPath);
-            if (blocked) return blocked;
-
-            const git = simpleGit(repoPath);
-
-            // Get tracked files only (respects .gitignore by definition)
-            const result = await git.raw(['ls-files']);
-            const ignoreDirs = ['node_modules', '.git', 'dist', 'build', '.next', '.cache', 'coverage', '.turbo', '__pycache__', '.tsbuildinfo'];
-
-            // Also parse .gitignore for extra patterns
-            const gitignorePatterns: string[] = [];
-            const gitignorePath = path.join(repoPath, '.gitignore');
-            if (existsSync(gitignorePath)) {
-                try {
-                    const content = readFileSync(gitignorePath, 'utf-8');
-                    content.split('\n').forEach(line => {
-                        line = line.trim();
-                        if (line && !line.startsWith('#')) {
-                            // Normalize: remove trailing slashes for dir matching
-                            const clean = line.replace(/\/+$/, '');
-                            if (clean) gitignorePatterns.push(clean);
-                        }
-                    });
-                } catch (e) { /* ignore */ }
-            }
-
-            const filePaths = result.trim().split('\n').filter(fp => {
-                if (!fp) return false;
-                // Filter out known heavy directories
-                if (ignoreDirs.some(d => fp.startsWith(d + '/') || fp.startsWith(d + '\\'))) return false;
-                // Filter out files matching gitignore patterns (extra safety)
-                for (const pattern of gitignorePatterns) {
-                    if (fp.startsWith(pattern + '/') || fp.startsWith(pattern + '\\')) return false;
-                    if (fp === pattern) return false;
-                    // Simple glob: *.ext
-                    if (pattern.startsWith('*.')) {
-                        const ext = pattern.substring(1); // .ext
-                        if (fp.endsWith(ext)) return false;
-                    }
-                }
-                return true;
-            });
-
-            const files = filePaths.map(filePath => {
-                const parts = filePath.split('/');
-                const name = parts[parts.length - 1];
-                const ext = name.includes('.') ? name.split('.').pop()!.toLowerCase() : '';
-
-                let content = null;
-                let lines = 0;
-                let isBinary = BINARY_EXTS.has(ext);
-
-                if (!isBinary) {
-                    try {
-                        const fullPath = path.join(repoPath, filePath);
-                        const raw = readFileSync(fullPath, 'utf-8');
-                        const allLines = raw.split('\n');
-                        lines = allLines.length;
-                        // Send full content for small/medium files, truncate very large ones
-                        if (allLines.length > 10000) {
-                            content = allLines.slice(0, 10000).join('\n');
-                        } else {
-                            content = raw;
-                        }
-                    } catch (e) {
-                        content = null;
-                    }
-                }
-
-                return {
-                    path: filePath,
-                    name,
-                    ext,
-                    type: 'file',
-                    content,
-                    lines,
-                    isBinary
-                };
-            });
-
-            return Response.json({ files, total: files.length });
-        } catch (error: any) {
-            console.error('api:repo:tree:error', error);
-            return new Response(`Error: ${error.message}`, { status: 500 });
-        }
-    });
-}
+import { measure } from 'measure-fn';
+import simpleGit from 'simple-git';
+import { readFileSync, existsSync, readdirSync, statSync } from 'fs';
+import path from 'path';
+import { validateRepoPath } from '../validate-path';
+
+const BINARY_EXTS = new Set(['png', 'jpg', 'jpeg', 'gif', 'bmp', 'ico', 'svg', 'webp', 'mp3', 'mp4', 'wav', 'ogg', 'avi', 'mov', 'zip', 'tar', 'gz', 'rar', '7z', 'pdf', 'doc', 'docx', 'xls', 'xlsx', 'ppt', 'exe', 'dll', 'so', 'dylib', 'woff', 'woff2', 'ttf', 'eot', 'otf', 'lock']);
+const IMAGE_EXTS = new Set(['png', 'jpg', 'jpeg', 'gif', 'bmp', 'ico', 'svg', 'webp']);
+const PDF_EXTS = new Set(['pdf']);
+const MAX_READ_SIZE = 2 * 1024 * 1024;
+
+export async function POST(req: Request) {
+    return measure('api:repo:tree', async () => {
+        try {
+            const body = await req.json();
+            const repoPath = body.path;
+            const stream = body.stream === true;
+            const includeAll = body.includeAll === true;
+
+            if (!repoPath) {
+                return new Response('Repository path is required', { status: 400 });
+            }
+
+            const blocked = validateRepoPath(repoPath);
+            if (blocked) return blocked;
+
+            const ignoreDirs = new Set(['node_modules', '.git', 'dist', 'build', '.next', '.cache', 'coverage', '.turbo', '__pycache__', '.tsbuildinfo']);
+
+            // Recursively scan filesystem for all files (ignoring standard dirs)
+            function scanDir(dir: string, prefix: string): string[] {
+                const results: string[] = [];
+                try {
+                    const entries = readdirSync(dir);
+                    for (const entry of entries) {
+                        if (ignoreDirs.has(entry)) continue;
+                        const fullPath = path.join(dir, entry);
+                        const relativePath = prefix ? `${prefix}/${entry}` : entry;
+                        try {
+                            const stats = statSync(fullPath);
+                            if (stats.isDirectory()) {
+                                results.push(...scanDir(fullPath, relativePath));
+                            } else if (stats.isFile()) {
+                                results.push(relativePath);
+                            }
+                        } catch (e: any) {
+                            console.warn(`[tree:scanDir] stat error: ${fullPath}: ${e.message}`);
+                        }
+                    }
+                } catch (e: any) {
+                    console.warn(`[tree:scanDir] readdir error: ${dir}: ${e.message}`);
+                }
+                return results;
+            }
+
+            let filePaths: string[];
+
+            const git = simpleGit(repoPath);
+            const isRepo = await git.checkIsRepo().catch(() => false);
+
+            if (!isRepo || includeAll) {
+                // Not a git repo or explicit all-files mode: scan filesystem
+                filePaths = scanDir(repoPath, '');
+            } else {
+                // Get tracked files
+                const result = await git.raw(['ls-files']);
+                const trackedPaths = result.trim().split('\n').filter(fp => {
+                    if (!fp) return false;
+                    if (Array.from(ignoreDirs).some(d => fp.startsWith(d + '/') || fp.startsWith(d + '\\'))) return false;
+                    return true;
+                });
+
+                // If very few tracked files, also scan filesystem for untracked content
+                // This catches repos where most content (PDFs, images) is gitignored
+                if (trackedPaths.length < 50) {
+                    const allPaths = scanDir(repoPath, '');
+                    console.log(`[tree] ${trackedPaths.length} tracked, ${allPaths.length} on disk`);
+                    if (allPaths.length > trackedPaths.length * 5) {
+                        // Lots of untracked content — include everything
+                        const trackedSet = new Set(trackedPaths);
+                        filePaths = allPaths;
+                        // But still filter out obvious junk from non-tracked scan
+                        filePaths = filePaths.filter(fp => {
+                            if (Array.from(ignoreDirs).some(d => fp.startsWith(d + '/') || fp.startsWith(d + '\\'))) return false;
+                            return true;
+                        });
+                        console.log(`[tree] ${trackedPaths.length} tracked, ${allPaths.length} on disk → including all ${filePaths.length} files`);
+                    } else {
+                        filePaths = trackedPaths;
+                    }
+                } else {
+                    filePaths = trackedPaths;
+                }
+            }
+
+            function readFile(filePath: string) {
+                const parts = filePath.split('/');
+                const name = parts[parts.length - 1];
+                const ext = name.includes('.') ? name.split('.').pop()!.toLowerCase() : '';
+
+                let content = null;
+                let lines = 0;
+                let size = 0;
+                let isBinary = BINARY_EXTS.has(ext);
+                const isImage = IMAGE_EXTS.has(ext);
+                const isPdf = PDF_EXTS.has(ext);
+
+                if (!isBinary) {
+                    try {
+                        const fullPath = path.join(repoPath, filePath);
+                        const file = Bun.file(fullPath);
+                        size = file.size;
+
+                        // Skip reading content for very large files
+                        if (size > MAX_READ_SIZE) {
+                            isBinary = true;
+                        } else {
+                            const raw = readFileSync(fullPath, 'utf-8');
+                            size = raw.length;
+                            const allLines = raw.split('\n');
+                            lines = allLines.length;
+                            if (allLines.length > 10000) {
+                                content = allLines.slice(0, 10000).join('\n');
+                            } else {
+                                content = raw;
+                            }
+                        }
+                    } catch (e) {
+                        content = null;
+                    }
+                } else {
+                    // For binary files, at least get the file size
+                    try {
+                        const fullPath = path.join(repoPath, filePath);
+                        size = Bun.file(fullPath).size;
+                    } catch (_) {}
+                }
+
+                return { path: filePath, name, ext, type: 'file', content, lines, size, isBinary, isImage, isPdf };
+            }
+
+            // ── Streaming mode: NDJSON with total header ──
+            if (stream) {
+                const total = filePaths.length;
+                const BATCH_SIZE = 20;
+                const encoder = new TextEncoder();
+
+                const readable = new ReadableStream({
+                    start(controller) {
+                        // First line: total count
+                        controller.enqueue(encoder.encode(JSON.stringify({ total }) + '\n'));
+
+                        let i = 0;
+                        function nextBatch() {
+                            const end = Math.min(i + BATCH_SIZE, total);
+                            const batch: any[] = [];
+                            for (; i < end; i++) {
+                                batch.push(readFile(filePaths[i]));
+                            }
+                            controller.enqueue(encoder.encode(JSON.stringify({ files: batch, loaded: i }) + '\n'));
+
+                            if (i < total) {
+                                // Yield to event loop between batches
+                                setTimeout(nextBatch, 0);
+                            } else {
+                                controller.close();
+                            }
+                        }
+                        nextBatch();
+                    }
+                });
+
+                return new Response(readable, {
+                    headers: {
+                        'Content-Type': 'application/x-ndjson',
+                        'Cache-Control': 'no-cache',
+                    }
+                });
+            }
+
+            // ── Legacy non-streaming mode ──
+            const files = filePaths.map(readFile);
+            return Response.json({ files, total: files.length });
+        } catch (error: any) {
+            console.error('api:repo:tree:error', error);
+            return new Response(`Error: ${error.message}`, { status: 500 });
+        }
+    });
+}

package/app/api/version/route.ts

@@ -0,0 +1,26 @@
+import path from 'path';
+
+function runGit(args: string[]): string {
+    const repoRoot = path.resolve(import.meta.dir, '../../..');
+    const proc = Bun.spawnSync(['git', ...args], {
+        cwd: repoRoot,
+        stdout: 'pipe',
+        stderr: 'pipe',
+    });
+
+    if (proc.exitCode !== 0) {
+        return '';
+    }
+
+    return proc.stdout.toString().trim();
+}
+
+export async function GET() {
+    const commit = process.env.GIT_COMMIT_HASH || runGit(['rev-parse', '--short', 'HEAD']) || 'unknown';
+    const commitDate = process.env.GIT_COMMIT_DATE || runGit(['log', '-1', '--format=%cs']) || '';
+
+    return Response.json({
+        commit,
+        commitDate,
+    });
+}