mirror of https://github.com/breard-r/acmed.git
Rodolphe Bréard
2 months ago
Failed to extract signature
5 changed files with 462 additions and 6 deletions
-
157Cargo.lock
-
1Cargo.toml
-
259src/cli.rs
-
30src/log.rs
-
21src/main.rs
@ -0,0 +1,259 @@ |
|||
use crate::log::Level;
|
|||
use clap::{Args, Parser};
|
|||
use std::path::{Path, PathBuf};
|
|||
|
|||
#[derive(Parser, Debug)]
|
|||
#[command(version, about, long_about = None)]
|
|||
pub struct CliArgs {
|
|||
/// Path to the main configuration file
|
|||
#[arg(short, long, value_name = "FILE", default_value = crate::DEFAULT_CONFIG_PATH)]
|
|||
pub config: PathBuf,
|
|||
|
|||
/// Specify the log level
|
|||
#[arg(long, value_name = "LEVEL", value_enum, default_value_t = crate::DEFAULT_LOG_LEVEL)]
|
|||
pub log_level: Level,
|
|||
|
|||
#[command(flatten)]
|
|||
pub log: Log,
|
|||
|
|||
/// Runs in the foreground
|
|||
#[arg(short, long, default_value_t = false)]
|
|||
pub foreground: bool,
|
|||
|
|||
#[command(flatten)]
|
|||
pub pid: Pid,
|
|||
|
|||
/// Add a root certificate to the trust store (can be set multiple times)
|
|||
#[arg(long, value_name = "FILE")]
|
|||
pub root_cert: Vec<PathBuf>,
|
|||
}
|
|||
|
|||
#[derive(Args, Debug)]
|
|||
#[group(multiple = false)]
|
|||
pub struct Log {
|
|||
/// Sends log messages via syslog
|
|||
#[arg(long)]
|
|||
pub log_syslog: bool,
|
|||
|
|||
/// Prints log messages to the standard error output
|
|||
#[arg(long)]
|
|||
pub log_stderr: bool,
|
|||
}
|
|||
|
|||
#[derive(Args, Debug)]
|
|||
#[group(multiple = false)]
|
|||
pub struct Pid {
|
|||
/// Path to the PID file
|
|||
#[arg(long, value_name = "FILE", default_value = crate::DEFAULT_PID_FILE)]
|
|||
pid_file: PathBuf,
|
|||
|
|||
/// Do not create any PID file
|
|||
#[arg(long)]
|
|||
no_pid_file: bool,
|
|||
}
|
|||
|
|||
impl Pid {
|
|||
pub fn get_pid_file(&self) -> Option<&Path> {
|
|||
if !self.no_pid_file {
|
|||
Some(self.pid_file.as_path())
|
|||
} else {
|
|||
None
|
|||
}
|
|||
}
|
|||
}
|
|||
|
|||
#[cfg(test)]
|
|||
mod tests {
|
|||
use super::*;
|
|||
use clap::CommandFactory;
|
|||
|
|||
#[test]
|
|||
fn verify_cli() {
|
|||
CliArgs::command().debug_assert();
|
|||
}
|
|||
|
|||
#[test]
|
|||
fn no_args() {
|
|||
let args: &[&str] = &[];
|
|||
let pa = CliArgs::try_parse_from(args).unwrap();
|
|||
assert_eq!(pa.config, PathBuf::from(crate::DEFAULT_CONFIG_PATH));
|
|||
assert_eq!(pa.log_level, Level::Warn);
|
|||
assert_eq!(pa.log.log_syslog, false);
|
|||
assert_eq!(pa.log.log_stderr, false);
|
|||
assert_eq!(pa.foreground, false);
|
|||
assert_eq!(pa.pid.pid_file, PathBuf::from(crate::DEFAULT_PID_FILE));
|
|||
assert_eq!(pa.pid.no_pid_file, false);
|
|||
assert_eq!(
|
|||
pa.pid.get_pid_file(),
|
|||
Some(PathBuf::from(crate::DEFAULT_PID_FILE).as_path())
|
|||
);
|
|||
assert!(pa.root_cert.is_empty());
|
|||
}
|
|||
|
|||
#[test]
|
|||
fn all_args_long_1() {
|
|||
let argv: &[&str] = &[
|
|||
"acmed",
|
|||
"--config",
|
|||
"/tmp/test.toml",
|
|||
"--log-level",
|
|||
"debug",
|
|||
"--log-syslog",
|
|||
"--foreground",
|
|||
"--pid-file",
|
|||
"/tmp/debug/acmed.pid",
|
|||
"--root-cert",
|
|||
"/tmp/certs/root_01.pem",
|
|||
"--root-cert",
|
|||
"/tmp/certs/root_02.pem",
|
|||
"--root-cert",
|
|||
"/tmp/certs/root_03.pem",
|
|||
];
|
|||
let pa = CliArgs::try_parse_from(argv).unwrap();
|
|||
assert_eq!(pa.config, PathBuf::from("/tmp/test.toml"));
|
|||
assert_eq!(pa.log_level, Level::Debug);
|
|||
assert_eq!(pa.log.log_syslog, true);
|
|||
assert_eq!(pa.log.log_stderr, false);
|
|||
assert_eq!(pa.foreground, true);
|
|||
assert_eq!(
|
|||
pa.pid.get_pid_file(),
|
|||
Some(PathBuf::from("/tmp/debug/acmed.pid").as_path())
|
|||
);
|
|||
assert_eq!(
|
|||
pa.root_cert,
|
|||
vec![
|
|||
PathBuf::from("/tmp/certs/root_01.pem"),
|
|||
PathBuf::from("/tmp/certs/root_02.pem"),
|
|||
PathBuf::from("/tmp/certs/root_03.pem")
|
|||
]
|
|||
);
|
|||
}
|
|||
|
|||
#[test]
|
|||
fn all_args_long_2() {
|
|||
let argv: &[&str] = &[
|
|||
"acmed",
|
|||
"--config",
|
|||
"/tmp/test.toml",
|
|||
"--log-level",
|
|||
"debug",
|
|||
"--log-stderr",
|
|||
"--foreground",
|
|||
"--no-pid-file",
|
|||
"--root-cert",
|
|||
"/tmp/certs/root_01.pem",
|
|||
"--root-cert",
|
|||
"/tmp/certs/root_02.pem",
|
|||
"--root-cert",
|
|||
"/tmp/certs/root_03.pem",
|
|||
];
|
|||
let pa = CliArgs::try_parse_from(argv).unwrap();
|
|||
assert_eq!(pa.config, PathBuf::from("/tmp/test.toml"));
|
|||
assert_eq!(pa.log_level, Level::Debug);
|
|||
assert_eq!(pa.log.log_syslog, false);
|
|||
assert_eq!(pa.log.log_stderr, true);
|
|||
assert_eq!(pa.foreground, true);
|
|||
assert_eq!(pa.pid.get_pid_file(), None);
|
|||
assert_eq!(
|
|||
pa.root_cert,
|
|||
vec![
|
|||
PathBuf::from("/tmp/certs/root_01.pem"),
|
|||
PathBuf::from("/tmp/certs/root_02.pem"),
|
|||
PathBuf::from("/tmp/certs/root_03.pem")
|
|||
]
|
|||
);
|
|||
}
|
|||
|
|||
#[test]
|
|||
fn all_args_short_1() {
|
|||
let argv: &[&str] = &[
|
|||
"acmed",
|
|||
"-c",
|
|||
"/tmp/test.toml",
|
|||
"--log-level",
|
|||
"debug",
|
|||
"--log-syslog",
|
|||
"-f",
|
|||
"--pid-file",
|
|||
"/tmp/debug/acmed.pid",
|
|||
"--root-cert",
|
|||
"/tmp/certs/root_01.pem",
|
|||
"--root-cert",
|
|||
"/tmp/certs/root_02.pem",
|
|||
"--root-cert",
|
|||
"/tmp/certs/root_03.pem",
|
|||
];
|
|||
let pa = CliArgs::try_parse_from(argv).unwrap();
|
|||
assert_eq!(pa.config, PathBuf::from("/tmp/test.toml"));
|
|||
assert_eq!(pa.log_level, Level::Debug);
|
|||
assert_eq!(pa.log.log_syslog, true);
|
|||
assert_eq!(pa.log.log_stderr, false);
|
|||
assert_eq!(pa.foreground, true);
|
|||
assert_eq!(
|
|||
pa.pid.get_pid_file(),
|
|||
Some(PathBuf::from("/tmp/debug/acmed.pid").as_path())
|
|||
);
|
|||
assert_eq!(
|
|||
pa.root_cert,
|
|||
vec![
|
|||
PathBuf::from("/tmp/certs/root_01.pem"),
|
|||
PathBuf::from("/tmp/certs/root_02.pem"),
|
|||
PathBuf::from("/tmp/certs/root_03.pem")
|
|||
]
|
|||
);
|
|||
}
|
|||
|
|||
#[test]
|
|||
fn all_args_short_2() {
|
|||
let argv: &[&str] = &[
|
|||
"acmed",
|
|||
"-c",
|
|||
"/tmp/test.toml",
|
|||
"--log-level",
|
|||
"debug",
|
|||
"--log-stderr",
|
|||
"-f",
|
|||
"--no-pid-file",
|
|||
"--root-cert",
|
|||
"/tmp/certs/root_01.pem",
|
|||
"--root-cert",
|
|||
"/tmp/certs/root_02.pem",
|
|||
"--root-cert",
|
|||
"/tmp/certs/root_03.pem",
|
|||
];
|
|||
let pa = CliArgs::try_parse_from(argv).unwrap();
|
|||
assert_eq!(pa.config, PathBuf::from("/tmp/test.toml"));
|
|||
assert_eq!(pa.log_level, Level::Debug);
|
|||
assert_eq!(pa.log.log_syslog, false);
|
|||
assert_eq!(pa.log.log_stderr, true);
|
|||
assert_eq!(pa.foreground, true);
|
|||
assert_eq!(pa.pid.get_pid_file(), None);
|
|||
assert_eq!(
|
|||
pa.root_cert,
|
|||
vec![
|
|||
PathBuf::from("/tmp/certs/root_01.pem"),
|
|||
PathBuf::from("/tmp/certs/root_02.pem"),
|
|||
PathBuf::from("/tmp/certs/root_03.pem")
|
|||
]
|
|||
);
|
|||
}
|
|||
|
|||
#[test]
|
|||
fn err_log_output() {
|
|||
let argv: &[&str] = &["acmed", "--log-stderr", "--log-syslog"];
|
|||
let pa = CliArgs::try_parse_from(argv);
|
|||
assert!(pa.is_err());
|
|||
}
|
|||
|
|||
#[test]
|
|||
fn err_pid_file() {
|
|||
let argv: &[&str] = &[
|
|||
"acmed",
|
|||
"--pid-file",
|
|||
"/tmp/debug/acmed.pid",
|
|||
"--no-pid-file",
|
|||
];
|
|||
let pa = CliArgs::try_parse_from(argv);
|
|||
assert!(pa.is_err());
|
|||
}
|
|||
}
|
|||
@ -0,0 +1,30 @@ |
|||
use clap::ValueEnum;
|
|||
use tracing_subscriber::FmtSubscriber;
|
|||
|
|||
#[derive(Copy, Clone, Debug, PartialEq, Eq, PartialOrd, Ord, ValueEnum)]
|
|||
pub enum Level {
|
|||
Error,
|
|||
Warn,
|
|||
Info,
|
|||
Debug,
|
|||
Trace,
|
|||
}
|
|||
|
|||
impl Level {
|
|||
fn tracing(&self) -> tracing::Level {
|
|||
match self {
|
|||
Self::Error => tracing::Level::ERROR,
|
|||
Self::Warn => tracing::Level::WARN,
|
|||
Self::Info => tracing::Level::INFO,
|
|||
Self::Debug => tracing::Level::DEBUG,
|
|||
Self::Trace => tracing::Level::TRACE,
|
|||
}
|
|||
}
|
|||
}
|
|||
|
|||
pub fn init(level: Level, is_syslog: bool) {
|
|||
let subscriber = FmtSubscriber::builder()
|
|||
.with_max_level(level.tracing())
|
|||
.finish();
|
|||
tracing::subscriber::set_global_default(subscriber).expect("setting default subscriber failed");
|
|||
}
|
|||
Write
Preview
Loading…
Cancel
Save
Reference in new issue