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

use std::{fs::File, path::Path};

use kimchi::precomputed_srs::TestSRS;
use poly_commitment::{commitment::CommitmentCurve, ipa::SRS};
use time::macros::format_description;
use tracing::debug;
use tracing_subscriber::{
    fmt::{format::FmtSpan, time::UtcTime},
    EnvFilter,
};

pub fn get_srs_from_cache<G: CommitmentCurve>(cache: String) -> SRS<G> {
    debug!("Loading SRS from cache {}", cache);
    let file_path = Path::new(&cache);
    let file = File::open(file_path).expect("Error opening SRS cache file");
    let srs: SRS<G> = {
        // By convention, proof systems serializes a TestSRS with filename 'test_<CURVE_NAME>.srs'.
        // The benefit of using this is you don't waste time verifying the SRS.
        if file_path
            .file_name()
            .unwrap()
            .to_str()
            .unwrap()
            .starts_with("test_")
        {
            let test_srs: TestSRS<G> = rmp_serde::from_read(&file).unwrap();
            From::from(test_srs)
        } else {
            rmp_serde::from_read(&file).unwrap()
        }
    };
    debug!("SRS loaded successfully from cache");
    srs
}

pub fn init_console_subscriber() {
    let timer = UtcTime::new(format_description!(
        "[year]-[month]-[day]T[hour repr:24]:[minute]:[second].[subsecond digits:3]Z"
    ));
    tracing_subscriber::fmt()
        .with_env_filter(EnvFilter::from_default_env())
        .with_span_events(FmtSpan::CLOSE)
        .with_timer(timer)
        .with_target(true)
        .with_thread_ids(false)
        .with_line_number(false)
        .with_file(false)
        .with_level(true)
        .with_ansi(true)
        .with_writer(std::io::stdout)
        .init();
}

#[cfg(test)]
#[ctor::ctor]
fn init_test_logging() {
    init_console_subscriber();
}