use clap::Parser;
use std::path::PathBuf;
use crate::pipeline::MIN_TRANSCODE_DURATION_SECS;
#[derive(Parser)]
#[command(name = "hvac", version, about = "GPU-accelerated media transcoder")]
pub struct Cli {
#[arg(required_unless_present_any = ["dump_config", "setup_launchdarkly"])]
pub path: Option<PathBuf>,
#[arg(short, long)]
pub config: Option<PathBuf>,
#[arg(long)]
pub dump_config: bool,
#[arg(short, long)]
pub quiet: bool,
#[arg(long, default_value_t = false)]
pub no_overwrite: bool,
#[arg(long, default_value_t = false)]
pub dry_run: bool,
#[arg(short, long)]
pub jobs: Option<usize>,
#[arg(short, long)]
pub output_dir: Option<PathBuf>,
#[arg(long, default_value_t = false)]
pub replace: bool,
#[arg(long, default_value_t = MIN_TRANSCODE_DURATION_SECS)]
pub min_duration: f64,
#[arg(long, default_value_t = 30)]
pub probe_timeout: u64,
#[arg(long, default_value_t = false)]
pub skip_ambiguous_audio: bool,
#[arg(long, value_name = "KEY")]
pub launchdarkly_sdk_key: Option<String>,
#[arg(long, default_value_t = false)]
pub setup_launchdarkly: bool,
#[arg(long, value_name = "KEY")]
pub ld_api_key: Option<String>,
}
impl Cli {
pub fn overwrite(&self) -> bool {
!self.no_overwrite
}
}
#[cfg(test)]
mod tests {
use super::*;
use clap::CommandFactory;
#[test]
fn cli_command_factory_parses() {
let cmd = Cli::command();
let _ = cmd.get_name();
}
#[test]
fn overwrite_default_is_true() {
let cli = Cli::try_parse_from(["hvac", "/some/path"]).unwrap();
assert!(cli.overwrite());
assert!(!cli.no_overwrite);
}
#[test]
fn overwrite_false_when_no_overwrite_set() {
let cli = Cli::try_parse_from(["hvac", "--no-overwrite", "/some/path"]).unwrap();
assert!(!cli.overwrite());
}
#[test]
fn min_duration_defaults_to_constant() {
let cli = Cli::try_parse_from(["hvac", "/some/path"]).unwrap();
assert_eq!(cli.min_duration, MIN_TRANSCODE_DURATION_SECS);
}
#[test]
fn probe_timeout_default_is_30() {
let cli = Cli::try_parse_from(["hvac", "/some/path"]).unwrap();
assert_eq!(cli.probe_timeout, 30);
}
#[test]
fn skip_ambiguous_audio_default_is_false() {
let cli = Cli::try_parse_from(["hvac", "/some/path"]).unwrap();
assert!(!cli.skip_ambiguous_audio);
}
#[test]
fn skip_ambiguous_audio_flag_sets_true() {
let cli = Cli::try_parse_from(["hvac", "--skip-ambiguous-audio", "/some/path"]).unwrap();
assert!(cli.skip_ambiguous_audio);
}
#[test]
fn launchdarkly_sdk_key_takes_no_env_var() {
unsafe {
std::env::set_var("LAUNCHDARKLY_SDK_KEY", "should-not-be-read");
}
let cli = Cli::try_parse_from(["hvac", "/some/path"]).unwrap();
unsafe {
std::env::remove_var("LAUNCHDARKLY_SDK_KEY");
}
assert!(cli.launchdarkly_sdk_key.is_none());
}
#[test]
fn setup_launchdarkly_does_not_require_path() {
let cli =
Cli::try_parse_from(["hvac", "--setup-launchdarkly", "--ld-api-key", "k"]).unwrap();
assert!(cli.path.is_none());
assert!(cli.setup_launchdarkly);
}
#[test]
fn dump_config_does_not_require_path() {
let cli = Cli::try_parse_from(["hvac", "--dump-config"]).unwrap();
assert!(cli.path.is_none());
assert!(cli.dump_config);
}
}