bazzar/api/src/main.rs

#![feature(drain_filter)]

use std::io::Write;

use actix::Actor;
use actix_session::storage::RedisActorSessionStore;
use actix_session::SessionMiddleware;
use actix_web::middleware::Logger;
use actix_web::web::Data;
use actix_web::{App, HttpServer};
use config::UpdateConfig;
use email_manager::TestMail;
use jemallocator::Jemalloc;
use model::{Email, Encrypt, Login, PassHash, Password, Role};
use opts::{
    Command, CreateAccountCmd, CreateAccountOpts, GenerateHashOpts, MigrateOpts, Opts, ServerOpts,
    TestMailerOpts,
};
use validator::{validate_email, validate_length};

mod opts;
pub mod routes;

#[global_allocator]
static GLOBAL: Jemalloc = Jemalloc;

#[derive(Debug, thiserror::Error)]
pub enum Error {
    #[error("Failed to boot. {0:?}")]
    Boot(std::io::Error),
    #[error("Unable to read password file. {0:?}")]
    PassFile(std::io::Error),
    #[error("Unable to read password from STDIN. {0:?}")]
    ReadPass(std::io::Error),
    #[error("{0}")]
    Database(#[from] database_manager::Error),
}

pub type Result<T> = std::result::Result<T, Error>;

async fn server(opts: ServerOpts) -> Result<()> {
    let redis_connection_string = "127.0.0.1:6379";

    let app_config = config::default_load(&opts);

    let db = database_manager::Database::build(app_config.clone())
        .await?
        .start();
    let token_manager = token_manager::TokenManager::new(app_config.clone(), db.clone()).start();
    let order_manager = order_manager::OrderManager::new(app_config.clone(), db.clone()).start();
    let payment_manager = payment_manager::PaymentManager::build(app_config.clone(), db.clone())
        .await
        .expect("Failed to start payment manager")
        .start();
    let search_manager = search_manager::SearchManager::new(app_config.clone());
    let fs_manager = fs_manager::FsManager::build(app_config.clone())
        .await
        .expect("Failed to initialize file system storage");
    let addr = {
        let l = app_config.lock();
        let w = l.web();
        (w.bind().unwrap_or(opts.bind), w.port().unwrap_or(opts.port))
    };

    println!("Listen at {:?}", addr);

    HttpServer::new(move || {
        let config = app_config.clone();
        App::new()
            .wrap(Logger::default())
            .wrap(SessionMiddleware::new(
                RedisActorSessionStore::new(redis_connection_string),
                {
                    let l = config.lock();
                    l.web().session_secret()
                },
            ))
            .wrap(actix_web::middleware::Compress::default())
            .wrap(actix_web::middleware::NormalizePath::trim())
            .app_data(Data::new(config))
            .app_data(Data::new(db.clone()))
            .app_data(Data::new(token_manager.clone()))
            .app_data(Data::new(order_manager.clone()))
            .app_data(Data::new(payment_manager.clone()))
            .app_data(Data::new(search_manager.clone()))
            .app_data(Data::new(fs_manager.clone()))
            .configure(routes::configure)
            .service({
                let l = app_config.lock();
                actix_files::Files::new(&l.files().public_path(), l.files().local_path())
            })
            .default_service(actix_web::web::to(actix_web::HttpResponse::Ok))
    })
    .bind(addr)
    .map_err(Error::Boot)?
    .run()
    .await
    .map_err(Error::Boot)
}

async fn migrate(opts: MigrateOpts) -> Result<()> {
    use sqlx::migrate::MigrateError;

    let config = config::default_load(&opts);
    let db = database_manager::Database::build(config).await?;
    let res: std::result::Result<(), MigrateError> =
        sqlx::migrate!("../migrations").run(db.pool()).await;
    match res {
        Ok(()) => Ok(()),
        Err(e) => {
            eprintln!("{e}");
            std::process::exit(1);
        }
    }
}

async fn generate_hash(_opts: GenerateHashOpts) -> Result<()> {
    model::print_hash();
    Ok(())
}

async fn create_account(opts: CreateAccountOpts) -> Result<()> {
    let (role, opts) = match opts.cmd.expect("Account type is mandatory") {
        CreateAccountCmd::Admin(opts) => (Role::Admin, opts),
        CreateAccountCmd::User(opts) => (Role::User, opts),
    };
    if !validate_email(&opts.email) {
        panic!("Invalid email address");
    }
    if !validate_length(&opts.login, Some(4), Some(100), None) {
        panic!("Login must have at least 4 characters and no more than 100");
    }
    let config = config::default_load(&opts);
    let db = database_manager::Database::build(config.clone())
        .await?
        .start();
    let pass = match opts.pass_file {
        Some(path) => std::fs::read_to_string(path).map_err(Error::PassFile)?,
        None => {
            let mut s = String::with_capacity(100);
            {
                let mut std_out = std::io::stdout();
                let std_in = std::io::stdin();

                std_out
                    .write_all(b"PASS > ")
                    .expect("Failed to write to stdout");
                std_out.flush().expect("Failed to write to stdout");

                std_in.read_line(&mut s).map_err(Error::ReadPass)?;
            }
            if let Some(pos) = s.chars().position(|c| c == '\n') {
                s.remove(pos);
            }
            s
        }
    };
    if pass.trim().is_empty() {
        panic!("Password cannot be empty!");
    }
    let hash = Password::from(pass)
        .encrypt(&config.lock().web().pass_salt())
        .unwrap();

    db.send(database_manager::CreateAccount {
        email: Email::from(opts.email),
        login: Login::from(opts.login),
        pass_hash: PassHash::from(hash),
        role,
    })
    .await
    .unwrap()
    .unwrap();
    Ok(())
}

async fn test_mailer(opts: TestMailerOpts) -> Result<()> {
    let config = config::default_load(&opts);
    opts.update_config(&mut *config.lock());

    let manager = email_manager::EmailManager::build(config)
        .expect("Invalid email manager config")
        .start();
    if manager
        .send(TestMail {
            receiver: opts.receiver.expect("e-mail address is required"),
        })
        .await
        .expect("Failed to execute actor")
        .expect("Failed to send email")
        .success
    {
        println!("Success!");
    } else {
        eprintln!("Failure!");
        std::process::exit(1);
    }
    Ok(())
}

#[actix_web::main]
async fn main() -> Result<()> {
    human_panic::setup_panic!();

    dotenv::dotenv().ok();
    pretty_env_logger::init();

    let opts: Opts = gumdrop::Options::parse_args_default_or_exit();
    match opts.cmd.unwrap_or_default() {
        Command::Migrate(opts) => migrate(opts).await,
        Command::Server(opts) => server(opts).await,
        Command::GenerateHash(opts) => generate_hash(opts).await,
        Command::CreateAccount(opts) => create_account(opts).await,
        Command::TestMailer(opts) => test_mailer(opts).await,
        Command::ConfigInfo(_) => {
            config::config_info().await.unwrap();
            Ok(())
        }
    }
}