yggdrasil-cli 0.2.5

Yggdrasil is a project flattener and diff engine that turns any subset of your codebase into a single AI-ready codex (index + contents), or compares snapshots with annotated diffs.
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121

use crate::cli::Args;
use crate::scanner::collect_files;
use crate::snapshot::filelist::prepare_file_list;
use crate::snapshot::writer::{open_writer, OutputTarget};
use crate::snapshot::format_selection::select_formatter;

use std::fs::File;
use std::io::Write;

/// Inject FUR-style stats into the markdown buffer
fn finalize_markdown(buf: &[u8], out_path: &str) {
    let text = String::from_utf8_lossy(buf);

    let word_count = text.split_whitespace().count();
    let token_est = ((word_count as f32) * (1.33)).round() as usize;

    // Insert stats *above* INDEX
    let inject = format!(
        "> ✍️ Words: {}\n> 🪙 Tokens (est.): {}\n\n## INDEX",
        word_count, token_est
    );

    let final_text = text.replacen("## INDEX", &inject, 1);

    let mut file = File::create(out_path).expect("Failed to write final markdown file");
    file.write_all(final_text.as_bytes()).unwrap();
}

/// Run the project snapshot (default command)
pub fn run_snapshot(mut args: Args) {

    //
    // ============================================================
    // 1. HANDLE --whited (legacy) AND --printed (new)
    // ============================================================
    //

    // Legacy: --whited
    if let Some(opt) = &args.whited {
        args.contents = true;

        args.out = Some(match opt {
            Some(name) => name.clone(),
            None => "SHOW.md".to_string(),
        });

        if args.white.is_none() {
            args.white = Some(None); // triggers stdin pattern prompt
        }
    }

    // New: --printed
    if let Some(opt) = &args.printed {
        args.contents = true;

        args.out = Some(match opt {
            Some(name) => name.clone(),
            None => "SHOW.md".to_string(),
        });
    }

    //
    // ============================================================
    // 2. Run scan
    // ============================================================
    //

    let root = args.dir.clone();
    let mut writer = open_writer(&args);

    let files = collect_files(&args);
    let prepared = prepare_file_list(files);

    let fmt = select_formatter(&args);

    //
    // ============================================================
    // 3. Render snapshot
    // ============================================================
    //

    match &mut writer {
        //
        //  A) Markdown to file → buffer → inject → write at end
        //
        OutputTarget::Memory(buf) => {
            // Write into the buffer
            fmt.print_preamble(&root, buf);
            fmt.print_index(&prepared, buf);

            if args.contents {
                fmt.print_contents(&prepared, buf);
            }

            // Finalize and write out
            let out_path = args.out.as_ref().unwrap();
            finalize_markdown(buf.as_slice(), out_path);
        }

        //
        //  B) stdout (no injection)
        //
        OutputTarget::Stdout => {
            let out = &mut std::io::stdout();

            fmt.print_preamble(&root, out);
            fmt.print_index(&prepared, out);

            if args.contents {
                fmt.print_contents(&prepared, out);
            }
        }

        //
        //  C) Should never occur
        //
        OutputTarget::File(_) => {
            panic!("OutputTarget::File should not be used; use Memory for markdown.");
        }
    }
}