Plume/plume-cli/src/main.rs

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
}