5 files changed, 125 insertions, 0 deletions
diff --git a/schist_backend/src/databases.rs b/schist_backend/src/databases.rs
new file mode 100644
index 0000000..7202e78
--- /dev/null
+++ b/schist_backend/src/databases.rs
@@ -0,0 +1,11 @@
+use rocket_sync_db_pools::{database, diesel};
+
+cfg_if::cfg_if! {
+    if #[cfg(test)] {
+        #[database("user_data.test")]
+        pub struct UserDataDb(diesel::SqliteConnection);
+    } else {
+        #[database("user_data")]
+        pub struct UserDataDb(diesel::SqliteConnection);
+    }
+}
diff --git a/schist_backend/src/main.rs b/schist_backend/src/main.rs
new file mode 100644
index 0000000..8395681
--- /dev/null
+++ b/schist_backend/src/main.rs
@@ -0,0 +1,48 @@
+use schist_schema::migrations::MIGRATIONS;
+use diesel::SqliteConnection;
+use diesel_migrations::MigrationHarness;
+use rocket::fairing::AdHoc;
+use rocket::http::ContentType;
+use rocket::{launch, routes, Build, Rocket};
+
+use crate::databases::UserDataDb;
+
+pub mod databases;
+pub mod routes;
+
+fn run_database_migrations(connection: &mut SqliteConnection) {
+    connection
+        .run_pending_migrations(MIGRATIONS)
+        .expect("Error running database migrations");
+}
+
+#[launch]
+async fn rocket() -> Rocket<Build> {
+    rocket::build()
+        .attach(UserDataDb::fairing())
+        .attach(AdHoc::on_liftoff("Database Migration", |rocket| {
+            Box::pin(async move {
+                let user_data_db = UserDataDb::get_one(rocket).await.expect(
+                    "Failed to get connection to user data database in order to run migrations",
+                );
+                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::category::get_all_categories,
+            ],
+        )
+}
diff --git a/schist_backend/src/routes/account.rs b/schist_backend/src/routes/account.rs
new file mode 100644
index 0000000..83d4d1b
--- /dev/null
+++ b/schist_backend/src/routes/account.rs
@@ -0,0 +1,48 @@
+use schist_models::account::Account;
+use schist_queries::accounts::{get_all_accounts as get_all_accounts_query, insert_accounts};
+use rocket::http::Status;
+use rocket::serde::json::Json;
+use rocket::{get, post, Responder};
+
+use crate::databases::UserDataDb;
+
+#[get("/account")]
+pub async fn get_all_accounts(user_data_db: UserDataDb) -> Result<Json<Vec<Account>>, (Status, String)> {
+    user_data_db
+        .run(get_all_accounts_query)
+        .await
+        .map(Json)
+        .map_err(|err| (Status::InternalServerError, err.to_string()))
+}
+
+#[derive(Responder)]
+pub enum CreateAccountResponder {
+    #[response(status = 500, content_type = "text")]
+    InternalServerError(String),
+
+    #[response(status = 202, content_type = "json")]
+    Created(Json<Account>),
+
+    #[response(status = 400, content_type = "text")]
+    BadRequest(String),
+}
+
+#[post("/account", data = "<account>", format = "json")]
+pub async fn create_account(
+    account: Json<Account>,
+    user_data_db: UserDataDb,
+) -> CreateAccountResponder {
+    let account = account.into_inner();
+    let accounts_vec = vec![account.clone()];
+
+    let num_rows_inserted = user_data_db
+        .run(move |connection| insert_accounts(&accounts_vec, connection))
+        .await;
+
+    match num_rows_inserted {
+        Ok(_) => CreateAccountResponder::Created(Json(account)),
+        Err(err) => CreateAccountResponder::InternalServerError(format!(
+                "Error creating account: {err}"
+            )),
+    }
+}
diff --git a/schist_backend/src/routes/category.rs b/schist_backend/src/routes/category.rs
new file mode 100644
index 0000000..b2412ac
--- /dev/null
+++ b/schist_backend/src/routes/category.rs
@@ -0,0 +1,16 @@
+use rocket::get;
+use rocket::http::Status;
+use rocket::serde::json::Json;
+use schist_models::category::Category;
+use schist_queries::categories::get_all_categories as get_all_categories_query;
+
+use crate::databases::UserDataDb;
+
+#[get("/category")]
+pub async fn get_all_categories(user_data_db: UserDataDb) -> Result<Json<Vec<Category>>, Status> {
+    user_data_db
+        .run(get_all_categories_query)
+        .await
+        .map(Json)
+        .map_err(|_| Status { code: 500 })
+}
diff --git a/schist_backend/src/routes/mod.rs b/schist_backend/src/routes/mod.rs
new file mode 100644
index 0000000..e22565c
--- /dev/null
+++ b/schist_backend/src/routes/mod.rs
@@ -0,0 +1,2 @@
+pub mod account;
+pub mod category;