From 8d15e786ace69ed83aa1b17f716d64b2ebf5ca9a Mon Sep 17 00:00:00 2001
From: Matthias Lodner <matthias@lodner.de>
Date: Sat, 31 Jan 2026 04:10:51 +0100
Subject: [PATCH] Added data label for backend.

---
 common/src/models/article.rs                  |   2 +-
 common/src/models/label.rs                    | 148 ++++++++++++++++++
 common/src/models/mod.rs                      |   1 +
 common/src/models/variant.rs                  |  45 +++++-
 .../app/components/variant/variantRow.vue     |   7 +-
 frontend/app/types/label.ts                   |   9 ++
 frontend/app/types/variant.ts                 |   9 +-
 .../migrations/20260131020351_add_label.sql   |  17 ++
 server/src/main.rs                            |   4 +
 9 files changed, 231 insertions(+), 11 deletions(-)
 create mode 100644 common/src/models/label.rs
 create mode 100644 frontend/app/types/label.ts
 create mode 100644 server/migrations/20260131020351_add_label.sql

diff --git a/common/src/models/article.rs b/common/src/models/article.rs
index 4885fd9..4eb9ce6 100644
--- a/common/src/models/article.rs
+++ b/common/src/models/article.rs
@@ -33,7 +33,7 @@ fn from_postgres_row_small(row: PgRow) -> Article {
 }
 
 async fn from_postgres_row(row: PgRow, pool: PgPool) -> Article {
-    let variants_option = load_article_variants(row.get(0), pool).await;
+    let variants_option = load_article_variants(row.get(0), &pool).await;
 
     let variants = match variants_option {
         Some(variants_inner) => {
diff --git a/common/src/models/label.rs b/common/src/models/label.rs
new file mode 100644
index 0000000..b2b7aad
--- /dev/null
+++ b/common/src/models/label.rs
@@ -0,0 +1,148 @@
+use log::{error, info};
+use serde::{Deserialize, Serialize};
+use sqlx::{Arguments, Decode, Encode, FromRow, PgPool, Row};
+use sqlx::postgres::{PgArguments, PgRow};
+
+#[derive(Debug, Deserialize, Serialize, Clone, FromRow, Decode, Encode)]
+pub struct Label {
+    pub id: i64,
+    pub name: String,
+}
+
+fn from_postgres_row_small(row: PgRow) -> Label {
+    Label {
+        id: row.get(0),
+        name: row.get(1),
+    }
+}
+
+pub async fn load_labels(pool: PgPool) -> Option<Vec<Label>> {
+    let statement = format!(
+        "SELECT * FROM {}",
+        "label",
+    );
+
+    let res = sqlx::query(
+        statement.as_str())
+        .fetch_all(&pool).await;
+
+    let mut labels: Vec<Label> = Vec::new();
+
+    match res {
+        Ok(records) => {
+            for record in records {
+                labels.push(
+                    from_postgres_row_small(record)
+                );
+            }
+
+            Some(labels)
+        }
+        Err(err) => {
+            error!("Error while loading all labels");
+            None
+        }
+    }
+}
+
+pub async fn load_label(id: i64, pool: &PgPool) -> Option<Label> {
+    let statement = format!(
+        "SELECT * FROM {} WHERE $1 = id",
+        "label",
+    );
+
+    let mut args = PgArguments::default();
+    args.add(id);
+
+    let res = sqlx::query_with(
+        statement.as_str(), args)
+        .fetch_one(pool).await;
+    
+    match res {
+        Ok(record) => {
+            Some(from_postgres_row_small(record))
+        }
+        Err(err) => {
+            error!("Error while loading label: {}", id);
+            None
+        }
+    }
+}
+
+pub async fn insert_label(label: &Label, pool: PgPool) -> Option<Label> {
+    let mut args = PgArguments::default();
+    args.add(&label.name);
+
+    let statement = format!(
+        "INSERT INTO {} (name) VALUES ($1) RETURNING *",
+        "label",
+    );
+
+    let res = sqlx::query_with(
+        statement.as_str(),
+        args)
+        .fetch_one(&pool).await;
+
+    info!("{:#?}", res);
+
+    match res {
+        Ok(record) => {
+            let label = from_postgres_row_small(record);
+
+            Some(label)
+        }
+        Err(err) => {
+            error!("Error while loading newly created label: {}", label.name);
+            None
+        }
+    }
+}
+
+pub async fn update_label(label_id: i64, label: &Label, pool: PgPool) -> Option<Label> {
+    let mut args = PgArguments::default();
+    args.add(&label.id);
+    args.add(&label.name);
+
+    let statement = format!(
+        "UPDATE {} SET name = $2 WHERE id = $1 RETURNING *",
+        "label",
+    );
+
+    let res = sqlx::query_with(
+        statement.as_str(),
+        args)
+        .fetch_one(&pool).await;
+
+    info!("{:#?}", res);
+
+    match res {
+        Ok(record) => {
+            let label = from_postgres_row_small(record);
+            Some(label)
+        }
+        Err(err) => {
+            error!("Error while updating label: {}", label_id);
+            None
+        }
+    }
+}
+
+pub async fn delete_label(label_id: i64, pool: PgPool) -> Option<()> {
+    let result = sqlx::query(
+        format!("DELETE FROM {} WHERE id = {}",
+                "label",
+                label_id
+        ).as_str()
+    ).execute(&pool).await;
+
+    match result {
+        Ok(_) => {
+            Some(())
+        }
+        Err(err) => {
+            println!("Error deleting label: {:#?}", err);
+
+            None
+        }
+    }
+}
diff --git a/common/src/models/mod.rs b/common/src/models/mod.rs
index 3a7da5f..3b5b6fb 100644
--- a/common/src/models/mod.rs
+++ b/common/src/models/mod.rs
@@ -2,3 +2,4 @@ pub mod origin;
 pub mod article;
 pub mod variant;
 pub mod unit;
+pub mod label;
diff --git a/common/src/models/variant.rs b/common/src/models/variant.rs
index 88dce9e..23d6561 100644
--- a/common/src/models/variant.rs
+++ b/common/src/models/variant.rs
@@ -2,6 +2,7 @@ use log::{error, info};
 use serde::{Deserialize, Serialize};
 use sqlx::{Arguments, FromRow, PgPool, Row};
 use sqlx::postgres::{PgArguments, PgRow};
+use crate::models::label::{load_label, Label};
 use crate::models::unit::Unit;
 
 #[derive(Debug, Deserialize, Serialize, Clone, FromRow)]
@@ -13,9 +14,13 @@ pub struct Variant {
     pub description: Option<String>,
     pub ingredients: Option<String>,
     pub unit: Unit,
+    pub label: Option<Label>,
+    pub variant_description: Option<String>,
 }
 
-fn from_postgres_row_small(row: PgRow) -> Variant {
+async fn from_postgres_row(row: PgRow, pool: &PgPool) -> Variant {
+    let label = load_label(row.get(0), pool).await;
+
     Variant {
         id: row.get(0),
         weight: row.get(2),
@@ -24,10 +29,12 @@ fn from_postgres_row_small(row: PgRow) -> Variant {
         description: row.get(5),
         ingredients: row.get(6),
         unit: row.get(7),
+        label: label,
+        variant_description: row.get(9),
     }
 }
 
-pub async fn load_article_variants(article_id: i64, pool: PgPool) -> Option<Vec<Variant>> {
+pub async fn load_article_variants(article_id: i64, pool: &PgPool) -> Option<Vec<Variant>> {
     let statement = format!(
         "SELECT * FROM {} WHERE article_id = $1",
         "variant",
@@ -38,7 +45,7 @@ pub async fn load_article_variants(article_id: i64, pool: PgPool) -> Option<Vec<
 
     let res = sqlx::query_with(
         statement.as_str(), args)
-        .fetch_all(&pool).await;
+        .fetch_all(pool).await;
 
     let mut variants: Vec<Variant> = Vec::new();
 
@@ -46,7 +53,7 @@ pub async fn load_article_variants(article_id: i64, pool: PgPool) -> Option<Vec<
         Ok(records) => {
             for record in records {
                 variants.push(
-                    from_postgres_row_small(record)
+                    from_postgres_row(record, &pool).await
                 );
             }
 
@@ -60,6 +67,15 @@ pub async fn load_article_variants(article_id: i64, pool: PgPool) -> Option<Vec<
 }
 
 pub async fn insert_variant(article_id: i64, variant: &Variant, pool: PgPool) -> Option<Variant> {
+    let label_id = match &variant.label {
+        Some(label) => {
+            Some(label.id)
+        }
+        None => {
+            None
+        }
+    };
+
     let mut args = PgArguments::default();
     args.add(&article_id);
     args.add(&variant.weight);
@@ -68,9 +84,11 @@ pub async fn insert_variant(article_id: i64, variant: &Variant, pool: PgPool) ->
     args.add(&variant.description);
     args.add(&variant.ingredients);
     args.add(&variant.unit);
+    args.add(label_id);
+    args.add(&variant.variant_description);
 
     let statement = format!(
-        "INSERT INTO {} (article_id, weight, ean, name, description, ingredients, unit_id) VALUES ($1, $2, $3, $4, $5, $6, $7) RETURNING *",
+        "INSERT INTO {} (article_id, weight, ean, name, description, ingredients, unit_id, label_id, variant_description) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9) RETURNING *",
         "variant",
     );
 
@@ -83,7 +101,7 @@ pub async fn insert_variant(article_id: i64, variant: &Variant, pool: PgPool) ->
 
     match res {
         Ok(record) => {
-            let variant = from_postgres_row_small(record);
+            let variant = from_postgres_row(record, &pool).await;
             Some(variant)
         }
         Err(err) => {
@@ -94,6 +112,15 @@ pub async fn insert_variant(article_id: i64, variant: &Variant, pool: PgPool) ->
 }
 
 pub async fn update_variant(variant_id: i64, variant: &Variant, pool: PgPool) -> Option<Variant> {
+    let label_id = match &variant.label {
+        Some(label) => {
+            Some(label.id)
+        }
+        None => {
+            None
+        }
+    };
+
     let mut args = PgArguments::default();
     args.add(&variant.id);
     args.add(&variant.weight);
@@ -102,11 +129,13 @@ pub async fn update_variant(variant_id: i64, variant: &Variant, pool: PgPool) ->
     args.add(&variant.description);
     args.add(&variant.ingredients);
     args.add(&variant.unit);
+    args.add(label_id);
+    args.add(&variant.variant_description);
 
     info!("Variant: {:#?}", variant);
 
     let statement = format!(
-        "UPDATE {} SET weight = $2, ean = $3, name = $4, description = $5, ingredients = $6, unit_id = $7 WHERE id = $1 RETURNING *",
+        "UPDATE {} SET weight = $2, ean = $3, name = $4, description = $5, ingredients = $6, unit_id = $7, label_id = $8, variant_description = $9 WHERE id = $1 RETURNING *",
         "variant",
     );
 
@@ -119,7 +148,7 @@ pub async fn update_variant(variant_id: i64, variant: &Variant, pool: PgPool) ->
 
     match res {
         Ok(record) => {
-            let variant = from_postgres_row_small(record);
+            let variant = from_postgres_row(record, &pool).await;
             Some(variant)
         }
         Err(err) => {
diff --git a/frontend/app/components/variant/variantRow.vue b/frontend/app/components/variant/variantRow.vue
index 273a272..6343b8e 100644
--- a/frontend/app/components/variant/variantRow.vue
+++ b/frontend/app/components/variant/variantRow.vue
@@ -4,6 +4,7 @@
   import {Variant} from "~/types/variant";
   import DeleteConfirm from "~/components/ui/deleteConfirm.vue";
   import type {Unit} from "~/types/unit";
+  import type {Label} from "~/types/label";
 
   let store = useLabelEditorStore();
 
@@ -19,6 +20,8 @@
   let unit: Ref<Unit | null> = ref<Unit | null>(JSON.parse(JSON.stringify(props.variant.unit)));
   let description: Ref<string | null> = ref<string | null>(JSON.parse(JSON.stringify(props.variant.description)));
   let ingredients: Ref<string | null> = ref<string | null>(JSON.parse(JSON.stringify(props.variant.ingredients)));
+  let label: Ref<Label | null> = ref<Label | null>(JSON.parse(JSON.stringify(props.variant.label)));
+  let variant_description: Ref<string | null> = ref<string | null>(JSON.parse(JSON.stringify(props.variant.variant_description)));
 
   function toggleEdit() {
     if (edit.value) {
@@ -44,7 +47,9 @@
 
   async function updateVariant() {
     if (weight.value != null && unit.value != null) {
-      let updateVariant: Variant = new Variant(props.variant.id, weight.value, ean.value, name.value, description.value, ingredients.value, unit.value);
+      let updateVariant: Variant = new Variant(props.variant.id, weight.value,
+          ean.value, name.value, description.value, ingredients.value,
+          unit.value, label.value, variant_description.value);
 
       await store.updateVariant(updateVariant);
 
diff --git a/frontend/app/types/label.ts b/frontend/app/types/label.ts
new file mode 100644
index 0000000..c672ae8
--- /dev/null
+++ b/frontend/app/types/label.ts
@@ -0,0 +1,9 @@
+export class Label {
+    readonly id: number;
+    name: string;
+    
+    constructor(id: number, name: string) {
+        this.id = id;
+        this.name = name;
+    }
+}
diff --git a/frontend/app/types/variant.ts b/frontend/app/types/variant.ts
index dced6bd..0fafcc3 100644
--- a/frontend/app/types/variant.ts
+++ b/frontend/app/types/variant.ts
@@ -1,4 +1,5 @@
 import type {Unit} from "~/types/unit";
+import type {Label} from "~/types/label";
 
 export class Variant {
   readonly id: number;
@@ -8,8 +9,12 @@ export class Variant {
   description: string | null;
   ingredients: string | null;
   unit: Unit;
+  label: Label | null;
+  variant_description: string | null;
 
-  constructor(id: number, weight: string, ean: string|null, name: string|null, description:string|null, ingredients: string|null, unit: Unit) {
+  constructor(id: number, weight: string, ean: string|null, name: string|null,
+              description:string|null, ingredients: string|null, unit: Unit,
+              label: Label | null, variant_description: string | null) {
     this.id = id;
     this.weight = weight;
     this.ean = ean;
@@ -17,5 +22,7 @@ export class Variant {
     this.description = description;
     this.ingredients = ingredients;
     this.unit = unit;
+    this.label = label;
+    this.variant_description = variant_description;
   }
 }
diff --git a/server/migrations/20260131020351_add_label.sql b/server/migrations/20260131020351_add_label.sql
new file mode 100644
index 0000000..a8c64f2
--- /dev/null
+++ b/server/migrations/20260131020351_add_label.sql
@@ -0,0 +1,17 @@
+-- Add migration script here
+
+-- Create Label
+CREATE TABLE label
+(
+    id bigint NOT NULL GENERATED ALWAYS AS IDENTITY ( INCREMENT 1 START 1 ),
+    name text NOT NULL
+);
+
+-- Alter variant table
+ALTER TABLE public.variant
+    ADD label_id bigint,
+    ADD variant_description text,
+    ADD CONSTRAINT variant_label_id_fkey
+       foreign key (label_id)
+           references unit(id)
+           ON DELETE NO ACTION;
diff --git a/server/src/main.rs b/server/src/main.rs
index 870802b..6296724 100644
--- a/server/src/main.rs
+++ b/server/src/main.rs
@@ -211,6 +211,8 @@ async fn add_variant(
                 description: None,
                 ingredients: None,
                 unit: Unit::G,
+                label: None,
+                variant_description: None,
             };
 
             (StatusCode::INTERNAL_SERVER_ERROR, Json(variant))
@@ -240,6 +242,8 @@ async fn update_variant_route(
                 description: None,
                 ingredients: None,
                 unit: Unit::G,
+                label: None,
+                variant_description: None,
             };
 
             (StatusCode::INTERNAL_SERVER_ERROR, Json(variant))