patch: Fix cors

dustinrouillard · May 24, 2024 · 0a98036 · 0a98036
1 parent ffbfe69
commit 0a98036
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -34,6 +34,7 @@ once_cell = "1.19.0"
 sha1 = "0.10.6"
 hex = "0.4.3"
 gql_client = "1.0.7"
+actix-cors = "0.7.0"
 
 [profile.release]
 lto = true

diff --git a/src/main.rs b/src/main.rs
@@ -6,6 +6,7 @@ pub mod structs;
 
 use std::{error::Error, time::Duration};
 
+use actix_cors::Cors;
 use actix_multipart::form::MultipartFormConfig;
 use actix_web::{middleware, web, App, HttpServer};
 use connectivity::{
@@ -79,13 +80,22 @@ async fn main() -> Result<(), Box<dyn Error>> {
   }
 
   HttpServer::new(move || {
+    let cors = Cors::default()
+      .allowed_origin_fn(|origin, _req_head| {
+        origin.as_bytes().ends_with(b".dstn.to")
+      })
+      .allowed_origin_fn(|origin, _req_head| {
+        origin.as_bytes().ends_with(b":3000")
+      });
+
     App::new()
       .app_data(web::Data::clone(&data_http))
       .app_data(
         MultipartFormConfig::default()
           .total_limit(10 * 1024 * 1024)
           .memory_limit(10 * 1024 * 1024),
       )
+      .wrap(cors)
       .wrap(middleware::NormalizePath::default())
       .wrap(TracingLogger::default())
       .default_service(web::to(services::base::index))