sarah 0.1.3

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

use std::fmt::Write as _;

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

impl Report {
    pub fn to_markdown(&self, top: usize) -> String {
        [
            self.markdown_summary(),
            self.markdown_languages(),
            self.markdown_files(top),
            self.markdown_functions(top),
            "\n".to_string(),
            self.warnings_block("## Refactor candidates"),
        ]
        .concat()
    }

    fn markdown_summary(&self) -> String {
        let t = &self.totals;
        format!(
            "# Repo size report\n\n**{} files**, {} lines ({} code, {} comments, {} blank), **{} tokens** (tokenizer: {})\n\n{}\n",
            t.files,
            t.lines,
            t.code,
            t.comments,
            t.blanks,
            t.tokens,
            self.tokenizer,
            self.skipped_line()
        )
    }

    fn markdown_languages(&self) -> String {
        let mut out = String::from(
            "## Languages\n\n| Language | Files | Lines | Code | Tokens |\n|---|---:|---:|---:|---:|\n",
        );
        for (name, l) in self.languages_by_tokens() {
            let _ = writeln!(
                out,
                "| {name} | {} | {} | {} | {} |",
                l.files, l.lines, l.code, l.tokens
            );
        }
        out
    }

    fn markdown_files(&self, top: usize) -> String {
        let mut out = format!(
            "\n## Files (top {top})\n\n| File | Lines | Code | Tokens | Fns |\n|---|---:|---:|---:|---:|\n"
        );
        for f in self.files.iter().take(top) {
            let _ = writeln!(
                out,
                "| `{}` | {} | {} | {} | {} |",
                f.path.display(),
                f.lines,
                f.code,
                f.tokens,
                RenderHelpers::fn_count(f)
            );
        }
        out
    }

    fn markdown_functions(&self, top: usize) -> String {
        if self.functions.is_empty() {
            return String::new();
        }
        let mut out = format!(
            "\n## Functions (top {top} by lines)\n\n| Function | Location | Lines | Tokens |\n|---|---|---:|---:|\n"
        );
        for f in self.functions.iter().take(top) {
            let _ = writeln!(
                out,
                "| `{}` | `{}:{}` | {} | {} |",
                f.name,
                f.path.display(),
                f.start_line,
                f.lines,
                f.tokens
            );
        }
        out
    }
}