feat: initial draft of generating a schema from one image

fix: error formatting

backend/models/lists.go

@@ -3,10 +3,12 @@ package models
 import (
 	"context"
 	"database/sql"
-	. "github.com/go-jet/jet/v2/postgres"
+	"fmt"
 	"screenmark/screenmark/.gen/haystack/haystack/model"
 	. "screenmark/screenmark/.gen/haystack/haystack/table"
 
+	. "github.com/go-jet/jet/v2/postgres"
+
 	"github.com/google/uuid"
 )
 
@@ -14,15 +16,74 @@ type ListModel struct {
 	dbPool *sql.DB
 }
 
-func (m ListModel) Save(ctx context.Context, userId uuid.UUID, name string, description string) (model.Lists, error) {
+type ListWithItems struct {
+	model.Lists
+
+	Schema struct {
+		model.Schemas
+
+		SchemaItems []model.SchemaItems
+	}
+}
+
+func (m ListModel) Save(ctx context.Context, userId uuid.UUID, name string, description string, schemaItems []model.SchemaItems) (ListWithItems, error) {
+	tx, err := m.dbPool.BeginTx(ctx, nil)
+
 	stmt := Lists.INSERT(Lists.UserID, Lists.Name, Lists.Description).
 		VALUES(userId, name, description).
 		RETURNING(Lists.ID, Lists.Name, Lists.Description)
 
 	newList := model.Lists{}
-	err := stmt.QueryContext(ctx, m.dbPool, &newList)
+	err = stmt.QueryContext(ctx, tx, &newList)
 
-	return newList, err
+	if err != nil {
+		tx.Rollback()
+		return ListWithItems{}, fmt.Errorf("Could not save new list. %s", err)
+	}
+
+	insertSchemaStmt := Schemas.INSERT(Schemas.ListID).
+		VALUES(newList.ID).
+		RETURNING(Schemas.ID)
+
+	newSchema := model.Schemas{}
+	err = insertSchemaStmt.QueryContext(ctx, tx, &newSchema)
+
+	if err != nil {
+		tx.Rollback()
+		return ListWithItems{}, fmt.Errorf("Could not save new schema. %s", err)
+	}
+
+	// This is very interesting...
+	for i := range schemaItems {
+		schemaItems[i].SchemaID = newSchema.ID
+	}
+
+	insertSchemaItemsStmt := SchemaItems.INSERT(SchemaItems.Item, SchemaItems.Value, SchemaItems.Description, SchemaItems.SchemaID).
+		MODELS(schemaItems)
+	_, err = insertSchemaItemsStmt.ExecContext(ctx, tx)
+
+	if err != nil {
+		tx.Rollback()
+		return ListWithItems{}, fmt.Errorf("Could not save schema items. %s", err)
+	}
+
+	err = tx.Commit()
+	if err != nil {
+		return ListWithItems{}, fmt.Errorf("Could not commit transaction. %s", err)
+	}
+
+	getListAndItems := SELECT(Lists.AllColumns, Schemas.AllColumns, SchemaItems.AllColumns).
+		FROM(
+			Lists.
+				INNER_JOIN(Schemas, Schemas.ListID.EQ(Lists.ID)).
+				INNER_JOIN(SchemaItems, SchemaItems.SchemaID.EQ(Schemas.ID)),
+		).
+		WHERE(Lists.ID.EQ(UUID(newList.ID)))
+
+	listWithItems := ListWithItems{}
+	err = getListAndItems.QueryContext(ctx, m.dbPool, &listWithItems)
+
+	return listWithItems, err
 }
 
 func (m ListModel) List(ctx context.Context, userId uuid.UUID) ([]model.Lists, error) {
@@ -35,12 +96,50 @@ func (m ListModel) List(ctx context.Context, userId uuid.UUID) ([]model.Lists, e
 	return lists, err
 }
 
-func (m ListModel) SaveInto(ctx context.Context, listId uuid.UUID, imageId uuid.UUID) error {
+type IDValue struct {
+	ID    string `json:"id"`
+	Value string `json:"value"`
+}
+
+func (m ListModel) SaveInto(ctx context.Context, listId uuid.UUID, imageId uuid.UUID, schemaValues []IDValue) error {
+	imageSchemaItems := make([]model.ImageSchemaItems, len(schemaValues))
+
+	for i, v := range schemaValues {
+		parsedId, err := uuid.Parse(v.ID)
+		if err != nil {
+			return err
+		}
+
+		imageSchemaItems[i].SchemaItemID = parsedId
+		imageSchemaItems[i].ImageID = imageId
+		imageSchemaItems[i].Value = &v.Value
+	}
+
+	tx, err := m.dbPool.BeginTx(ctx, nil)
+	if err != nil {
+		return err
+	}
+
 	stmt := ImageLists.INSERT(ImageLists.ListID, ImageLists.ImageID).
 		VALUES(listId, imageId)
 
-	_, err := stmt.ExecContext(ctx, m.dbPool)
+	_, err = stmt.ExecContext(ctx, tx)
+	if err != nil {
+		tx.Rollback()
+		return fmt.Errorf("Could not insert new list. %s", err)
+	}
 
+	insertSchemaItemsStmt := ImageSchemaItems.
+		INSERT(ImageSchemaItems.Value, ImageSchemaItems.SchemaItemID, ImageSchemaItems.ImageID).
+		MODELS(imageSchemaItems)
+
+	_, err = insertSchemaItemsStmt.ExecContext(ctx, tx)
+	if err != nil {
+		tx.Rollback()
+		return fmt.Errorf("Could not insert schema items. %s", err)
+	}
+
+	err = tx.Commit()
 	return err
 }