sourcepile 0.1.0

Flatten a folder of source files into a single annotated dump for LLMs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

mod cli;
mod filter;
mod formatter;
mod walker;

use anyhow::{Context, Result};
use clap::Parser;
use cli::Cli;
use std::{fs::File, io::Write, path::PathBuf};

fn main() -> Result<()> {
    // Parse flags / args
    let cli = Cli::parse();

    // Canonicalise the target directory
    let root = cli
        .path
        .canonicalize()
        .with_context(|| format!("`{}` is not a valid directory", cli.path.display()))?;

    // Discover files
    let files = walker::collect_files(&root, cli.all, !cli.ignore_gitignore)?;

    // Concatenate them all
    let dump = formatter::format_files(&root, &files, cli.max_lines)?;

    // Determine output path (default: ./sourcepile.txt)
    let outfile: PathBuf = cli
        .output
        .unwrap_or_else(|| std::env::current_dir().unwrap().join("sourcepile.txt"));

    // Write
    let mut f = File::create(&outfile)
        .with_context(|| format!("failed to create {}", outfile.display()))?;
    f.write_all(dump.as_bytes())
        .with_context(|| format!("failed to write {}", outfile.display()))?;

    println!("Wrote {} files → {}", files.len(), outfile.display());
    Ok(())
}