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

use std::{self, fs::read_to_string, path::PathBuf, process::Command};

use once_cell::sync::Lazy;
use tracing::Level;
use tracing_subscriber::prelude::*;

pub fn get_git_root() -> PathBuf {
    static DIR: Lazy<PathBuf> = Lazy::new(|| {
        let output = Command::new("git")
            .arg("rev-parse")
            .arg("--show-toplevel")
            .output()
            .expect("failed to get root git directory");

        assert!(output.status.success());
        String::from_utf8_lossy(&output.stdout).trim().to_string().into()
    });

    DIR.clone()
}

/// Used for loading golden txt files.
pub fn load_txt(path: &str) -> Vec<String> {
    let s = read_to_string(path).expect("failed to load txt file");
    s.lines()
        .map(|s| s.trim())
        .filter(|&s| !s.is_empty())
        .map(|s| s.to_string())
        .collect()
}

pub fn logger(level: Level) -> impl tracing::Subscriber {
    tracing_subscriber::FmtSubscriber::builder()
        .without_time()
        .with_target(false)
        .with_ansi(true)
        .with_max_level(level)
        .with_test_writer()
        .pretty()
        .finish()
}

pub fn init_logger(level: Level) -> tracing::subscriber::DefaultGuard {
    tracing::subscriber::set_default(logger(level))
}

pub fn init_tracing(name: String) -> tracing::subscriber::DefaultGuard {
    let logger = tracing_subscriber::FmtSubscriber::builder()
        .without_time()
        .with_target(false)
        .with_ansi(true)
        .with_max_level(Level::INFO)
        .with_test_writer()
        .pretty()
        .finish();

    if cfg!(debug_assertions) {
        tracing::subscriber::set_default(logger)
    } else {
        // Install a new OpenTelemetry trace pipeline
        let tracer = opentelemetry_jaeger::new_pipeline()
            .with_service_name(name)
            .install_simple()
            .expect("failed to create open telemetry pipeline");

        // Create a tracing subscriber with the configured tracer
        let telemetry = tracing_opentelemetry::layer().with_tracer(tracer);

        let collector = logger.with(telemetry);
        tracing::subscriber::set_default(collector)
    }
}