sarah 0.1.9

Count tokens, lines and function sizes per file to help AI agents decide what to refactor
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

use super::View;
use super::common::RenderHelpers;
use crate::report::model::Report;

impl Report {
    pub fn to_table(&self, view: View) -> String {
        let View { top, dirs } = view;
        [
            self.table_summary(),
            self.table_languages(),
            self.table_files(top),
            self.table_functions(top),
            if dirs {
                self.table_dirs(top)
            } else {
                String::new()
            },
            self.warnings_block("Warnings"),
        ]
        .concat()
    }

    fn table_dirs(&self, top: usize) -> String {
        let mut table = RenderHelpers::new_table(&[
            "Folder",
            "Files",
            "Dirs",
            "Total files",
            "Lines",
            "Tokens",
        ]);
        for d in self.dirs_by_tokens().into_iter().take(top) {
            table.add_row(vec![
                d.path.display().to_string(),
                d.files.to_string(),
                d.dirs.to_string(),
                d.total_files.to_string(),
                d.lines.to_string(),
                d.tokens.to_string(),
            ]);
        }
        format!("Folders (top {top} by tokens):\n{table}\n\n")
    }

    fn table_summary(&self) -> String {
        let t = &self.totals;
        format!(
            "{} files, {} lines ({} code, {} comments, {} blank), {} tokens [{}]\n{}\n",
            t.files,
            t.lines,
            t.code,
            t.comments,
            t.blanks,
            t.tokens,
            self.tokenizer,
            self.skipped_line()
        )
    }

    fn table_languages(&self) -> String {
        let mut langs = RenderHelpers::new_table(&["Language", "Files", "Lines", "Code", "Tokens"]);
        for (name, l) in self.languages_by_tokens() {
            langs.add_row(vec![
                name.clone(),
                l.files.to_string(),
                l.lines.to_string(),
                l.code.to_string(),
                l.tokens.to_string(),
            ]);
        }
        format!("{langs}\n\n")
    }

    fn table_files(&self, top: usize) -> String {
        let mut files = RenderHelpers::new_table(&["File", "Lines", "Code", "Tokens", "Fns"]);
        for f in self.files.iter().take(top) {
            files.add_row(vec![
                f.path.display().to_string(),
                f.lines.to_string(),
                f.code.to_string(),
                f.tokens.to_string(),
                RenderHelpers::fn_count(f),
            ]);
        }
        format!("Files (top {top}):\n{files}\n\n")
    }

    fn table_functions(&self, top: usize) -> String {
        if self.functions.is_empty() {
            return String::new();
        }
        let mut fns = RenderHelpers::new_table(&["Function", "Location", "Lines", "Tokens"]);
        for f in self.functions.iter().take(top) {
            fns.add_row(vec![
                f.name.clone(),
                format!("{}:{}", f.path.display(), f.start_line),
                f.lines.to_string(),
                f.tokens.to_string(),
            ]);
        }
        format!("Functions (top {top} by lines):\n{fns}\n\n")
    }
}