1
import os
2

3
def get_project_context(root_dir, ignore_dirs=['.git', 'node_modules', '__pycache__']):
4
    context = "### PROJECT STRUCTURE\n"
5
    file_contents = "\n### FILE CONTENTS\n"
6

7
    for root, dirs, files in os.walk(root_dir):
8
        dirs[:] = [d for d in dirs if d not in ignore_dirs]
9
        level = root.replace(root_dir, '').count(os.sep)
10
        indent = ' ' * 4 * level
11
        context += f"{indent}{os.path.basename(root)}/\n"
12

13
        for f in files:
14
            if f.endswith(('.py', '.ts', '.mdx', '.js', '.java')):
15
                full_path = os.path.join(root, f)
16
                with open(full_path, 'r') as content_file:
17
                    file_contents += f"\n-- FILE: {f} --\n{content_file.read()}\n"
18
    return context + file_contents

1
import { promises as fs } from 'fs';
2
import path from 'path';
3

4
async function getProjectContext(dir, ignore = ['.git', 'node_modules']) {
5
    let context = "### PROJECT STRUCTURE\n";
6
    let contents = "\n### FILE CONTENTS\n";
7

8
    async function walk(currentDir) {
9
        const files = await fs.readdir(currentDir, { withFileTypes: true });
10
        for (const file of files) {
11
            if (ignore.includes(file.name)) continue;
12
            const res = path.resolve(currentDir, file.name);
13
            if (file.isDirectory()) {
14
                context += `DIR: ${file.name}\n`;
15
                await walk(res);
16
            } else if (file.name.match(/\.(py|ts|mdx|js|java)$/)) {
17
                context += `FILE: ${file.name}\n`;
18
                const data = await fs.readFile(res, 'utf8');
19
                contents += `\n-- FILE: ${file.name} --\n${data}\n`;
20
            }
21
        }
22
    }
23
    await walk(dir);
24
    return context + contents;
25
}
26
export { getProjectContext };

1
import java.io.IOException;
2
import java.nio.file.*;
3
import java.util.stream.Collectors;
4

5
public class Bridge {
6
    public static String getProjectContext(String rootDir) throws IOException {
7
        StringBuilder context = new StringBuilder("### PROJECT STRUCTURE\n");
8
        StringBuilder contents = new StringBuilder("\n### FILE CONTENTS\n");
9

10
        Files.walk(Paths.get(rootDir))
11
            .filter(p -> !p.toString().contains(".git") && !p.toString().contains("node_modules"))
12
            .forEach(path -> {
13
                if (Files.isDirectory(path)) {
14
                    context.append("DIR: ").append(path.getFileName()).append("\n");
15
                } else if (path.toString().matches(".*\\.(py|ts|mdx|js|java)$")) {
16
                    context.append("FILE: ").append(path.getFileName()).append("\n");
17
                    try {
18
                        contents.append("\n-- FILE: ").append(path.getFileName())
19
                                .append(" --\n").append(Files.readString(path)).append("\n");
20
                    } catch (IOException e) { e.printStackTrace(); }
21
                }
22
            });
23
        return context.toString() + contents.toString();
24
    }
25
}