initial commit

backend/src/main.rs

@@ -0,0 +1,95 @@
+use axum::{
+    extract::State,
+    http::StatusCode,
+    response::Json,
+    routing::post,
+    Router,
+};
+use serde::{Deserialize, Serialize};
+use sqlx::{postgres::PgPoolOptions, PgPool, Row};
+use tower_http::cors::{Any, CorsLayer};
+use uuid::Uuid;
+
+#[derive(Clone)]
+struct AppState {
+    db: PgPool,
+}
+
+#[derive(Deserialize)]
+struct SubmitRequest {
+    name: String,
+    alcohol: Option<String>,
+    food: Option<String>,
+    with_partner: bool,
+    partner_name: Option<String>,
+    partner_alcohol: Option<String>,
+    partner_food: Option<String>,
+}
+
+#[derive(Serialize)]
+struct SubmitResponse {
+    id: Uuid,
+}
+
+async fn submit(
+    State(state): State<AppState>,
+    Json(body): Json<SubmitRequest>,
+) -> Result<Json<SubmitResponse>, StatusCode> {
+    let row = sqlx::query(
+        "INSERT INTO responses (name, alcohol, food, with_partner, partner_name, partner_alcohol, partner_food)
+         VALUES ($1, $2, $3, $4, $5, $6, $7)
+         RETURNING id",
+    )
+    .bind(&body.name)
+    .bind(&body.alcohol)
+    .bind(&body.food)
+    .bind(body.with_partner)
+    .bind(&body.partner_name)
+    .bind(&body.partner_alcohol)
+    .bind(&body.partner_food)
+    .fetch_one(&state.db)
+    .await
+    .map_err(|e| {
+        tracing::error!("DB error: {e}");
+        StatusCode::INTERNAL_SERVER_ERROR
+    })?;
+
+    let id: Uuid = row.get("id");
+
+    Ok(Json(SubmitResponse { id }))
+}
+
+#[tokio::main]
+async fn main() {
+    tracing_subscriber::fmt::init();
+    dotenvy::dotenv().ok();
+
+    let database_url = std::env::var("DATABASE_URL").expect("DATABASE_URL must be set");
+    let port = std::env::var("PORT").unwrap_or_else(|_| "8080".to_string());
+
+    let db = PgPoolOptions::new()
+        .max_connections(5)
+        .connect(&database_url)
+        .await
+        .expect("Failed to connect to database");
+
+    sqlx::migrate!()
+        .run(&db)
+        .await
+        .expect("Failed to run migrations");
+
+    let cors = CorsLayer::new()
+        .allow_origin(Any)
+        .allow_methods(Any)
+        .allow_headers(Any);
+
+    let app = Router::new()
+        .route("/api/submit", post(submit))
+        .layer(cors)
+        .with_state(AppState { db });
+
+    let addr = format!("0.0.0.0:{port}");
+    tracing::info!("Listening on {addr}");
+    let listener = tokio::net::TcpListener::bind(&addr).await.unwrap();
+    axum::serve(listener, app).await.unwrap();
+}