From cc8f6c67f2bd86cc560430a3a9ecc1996ad46224 Mon Sep 17 00:00:00 2001
From: Dubzer <me@dubzer.dev>
Date: Thu, 12 Oct 2023 22:06:03 +0300
Subject: [PATCH] Support redirects

---
 src/handlers.rs | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)
diff --git a/src/handlers.rs b/src/handlers.rs
index 6726a30..0061b90 100644
--- a/src/handlers.rs
+++ b/src/handlers.rs
@@ -45,7 +45,10 @@ pub async fn handle(mut req: Request<Body>, origin: HeaderValue) -> Result<Respo
         .ok_or_else(|| HandlerError::new_with_origin("sowwy, you might have fowgotten host in your destination", StatusCode::BAD_REQUEST, origin.clone()))?;
 
 
-    req.headers_mut().insert("Host", HeaderValue::from_str(destination_host.as_str()).unwrap());
+    let cors_host = req.headers_mut()
+        .insert("Host", HeaderValue::from_str(destination_host.as_str()).unwrap())
+        .ok_or_else(|| HandlerError::new_with_origin("sowwy, you might have fowgotten host headew", StatusCode::BAD_REQUEST, origin.clone()))?;
+    
     *req.uri_mut() = destination_uri;
 
     let client_response = GLOBAL_CLIENT.request(req).await
@@ -55,6 +58,10 @@ pub async fn handle(mut req: Request<Body>, origin: HeaderValue) -> Result<Respo
     let (mut parts, body) = client_response.into_parts();
     parts.headers.extend(get_default_cors(origin.clone()));
 
+    if let Some(location) = parts.headers.get_mut("Location") {
+        *location = HeaderValue::from_str(format!("https://{}/{}", cors_host.to_str()?, location.to_str()?).as_str())?;
+    }
+
     return Ok(Response::from_parts(parts, body));
 }
 
