fix: version redirect

think-in-universe · think-in-universe · commit 2d719cfb3500 · 2025-11-26T18:01:27.000+08:00
diff --git a/crates/api/src/routes/mod.rs b/crates/api/src/routes/mod.rs
@@ -4,7 +4,7 @@ pub mod attestation;
 pub mod oauth;
 pub mod users;
 
-use axum::{middleware::from_fn_with_state, response::Redirect, routing::get, Json, Router};
+use axum::{middleware::from_fn_with_state, routing::get, Json, Router};
 use serde::Serialize;
 use tower_http::cors::{Any, CorsLayer};
 use utoipa::ToSchema;
@@ -78,8 +78,6 @@ pub fn create_router_with_cors(app_state: AppState, allowed_origins: Vec<String>
 
     // Build the base router
     let router = Router::new()
-        // Redirect root to /?v=1 to workaround the cache issue
-        .route("/", get(|| async { Redirect::temporary("/?v=1") }))
         .route("/health", get(health_check))
         .nest("/v1/auth", auth_routes)
         .nest("/v1/users", user_routes)
diff --git a/crates/api/src/static_files.rs b/crates/api/src/static_files.rs
@@ -1,7 +1,7 @@
 use axum::{
     body::Body,
     http::{header, Request, StatusCode},
-    response::{IntoResponse, Response},
+    response::{IntoResponse, Redirect, Response},
 };
 use std::{env, path::PathBuf};
 use tower::ServiceExt;
@@ -19,6 +19,17 @@ fn frontend_dir() -> PathBuf {
 /// Serve static files with SPA fallback and proper cache headers.
 pub async fn static_handler(req: Request<Body>) -> Response {
     let frontend_dir = frontend_dir();
+    let uri = req.uri().clone();
+
+    let needs_version_redirect = uri.path() == "/"
+        && uri
+            .query()
+            .map(|q| !q.split('&').any(|pair| pair.starts_with("v=")))
+            .unwrap_or(true);
+
+    if needs_version_redirect {
+        return Redirect::temporary("/?v=1").into_response();
+    }
 
     let has_extension = req
         .uri()
