aboutsummaryrefslogtreecommitdiff
path: root/src/main.rs
blob: 3b3f868cb63b3b9aabaa647e0f5776819f0b2386 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
mod auth;
mod error;
mod extract;
mod response;
mod routers;
mod state;

pub use auth::{JwtClaims, create_jwt, create_password, validate_jwt, validate_password};
pub use error::{ApiError, ApiResult};
pub use response::{ErrorResponse, FailResponse, SuccessResponse};
pub use state::AppState;

use sea_orm::Database;
use tokio::net::TcpListener;
use tower_http::trace::TraceLayer;
use tracing::info;
use tracing_subscriber::EnvFilter;

async fn app_state() -> AppState {
    let db_url = std::env::var("DATABASE_URL").expect("DATABASE_URL must be set");
    let secret = std::env::var("SECRET").expect("SECRET must be set");

    AppState {
        db: Database::connect(db_url).await.unwrap(),
        secret: secret,
    }
}

async fn listener() -> TcpListener {
    let addr = std::env::var("SERVER_BIND").expect("SERVER_BIND must be set");
    TcpListener::bind(addr).await.unwrap()
}

#[tokio::main]
async fn main() {
    tracing_subscriber::fmt()
        .with_env_filter(EnvFilter::new("info,sqlx=warn,tower_http=debug"))
        .init();

    let state = app_state().await;
    let router = routers::router()
        .layer(TraceLayer::new_for_http())
        .with_state(state);
    let listener = listener().await;

    info!(
        "Start listening server on http://{:?}",
        listener.local_addr().unwrap()
    );

    axum::serve(listener, router).await.unwrap();
}