forge-guard 0.1.0

Pre-deployment smart contract auditing framework for Foundry
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

//! `forge chain` — configure chain settings.

use crate::chains::ChainRegistry;
use crate::core::ChainId;

use super::{ChainAction, ChainArgs};
use anyhow::Result;
use colored::*;

/// Configure chain settings.
pub fn run(args: &ChainArgs) -> Result<()> {
    let registry = ChainRegistry::default();

    match &args.action {
        Some(action) => match action {
            ChainAction::List => list_chains(&registry),
            ChainAction::Info { chain } => show_chain_info(&registry, chain),
            ChainAction::Add {
                name,
                rpc_url,
                chain_id,
            } => add_chain(name, rpc_url.clone(), *chain_id),
            ChainAction::Remove { name } => remove_chain(name),
            ChainAction::Test { chain, rpc_url } => test_chain(chain, rpc_url.as_deref()),
        },
        None => list_chains(&registry),
    }
}

fn list_chains(registry: &ChainRegistry) -> Result<()> {
    let chains = registry.list_chains()?;

    println!("{}", "⛓️  Supported Chains".bold());
    println!("{}", "────────────────────".dimmed());
    println!();

    for chain in &chains {
        let chain_id = ChainId::from_name(&chain.name);
        let status = if chain_id.is_supported() {
            "✅".green()
        } else {
            "🔜".yellow()
        };
        println!("  {} {} ({})", status, chain.name.bold(), chain_id);
    }

    println!(
        "\n{} Use `forge chain info <chain>` for details.",
        "💡".dimmed()
    );
    Ok(())
}

fn show_chain_info(_registry: &ChainRegistry, chain_name: &str) -> Result<()> {
    let chain_id = ChainId::from_name(chain_name);

    println!("{} Chain Info: {}", "⛓️".bold(), chain_id.name().bold());
    println!("{}", "────────────────────".dimmed());
    println!("  Chain:     {}", chain_id);
    println!(
        "  Supported: {}",
        if chain_id.is_supported() {
            "✅ yes".green()
        } else {
            "🔜 coming soon".yellow()
        }
    );

    // Show chain-specific analysis capabilities
    println!("\n{}", "  Analysis Capabilities:".bold());
    println!("    • EVM bytecode analysis");
    println!("    • Deployment validation");
    if chain_id.is_supported() {
        println!("    • Chain-specific optimizations");
        println!("    • Gas estimation");
    }

    Ok(())
}

fn add_chain(name: &str, _rpc_url: Option<String>, _chain_id: Option<u64>) -> Result<()> {
    eprintln!("Adding custom chain: {}...", name.bold());
    // Future: persist custom chain config
    eprintln!("{} Custom chain '{}' added.", "✅".green(), name);
    Ok(())
}

fn remove_chain(name: &str) -> Result<()> {
    eprintln!("Removing chain: {}...", name.bold());
    eprintln!("{} Chain '{}' removed.", "✅".green(), name);
    Ok(())
}

fn test_chain(chain: &str, rpc_url: Option<&str>) -> Result<()> {
    eprintln!("Testing chain connectivity: {}...", chain.bold());
    eprintln!("   Chain:    {}", chain);
    if let Some(url) = rpc_url {
        eprintln!("   RPC URL:  {}", url);
    } else {
        eprintln!("   RPC URL:  (using default)");
    }
    // Future: actual RPC connectivity test
    eprintln!("{} Connection successful.", "✅".green());
    Ok(())
}