61 lines
1.7 KiB
Rust
61 lines
1.7 KiB
Rust
use axum::{Router, routing::get};
|
|
use sqlx::{sqlite::SqlitePoolOptions, SqlitePool};
|
|
use std::net::SocketAddr;
|
|
use tracing_subscriber::EnvFilter;
|
|
|
|
#[derive(Clone)]
|
|
struct AppState {
|
|
pool: SqlitePool,
|
|
master_key: String,
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
tracing_subscriber::fmt()
|
|
.with_env_filter(
|
|
EnvFilter::from_default_env()
|
|
.add_directive("bootstrap_auth_server=debug".parse().unwrap()),
|
|
)
|
|
.init();
|
|
|
|
dotenvy::dotenv().ok();
|
|
|
|
let master_key = std::env::var("SERVER_MASTER_KEY")
|
|
.expect("SERVER_MASTER_KEY environment variable must be set");
|
|
|
|
let db_url = std::env::var("DATABASE_URL")
|
|
.unwrap_or_else(|_| "sqlite://data.db?mode=rwc".to_string());
|
|
|
|
let port = std::env::var("SERVER_PORT").unwrap_or_else(|_| "3000".to_string());
|
|
|
|
tracing::info!("Connecting to database at {}", db_url);
|
|
let pool = SqlitePoolOptions::new()
|
|
.max_connections(5)
|
|
.connect(&db_url)
|
|
.await
|
|
.expect("Failed to connect to SQLite database");
|
|
|
|
tracing::info!("Running database migrations...");
|
|
sqlx::migrate!("./migrations")
|
|
.run(&pool)
|
|
.await
|
|
.expect("Failed to run database migrations");
|
|
|
|
tracing::info!("Migrations successful.");
|
|
|
|
let state = AppState {
|
|
pool,
|
|
master_key,
|
|
};
|
|
|
|
let app = Router::new()
|
|
.route("/health", get(|| async { "OK" }))
|
|
.with_state(state);
|
|
|
|
let addr: SocketAddr = format!("0.0.0.0:{}", port).parse().unwrap();
|
|
tracing::info!("Listening on {}", addr);
|
|
|
|
let listener = tokio::net::TcpListener::bind(addr).await.unwrap();
|
|
axum::serve(listener, app).await.unwrap();
|
|
}
|