provekit-cli 1.0.0

ProveKit CLI for generating and verifying zero-knowledge proofs
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

#![allow(missing_docs)]
mod cmd;
#[cfg(feature = "profiling-allocator")]
mod profiling_alloc;
mod span_stats;

#[cfg(feature = "profiling-allocator")]
use crate::profiling_alloc::ProfilingAllocator;
#[cfg(feature = "tracy")]
use tracing::{info, warn};
use {
    self::cmd::Command,
    anyhow::Result,
    span_stats::SpanStats,
    tracing::subscriber,
    tracing_subscriber::{self, filter::LevelFilter, layer::SubscriberExt as _, Layer, Registry},
};

#[cfg(feature = "profiling-allocator")]
#[global_allocator]
static ALLOCATOR: ProfilingAllocator = ProfilingAllocator::new();

fn main() -> Result<()> {
    let args = argh::from_env::<cmd::Args>();
    // Debug builds: track ALL spans for detailed profiling.
    // Release builds: only INFO+ to reduce overhead.
    #[cfg(debug_assertions)]
    let level = LevelFilter::TRACE;
    #[cfg(not(debug_assertions))]
    let level = LevelFilter::INFO;
    let subscriber = Registry::default().with(SpanStats.with_filter(level));

    #[cfg(feature = "tracy")]
    let subscriber = {
        if args.tracy {
            if let Some(depth) = args.tracy_allocations {
                info!("Tracy profiling enabled with allocation tracking (depth {depth}).");
                ALLOCATOR.enable_tracy(depth);
            } else {
                info!("Tracy profiling enabled (without allocation tracking).");
            }
        } else {
            if args.tracy_allocations.is_some() {
                warn!("--tracy-allocations specified without --tracy, ignoring.");
            }
            if args.tracy_keepalive {
                warn!("--tracy-keepalive specified without --tracy, ignoring.");
            }
        }
        subscriber.with(args.tracy.then(tracing_tracy::TracyLayer::default))
    };

    subscriber::set_global_default(subscriber)?;

    // Run CLI command
    let res = args.run();

    #[cfg(feature = "tracy")]
    if args.tracy_keepalive {
        use std::io::{stderr, stdin, stdout, Write};
        eprintln!("Tracy keepalive enabled, press Enter to exit.");
        stdout().flush()?;
        stderr().flush()?;
        stdin().read_line(&mut String::new())?;
    }

    res
}