1 files changed, 67 insertions, 0 deletions
diff --git a/crates/rocie-server/src/api/set/auth/unit.rs b/crates/rocie-server/src/api/set/auth/unit.rs
new file mode 100644
index 0000000..21d1e11
--- /dev/null
+++ b/crates/rocie-server/src/api/set/auth/unit.rs
@@ -0,0 +1,67 @@
+use actix_identity::Identity;
+use actix_web::{HttpResponse, Responder, Result, post, web};
+use serde::Deserialize;
+use utoipa::ToSchema;
+
+use crate::{
+    app::App,
+    storage::sql::{
+        insert::Operations,
+        unit::{Unit, UnitId},
+        unit_property::UnitPropertyId,
+    },
+};
+
+#[derive(Deserialize, ToSchema)]
+struct UnitStub {
+    full_name_plural: String,
+    full_name_singular: String,
+    short_name: String,
+    unit_property: UnitPropertyId,
+
+    #[schema(nullable = false)]
+    description: Option<String>,
+}
+
+/// Register an Unit
+#[utoipa::path(
+    responses(
+        (
+            status = OK,
+            description = "Unit successfully registered in database",
+            body = UnitId,
+        ),
+        (
+            status = UNAUTHORIZED,
+            description = "You did not login before calling this endpoint",
+        ),
+        (
+            status = INTERNAL_SERVER_ERROR,
+            description = "Server encountered error",
+            body = String,
+        )
+    ),
+    request_body = UnitStub,
+)]
+#[post("/unit/new")]
+pub(crate) async fn register_unit(
+    app: web::Data<App>,
+    unit: web::Json<UnitStub>,
+    _user: Identity,
+) -> Result<impl Responder> {
+    let unit = unit.into_inner();
+    let mut ops = Operations::new("register unit");
+
+    let unit = Unit::register(
+        unit.full_name_singular,
+        unit.full_name_plural,
+        unit.short_name,
+        unit.description,
+        unit.unit_property,
+        &mut ops,
+    );
+
+    ops.apply(&app).await?;
+
+    Ok(HttpResponse::Ok().json(unit.id))
+}