Add AppState to Controller and split up models

emgauwa-lib/src/models/controller.rs

@@ -0,0 +1,51 @@
+use actix::MessageResponse;
+use futures::executor::block_on;
+use serde_derive::{Deserialize, Serialize};
+use sqlx::pool::PoolConnection;
+use sqlx::Sqlite;
+
+use crate::db::DbController;
+use crate::errors::{DatabaseError, EmgauwaError};
+use crate::models::{convert_db_list_cache, FromDbModel, Relay};
+
+#[derive(Serialize, Deserialize, Debug, Clone, MessageResponse)]
+pub struct Controller {
+	#[serde(flatten)]
+	pub c: DbController,
+	pub relays: Vec<Relay>,
+}
+
+impl FromDbModel for Controller {
+	type DbModel = DbController;
+	type DbModelCache = Vec<Relay>;
+
+	fn from_db_model(
+		conn: &mut PoolConnection<Sqlite>,
+		db_model: Self::DbModel,
+	) -> Result<Self, DatabaseError> {
+		let relays_db = block_on(db_model.get_relays(conn))?;
+		let cache = convert_db_list_cache(conn, relays_db, db_model.clone())?;
+		Self::from_db_model_cache(conn, db_model, cache)
+	}
+
+	fn from_db_model_cache(
+		_conn: &mut PoolConnection<Sqlite>,
+		db_model: Self::DbModel,
+		cache: Self::DbModelCache,
+	) -> Result<Self, DatabaseError> {
+		Ok(Controller {
+			c: db_model,
+			relays: cache,
+		})
+	}
+}
+
+impl Controller {
+	pub fn reload(&mut self, conn: &mut PoolConnection<Sqlite>) -> Result<(), EmgauwaError> {
+		self.c = block_on(self.c.reload(conn))?;
+		for relay in &mut self.relays {
+			relay.reload(conn)?;
+		}
+		Ok(())
+	}
+}