From 08a1a373ecb0dcb329c601415d32c3bdd9262c80 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Moritz=20H=C3=B6lting?=
 <87192362+moritz-hoelting@users.noreply.github.com>
Date: Tue, 2 Jul 2024 21:41:32 +0200
Subject: [PATCH] implement cors

---
 Cargo.lock  | 16 ++++++++++++++++
 Cargo.toml  |  1 +
 src/main.rs | 28 ++++++++++++++++++++++++----
 3 files changed, 41 insertions(+), 4 deletions(-)

diff --git a/Cargo.lock b/Cargo.lock
index f66cbf0..c78d7b5 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -19,6 +19,21 @@ dependencies = [
  "tracing",
 ]
 
+[[package]]
+name = "actix-cors"
+version = "0.7.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f9e772b3bcafe335042b5db010ab7c09013dad6eac4915c91d8d50902769f331"
+dependencies = [
+ "actix-utils",
+ "actix-web",
+ "derive_more",
+ "futures-util",
+ "log",
+ "once_cell",
+ "smallvec",
+]
+
 [[package]]
 name = "actix-governor"
 version = "0.5.0"
@@ -1195,6 +1210,7 @@ checksum = "78ca9ab1a0babb1e7d5695e3530886289c18cf2f87ec19a575a0abdce112e3a3"
 name = "mensa-upb-api"
 version = "0.1.0"
 dependencies = [
+ "actix-cors",
  "actix-governor",
  "actix-web",
  "anyhow",
diff --git a/Cargo.toml b/Cargo.toml
index 4999cc8..32dd67e 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -5,6 +5,7 @@ version = "0.1.0"
 edition = "2021"
 
 [dependencies]
+actix-cors = "0.7.0"
 actix-governor = { version = "0.5.0", features = ["log"] }
 actix-web = "4.8.0"
 anyhow = "1.0.86"
diff --git a/src/main.rs b/src/main.rs
index be4ad2c..9635cea 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,5 +1,6 @@
 use std::{env, io, str::FromStr};
 
+use actix_cors::Cors;
 use actix_governor::{Governor, GovernorConfigBuilder};
 use actix_web::{get, web, App, HttpResponse, HttpServer, Responder};
 use chrono::{Duration as CDuration, Utc};
@@ -11,8 +12,10 @@ use strum::IntoEnumIterator;
 
 #[actix_web::main]
 async fn main() -> io::Result<()> {
-    if dotenvy::dotenv().is_ok() {
-        println!("Loaded .env file");
+    match dotenvy::dotenv() {
+        Ok(_) => println!("Loaded .env file"),
+        Err(dotenvy::Error::LineParse(..)) => eprintln!("Malformed .env file"),
+        Err(_) => {}
     }
 
     let interface = env::var("API_INTERFACE").unwrap_or("127.0.0.1".to_string());
@@ -29,6 +32,15 @@ async fn main() -> io::Result<()> {
         .and_then(|s| s.parse::<u32>().ok())
         .unwrap_or(5);
 
+    let allowed_cors = env::var("API_CORS_ALLOWED")
+        .map(|val| {
+            val.split(',')
+                .map(|domain| domain.trim().to_string())
+                .collect_vec()
+        })
+        .ok()
+        .unwrap_or_default();
+
     let governor_conf = GovernorConfigBuilder::default()
         .per_second(seconds_replenish)
         .burst_size(burst_size)
@@ -38,8 +50,16 @@ async fn main() -> io::Result<()> {
     let menu_cache = MenuCache::default();
 
     HttpServer::new(move || {
+        let cors = allowed_cors
+            .iter()
+            .fold(Cors::default(), |cors, domain| cors.allowed_origin(domain))
+            .send_wildcard()
+            .allow_any_method()
+            .allow_any_header()
+            .max_age(3600);
         App::new()
             .wrap(Governor::new(&governor_conf))
+            .wrap(cors)
             .app_data(web::Data::new(menu_cache.clone()))
             .service(index)
             .service(menu_today)
@@ -84,9 +104,9 @@ async fn menu_today(
 
         if let Ok(days_ahead) = days_ahead {
             let date = (Utc::now() + CDuration::days(days_ahead)).date_naive();
-        let menu = cache.get_combined(&canteens, date).await;
+            let menu = cache.get_combined(&canteens, date).await;
 
-        HttpResponse::Ok().json(menu)
+            HttpResponse::Ok().json(menu)
         } else {
             HttpResponse::BadRequest().json(json!({
                 "error": "Invalid days query"