pmat 3.11.0

PMAT - Zero-config AI context generation and code quality toolkit (CLI, MCP, HTTP)
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

//! Example demonstrating GitHub repository size checking
//!
//! This example shows how to check the size of GitHub repositories
//! before cloning them, which is useful for CI/CD systems with
//! limited disk space.
//!
//! Requires the `git-lib` feature to be enabled.

#[cfg(not(feature = "git-lib"))]
fn main() {
    println!("This example requires the 'git-lib' feature to be enabled.");
    println!("Run with: cargo run --example check_github_repo --features git-lib");
}

#[cfg(feature = "git-lib")]
use anyhow::Result;
#[cfg(feature = "git-lib")]
use pmat::services::git_clone::{GitCloner, ParsedGitHubUrl};

#[cfg(feature = "git-lib")]
#[tokio::main]
async fn main() -> Result<()> {
    println!("📦 GitHub Repository Size Checker\n");

    // Get repository URL from command line or use defaults
    let args: Vec<String> = std::env::args().collect();

    let repos = if args.len() > 1 {
        // Parse command line arguments as owner/repo pairs
        args[1..]
            .iter()
            .filter_map(|arg| {
                let parts: Vec<&str> = arg.split('/').collect();
                if parts.len() == 2 {
                    Some((parts[0].to_string(), parts[1].to_string()))
                } else {
                    eprintln!("⚠️  Invalid format '{}', expected 'owner/repo'", arg);
                    None
                }
            })
            .collect()
    } else {
        // Default repositories to check
        vec![
            ("github".to_string(), "gitignore".to_string()),
            ("rust-lang".to_string(), "mdBook".to_string()),
            ("BurntSushi".to_string(), "ripgrep".to_string()),
        ]
    };

    if repos.is_empty() {
        println!("Usage: {} [owner/repo] [owner/repo] ...", args[0]);
        println!("Example: {} github/gitignore rust-lang/rust", args[0]);
        return Ok(());
    }

    let temp_dir = tempfile::tempdir()?;
    let git_clone = GitCloner::new(temp_dir.path().to_path_buf());
    let mut total_size = 0u64;

    println!("Checking {} repositories...\n", repos.len());

    for (owner, repo) in repos {
        let parsed_url = ParsedGitHubUrl {
            owner: owner.clone(),
            repo: repo.clone(),
        };

        print!("📊 {}/{}... ", owner, repo);
        std::io::Write::flush(&mut std::io::stdout())?;

        match git_clone.check_repo_size(&parsed_url).await {
            Ok(size_kb) => {
                total_size += size_kb;
                let size_mb = size_kb as f64 / 1024.0;

                let size_str = if size_mb < 1.0 {
                    format!("{} KB", size_kb)
                } else if size_mb < 1024.0 {
                    format!("{:.1} MB", size_mb)
                } else {
                    format!("{:.1} GB", size_mb / 1024.0)
                };

                let emoji = match size_kb {
                    s if s < 1024 => "🟢",   // < 1 MB
                    s if s < 10240 => "🟡",  // < 10 MB
                    s if s < 102400 => "🟠", // < 100 MB
                    _ => "🔴",               // > 100 MB
                };

                println!("{} {}", emoji, size_str);
            }
            Err(e) => {
                println!("❌ Error: {}", e);
                if e.to_string().contains("404") {
                    println!("   → Repository not found or private");
                } else if e.to_string().contains("403") {
                    println!(
                        "   → API rate limit exceeded (set GITHUB_TOKEN environment variable)"
                    );
                }
            }
        }
    }

    println!("\n📈 Summary:");
    println!("   Total size: {:.1} MB", total_size as f64 / 1024.0);

    // Provide recommendations based on total size
    if total_size > 1024 * 1024 {
        println!("   ⚠️  Warning: Total size exceeds 1 GB");
        println!("   💡 Consider cloning with --depth=1 for large repositories");
    } else if total_size > 100 * 1024 {
        println!("   💡 Tip: These repos are moderately sized, ensure adequate disk space");
    } else {
        println!("   ✅ All repositories are reasonably sized for cloning");
    }

    // Check if GitHub token is set
    if std::env::var("GITHUB_TOKEN").is_err() {
        println!("\n💡 Tip: Set GITHUB_TOKEN environment variable to increase API rate limits");
    }

    Ok(())
}