1 files changed, 110 insertions, 0 deletions
diff --git a/crates/rocie-server/src/storage/sql/insert/unit/mod.rs b/crates/rocie-server/src/storage/sql/insert/unit/mod.rs
new file mode 100644
index 0000000..ba08487
--- /dev/null
+++ b/crates/rocie-server/src/storage/sql/insert/unit/mod.rs
@@ -0,0 +1,110 @@
+use serde::{Deserialize, Serialize};
+use sqlx::query;
+use uuid::Uuid;
+
+use crate::storage::sql::{insert::{Operations, Transactionable}, unit::{Unit, UnitId}};
+
+#[derive(Debug, Deserialize, Serialize)]
+pub(crate) enum Operation {
+    RegisterUnit {
+        id: UnitId,
+        full_name_singular: String,
+        full_name_plural: String,
+        short_name: String,
+        description: Option<String>,
+    },
+}
+
+impl Transactionable for Operation {
+    type ApplyError = apply::Error;
+    type UndoError = undo::Error;
+
+    async fn apply(self, txn: &mut sqlx::SqliteConnection) -> Result<(), apply::Error> {
+        match self {
+            Operation::RegisterUnit {
+                id,
+                full_name_singular,
+                full_name_plural,
+                short_name,
+                description,
+            } => {
+                query!(
+                    "
+                    INSERT INTO units (id, full_name_singular, full_name_plural, short_name, description)
+                    VALUES (?,?,?,?,?)
+",
+                    id, full_name_singular, full_name_plural, short_name, description,
+                )
+                .execute(txn)
+                .await?;
+            }
+        }
+        Ok(())
+    }
+
+    async fn undo(self, txn: &mut sqlx::SqliteConnection) -> Result<(), undo::Error> {
+        match self {
+            Operation::RegisterUnit {
+                id,
+                full_name_singular,
+                full_name_plural,
+                short_name,
+                description,
+            } => {
+                query!(
+                    "
+                    DELETE FROM units
+                    WHERE id = ? AND full_name_singular = ? AND full_name_plural = ? AND short_name = ? AND description = ?;
+",
+                    id, full_name_singular, full_name_plural, short_name, description,
+                )
+                .execute(txn)
+                .await?;
+            }
+        }
+        Ok(())
+    }
+}
+
+pub(crate) mod undo {
+    #[derive(thiserror::Error, Debug)]
+    pub(crate) enum Error {
+        #[error("Failed to execute undo sql statments: {0}")]
+        SqlError(#[from] sqlx::Error),
+    }
+}
+pub(crate) mod apply {
+    #[derive(thiserror::Error, Debug)]
+    pub(crate) enum Error {
+        #[error("Failed to execute apply sql statments: {0}")]
+        SqlError(#[from] sqlx::Error),
+    }
+}
+
+impl Unit {
+    pub(crate) fn register(
+        full_name_singular: String,
+        full_name_plural: String,
+        short_name: String,
+        description: Option<String>,
+        ops: &mut Operations<Operation>,
+    ) -> Self {
+        let id = UnitId::from(Uuid::new_v4());
+
+        ops.push(Operation::RegisterUnit {
+            id,
+            full_name_singular: full_name_singular.clone(),
+            full_name_plural: full_name_plural.clone(),
+            short_name: short_name.clone(),
+            description: description.clone(),
+        });
+
+        Self {
+            id,
+            full_name_singular,
+            full_name_plural,
+            short_name,
+            description,
+        }
+    }
+}