User creation CLI tool

This commit is contained in:
Bat 2018-10-05 13:09:04 +01:00
parent 0da5d2e927
commit d48a7796f1
3 changed files with 83 additions and 1 deletions

View File

@ -10,6 +10,7 @@ path = "src/main.rs"
[dependencies] [dependencies]
clap = "2.32" clap = "2.32"
dotenv = "0.13" dotenv = "0.13"
rpassword = "2.0"
[dependencies.diesel] [dependencies.diesel]
features = ["postgres", "r2d2", "chrono"] features = ["postgres", "r2d2", "chrono"]

View File

@ -2,6 +2,7 @@ extern crate clap;
extern crate diesel; extern crate diesel;
extern crate dotenv; extern crate dotenv;
extern crate plume_models; extern crate plume_models;
extern crate rpassword;
use clap::App; use clap::App;
use diesel::{Connection, PgConnection}; use diesel::{Connection, PgConnection};
@ -9,13 +10,15 @@ use std::io::{self, prelude::*};
use plume_models::DB_URL; use plume_models::DB_URL;
mod instance; mod instance;
mod users;
fn main() { fn main() {
let mut app = App::new("Plume CLI") let mut app = App::new("Plume CLI")
.bin_name("plm") .bin_name("plm")
.version("0.2.0") .version("0.2.0")
.about("Collection of tools to manage your Plume instance.") .about("Collection of tools to manage your Plume instance.")
.subcommand(instance::command()); .subcommand(instance::command())
.subcommand(users::command());
let matches = app.clone().get_matches(); let matches = app.clone().get_matches();
dotenv::dotenv().ok(); dotenv::dotenv().ok();
@ -23,6 +26,7 @@ fn main() {
match matches.subcommand() { match matches.subcommand() {
("instance", Some(args)) => instance::run(args, &conn.expect("Couldn't connect to the database.")), ("instance", Some(args)) => instance::run(args, &conn.expect("Couldn't connect to the database.")),
("users", Some(args)) => users::run(args, &conn.expect("Couldn't connect to the database.")),
_ => app.print_help().unwrap() _ => app.print_help().unwrap()
}; };
} }

77
plume-cli/src/users.rs Normal file
View File

@ -0,0 +1,77 @@
use clap::{Arg, ArgMatches, App, SubCommand};
use diesel::PgConnection;
use rpassword;
use std::io::{self, Write};
use plume_models::{
users::*,
};
pub fn command<'a, 'b>() -> App<'a, 'b> {
SubCommand::with_name("users")
.about("Manage users")
.subcommand(SubCommand::with_name("new")
.arg(Arg::with_name("name")
.short("n")
.long("name")
.alias("username")
.takes_value(true)
.help("The username of the new user")
).arg(Arg::with_name("display-name")
.short("N")
.long("display-name")
.takes_value(true)
.help("The display name of the new user")
).arg(Arg::with_name("biography")
.short("b")
.long("bio")
.alias("biography")
.takes_value(true)
.help("The biography of the new user")
).arg(Arg::with_name("email")
.short("e")
.long("email")
.takes_value(true)
.help("Email address of the new user")
).arg(Arg::with_name("password")
.short("p")
.long("password")
.takes_value(true)
.help("The password of the new user")
).arg(Arg::with_name("admin")
.short("a")
.long("admin")
.help("Makes the user an administrator of the instance")
).about("Create a new user on this instance"))
}
pub fn run<'a>(args: &ArgMatches<'a>, conn: &PgConnection) {
let conn = conn;
match args.subcommand() {
("new", Some(x)) => new(x, conn),
_ => println!("Unknwon subcommand"),
}
}
fn new<'a>(args: &ArgMatches<'a>, conn: &PgConnection) {
let username = args.value_of("name").map(String::from).unwrap_or_else(|| super::ask_for("Username"));
let display_name = args.value_of("display-name").map(String::from).unwrap_or_else(|| super::ask_for("Display name"));
let admin = args.is_present("admin");
let bio = args.value_of("biography").unwrap_or("").to_string();
let email = args.value_of("email").map(String::from).unwrap_or_else(|| super::ask_for("Email address"));
let password = args.value_of("password").map(String::from).unwrap_or_else(|| {
print!("Password: ");
io::stdout().flush().expect("Couldn't flush STDOUT");
rpassword::read_password().expect("Couldn't read your password.")
});
NewUser::new_local(
conn,
username,
display_name,
admin,
bio,
email,
User::hash_pass(password),
).update_boxes(conn);
}