Init

1 files changed, 65 insertions, 0 deletions

diff --git a/src/main.rs b/src/main.rs
new file mode 100644
index 0000000..8643d71
--- /dev/null
+++ b/src/main.rs
@@ -0,0 +1,65 @@
+mod config;
+mod errors;
+mod graphql;
+mod logger;
+mod routes;
+use std::{net::SocketAddr, time::Duration};
+
+use crate::config::CONFIG;
+use axum::{
+    http::{header, Request},
+    Router,
+};
+use tokio::net::TcpListener;
+use tower_http::{
+    classify::ServerErrorsFailureClass, sensitive_headers::SetSensitiveHeadersLayer,
+    trace::TraceLayer,
+};
+
+use tracing::Span;
+
+/// Create the app: setup everything and returns a `Router`
+async fn create_app() -> Router {
+    logger::setup();
+    // let _ = db::setup().await;
+
+    Router::new()
+        .nest("/graphql", graphql::create_route())
+        .fallback(crate::routes::page_404)
+        // Mark the `Authorization` request header as sensitive so it doesn't
+        // show in logs.
+        .layer(SetSensitiveHeadersLayer::new(std::iter::once(
+            header::AUTHORIZATION,
+        )))
+        // Use a layer for `TraceLayer`
+        .layer(
+            TraceLayer::new_for_http()
+                .on_request(|request: &Request<_>, _span: &Span| {
+                    tracing::info!("{} {}", request.method(), request.uri());
+                })
+                .on_failure(
+                    |error: ServerErrorsFailureClass, latency: Duration, _span: &Span| {
+                        tracing::error!("{} | {} s", error, latency.as_secs());
+                    },
+                ),
+        )
+}
+
+#[tokio::main(flavor = "current_thread")]
+async fn main() {
+    let app = create_app().await;
+
+    let host = &CONFIG.allowed_host;
+
+    let addr = match host.parse::<SocketAddr>() {
+        Ok(addr) => addr,
+        Err(e) => {
+            panic!("`{}` {}", host, e);
+        }
+    };
+    tracing::info!("Listening on {}", addr);
+
+    axum::serve(TcpListener::bind(&addr).await.unwrap(), app)
+        .await
+        .unwrap();
+}