snippy 0.1.1

A command-line tool for that makes using LLMs for code generation a breeze
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
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
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138

use crate::trie::{Trie, TrieNode};
use crate::ClipboardError;
use std::collections::HashMap;
use std::path::{Path, PathBuf};
use tracing::{debug, info, trace};

/// Prints the statistics of token counts for files.
pub fn print_stats(token_counts: &HashMap<PathBuf, usize>) -> Result<(), ClipboardError> {
    debug!("Printing statistics for token counts");
    let mut trie = Trie::new();
    let mut total_tokens = 0;

    for (path, &token_count) in token_counts {
        trace!(
            "Inserting path into trie: {:?} with {} tokens",
            path,
            token_count
        );
        trie.insert(path, token_count)?;
        total_tokens += token_count;
    }

    info!("Overall ({} tokens)", total_tokens);
    print_tree(&trie.get_root(), "", true)?;
    Ok(())
}

fn print_tree(node: &TrieNode, prefix: &str, is_last: bool) -> Result<(), ClipboardError> {
    let connector = if is_last { "┗━━" } else { "┣━━" };

    let mut children: Vec<_> = node.children.iter().collect();
    children.sort_by(|a, b| a.0.cmp(b.0));

    for (i, (name, child)) in children.iter().enumerate() {
        let is_last_child = i == children.len() - 1;
        let new_prefix = format!("{}{}    ", prefix, if is_last_child { " " } else { "┃" });

        if child.token_count.is_some() {
            info!(
                "{}{} {} {} ({} tokens)",
                prefix,
                connector,
                get_file_icon(&Path::new(name))
                    .map_err(|e| ClipboardError::FileReadError(e.to_string()))?,
                name,
                child.token_count.unwrap()
            );
        } else {
            let total_tokens = child.calculate_total_tokens();
            info!(
                "{}{}📂 {} ({} tokens)",
                prefix, connector, name, total_tokens
            );
            print_tree(child, &new_prefix, is_last_child)?;
        }
    }
    Ok(())
}

pub fn get_file_icon(path: &Path) -> Result<&'static str, ClipboardError> {
    debug!("Getting file icon for path: {:?}", path);
    match path.extension().and_then(|e| e.to_str()) {
        // Programming Languages
        Some("py") | Some("pyw") | Some("pyc") | Some("pyd") | Some("pyo") | Some("pyi") => {
            Ok("🐍")
        }
        Some("js") => Ok("🟨"),
        Some("ts") | Some("tsx") => Ok("🔷"),
        Some("html") | Some("htm") => Ok("🌐"),
        Some("css") | Some("scss") | Some("sass") => Ok("🎨"),
        Some("java") | Some("class") | Some("jar") => Ok("☕"),
        Some("c") | Some("h") => Ok("🇨"),
        Some("cpp") | Some("hpp") | Some("cc") | Some("cxx") => Ok("🇨➕"),
        Some("cs") => Ok("🔷"),
        Some("go") => Ok("🐹"),
        Some("rb") | Some("erb") => Ok("💎"),
        Some("php") => Ok("🐘"),
        Some("swift") => Ok("🕊️"),
        Some("kt") | Some("kts") => Ok("🇰"),
        Some("rs") | Some("rlib") => Ok("🦀"),

        // Data and Config Files
        Some("json") => Ok("🔖"),
        Some("yaml") | Some("yml") => Ok("🗂️"),
        Some("xml") => Ok("📰"),
        Some("csv") => Ok("📊"),
        Some("ini") | Some("conf") | Some("toml") => Ok("⚙️"),
        Some("lock") => Ok("🔒"),

        // Documentation and Text
        Some("md") | Some("markdown") => Ok("📝"),
        Some("txt") => Ok("📄"),
        Some("pdf") => Ok("📕"),
        Some("doc") | Some("docx") => Ok("📘"),

        // Images
        Some("jpg") | Some("jpeg") | Some("png") | Some("gif") | Some("svg") | Some("bmp") => {
            Ok("🖼️")
        }

        // Archives
        Some("zip") | Some("tar") | Some("gz") | Some("7z") | Some("rar") => Ok("🗜️"),

        // Executables and Scripts
        Some("exe") | Some("dll") | Some("so") | Some("dylib") => Ok("⚙️"),
        Some("sh") | Some("bash") | Some("zsh") | Some("fish") => Ok("🐚"),
        Some("bat") | Some("cmd") | Some("ps1") => Ok("🖥️"),

        // Version Control
        Some("gitignore") | Some("gitattributes") => Ok("🔒"),

        // Build and Package Management
        Some("dockerfile") => Ok("🐳"),
        Some("makefile") => Ok("🏗️"),
        Some("cmake") => Ok("🏗️"),

        // Project Files
        Some("sln") | Some("csproj") | Some("fsproj") | Some("vbproj") => Ok("🏗️"),
        Some("proj") => Ok("🏗️"),

        // .NET-specific
        Some("pdb") => Ok("🔷"),
        Some("resx") => Ok("🌐"),

        // Python-specific
        Some("ipynb") => Ok("📓"),

        // React-specific
        Some("jsx") => Ok("⚛️"),

        // TypeScript-specific
        Some("d") => Ok("📘"),

        // Fallback for directories and unknown types
        None => Ok("📁"),
        _ => Ok("📄"),
    }
}