Nelze vybrat více než 25 témat
Téma musí začínat písmenem nebo číslem, může obsahovat pomlčky („-“) a může být dlouhé až 35 znaků.
63 řádky
1.9 KiB
Rust
63 řádky
1.9 KiB
Rust
use dotenv;
|
|
|
|
use clap::App;
|
|
use plume_models::{db_conn::init_pool, instance::Instance};
|
|
use std::io::{self, prelude::*};
|
|
|
|
mod instance;
|
|
mod migration;
|
|
mod search;
|
|
mod users;
|
|
|
|
fn main() {
|
|
let mut app = App::new("Plume CLI")
|
|
.bin_name("plm")
|
|
.version(env!("CARGO_PKG_VERSION"))
|
|
.about("Collection of tools to manage your Plume instance.")
|
|
.subcommand(instance::command())
|
|
.subcommand(migration::command())
|
|
.subcommand(search::command())
|
|
.subcommand(users::command());
|
|
let matches = app.clone().get_matches();
|
|
|
|
match dotenv::dotenv() {
|
|
Ok(path) => println!("Configuration read from {}", path.display()),
|
|
Err(ref e) if e.not_found() => eprintln!("no .env was found"),
|
|
e => e.map(|_| ()).unwrap(),
|
|
}
|
|
let db_pool = init_pool()
|
|
.expect("Couldn't create a database pool, please check DATABASE_URL in your .env");
|
|
let _ = db_pool
|
|
.get()
|
|
.as_ref()
|
|
.map(|conn| Instance::cache_local(conn));
|
|
|
|
match matches.subcommand() {
|
|
("instance", Some(args)) => instance::run(
|
|
args,
|
|
&db_pool.get().expect("Couldn't connect to the database."),
|
|
),
|
|
("migration", Some(args)) => migration::run(
|
|
args,
|
|
&db_pool.get().expect("Couldn't connect to the database."),
|
|
),
|
|
("search", Some(args)) => search::run(args, db_pool),
|
|
("users", Some(args)) => users::run(
|
|
args,
|
|
&db_pool.get().expect("Couldn't connect to the database."),
|
|
),
|
|
_ => app.print_help().expect("Couldn't print help"),
|
|
};
|
|
}
|
|
|
|
pub fn ask_for(something: &str) -> String {
|
|
print!("{}: ", something);
|
|
io::stdout().flush().expect("Couldn't flush STDOUT");
|
|
let mut input = String::new();
|
|
io::stdin()
|
|
.read_line(&mut input)
|
|
.expect("Unable to read line");
|
|
input.retain(|c| c != '\n');
|
|
input
|
|
}
|