feat(locations): now working

backend/models/locations.go

@@ -3,27 +3,76 @@ package models
 import (
 	"context"
 	"database/sql"
+	"log"
 	"screenmark/screenmark/.gen/haystack/haystack/model"
 	. "screenmark/screenmark/.gen/haystack/haystack/table"
+
+	"github.com/google/uuid"
 )
 
 type LocationModel struct {
 	dbPool *sql.DB
 }
 
-func (m LocationModel) Save(ctx context.Context, location model.Locations) (model.Locations, error) {
+// This looks stupid
+func getValues(location model.Locations) []any {
+	arr := make([]any, 0)
+
+	if location.Address != nil {
+		arr = append(arr, *location.Address)
+	} else {
+		arr = append(arr, nil)
+	}
+
+	if location.Coordinates != nil {
+		arr = append(arr, *location.Coordinates)
+	} else {
+		arr = append(arr, nil)
+	}
+
+	if location.Description != nil {
+		arr = append(arr, *location.Description)
+	} else {
+		arr = append(arr, nil)
+	}
+
+	return arr
+}
+
+func (m LocationModel) Save(ctx context.Context, locations []model.Locations) (model.Locations, error) {
 	insertLocationStmt := Locations.
-		INSERT(Locations.Name, Locations.Address, Locations.Coordinates, Locations.Description).
-		VALUES(location.Name, location.Address, location.Coordinates, location.Description).
-		RETURNING(Locations.AllColumns)
+		INSERT(Locations.Name, Locations.Address, Locations.Coordinates, Locations.Description)
+
+	for _, location := range locations {
+		insertLocationStmt = insertLocationStmt.VALUES(location.Name, getValues(location)...)
+	}
+
+	insertLocationStmt = insertLocationStmt.RETURNING(Locations.AllColumns)
+
+	log.Println(insertLocationStmt.DebugSql())
 
 	insertedLocation := model.Locations{}
-
-	err := insertLocationStmt.QueryContext(ctx, m.dbPool, &insertLocationStmt)
+	err := insertLocationStmt.QueryContext(ctx, m.dbPool, &insertedLocation)
 
 	return insertedLocation, err
 }
 
+func (m LocationModel) SaveToImage(ctx context.Context, imageId uuid.UUID, locations []model.Locations) error {
+	location, err := m.Save(ctx, locations)
+
+	if err != nil {
+		return err
+	}
+
+	insertImageLocationStmt := ImageLocations.
+		INSERT(ImageLocations.ImageID, ImageLocations.LocationID).
+		VALUES(imageId, location.ID)
+
+	_, err = insertImageLocationStmt.ExecContext(ctx, m.dbPool)
+
+	return err
+}
+
 func NewLocationModel(db *sql.DB) LocationModel {
 	return LocationModel{dbPool: db}
 }

backend/models/tags.go

@@ -8,6 +8,7 @@ import (
 	. "screenmark/screenmark/.gen/haystack/haystack/table"
 
 	. "github.com/go-jet/jet/v2/postgres"
+
 	"github.com/google/uuid"
 )