use anyhow::Result;
use clap::Parser;
use crate::admin;
use crate::cli::{Cli, Command};
use crate::commands;
use crate::effects;
use super::doctor::handle_doctor;
use super::info::handle_info;
use super::scan::handle_scan;
async fn dispatch_command(command: Command) -> Result<()> {
match command {
Command::Scan {
timeout,
tapo_target,
} => handle_scan(timeout, tapo_target).await,
Command::Info { host } => handle_info(host).await,
Command::Doctor { host } => handle_doctor(host).await,
Command::On { host, outlet } => commands::handle_on(&host, outlet).await,
Command::Off { host, outlet } => commands::handle_off(&host, outlet).await,
Command::Toggle { host, outlet } => commands::handle_toggle(&host, outlet).await,
Command::Group {
action,
pattern,
dry_run,
concurrency,
} => commands::handle_group(&pattern, action, dry_run, concurrency as usize).await,
Command::Dim { host, level } => commands::handle_dim(&host, level).await,
Command::ColorTemp { host, kelvin } => commands::handle_color_temp(&host, kelvin).await,
Command::Color {
host,
hue,
saturation,
value,
} => commands::handle_color(&host, hue, saturation, value).await,
Command::Rainbow {
host,
step_delay_ms,
} => commands::handle_rainbow(&host, step_delay_ms).await,
Command::Energy {
host,
outlet,
command,
} => match command {
Some(crate::cli::EnergyCommand::Watch {
host,
outlet,
interval,
count,
format,
}) => commands::handle_energy_watch(&host, outlet, interval, count, format).await,
None => {
commands::handle_energy(
&host.ok_or_else(|| {
crate::error::error(
"invalid_arguments",
"energy requires DEVICE or watch DEVICE",
)
})?,
outlet,
)
.await
}
},
Command::EnergyDaily {
host,
month,
outlet,
} => commands::handle_energy_daily(&host, month, outlet).await,
Command::EnergyMonthly { host, year, outlet } => {
commands::handle_energy_monthly(&host, year, outlet).await
}
Command::Specs { host } => admin::handle_specs(&host).await,
Command::Presets { host } => admin::handle_presets(&host).await,
Command::Effects { host } => effects::handle_effects_command(&host).await,
Command::Effect { host, name } => effects::handle_effect_command(&host, &name).await,
Command::Schedules { host } => admin::handle_schedules(&host).await,
Command::Led { host, state } => {
let on = matches!(state, crate::cli::LedAction::On);
admin::handle_led(&host, on).await
}
Command::Clock { host } => admin::handle_clock(&host).await,
Command::Rename { host, name } => admin::handle_rename(&host, &name).await,
Command::Restart { host } => admin::handle_restart(&host).await,
Command::Outlets { host } => admin::handle_outlets(&host).await,
Command::OutletRename { host, outlet, name } => {
admin::handle_outlet_rename(&host, outlet, &name).await
}
Command::Alias {
name,
ip,
klap,
tapo,
} => admin::handle_alias(&name, &ip, klap, tapo),
Command::Unalias { name } => admin::handle_unalias(&name),
Command::Aliases => admin::handle_aliases(),
Command::Login { email, password } => admin::handle_login(&email, password),
Command::Completions { shell } => {
admin::handle_completions(shell);
Ok(())
}
}
}
pub async fn run() -> Result<()> {
let args: Vec<std::ffi::OsString> = std::env::args_os().collect();
let wants_json = args
.iter()
.take_while(|arg| *arg != "--")
.any(|arg| arg == "--json");
let cli = match Cli::try_parse_from(&args) {
Ok(cli) => cli,
Err(error) => {
if wants_json && error.use_stderr() {
let result = Err(crate::error::error("invalid_arguments", error.to_string()));
std::println!(
"{}",
crate::output::envelope("parse", &result, serde_json::Value::Null)
);
return result;
}
error.exit();
}
};
let streaming = matches!(
&cli.command,
Command::Energy {
command: Some(_),
..
}
) && !matches!(&cli.command, Command::Energy { command: Some(crate::cli::EnergyCommand::Watch { format: crate::cli::WatchFormat::Csv, .. }), .. } if cli.json);
let command = cli.command.name();
let (result, data) = crate::output::collect(cli.json, dispatch_command(cli.command)).await;
if cli.json && !streaming {
std::println!("{}", crate::output::envelope(command, &result, data));
}
result
}