54 lines
1.3 KiB
Rust
54 lines
1.3 KiB
Rust
use std::process::exit;
|
|
|
|
use clap::{Parser, Subcommand};
|
|
use figment::Figment;
|
|
|
|
use crate::config::Config;
|
|
use crate::cli::{NomiloCommand};
|
|
use crate::models::{LocalUser, CreateUserRequest, Role};
|
|
use crate::get_db_conn;
|
|
|
|
#[derive(Subcommand)]
|
|
pub enum UserCommand {
|
|
/// Add new user
|
|
Add(AddUserCommand),
|
|
}
|
|
|
|
|
|
#[derive(Parser)]
|
|
pub struct AddUserCommand {
|
|
#[clap(long = "--email", short = 'e')]
|
|
pub email: String,
|
|
#[clap(long = "--is-admin", short = 'a')]
|
|
pub is_admin: bool,
|
|
#[clap(long = "--password", short = 'p')]
|
|
pub password: Option<String>,
|
|
}
|
|
|
|
|
|
impl NomiloCommand for UserCommand {
|
|
fn run(self, figment: Figment, app_config: Config) {
|
|
match self {
|
|
UserCommand::Add(sub) => sub.run(figment, app_config),
|
|
};
|
|
}
|
|
}
|
|
|
|
impl NomiloCommand for AddUserCommand {
|
|
fn run(self, figment: Figment, _app_config: Config) {
|
|
let res = LocalUser::create_user(&get_db_conn(&figment), CreateUserRequest {
|
|
email: self.email,
|
|
role: Some(if self.is_admin { Role::Admin } else { Role::ZoneAdmin }),
|
|
password: self.password.unwrap(),
|
|
});
|
|
|
|
match res {
|
|
Ok(_) => println!("Successfully added user"),
|
|
Err(err) => {
|
|
eprintln!("Error while adding user: {:?}", err);
|
|
exit(1);
|
|
}
|
|
};
|
|
}
|
|
}
|