fix: improve export functionality and handle unsupported files

- Exclude default module directories like node_modules, build, out, etc., during codebase export.
- Fix 'Export Selected as Markdown' functionality to support multiple selected files and folders.
- Add logic to handle unsupported files by exporting only the file name and path without file contents.
- Refactor file traversal logic to skip ignored directories and files as specified in .gitignore and default ignore patterns.
- Improve overall performance and user experience with better error handling and clearer file structure in exported Markdown.

Author: alpha912

src/extension.ts

@@ -8,8 +8,9 @@ export function activate(context: vscode.ExtensionContext) {
     exportCodebase();
   });
 
-  let exportSelected = vscode.commands.registerCommand('codebaseMD.exportSelected', (uri: vscode.Uri) => {
-    exportSelectedFiles(uri);
+  let exportSelected = vscode.commands.registerCommand('codebaseMD.exportSelected', (uri: vscode.Uri, uris: vscode.Uri[]) => {
+    const selectedUris = uris && uris.length > 0 ? uris : [uri];
+    exportSelectedFiles(selectedUris);
   });
 
   context.subscriptions.push(exportAll);
@@ -29,8 +30,8 @@ async function exportCodebase() {
   saveMarkdownFile(markdownContent);
 }
 
-async function exportSelectedFiles(uri: vscode.Uri) {
-  const files = await getFilesFromUri(uri);
+async function exportSelectedFiles(uris: vscode.Uri[]) {
+  const files = await getFilesFromUris(uris);
   if (files.length === 0) {
     vscode.window.showErrorMessage('No files to export.');
     return;
@@ -55,6 +56,10 @@ async function getAllFiles(dir: string): Promise<string[]> {
       }
 
       if (entry.isDirectory()) {
+        // Check if directory is ignored
+        if (ig.ignores(relativePath + '/')) {
+          continue;
+        }
         await traverse(fullPath);
       } else {
         files.push(fullPath);
@@ -66,7 +71,7 @@ async function getAllFiles(dir: string): Promise<string[]> {
   return files;
 }
 
-async function getFilesFromUri(uri: vscode.Uri): Promise<string[]> {
+async function getFilesFromUris(uris: vscode.Uri[]): Promise<string[]> {
   const files: string[] = [];
   const rootPath = vscode.workspace.workspaceFolders![0].uri.fsPath;
   const ig = createIgnoreInstance(rootPath);
@@ -80,6 +85,10 @@ async function getFilesFromUri(uri: vscode.Uri): Promise<string[]> {
     }
 
     if (stat.type === vscode.FileType.Directory) {
+      // Check if directory is ignored
+      if (ig.ignores(relativePath + '/')) {
+        return;
+      }
       const entries = await vscode.workspace.fs.readDirectory(currentUri);
       for (const [name, fileType] of entries) {
         const childUri = vscode.Uri.joinPath(currentUri, name);
@@ -90,7 +99,10 @@ async function getFilesFromUri(uri: vscode.Uri): Promise<string[]> {
     }
   }
 
-  await processUri(uri);
+  for (const uri of uris) {
+    await processUri(uri);
+  }
+
   return files;
 }
 
@@ -101,6 +113,33 @@ function createIgnoreInstance(rootDir: string): Ignore {
     const gitignoreContent = fs.readFileSync(gitignorePath, 'utf8');
     ig.add(gitignoreContent);
   }
+  // Add default ignored directories and files
+  ig.add([
+    'node_modules/',
+    'build/',
+    'out/',
+    'dist/',
+    '.git/',
+    '.svn/',
+    '.hg/',
+    '.vscode/',
+    '.idea/',
+    'coverage/',
+    'logs/',
+    '*.log',
+    '*.exe',
+    '*.dll',
+    '*.bin',
+    '*.lock',
+    '*.zip',
+    '*.tar',
+    '*.tar.gz',
+    '*.tgz',
+    '*.jar',
+    '*.class',
+    '*.pyc',
+    '__pycache__/'
+  ]);
   return ig;
 }
 
@@ -109,6 +148,17 @@ function isLargeFile(fileName: string): boolean {
   return largeFiles.includes(fileName);
 }
 
+function isSupportedFile(fileName: string): boolean {
+  const supportedExtensions = [
+    '.js', '.jsx', '.ts', '.tsx', '.html', '.css', '.scss', '.json', '.md', '.txt',
+    '.py', '.java', '.c', '.cpp', '.cs', '.rb', '.go', '.php', '.sh', '.xml',
+    '.yaml', '.yml', '.ini', '.bat', '.sql', '.rs', '.swift', '.kt', '.dart',
+    '.lua', '.r', '.pl', '.hs', '.erl', '.ex', '.el', '.jl', '.scala'
+  ];
+  const ext = path.extname(fileName).toLowerCase();
+  return supportedExtensions.includes(ext);
+}
+
 async function generateMarkdown(files: string[], rootPath: string): Promise<string> {
   let markdown = `# Project Export\n\n`;
 
@@ -125,12 +175,18 @@ async function generateMarkdown(files: string[], rootPath: string): Promise<stri
   // File contents
   for (const file of files) {
     const relativePath = path.relative(rootPath, file);
-    const code = fs.readFileSync(file, 'utf8');
-    const ext = path.extname(file).substring(1);
+    const fileName = path.basename(file);
     markdown += `\n### ${relativePath}\n\n`;
-    markdown += '```' + ext + '\n';
-    markdown += code;
-    markdown += '\n```\n';
+
+    if (isSupportedFile(fileName)) {
+      const code = fs.readFileSync(file, 'utf8');
+      const ext = path.extname(file).substring(1);
+      markdown += '```' + ext + '\n';
+      markdown += code;
+      markdown += '\n```\n';
+    } else {
+      markdown += `*(Unsupported file type)*\n`;
+    }
   }
 
   return markdown;