From f0e3e0cfb183253e03b1a61ddecb7b9330d47fdb Mon Sep 17 00:00:00 2001 From: Joe Carstairs Date: Tue, 17 Sep 2024 11:28:56 +0100 Subject: Refactors backend --- backend/src/main.rs | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) (limited to 'backend/src/main.rs') diff --git a/backend/src/main.rs b/backend/src/main.rs index db47ceb..cdf4699 100644 --- a/backend/src/main.rs +++ b/backend/src/main.rs @@ -2,14 +2,17 @@ use diesel::SqliteConnection; use diesel_migrations::MigrationHarness; use diesel_migrations::{embed_migrations, EmbeddedMigrations}; use rocket::fairing::AdHoc; +use rocket::http::ContentType; use rocket::{launch, routes, Build, Rocket}; use crate::databases::UserDataDb; pub mod databases; pub mod models; +pub mod queries; pub mod routes; pub mod schema; +pub mod utils; pub const MIGRATIONS: EmbeddedMigrations = embed_migrations!("migrations"); @@ -31,11 +34,23 @@ async fn rocket() -> Rocket { user_data_db.run(run_database_migrations).await; }) })) + .attach(AdHoc::on_response("Add CORS headers", |_req, res| { + Box::pin(async move { + res.set_header(ContentType::JSON); + res.set_header(rocket::http::Header::new( + "Access-Control-Allow-Origin", + "http://localhost:5173", + )); + }) + })) .mount( "/", routes![ routes::account::get_all_accounts, - routes::account::create_account + routes::account::create_account, + routes::balance::get_all_category_balances, + routes::budget::get_all_category_budgets, + routes::category::get_all_categories, ], ) } -- cgit v1.2.3