fix: saving image schema items

backend/models/lists.go

@@ -3,6 +3,7 @@ package models
 import (
 	"context"
 	"database/sql"
+	"fmt"
 	"screenmark/screenmark/.gen/haystack/haystack/model"
 	. "screenmark/screenmark/.gen/haystack/haystack/table"
 
@@ -103,12 +104,40 @@ func (m StackModel) SaveItems(ctx context.Context, items []model.SchemaItems) er
 	return err
 }
 
-func (m StackModel) SaveImage(ctx context.Context, imageID uuid.UUID, stackID uuid.UUID) error {
+func (m StackModel) SaveImage(ctx context.Context, imageID uuid.UUID, stackID uuid.UUID) (model.ImageStacks, error) {
 	saveImageStmt := ImageStacks.
 		INSERT(ImageStacks.ImageID, ImageStacks.StackID).
-		VALUES(imageID, stackID)
+		VALUES(imageID, stackID).
+		RETURNING(ImageStacks.AllColumns)
 
-	_, err := saveImageStmt.ExecContext(ctx, m.dbPool)
+	imageStack := model.ImageStacks{}
+
+	err := saveImageStmt.QueryContext(ctx, m.dbPool, &imageStack)
+
+	return imageStack, err
+}
+
+func (m StackModel) SaveSchemaItems(ctx context.Context, imageID uuid.UUID, items []IDValue) error {
+	if len(items) == 0 {
+		return fmt.Errorf("items cannot be empty")
+	}
+
+	saveSchemaItemStmt := ImageSchemaItems.
+		INSERT(
+			ImageSchemaItems.ImageID,
+			ImageSchemaItems.SchemaItemID,
+			ImageSchemaItems.Value,
+		)
+
+	for _, item := range items {
+		saveSchemaItemStmt = saveSchemaItemStmt.VALUES(
+			imageID,
+			item.ID,
+			item.Value,
+		)
+	}
+
+	_, err := saveSchemaItemStmt.ExecContext(ctx, m.dbPool)
 
 	return err
 }