komodo/bin/cli/src/state.rs

use std::sync::OnceLock;

use clap::Parser;
use komodo_client::KomodoClient;
use merge_config_files::parse_config_file;

pub fn cli_args() -> &'static crate::args::CliArgs {
  static CLI_ARGS: OnceLock<crate::args::CliArgs> = OnceLock::new();
  CLI_ARGS.get_or_init(crate::args::CliArgs::parse)
}

pub fn komodo_client() -> &'static KomodoClient {
  static KOMODO_CLIENT: OnceLock<KomodoClient> = OnceLock::new();
  KOMODO_CLIENT.get_or_init(|| {
    let args = cli_args();
    let crate::args::CredsFile { url, key, secret } =
      match (&args.url, &args.key, &args.secret) {
        (Some(url), Some(key), Some(secret)) => {
          crate::args::CredsFile {
            url: url.clone(),
            key: key.clone(),
            secret: secret.clone(),
          }
        }
        (url, key, secret) => {
          let mut creds: crate::args::CredsFile =
            parse_config_file(cli_args().creds.as_str())
              .expect("failed to parse Komodo credentials");

          if let Some(url) = url {
            creds.url.clone_from(url);
          }
          if let Some(key) = key {
            creds.key.clone_from(key);
          }
          if let Some(secret) = secret {
            creds.secret.clone_from(secret);
          }

          creds
        }
      };
    futures::executor::block_on(
      KomodoClient::new(url, key, secret).with_healthcheck(),
    )
    .expect("failed to initialize Komodo client")
  })
}