add admin responses endpoint
All checks were successful
Build and Publish Images / build-and-push (push) Successful in 6s
All checks were successful
Build and Publish Images / build-and-push (push) Successful in 6s
This commit is contained in:
@@ -1,15 +1,18 @@
|
|||||||
use axum::{
|
use axum::{
|
||||||
extract::State,
|
extract::State,
|
||||||
http::StatusCode,
|
http::{HeaderMap, StatusCode},
|
||||||
response::Json,
|
response::Json,
|
||||||
routing::post,
|
routing::{get, post},
|
||||||
Router,
|
Router,
|
||||||
};
|
};
|
||||||
|
use chrono::{DateTime, Utc};
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
use sqlx::{postgres::PgPoolOptions, PgPool, Row};
|
use sqlx::{postgres::PgPoolOptions, PgPool, Row};
|
||||||
use tower_http::cors::{Any, CorsLayer};
|
use tower_http::cors::{Any, CorsLayer};
|
||||||
use uuid::Uuid;
|
use uuid::Uuid;
|
||||||
|
|
||||||
|
const ADMIN_PASSWORD: &str = "2123";
|
||||||
|
|
||||||
#[derive(Clone)]
|
#[derive(Clone)]
|
||||||
struct AppState {
|
struct AppState {
|
||||||
db: PgPool,
|
db: PgPool,
|
||||||
@@ -31,6 +34,19 @@ struct SubmitResponse {
|
|||||||
id: Uuid,
|
id: Uuid,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Serialize)]
|
||||||
|
struct Response {
|
||||||
|
id: Uuid,
|
||||||
|
name: String,
|
||||||
|
alcohol: Option<String>,
|
||||||
|
food: Option<String>,
|
||||||
|
with_partner: bool,
|
||||||
|
partner_name: Option<String>,
|
||||||
|
partner_alcohol: Option<String>,
|
||||||
|
partner_food: Option<String>,
|
||||||
|
created_at: DateTime<Utc>,
|
||||||
|
}
|
||||||
|
|
||||||
async fn submit(
|
async fn submit(
|
||||||
State(state): State<AppState>,
|
State(state): State<AppState>,
|
||||||
Json(body): Json<SubmitRequest>,
|
Json(body): Json<SubmitRequest>,
|
||||||
@@ -55,10 +71,51 @@ async fn submit(
|
|||||||
})?;
|
})?;
|
||||||
|
|
||||||
let id: Uuid = row.get("id");
|
let id: Uuid = row.get("id");
|
||||||
|
|
||||||
Ok(Json(SubmitResponse { id }))
|
Ok(Json(SubmitResponse { id }))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async fn admin_responses(
|
||||||
|
State(state): State<AppState>,
|
||||||
|
headers: HeaderMap,
|
||||||
|
) -> Result<Json<Vec<Response>>, StatusCode> {
|
||||||
|
let password = headers
|
||||||
|
.get("x-admin-password")
|
||||||
|
.and_then(|v| v.to_str().ok())
|
||||||
|
.unwrap_or("");
|
||||||
|
|
||||||
|
if password != ADMIN_PASSWORD {
|
||||||
|
return Err(StatusCode::UNAUTHORIZED);
|
||||||
|
}
|
||||||
|
|
||||||
|
let rows = sqlx::query(
|
||||||
|
"SELECT id, name, alcohol, food, with_partner, partner_name, partner_alcohol, partner_food, created_at
|
||||||
|
FROM responses ORDER BY created_at DESC",
|
||||||
|
)
|
||||||
|
.fetch_all(&state.db)
|
||||||
|
.await
|
||||||
|
.map_err(|e| {
|
||||||
|
tracing::error!("DB error: {e}");
|
||||||
|
StatusCode::INTERNAL_SERVER_ERROR
|
||||||
|
})?;
|
||||||
|
|
||||||
|
let responses = rows
|
||||||
|
.iter()
|
||||||
|
.map(|r| Response {
|
||||||
|
id: r.get("id"),
|
||||||
|
name: r.get("name"),
|
||||||
|
alcohol: r.get("alcohol"),
|
||||||
|
food: r.get("food"),
|
||||||
|
with_partner: r.get("with_partner"),
|
||||||
|
partner_name: r.get("partner_name"),
|
||||||
|
partner_alcohol: r.get("partner_alcohol"),
|
||||||
|
partner_food: r.get("partner_food"),
|
||||||
|
created_at: r.get("created_at"),
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
|
||||||
|
Ok(Json(responses))
|
||||||
|
}
|
||||||
|
|
||||||
#[tokio::main]
|
#[tokio::main]
|
||||||
async fn main() {
|
async fn main() {
|
||||||
tracing_subscriber::fmt::init();
|
tracing_subscriber::fmt::init();
|
||||||
@@ -85,6 +142,7 @@ async fn main() {
|
|||||||
|
|
||||||
let app = Router::new()
|
let app = Router::new()
|
||||||
.route("/api/submit", post(submit))
|
.route("/api/submit", post(submit))
|
||||||
|
.route("/api/admin/responses", get(admin_responses))
|
||||||
.layer(cors)
|
.layer(cors)
|
||||||
.with_state(AppState { db });
|
.with_state(AppState { db });
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user