Add minor fixes

src/db/mod.rs

@@ -0,0 +1,78 @@
+use log::{info, trace};
+use sqlx::migrate::Migrator;
+use sqlx::sqlite::SqlitePoolOptions;
+use sqlx::{Pool, Sqlite};
+
+use crate::db::errors::DatabaseError;
+use crate::db::model_utils::Period;
+use crate::db::schedules::{Periods, Schedule};
+use crate::types::EmgauwaUid;
+
+pub mod errors;
+pub mod models;
+pub mod schedules;
+pub mod tag;
+
+mod model_utils;
+
+static MIGRATOR: Migrator = sqlx::migrate!(); // defaults to "./migrations"
+
+pub async fn run_migrations(pool: &Pool<Sqlite>) {
+	info!("Running migrations");
+	MIGRATOR.run(pool).await.expect("Failed to run migrations.");
+}
+
+async fn init_schedule(
+	pool: &Pool<Sqlite>,
+	uid: &EmgauwaUid,
+	name: &str,
+	periods: Periods,
+) -> Result<(), DatabaseError> {
+	trace!("Initializing schedule {:?}", name);
+	match Schedule::get_by_uid(&mut pool.acquire().await.unwrap(), uid).await {
+		Ok(_) => Ok(()),
+		Err(err) => match err {
+			DatabaseError::NotFound => {
+				trace!("Schedule {:?} not found, inserting", name);
+				sqlx::query_as!(
+					Schedule,
+					"INSERT INTO schedules (uid, name, periods) VALUES (?, ?, ?) RETURNING *",
+					uid,
+					name,
+					periods,
+				)
+				.fetch_optional(pool)
+				.await?
+				.ok_or(DatabaseError::InsertGetError)
+				.map(|_| ())
+			}
+			_ => Err(err),
+		},
+	}
+}
+
+pub async fn init(db: &str) -> Pool<Sqlite> {
+	let pool: Pool<Sqlite> = SqlitePoolOptions::new()
+		.acquire_timeout(std::time::Duration::from_secs(1))
+		.max_connections(5)
+		.connect(db)
+		.await
+		.expect("Error connecting to database.");
+
+	run_migrations(&pool).await;
+
+	init_schedule(&pool, &EmgauwaUid::Off, "Off", Periods(vec![]))
+		.await
+		.expect("Error initializing schedule Off");
+
+	init_schedule(
+		&pool,
+		&EmgauwaUid::On,
+		"On",
+		Periods(vec![Period::new_on()]),
+	)
+	.await
+	.expect("Error initializing schedule On");
+
+	pool
+}