change other value parsers

This commit is contained in:
Isse 2023-05-05 15:05:57 +02:00
parent a74ec7a1ad
commit eed613a50f
2 changed files with 25 additions and 3 deletions

View File

@ -12,6 +12,29 @@ pub enum AdminRole {
Admin = 1, Admin = 1,
} }
impl core::str::FromStr for AdminRole {
type Err = &'static str;
fn from_str(s: &str) -> Result<Self, Self::Err> {
match s {
"mod" | "moderator" => Ok(AdminRole::Moderator),
"admin" => Ok(AdminRole::Admin),
_ => Err("Could not parse AdminRole"),
}
}
}
impl ToString for AdminRole {
fn to_string(&self) -> String {
match self {
AdminRole::Moderator => "moderator",
AdminRole::Admin => "admin",
}
.into()
}
}
#[derive(Copy, Clone, Debug, PartialEq, Eq, Serialize, Deserialize)] #[derive(Copy, Clone, Debug, PartialEq, Eq, Serialize, Deserialize)]
pub struct Admin(pub AdminRole); pub struct Admin(pub AdminRole);

View File

@ -2,7 +2,6 @@ use clap::Parser;
use common::comp; use common::comp;
use server::persistence::SqlLogMode; use server::persistence::SqlLogMode;
use std::sync::mpsc::Sender; use std::sync::mpsc::Sender;
use strum::VariantNames;
use tracing::error; use tracing::error;
#[derive(Clone, Debug, Parser)] #[derive(Clone, Debug, Parser)]
@ -12,7 +11,7 @@ pub enum Admin {
/// Name of the admin to whom to assign a role /// Name of the admin to whom to assign a role
username: String, username: String,
/// role to assign to the admin /// role to assign to the admin
#[arg(ignore_case = true, value_parser = clap::builder::PossibleValuesParser::new(comp::AdminRole::VARIANTS))] #[arg(ignore_case = true, value_parser = clap::value_parser!(comp::AdminRole))]
role: comp::AdminRole, role: comp::AdminRole,
}, },
Remove { Remove {
@ -63,7 +62,7 @@ pub enum Message {
}, },
/// Enable or disable sql logging /// Enable or disable sql logging
SqlLogMode { SqlLogMode {
#[arg(default_value_t, value_parser = SqlLogMode::variants())] #[arg(default_value_t, value_parser = clap::value_parser!(SqlLogMode))]
mode: SqlLogMode, mode: SqlLogMode,
}, },
/// Disconnects all connected clients /// Disconnects all connected clients