Configuration & Env

use clap::Parser;
use figment::{providers::{Env, Format, Serialized, Toml}, Figment};
use serde::Deserialize;
 
#[derive(Parser)]
struct Cli {
    #[arg(short, long)]
    config: Option<std::path::PathBuf>,
    #[arg(long)]
    port: Option<u16>,
}
 
#[derive(Debug, Deserialize)]
struct Settings {
    port: u16,
    host: String,
}
 
fn load(cli: &Cli) -> anyhow::Result<Settings> {
    let mut figment = Figment::new()
        .merge(Serialized::defaults(Settings { port: 8080, host: "127.0.0.1".into() }))
        .merge(Toml::file(cli.config.clone().unwrap_or_else(|| "config.toml".into())))
        .merge(Env::prefixed("APP_").split("_"));
    if let Some(port) = cli.port {
        figment = figment.merge(Serialized::from(("port", port)));
    }
    Ok(figment.extract()?)
}

use clap::Parser;
use serde::Deserialize;
use std::path::PathBuf;
 
#[derive(Parser)]
#[command(name = "worker")]
struct Cli {
    /// Path to TOML config (default: ./worker.toml)
    #[arg(short, long, env = "WORKER_CONFIG")]
    config: Option<PathBuf>,
 
    /// Override log level
    #[arg(long, env = "RUST_LOG")]
    log_level: Option<String>,
 
    /// Override concurrency (wins over config file)
    #[arg(long)]
    jobs: Option<usize>,
}
 
#[derive(Debug, Deserialize)]
struct FileConfig {
    jobs: usize,
    queue_url: String,
}
 
fn main() -> anyhow::Result<()> {
    let cli = Cli::parse();
    let path = cli.config.unwrap_or_else(|| PathBuf::from("worker.toml"));
    let file: FileConfig = toml::from_str(&std::fs::read_to_string(&path)?)?;
    let jobs = cli.jobs.unwrap_or(file.jobs);
    eprintln!("running with {jobs} jobs against {}", file.queue_url);
    Ok(())
}

// Never read secrets from config files committed to git
let api_key = std::env::var("API_KEY")
    .context("set API_KEY in the environment")?;