sapper-iq 1.1.28 → 1.1.30

package/sapper.mjs

@@ -316,6 +316,160 @@ function scanCodebase(dir = '.', depth = 0, maxDepth = 5) {
   return { files, totalSize };
 }
 
+// Scan directory for files (for @ file picker)
+function getFilesForPicker(dir = '.', prefix = '', maxFiles = 50) {
+  let files = [];
+  try {
+    const entries = fs.readdirSync(dir, { withFileTypes: true });
+    for (const entry of entries) {
+      if (files.length >= maxFiles) break;
+      if (IGNORE_DIRS.has(entry.name) || entry.name.startsWith('.')) continue;
+      
+      const fullPath = prefix ? `${prefix}/${entry.name}` : entry.name;
+      
+      if (entry.isDirectory()) {
+        files.push({ path: fullPath + '/', isDir: true });
+        // Recurse one level for common structures
+        const subFiles = getFilesForPicker(`${dir}/${entry.name}`, fullPath, 20);
+        files = files.concat(subFiles.slice(0, 15)); // Limit subdirectory files
+      } else {
+        const ext = entry.name.includes('.') ? '.' + entry.name.split('.').pop() : '';
+        if (CODE_EXTENSIONS.has(ext.toLowerCase()) || CODE_EXTENSIONS.has(entry.name)) {
+          try {
+            const stats = fs.statSync(`${dir}/${entry.name}`);
+            files.push({ path: fullPath, isDir: false, size: stats.size });
+          } catch (e) {
+            files.push({ path: fullPath, isDir: false, size: 0 });
+          }
+        }
+      }
+    }
+  } catch (e) {}
+  return files.slice(0, maxFiles);
+}
+
+// Interactive file picker with arrow keys
+async function pickFiles() {
+  const files = getFilesForPicker('.', '', 50).filter(f => !f.isDir);
+  
+  if (files.length === 0) {
+    console.log(chalk.yellow('No code files found in current directory.'));
+    return [];
+  }
+  
+  const selected = new Set();
+  let cursor = 0;
+  const pageSize = Math.min(15, process.stdout.rows - 10 || 15);
+  
+  // Enable raw mode for key capture
+  if (process.stdin.isTTY) {
+    process.stdin.setRawMode(true);
+  }
+  process.stdin.resume();
+  
+  const renderList = () => {
+    // Clear screen and move cursor to top
+    console.clear();
+    console.log(box(
+      `${chalk.cyan('↑↓')} Navigate  ${chalk.cyan('Space')} Toggle  ${chalk.cyan('a')} All  ${chalk.cyan('Enter')} Confirm  ${chalk.cyan('q/Esc')} Cancel`,
+      '📎 Select Files', 'cyan'
+    ));
+    console.log();
+    
+    // Calculate visible range (pagination)
+    const startIdx = Math.max(0, Math.min(cursor - Math.floor(pageSize / 2), files.length - pageSize));
+    const endIdx = Math.min(startIdx + pageSize, files.length);
+    
+    // Show scroll indicator if needed
+    if (startIdx > 0) {
+      console.log(chalk.gray('  ↑ more files above...'));
+    }
+    
+    for (let i = startIdx; i < endIdx; i++) {
+      const file = files[i];
+      const isSelected = selected.has(i);
+      const isCursor = i === cursor;
+      
+      const checkbox = isSelected ? chalk.green('◉') : chalk.gray('○');
+      const prefix = isCursor ? chalk.cyan('▸ ') : '  ';
+      const name = isCursor ? chalk.cyan.bold(file.path) : chalk.white(file.path);
+      const size = file.size ? chalk.gray(` (${Math.round(file.size/1024)}KB)`) : '';
+      
+      console.log(`${prefix}${checkbox} ${name}${size}`);
+    }
+    
+    if (endIdx < files.length) {
+      console.log(chalk.gray('  ↓ more files below...'));
+    }
+    
+    console.log();
+    console.log(chalk.gray(`  Selected: ${selected.size} file${selected.size !== 1 ? 's' : ''}`));
+  };
+  
+  return new Promise((resolve) => {
+    renderList();
+    
+    const onKeypress = (chunk, key) => {
+      if (!key) {
+        // Handle raw chunk for arrow keys
+        const str = chunk.toString();
+        if (str === '\x1b[A') key = { name: 'up' };
+        else if (str === '\x1b[B') key = { name: 'down' };
+        else if (str === '\x1b[C') key = { name: 'right' };
+        else if (str === '\x1b[D') key = { name: 'left' };
+        else if (str === ' ') key = { name: 'space' };
+        else if (str === '\r' || str === '\n') key = { name: 'return' };
+        else if (str === '\x1b' || str === 'q') key = { name: 'escape' };
+        else if (str === 'a' || str === 'A') key = { name: 'a' };
+        else if (str === '\x03') key = { name: 'c', ctrl: true }; // Ctrl+C
+      }
+      
+      if (!key) return;
+      
+      if (key.name === 'up' || key.name === 'k') {
+        cursor = cursor > 0 ? cursor - 1 : files.length - 1;
+        renderList();
+      } else if (key.name === 'down' || key.name === 'j') {
+        cursor = cursor < files.length - 1 ? cursor + 1 : 0;
+        renderList();
+      } else if (key.name === 'space' || key.name === 'right') {
+        if (selected.has(cursor)) {
+          selected.delete(cursor);
+        } else {
+          selected.add(cursor);
+        }
+        renderList();
+      } else if (key.name === 'a') {
+        // Toggle all
+        if (selected.size === files.length) {
+          selected.clear();
+        } else {
+          for (let i = 0; i < files.length; i++) selected.add(i);
+        }
+        renderList();
+      } else if (key.name === 'return') {
+        cleanup();
+        const selectedFiles = Array.from(selected).map(i => files[i].path);
+        console.log(chalk.green(`\n✓ Selected ${selectedFiles.length} file${selectedFiles.length !== 1 ? 's' : ''}`));
+        resolve(selectedFiles);
+      } else if (key.name === 'escape' || key.name === 'q' || (key.ctrl && key.name === 'c')) {
+        cleanup();
+        console.log(chalk.gray('\nCancelled.'));
+        resolve([]);
+      }
+    };
+    
+    const cleanup = () => {
+      process.stdin.removeListener('data', onKeypress);
+      if (process.stdin.isTTY) {
+        process.stdin.setRawMode(false);
+      }
+    };
+    
+    process.stdin.on('data', onKeypress);
+  });
+}
+
 // Format scan results for AI context
 function formatScanResults(scanResult) {
   let output = `\n══════════════════════════════════════\n`;
@@ -681,7 +835,8 @@ TOOL SYNTAX:
       if (input.toLowerCase() === '/help') {
         console.log();
         const helpContent = 
-          `${chalk.cyan('@file')}          ${chalk.gray('│')} Attach file to prompt (e.g., @src/app.js)\n` +
+          `${chalk.cyan('@')} or ${chalk.cyan('/attach')}  ${chalk.gray('│')} Pick files to attach (interactive)\n` +
+          `${chalk.cyan('@file')}          ${chalk.gray('│')} Attach file inline (e.g., @src/app.js)\n` +
           `${chalk.cyan('/scan')}          ${chalk.gray('│')} Scan codebase into context\n` +
           `${chalk.cyan('/recall')}        ${chalk.gray('│')} Search memory for relevant context\n` +
           `${chalk.cyan('/reset /clear')}  ${chalk.gray('│')} Clear all context\n` +
@@ -790,7 +945,56 @@ TOOL SYNTAX:
         continue;
       }
       
-      // Process @file attachments in prompt (e.g., "analyze @package.json" or "fix @src/index.js")
+      // Handle @ alone or /attach command - interactive file picker
+      if (input.trim() === '@' || input.toLowerCase() === '/attach') {
+        const selectedFiles = await pickFiles();
+        
+        if (selectedFiles.length === 0) continue;
+        
+        // Read and attach selected files
+        const fileAttachments = [];
+        for (const filePath of selectedFiles) {
+          try {
+            const stats = fs.statSync(filePath);
+            if (stats.size > MAX_FILE_SIZE) {
+              console.log(chalk.yellow(`⚠️  ${filePath} is too large, skipping`));
+              continue;
+            }
+            const content = fs.readFileSync(filePath, 'utf8');
+            fileAttachments.push({ path: filePath, content, size: stats.size });
+            console.log(chalk.green(`📎 Attached: ${filePath}`));
+          } catch (e) {
+            console.log(chalk.yellow(`⚠️  Could not read ${filePath}`));
+          }
+        }
+        
+        if (fileAttachments.length === 0) continue;
+        
+        // Ask for the prompt to go with these files
+        console.log();
+        const prompt = await safeQuestion(chalk.cyan('Your prompt for these files: '));
+        
+        if (!prompt.trim()) {
+          console.log(chalk.gray('Cancelled.'));
+          continue;
+        }
+        
+        // Build message with attachments
+        let attachedContent = '\n\n══════════════════════════════════════\n';
+        attachedContent += `📎 ATTACHED FILES (${fileAttachments.length})\n`;
+        attachedContent += '══════════════════════════════════════\n\n';
+        
+        for (const file of fileAttachments) {
+          attachedContent += `┌─── ${file.path} ───\n`;
+          attachedContent += file.content;
+          if (!file.content.endsWith('\n')) attachedContent += '\n';
+          attachedContent += `└─── END ${file.path} ───\n\n`;
+        }
+        
+        messages.push({ role: 'user', content: prompt + attachedContent });
+        // Continue to AI response (don't use 'continue' here)
+      } else {
+        // Process @file attachments in prompt (e.g., "analyze @package.json" or "fix @src/index.js")
       let processedInput = input;
       const fileAttachments = [];
       const attachRegex = /@([\w.\/\-_]+)/g;
@@ -835,6 +1039,7 @@ TOOL SYNTAX:
       }
       
       messages.push({ role: 'user', content: processedInput });
+      } // End of else block for non-@ input
 
       let toolRounds = 0; // Prevent infinite loops
       const MAX_TOOL_ROUNDS = 20;