Start rust rewrite

src/db.rs

@@ -0,0 +1,62 @@
+pub mod errors;
+pub mod models;
+pub mod schema;
+mod types;
+
+use diesel::prelude::*;
+
+use diesel::dsl::sql;
+use dotenv::dotenv;
+use std::env;
+
+use models::*;
+use schema::schedules::dsl::*;
+
+use diesel_migrations::embed_migrations;
+use errors::DatabaseError;
+use types::EmgauwaUid;
+
+embed_migrations!("migrations");
+
+fn get_connection() -> SqliteConnection {
+    dotenv().ok();
+
+    let database_url = env::var("DATABASE_URL").expect("DATABASE_URL must be set");
+    SqliteConnection::establish(&database_url)
+        .unwrap_or_else(|_| panic!("Error connecting to {}", database_url))
+}
+
+pub fn run_migrations() {
+    let connection = get_connection();
+    embedded_migrations::run(&connection).expect("Failed to run migrations.");
+}
+
+pub fn get_schedules() -> Vec<Schedule> {
+    let connection = get_connection();
+    schedules
+        .limit(5)
+        .load::<Schedule>(&connection)
+        .expect("Error loading schedules")
+}
+
+pub fn create_schedule(new_name: &str) -> Result<Schedule, DatabaseError> {
+    let connection = get_connection();
+
+    let new_schedule = NewSchedule {
+        uid: &EmgauwaUid::default(),
+        name: new_name,
+        periods: "",
+    };
+
+    diesel::insert_into(schedules)
+        .values(&new_schedule)
+        .execute(&connection)
+        .or(Err(DatabaseError::InsertError))?;
+
+    let result = schedules
+        .find(sql("last_insert_rowid()"))
+        .get_result::<Schedule>(&connection)
+        .or(Err(DatabaseError::InsertGetError))?;
+
+    Ok(result)
+}