scai 0.1.48 → 0.1.50

package/dist/commands/AskCmd.js

@@ -5,12 +5,12 @@ import { searchFiles, queryFiles, getFunctionsForFiles } from '../db/fileIndex.j
 import { sanitizeQueryForFts } from '../utils/sanitizeQuery.js';
 import { generate } from '../lib/generate.js';
 import { buildContextualPrompt } from '../utils/buildContextualPrompt.js';
-import { generateFileTree } from '../utils/fileTree.js';
+import { generateFocusedFileTree } from '../utils/fileTree.js';
 import { log } from '../utils/log.js';
 import { PROMPT_LOG_PATH, SCAI_HOME, INDEX_DIR, RELATED_FILES_LIMIT, MAX_SUMMARY_LINES } from '../constants.js';
 export async function runAskCommand(query) {
     if (!query) {
-        query = await promptOnce('🧠 Ask your question:\n> ');
+        query = await promptOnce('💬 Ask your question:\n');
     }
     query = query.trim();
     if (!query) {
@@ -103,7 +103,7 @@ export async function runAskCommand(query) {
     // 🟩 STEP 6: Generate file tree
     let fileTree = '';
     try {
-        fileTree = generateFileTree(INDEX_DIR, 2);
+        fileTree = generateFocusedFileTree(INDEX_DIR, filepath, 2);
     }
     catch (e) {
         console.warn('⚠️ Could not generate file tree:', e);
@@ -144,11 +144,12 @@ export async function runAskCommand(query) {
 // 🟩 Helper: Prompt once
 function promptOnce(promptText) {
     return new Promise(resolve => {
+        console.log(promptText); // Instead of putting it *in* rl.question
         const rl = readline.createInterface({
             input: process.stdin,
             output: process.stdout,
         });
-        rl.question(promptText, answer => {
+        rl.question('> ', answer => {
             rl.close();
             resolve(answer.trim());
         });

package/dist/commands/FindCmd.js

@@ -1,6 +1,7 @@
 import { queryFiles } from '../db/fileIndex.js';
 import { sanitizeQueryForFts } from '../utils/sanitizeQuery.js';
 import path from 'path';
+import os from 'os';
 export async function runFindCommand(query) {
     if (!query) {
         console.error('❌ Please provide a search query.\n👉 Usage: scai find "keyword"');
@@ -13,9 +14,15 @@ export async function runFindCommand(query) {
         console.log('⚠️ No matching files found.');
         return;
     }
-    console.log(`✅ Found ${results.length} result(s).`);
-    console.log();
+    console.log(`✅ Found ${results.length} result(s).\n`);
+    const homeDir = os.homedir();
     results.forEach((result, index) => {
-        console.log(`📄 [${index + 1}] ${path.relative(process.cwd(), result.path)}`);
+        let absPath = path.resolve(result.path); // ensure absolute path
+        if (absPath.startsWith(homeDir)) {
+            absPath = absPath.replace(homeDir, '~');
+        }
+        // Normalize to forward slashes (especially for Windows)
+        absPath = absPath.replace(/\\/g, '/');
+        console.log(`📄 [${index + 1}] ${absPath}`);
     });
 }

package/dist/daemon/daemonBatch.js

@@ -8,7 +8,12 @@ import { log } from '../utils/log.js';
 import lockfile from 'proper-lockfile';
 import { summaryModule } from '../pipeline/modules/summaryModule.js';
 import { classifyFile } from '../fileRules/classifyFile.js';
+import { markFileAsSkippedByPath, selectUnprocessedFiles, updateFileWithSummaryAndEmbedding, } from '../db/sqlTemplates.js';
 const MAX_FILES_PER_BATCH = 5;
+/**
+ * Acquires a lock on the database to ensure that only one daemon batch
+ * can modify it at a time.
+ */
 async function lockDb() {
     try {
         return await lockfile.lock(DB_PATH);
@@ -18,15 +23,18 @@ async function lockDb() {
         throw err;
     }
 }
+/**
+ * Runs a daemon batch to process up to MAX_FILES_PER_BATCH unprocessed files.
+ * This includes:
+ * - Verifying file existence and validity
+ * - Generating summaries and embeddings if needed
+ * - Extracting functions from source files
+ * - Marking skipped files as necessary
+ */
 export async function runDaemonBatch() {
     log('🟡 Starting daemon batch...');
-    const rows = db.prepare(`
-    SELECT id, path, type, summary, indexed_at, last_modified, processing_status
-    FROM files
-    WHERE processing_status = 'unprocessed' OR summary IS NULL OR summary = ''
-    ORDER BY last_modified DESC
-    LIMIT ?
-  `).all(MAX_FILES_PER_BATCH);
+    // Selects up to MAX_FILES_PER_BATCH files that haven't been processed yet
+    const rows = db.prepare(selectUnprocessedFiles).all(MAX_FILES_PER_BATCH);
     if (rows.length === 0) {
         log('✅ No files left to process.');
         return false;
@@ -34,72 +42,62 @@ export async function runDaemonBatch() {
     const release = await lockDb();
     for (const row of rows) {
         log(`📂 Processing file: ${row.path}`);
+        // Skip if file is missing from the file system
         if (!fsSync.existsSync(row.path)) {
             log(`⚠️ Skipped missing file: ${row.path}`);
-            db.prepare(`UPDATE files SET processing_status = 'skipped' WHERE path = @path`).run({ path: row.path });
+            db.prepare(markFileAsSkippedByPath).run({ path: row.path });
             continue;
         }
+        // Skip if file is classified as something we don't process
         const classification = classifyFile(row.path);
         if (classification !== 'valid') {
             log(`⏭️ Skipping (${classification}): ${row.path}`);
-            db.prepare(`UPDATE files SET processing_status = 'skipped' WHERE path = @path`).run({ path: row.path });
+            db.prepare(markFileAsSkippedByPath).run({ path: row.path });
             continue;
         }
         try {
             const content = await fs.readFile(row.path, 'utf-8');
-            // 🧠 Only re-generate summary if it's missing or outdated
+            // Determine whether the file needs to be re-summarized
             const needsResummary = !row.summary ||
                 !row.indexed_at ||
                 (row.last_modified && new Date(row.last_modified) > new Date(row.indexed_at));
             if (needsResummary) {
                 log(`📝 Generating summary for ${row.path}...`);
+                // Generate a summary using the summary pipeline
                 const summaryResult = await summaryModule.run({ content, filepath: row.path });
                 const summary = summaryResult?.summary?.trim() || null;
                 let embedding = null;
+                // Generate an embedding from the summary (if present)
                 if (summary) {
                     const vector = await generateEmbedding(summary);
                     if (vector) {
                         embedding = JSON.stringify(vector);
                     }
                 }
-                db.prepare(`
-          UPDATE files
-          SET summary = @summary,
-              embedding = @embedding,
-              indexed_at = datetime('now')
-          WHERE path = @path
-        `).run({ summary, embedding, path: row.path });
+                // Update the file record with the new summary and embedding
+                db.prepare(updateFileWithSummaryAndEmbedding).run({
+                    summary,
+                    embedding,
+                    path: row.path,
+                });
                 log(`✅ Updated summary & embedding for ${row.path}`);
             }
             else {
                 log(`⚡ Skipped summary (up-to-date) for ${row.path}`);
             }
-            // 🧩 Extract functions and update processing status
+            // Extract top-level functions from the file and update the DB
             const extracted = await indexFunctionsForFile(row.path, row.id);
             if (extracted) {
-                db.prepare(`
-          UPDATE files
-          SET processing_status = 'extracted',
-              functions_extracted_at = datetime('now')
-          WHERE id = @id
-        `).run({ id: row.id });
                 log(`✅ Function extraction complete for ${row.path}\n`);
             }
             else {
-                // If no functions were found, set processing status to 'skipped' or 'failed'
-                db.prepare(`
-          UPDATE files
-          SET processing_status = 'failed',
-              functions_extracted_at = datetime('now')
-          WHERE id = @id
-        `).run({ id: row.id });
                 log(`ℹ️ No functions extracted for ${row.path}\n`);
             }
         }
         catch (err) {
             log(`❌ Failed: ${row.path}: ${err instanceof Error ? err.message : String(err)}\n`);
-            db.prepare(`UPDATE files SET processing_status = 'failed' WHERE path = @path`).run({ path: row.path });
         }
+        // Add a small delay to throttle processing
         await new Promise(resolve => setTimeout(resolve, 200));
     }
     await release();

package/dist/db/functionExtractors/extractFromJava.js

@@ -1,4 +1,8 @@
-// extractFromJava.ts
-export async function extractFromJava(filePath, content, fileId) {
+import { db } from '../client.js';
+import { markFileAsSkippedTemplate } from '../sqlTemplates.js';
+export async function extractFromJava(filePath, _content, fileId) {
     console.warn(`⛔️ Java extraction not implemented: ${filePath}`);
+    // Mark the file as skipped with the relevant status update
+    db.prepare(markFileAsSkippedTemplate).run({ id: fileId });
+    return false;
 }

package/dist/db/functionExtractors/extractFromJs.js

@@ -4,6 +4,8 @@ import { generateEmbedding } from '../../lib/generateEmbedding.js';
 import { db } from '../client.js';
 import path from 'path';
 import { log } from '../../utils/log.js';
+import fs from 'fs';
+import { markFileAsSkippedTemplate, markFileAsExtractedTemplate, markFileAsFailedTemplate } from '../sqlTemplates.js';
 function getFunctionName(node, parent, fileName) {
     if (node.id?.name)
         return node.id.name;
@@ -18,96 +20,104 @@ function getFunctionName(node, parent, fileName) {
     return `${fileName}:<anon>`;
 }
 export async function extractFromJS(filePath, content, fileId) {
-    const ast = parse(content, {
-        ecmaVersion: 'latest',
-        sourceType: 'module',
-        locations: true,
-    });
-    const functions = [];
-    walkAncestor(ast, {
-        FunctionDeclaration(node, ancestors) {
-            const parent = ancestors[ancestors.length - 2];
-            const name = getFunctionName(node, parent, path.basename(filePath));
-            functions.push({
-                name,
-                start_line: node.loc?.start.line ?? -1,
-                end_line: node.loc?.end.line ?? -1,
-                content: content.slice(node.start, node.end),
-            });
-        },
-        FunctionExpression(node, ancestors) {
-            const parent = ancestors[ancestors.length - 2];
-            const name = getFunctionName(node, parent, path.basename(filePath));
-            functions.push({
-                name,
-                start_line: node.loc?.start.line ?? -1,
-                end_line: node.loc?.end.line ?? -1,
-                content: content.slice(node.start, node.end),
-            });
-        },
-        ArrowFunctionExpression(node, ancestors) {
-            const parent = ancestors[ancestors.length - 2];
-            const name = getFunctionName(node, parent, path.basename(filePath));
-            functions.push({
-                name,
-                start_line: node.loc?.start.line ?? -1,
-                end_line: node.loc?.end.line ?? -1,
-                content: content.slice(node.start, node.end),
-            });
-        },
-    });
-    if (functions.length === 0) {
-        log(`⚠️ No functions found in: ${filePath}`);
-        return false;
-    }
-    log(`🔍 Found ${functions.length} functions in ${filePath}`);
-    for (const fn of functions) {
-        const embedding = await generateEmbedding(fn.content);
-        const result = db.prepare(`
-      INSERT INTO functions (
-        file_id, name, start_line, end_line, content, embedding, lang
-      ) VALUES (
-        @file_id, @name, @start_line, @end_line, @content, @embedding, @lang
-      )
-    `).run({
-            file_id: fileId,
-            name: fn.name,
-            start_line: fn.start_line,
-            end_line: fn.end_line,
-            content: fn.content,
-            embedding: JSON.stringify(embedding),
-            lang: 'js'
-        });
-        const callerId = result.lastInsertRowid;
-        const fnAst = parse(fn.content, {
+    try {
+        const code = fs.readFileSync(filePath, 'utf-8');
+        console.log(`[Debug] Attempting to parse: ${filePath}`);
+        console.log(`[Debug] First 3 lines:\n${code.split('\n').slice(0, 3).join('\n')}`);
+        const ast = parse(content, {
             ecmaVersion: 'latest',
             sourceType: 'module',
             locations: true,
         });
-        const calls = [];
-        walkAncestor(fnAst, {
-            CallExpression(node) {
-                if (node.callee?.type === 'Identifier' && node.callee.name) {
-                    calls.push({ calleeName: node.callee.name });
-                }
-            }
+        const functions = [];
+        walkAncestor(ast, {
+            FunctionDeclaration(node, ancestors) {
+                const parent = ancestors[ancestors.length - 2];
+                const name = getFunctionName(node, parent, path.basename(filePath));
+                functions.push({
+                    name,
+                    start_line: node.loc?.start.line ?? -1,
+                    end_line: node.loc?.end.line ?? -1,
+                    content: content.slice(node.start, node.end),
+                });
+            },
+            FunctionExpression(node, ancestors) {
+                const parent = ancestors[ancestors.length - 2];
+                const name = getFunctionName(node, parent, path.basename(filePath));
+                functions.push({
+                    name,
+                    start_line: node.loc?.start.line ?? -1,
+                    end_line: node.loc?.end.line ?? -1,
+                    content: content.slice(node.start, node.end),
+                });
+            },
+            ArrowFunctionExpression(node, ancestors) {
+                const parent = ancestors[ancestors.length - 2];
+                const name = getFunctionName(node, parent, path.basename(filePath));
+                functions.push({
+                    name,
+                    start_line: node.loc?.start.line ?? -1,
+                    end_line: node.loc?.end.line ?? -1,
+                    content: content.slice(node.start, node.end),
+                });
+            },
         });
-        for (const call of calls) {
-            db.prepare(`
-        INSERT INTO function_calls (caller_id, callee_name)
-        VALUES (@caller_id, @callee_name)
+        if (functions.length === 0) {
+            log(`⚠️ No functions found in: ${filePath}`);
+            db.prepare(markFileAsSkippedTemplate).run({ id: fileId });
+            return false;
+        }
+        log(`🔍 Found ${functions.length} functions in ${filePath}`);
+        for (const fn of functions) {
+            const embedding = await generateEmbedding(fn.content);
+            const result = db.prepare(`
+        INSERT INTO functions (
+          file_id, name, start_line, end_line, content, embedding, lang
+        ) VALUES (
+          @file_id, @name, @start_line, @end_line, @content, @embedding, @lang
+        )
       `).run({
-                caller_id: callerId,
-                callee_name: call.calleeName
+                file_id: fileId,
+                name: fn.name,
+                start_line: fn.start_line,
+                end_line: fn.end_line,
+                content: fn.content,
+                embedding: JSON.stringify(embedding),
+                lang: 'js'
+            });
+            const callerId = result.lastInsertRowid;
+            const fnAst = parse(fn.content, {
+                ecmaVersion: 'latest',
+                sourceType: 'module',
+                locations: true,
+            });
+            const calls = [];
+            walkAncestor(fnAst, {
+                CallExpression(node) {
+                    if (node.callee?.type === 'Identifier' && node.callee.name) {
+                        calls.push({ calleeName: node.callee.name });
+                    }
+                }
             });
+            for (const call of calls) {
+                db.prepare(`
+          INSERT INTO function_calls (caller_id, callee_name)
+          VALUES (@caller_id, @callee_name)
+        `).run({
+                    caller_id: callerId,
+                    callee_name: call.calleeName
+                });
+            }
+            log(`📌 Indexed function: ${fn.name} with ${calls.length} calls`);
         }
-        log(`📌 Indexed function: ${fn.name} with ${calls.length} calls`);
+        db.prepare(markFileAsExtractedTemplate).run({ id: fileId });
+        log(`✅ Marked functions as extracted for ${filePath}`);
+        return true;
+    }
+    catch (err) {
+        log(`❌ Failed to extract from: ${filePath}`);
+        log(`   ↳ ${String(err.message)}`);
+        db.prepare(markFileAsFailedTemplate).run({ id: fileId });
+        return false;
     }
-    db.prepare(`
-    UPDATE files 
-    SET processing_status = 'extracted'
-    WHERE id = @fileId
-  `).run({ fileId });
-    log(`✅ Marked functions as extracted for ${filePath}`);
-    return true;
 }

package/dist/db/functionExtractors/extractFromXML.js

@@ -1,4 +1,8 @@
-// extractFromJava.ts
-export async function extractFromXML(filePath, content, fileId) {
+import { db } from '../client.js';
+import { markFileAsSkippedTemplate } from '../sqlTemplates.js';
+export async function extractFromXML(filePath, _content, fileId) {
     console.warn(`⛔️ XML extraction not implemented: ${filePath}`);
+    // Mark the file as skipped with the relevant status update
+    db.prepare(markFileAsSkippedTemplate).run({ id: fileId });
+    return false;
 }

package/dist/db/functionExtractors/index.js

@@ -4,6 +4,7 @@ import { extractFromJava } from './extractFromJava.js';
 import { extractFromJS } from './extractFromJs.js';
 import { extractFromXML } from './extractFromXML.js';
 import { db } from '../client.js';
+import { markFileAsFailedTemplate, markFileAsSkippedByPath } from '../sqlTemplates.js';
 /**
  * Detects file type and delegates to the appropriate extractor.
  */
@@ -11,38 +12,27 @@ export async function extractFunctionsFromFile(filePath, content, fileId) {
     const type = detectFileType(filePath).trim().toLowerCase();
     try {
         if (type === 'js' || type === 'ts' || type === 'javascript' || type === 'typescript') {
-            log(`✅ Attempting to extract JS functions from ${filePath}\n`);
+            log(`✅ Attempting to extract JS functions from ${filePath}`);
             return await extractFromJS(filePath, content, fileId);
         }
         if (type === 'java') {
             log(`❌ Nothing extracted for ${filePath} due to missing implementation`);
             await extractFromJava(filePath, content, fileId);
-            // move into extract file
-            db.prepare(`
-        UPDATE files SET processing_status = 'skipped' WHERE id = @id
-      `).run({ id: fileId });
             return false;
         }
         if (type === 'xml') {
             log(`❌ Nothing extracted for ${filePath} due to missing implementation`);
             await extractFromXML(filePath, content, fileId);
-            // move into extract file
-            db.prepare(`
-        UPDATE files SET processing_status = 'skipped' WHERE id = @id
-      `).run({ id: fileId });
             return false;
         }
         log(`⚠️ Unsupported file type: ${type} for function extraction. Skipping ${filePath}`);
-        db.prepare(`
-      UPDATE files SET processing_status = 'skipped' WHERE id = @id
-    `).run({ id: fileId });
+        db.prepare(markFileAsSkippedByPath).run({ path: filePath });
         return false;
     }
     catch (error) {
         log(`❌ Failed to extract functions from ${filePath}: ${error instanceof Error ? error.message : error}`);
-        db.prepare(`
-      UPDATE files SET processing_status = 'failed' WHERE id = @id
-    `).run({ id: fileId });
+        // Use the sqlTemplate to mark the file as 'failed'
+        db.prepare(markFileAsFailedTemplate).run({ id: fileId });
         return false;
     }
 }

package/dist/db/sqlTemplates.js

@@ -51,3 +51,51 @@ export const insertFunctionCallTemplate = `
   INSERT INTO function_calls (caller_id, callee_name)
   VALUES (:caller_id, :callee_name)
 `;
+// Mark a file as unprocessed
+export const markFileAsUnprocessedTemplate = `
+  UPDATE files
+  SET processing_status = 'unprocessed',
+      functions_extracted_at = NULL
+  WHERE id = :id
+`;
+// Mark a file as extracted
+export const markFileAsExtractedTemplate = `
+  UPDATE files
+  SET processing_status = 'extracted',
+      functions_extracted_at = CURRENT_TIMESTAMP
+  WHERE id = :id
+`;
+// Mark a file as skipped (not extractable)
+export const markFileAsSkippedTemplate = `
+  UPDATE files
+  SET processing_status = 'skipped',
+      functions_extracted_at = NULL
+  WHERE id = :id
+`;
+// Mark a file as failed
+export const markFileAsFailedTemplate = `
+  UPDATE files
+  SET processing_status = 'failed',
+      functions_extracted_at = NULL
+  WHERE id = :id
+`;
+export const selectUnprocessedFiles = `
+  SELECT id, path, type, summary, indexed_at, last_modified, processing_status
+  FROM files
+  WHERE processing_status = 'unprocessed' OR summary IS NULL OR summary = ''
+  ORDER BY last_modified DESC
+  LIMIT ?
+`;
+export const markFileAsSkippedByPath = `
+  UPDATE files
+  SET processing_status = 'skipped',
+    functions_extracted_at = NULL
+  WHERE path = @path
+`;
+export const updateFileWithSummaryAndEmbedding = `
+  UPDATE files
+  SET summary = @summary,
+      embedding = @embedding,
+      indexed_at = datetime('now')
+  WHERE path = @path
+`;

package/dist/fileRules/fileClassifier.js

@@ -1,9 +1,48 @@
 // utils/fileClassifier.ts
 import path from 'path';
+/**
+ * Determines whether a file is likely to be a *generated* or *bundled* file,
+ * rather than handwritten source code.
+ *
+ * This helps filter out files that shouldn't be analyzed for user-authored logic,
+ * like minified JS bundles, Webpack chunks, TypeScript output, etc.
+ */
 export function isGeneratedOrBundledFile(filePath) {
     const base = path.basename(filePath);
-    const isMinified = /\.min\.(js|ts)$/.test(base); // ✅ New check
+    /**
+     * Minified file detection:
+     * Matches file names like `something.min.js` or `app.min.ts`.
+     * These are typically compiled output intended for production and are not original source code.
+     */
+    const isMinified = /\.min\.(js|ts)$/.test(base);
+    /**
+     * Hash-named or chunk file detection:
+     * Matches file names like `bundle.839abc.js`, `chunk.123abc.ts`, or `main-worker.js`.
+     * These are often created by bundlers like Webpack, Vite, or Rollup.
+     */
     const isHashNamed = /[-_.](worker|bundle|chunk|[a-f0-9]{6,})\.(js|ts)$/.test(base);
-    const isInOutputFolder = /[\\/]dist[\\/]|[\\/]build[\\/]|[\\/]assets[\\/]|[\\/]node_modules[\\/]/i.test(filePath);
-    return isMinified || isHashNamed || isInOutputFolder;
+    /**
+     * Output folder detection:
+     * These folders are commonly used to store compiled or bundled output.
+     * Examples: `dist/`, `build/`, `assets/`, `node_modules/`, `plugins/`
+     *
+     * If a file is inside any of these folders, we consider it generated/bundled.
+     */
+    const outputDirs = ['dist', 'build', 'assets', 'node_modules', 'plugins'];
+    const isInKnownOutputFolder = outputDirs.some(dir => new RegExp(`[\\\\/]${dir}[\\\\/]`, 'i').test(filePath));
+    /**
+     * Special case: `lib/` folder
+     * The `lib` folder may contain either handwritten code or compiled output, depending on the project.
+     * To avoid over-filtering, we only treat files in `lib/` as generated if they also look minified or hashed.
+     */
+    const isInLib = /[\\/]lib[\\/]/i.test(filePath);
+    const isLikelyBundledLib = isInLib && (isMinified || isHashNamed);
+    /**
+     * Return true if *any* of the following conditions are met:
+     *  - The file looks minified (e.g., `.min.js`)
+     *  - The file has a hash or is a known bundle/chunk/worker
+     *  - The file is located in a known output directory
+     *  - The file is in `lib/` and has signs of being generated (minified or hashed)
+     */
+    return isMinified || isHashNamed || isInKnownOutputFolder || isLikelyBundledLib;
 }

package/dist/utils/fileTree.js

@@ -1,30 +1,31 @@
 import fs from 'fs';
 import path from 'path';
-const IGNORED_DIRS = new Set([
-    'node_modules', 'dist', '.git', '.vscode', 'coverage', 'build', 'out', 'logs', 'tmp'
-]);
-export function generateFileTree(dir, maxDepth = 3, currentDepth = 0, prefix = '') {
-    if (currentDepth > maxDepth)
+export function generateFocusedFileTree(rootDir, focusPath, maxDepth = 2) {
+    const absoluteFocus = path.resolve(focusPath);
+    const parentDir = path.dirname(absoluteFocus);
+    const relativeTitle = path.relative(rootDir, parentDir).replace(/\\/g, '/');
+    const tree = generateFileTree(parentDir, maxDepth, absoluteFocus);
+    return `📂 ${relativeTitle || '.'}\n${tree}`;
+}
+function generateFileTree(dir, depth, highlightPath, prefix = '') {
+    if (depth < 0)
         return '';
     let output = '';
-    const items = fs.readdirSync(dir, { withFileTypes: true })
-        .filter(item => !IGNORED_DIRS.has(item.name))
-        .sort((a, b) => {
-        // Directories first
-        if (a.isDirectory() && !b.isDirectory())
-            return -1;
-        if (!a.isDirectory() && b.isDirectory())
-            return 1;
-        return a.name.localeCompare(b.name);
-    });
-    for (const [i, item] of items.entries()) {
-        const isLast = i === items.length - 1;
+    const entries = fs.readdirSync(dir, { withFileTypes: true });
+    const sorted = entries.sort((a, b) => Number(b.isDirectory()) - Number(a.isDirectory()));
+    sorted.forEach((entry, index) => {
+        const isLast = index === sorted.length - 1;
         const connector = isLast ? '└── ' : '├── ';
-        const childPrefix = prefix + (isLast ? '    ' : '│   ');
-        output += `${prefix}${connector}${item.name}\n`;
-        if (item.isDirectory()) {
-            output += generateFileTree(path.join(dir, item.name), maxDepth, currentDepth + 1, childPrefix);
+        const fullPath = path.join(dir, entry.name);
+        const isHighlighted = highlightPath && path.resolve(fullPath) === path.resolve(highlightPath);
+        if (entry.isDirectory()) {
+            output += `${prefix}${connector}${entry.name}/\n`;
+            output += generateFileTree(fullPath, depth - 1, highlightPath, prefix + (isLast ? '    ' : '│   '));
         }
-    }
+        else {
+            const name = isHighlighted ? `➡️ ${entry.name}` : entry.name;
+            output += `${prefix}${connector}${name}\n`;
+        }
+    });
     return output;
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "scai",
-  "version": "0.1.48",
+  "version": "0.1.50",
   "type": "module",
   "bin": {
     "scai": "./dist/index.js"