codebank 0.4.5

A powerful code documentation generator that creates structured markdown documentation from your codebase. Supports multiple languages including Rust, Python, TypeScript, C, and Go with intelligent parsing and formatting. Features test code filtering, summary generation, and customizable documentation strategies.
Documentation 
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62

use anyhow::Result;
use clap::{Parser, ValueEnum};
use codebank::{Bank, BankConfig, BankStrategy, CodeBank};
use std::fs;
use std::path::PathBuf;

#[derive(Parser, Debug)]
#[clap(
    name = "codebank",
    about = "A tool to generate code banks from source code",
    version
)]
struct Cli {
    input: PathBuf,

    /// Output file for the generated code bank (stdout if not provided)
    #[clap(short, long)]
    output: Option<PathBuf>,

    /// Strategy to use for generating the code bank
    #[clap(short, long, value_enum, default_value_t = OutputStrategy::Default)]
    strategy: OutputStrategy,
}

#[derive(Copy, Clone, Debug, ValueEnum)]
enum OutputStrategy {
    /// Include all code with full content
    Default,
    /// Include all code except tests
    NoTests,
    /// Include only public interfaces, not full implementations
    Summary,
}

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

    // Create a new code bank generator
    let code_bank = CodeBank::try_new()?;

    // Convert the CLI strategy to BankStrategy
    let strategy = match cli.strategy {
        OutputStrategy::Default => BankStrategy::Default,
        OutputStrategy::NoTests => BankStrategy::NoTests,
        OutputStrategy::Summary => BankStrategy::Summary,
    };

    let config = BankConfig::new(cli.input, strategy, vec![]);

    // Generate the code bank
    let content = code_bank.generate(&config)?;

    // Output to file or stdout
    if let Some(output_file) = cli.output {
        fs::write(&output_file, content)?;
        println!("Code bank written to {}", output_file.display());
    } else {
        println!("{}", content);
    }

    Ok(())
}