From 1280c745c7c55088c62f63ece636b501261f776b Mon Sep 17 00:00:00 2001
From: AINDUSTRIES <contact.aindustries@gmail.com>
Date: Mon, 17 Mar 2025 15:00:02 +0100
Subject: [PATCH] Added default headers (cors)

---
 src/main.rs | 19 ++++++++++++-------
 1 file changed, 12 insertions(+), 7 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 144c2ac..c0a26f4 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,25 +1,25 @@
-mod users;
 mod cases;
 mod items;
 mod types;
+mod users;
 mod utils;
 
-use users::*;
 use cases::*;
 use items::*;
+use users::*;
 use utils::*;
 
 use actix_web::web::Data;
-use actix_web::{App, HttpServer};
+use actix_web::{App, HttpServer, middleware::DefaultHeaders};
 use sqlx::sqlite::SqlitePool;
 
 #[derive(Clone)]
 struct AppState {
     database: SqlitePool,
     token_expiration: u64,
+    allow_origins: Vec<&'static str>,
 }
 
-
 #[actix_web::main]
 async fn main() -> std::io::Result<()> {
     let pool = SqlitePool::connect("sqlite:database.db")
@@ -28,9 +28,14 @@ async fn main() -> std::io::Result<()> {
     let app_state = Data::new(AppState {
         database: pool,
         token_expiration: 86400,
+        allow_origins: vec!["http://localhost:5173"],
     });
     HttpServer::new(move || {
         App::new()
+            .wrap(DefaultHeaders::new().add((
+                "Access-Control-Allow-Origin",
+                app_state.allow_origins.join(","),
+            )))
             .service(login)
             .service(register)
             .service(logout)
@@ -43,7 +48,7 @@ async fn main() -> std::io::Result<()> {
             .service(options)
             .app_data(app_state.clone())
     })
-        .bind(("127.0.0.1", 8000))?
-        .run()
-        .await
+    .bind(("127.0.0.1", 8000))?
+    .run()
+    .await
 }