forked from 2sys/phonograph
55 lines
1.2 KiB
Rust
55 lines
1.2 KiB
Rust
use clap::Parser as _;
|
|
use dotenvy::dotenv;
|
|
use tracing_subscriber::EnvFilter;
|
|
|
|
use crate::{
|
|
app_state::{App, AppState},
|
|
cli::{serve_command, worker_command, Cli, Commands},
|
|
settings::Settings,
|
|
};
|
|
|
|
mod app_error;
|
|
mod app_state;
|
|
mod auth;
|
|
mod base_pooler;
|
|
mod base_user_perms;
|
|
mod bases;
|
|
mod cli;
|
|
mod data_layer;
|
|
mod db_conns;
|
|
mod middleware;
|
|
mod pg_acls;
|
|
mod pg_attributes;
|
|
mod pg_classes;
|
|
mod pg_databases;
|
|
mod pg_roles;
|
|
mod router;
|
|
mod routes;
|
|
mod sessions;
|
|
mod settings;
|
|
mod users;
|
|
mod worker;
|
|
|
|
/// Run CLI
|
|
#[tokio::main]
|
|
async fn main() {
|
|
// Attempt to pre-load .env in case it contains a RUST_LOG variable
|
|
dotenv().ok();
|
|
tracing_subscriber::fmt()
|
|
.with_env_filter(EnvFilter::from_default_env())
|
|
.init();
|
|
|
|
let settings = Settings::load().unwrap();
|
|
|
|
let state: AppState = App::from_settings(settings.clone()).await.unwrap().into();
|
|
|
|
if settings.run_database_migrations == Some(1) {
|
|
sqlx::migrate!().run(&state.app_db).await.unwrap();
|
|
}
|
|
|
|
let cli = Cli::parse();
|
|
match &cli.command {
|
|
Commands::Serve => serve_command(state).await.unwrap(),
|
|
Commands::Worker(args) => worker_command(args, state).await.unwrap(),
|
|
}
|
|
}
|