plane_core/
cli.rs

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
use anyhow::Result;
use clap::Parser;
use serde::{de::DeserializeOwned, Serialize};

#[derive(Parser)]
struct CliArgs {
    #[clap(short, long)]
    dump_config: bool,

    config_file: Option<String>,
}

pub fn init_cli<C: Serialize + DeserializeOwned>() -> Result<C> {
    let cli_args = CliArgs::parse();

    let mut config_builder = config::Config::builder();
    if let Some(config_file) = cli_args.config_file {
        config_builder =
            config_builder.add_source(config::File::new(&config_file, config::FileFormat::Toml));
    }
    config_builder = config_builder.add_source(
        config::Environment::with_prefix("PLANE")
            .separator("__")
            .prefix_separator("_")
            .try_parsing(true)
            .list_separator(",")
            .with_list_parse_key("nats.hosts"),
    );
    let config: C = config_builder.build()?.try_deserialize()?;

    if cli_args.dump_config {
        println!("{}", serde_json::to_string_pretty(&config)?);
        // return Ok(());
        todo!("exit gracefully");
    }

    Ok(config)
}