komodo/bin/cli/src/main.rs

#[macro_use]
extern crate tracing;

use std::{io::Read, path::PathBuf, str::FromStr, sync::OnceLock};

use anyhow::Context;
use clap::{Parser, Subcommand};
use colored::Colorize;
use monitor_client::{api::read, MonitorClient};
use serde::{de::DeserializeOwned, Deserialize};

mod execution;
mod maps;
mod sync;

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

#[derive(Parser, Debug)]
#[command(version, about, long_about = None)]
struct CliArgs {
  /// Sync or Exec
  #[command(subcommand)]
  command: Command,
  /// The path to a creds file.
  #[arg(long, default_value_t = String::from("./creds.toml"))]
  creds: String,
  /// Log less (just resource names).
  #[arg(long, default_value_t = false)]
  quiet: bool,
}

#[derive(Debug, Clone, Subcommand)]
enum Command {
  /// Runs syncs on resource files
  Sync {
    /// The path of the resource folder / file
    /// Folder paths will recursively incorporate all the resources it finds under the folder
    #[arg(default_value_t = default_path())]
    path: String,
  },

  /// Runs execution files
  Exec {
    /// The path of the exec file
    path: PathBuf,
  },
}

fn default_path() -> String {
  let home = std::env::var("HOME")
    .expect("no HOME env var. cannot get default config path.");
  format!("{home}/.config/monitor/creds.toml")
}

#[derive(Debug, Deserialize)]
struct CredsFile {
  url: String,
  key: String,
  secret: String,
}

fn monitor_client() -> &'static MonitorClient {
  static MONITOR_CLIENT: OnceLock<MonitorClient> = OnceLock::new();
  MONITOR_CLIENT.get_or_init(|| {
    let CredsFile { url, key, secret } =
      parse_toml_file(&cli_args().creds)
        .expect("failed to parse monitor credentials");
    futures::executor::block_on(MonitorClient::new(url, key, secret))
      .expect("failed to initialize monitor client")
  })
}

#[tokio::main]
async fn main() -> anyhow::Result<()> {
  tracing_subscriber::fmt::init();

  let version =
    monitor_client().read(read::GetVersion {}).await?.version;
  info!("monitor version: {version}");

  match &cli_args().command {
    Command::Exec { path } => execution::run_execution(path).await?,
    Command::Sync { path } => {
      sync::run_sync(&PathBuf::from_str(path)?).await?
    }
  }

  Ok(())
}

fn parse_toml_file<T: DeserializeOwned>(
  path: impl AsRef<std::path::Path>,
) -> anyhow::Result<T> {
  let contents = std::fs::read_to_string(path)
    .context("failed to read file contents")?;
  toml::from_str(&contents).context("failed to parse toml contents")
}

fn wait_for_enter(press_enter_to: &str) -> anyhow::Result<()> {
  println!(
    "\nPress {} to {}\n",
    "ENTER".green(),
    press_enter_to.bold()
  );
  let buffer = &mut [0u8];
  std::io::stdin()
    .read_exact(buffer)
    .context("failed to read ENTER")?;
  Ok(())
}