1
0
Fork 0
mirror of https://codeberg.org/icewind/haze.git synced 2026-06-03 17:14:08 +02:00

better service parsing

This commit is contained in:
Robin Appelman 2021-06-25 16:43:23 +02:00
commit 3c97fc37e1
2 changed files with 26 additions and 8 deletions

View file

@ -55,7 +55,7 @@ pub enum ExecService {
impl HazeArgs {
pub fn parse<I, S>(mut args: I) -> Result<HazeArgs>
where
S: AsRef<str> + Into<String> + Display + PartialEq<str>,
S: AsRef<str> + Into<String> + Display,
I: Iterator<Item = S>,
{
let _bin = args.next();
@ -107,7 +107,7 @@ impl HazeArgs {
let mut args = args.peekable();
let service = match args.peek() {
Some(arg) if arg == "db" => {
Some(arg) if arg.as_ref() == "db" => {
args.next();
Some(ExecService::Db)
}
@ -127,7 +127,7 @@ impl HazeArgs {
}),
HazeCommand::Db => Ok(HazeArgs::Db {
filter,
root: args.next().filter(|arg| arg == "root").is_some(),
root: args.next().filter(|arg| arg.as_ref() == "root").is_some(),
}),
HazeCommand::Clean => Ok(HazeArgs::Clean),
HazeCommand::Logs => {
@ -259,7 +259,7 @@ fn test_arg_parse() {
HazeArgs::parse(vec!["haze", "asdasd", "exec", "db", "foo", "bar"].into_iter()).unwrap(),
HazeArgs::Exec {
filter: Some("asdasd".to_string()),
service: Some("db".to_string()),
service: Some(ExecService::Db),
command: vec!["foo".to_string(), "bar".to_string()],
}
);

View file

@ -55,10 +55,6 @@ impl CloudOptions {
} else {
break;
}
if db.is_some() && php.is_some() {
break;
}
}
Ok(CloudOptions {
@ -71,6 +67,8 @@ impl CloudOptions {
#[test]
fn test_option_parse() {
use crate::service::{LDAPAdmin, LDAP};
let mut args = vec![].into_iter().peekable();
assert_eq!(
CloudOptions::parse::<_, &str>(&mut args).unwrap(),
@ -108,6 +106,26 @@ fn test_option_parse() {
..Default::default()
}
);
let mut args = vec!["7", "ldap", "pgsql"].into_iter().peekable();
assert_eq!(
CloudOptions::parse(&mut args).unwrap(),
CloudOptions {
php: PhpVersion::Php74,
db: Database::Postgres,
services: vec![Service::Ldap(LDAP), Service::LdapAdmin(LDAPAdmin)],
..Default::default()
}
);
let mut args = vec!["7", "pgsql", "ldap"].into_iter().peekable();
assert_eq!(
CloudOptions::parse(&mut args).unwrap(),
CloudOptions {
php: PhpVersion::Php74,
db: Database::Postgres,
services: vec![Service::Ldap(LDAP), Service::LdapAdmin(LDAPAdmin)],
..Default::default()
}
);
}
#[derive(Debug)]