use std::ffi::OsString;
use clap::{Parser, Subcommand};
use netavark::commands::dhcp_proxy;
use netavark::commands::firewalld_reload;
use netavark::commands::setup;
use netavark::commands::teardown;
use netavark::commands::update;
use netavark::commands::version;
#[derive(Parser, Debug)]
#[clap(version = env!("CARGO_PKG_VERSION"))]
struct Opts {
#[clap(short, long)]
file: Option<OsString>,
#[clap(short, long)]
config: Option<OsString>,
#[clap(short, long)]
rootless: Option<bool>,
#[clap(short, long)]
aardvark_binary: Option<OsString>,
#[clap(long, long = "plugin-directory")]
plugin_directories: Option<Vec<OsString>>,
#[clap(subcommand)]
subcmd: SubCommand,
}
#[derive(Subcommand, Debug)]
enum SubCommand {
Setup(setup::Setup),
Update(update::Update),
Teardown(teardown::Teardown),
Version(version::Version),
DHCPProxy(dhcp_proxy::Opts),
#[command(name = "firewalld-reload")]
FirewallDReload,
}
fn main() {
env_logger::builder().format_timestamp(None).init();
let opts = Opts::parse();
let config = opts.config;
let rootless = opts.rootless.unwrap_or(false);
let aardvark_bin = opts
.aardvark_binary
.unwrap_or_else(|| OsString::from("/usr/libexec/podman/aardvark-dns"));
let result = match opts.subcmd {
SubCommand::Setup(setup) => setup.exec(
opts.file,
config,
aardvark_bin,
opts.plugin_directories,
rootless,
),
SubCommand::Teardown(teardown) => teardown.exec(
opts.file,
config,
aardvark_bin,
opts.plugin_directories,
rootless,
),
SubCommand::Update(mut update) => update.exec(config, aardvark_bin, rootless),
SubCommand::Version(version) => version.exec(),
SubCommand::DHCPProxy(proxy) => dhcp_proxy::serve(proxy),
SubCommand::FirewallDReload => firewalld_reload::listen(config),
};
match result {
Ok(_) => {}
Err(err) => {
err.print_json();
std::process::exit(err.get_exit_code());
}
}
}
#[cfg(test)]
mod test;