refactor: moving all files to backend

backend/models/database.go

@@ -0,0 +1,19 @@
+package models
+
+import (
+	"database/sql"
+
+	_ "github.com/lib/pq"
+)
+
+const CONNECTION = "postgresql://localhost:5432/haystack?sslmode=disable"
+
+var db *sql.DB
+
+func InitDatabase() error {
+	database, err := sql.Open("postgres", CONNECTION)
+
+	db = database
+
+	return err
+}

backend/models/image.go

@@ -0,0 +1,33 @@
+package models
+
+import (
+	"errors"
+	"fmt"
+
+	. "github.com/go-jet/jet/v2/postgres"
+	"screenmark/screenmark/.gen/haystack/haystack/model"
+	. "screenmark/screenmark/.gen/haystack/haystack/table"
+
+	"github.com/google/uuid"
+)
+
+func SaveImage(userId string, imageName string, imageData []byte) error {
+	stmt := UserImages.INSERT(UserImages.UserID, UserImages.ImageName, UserImages.Image).VALUES(userId, imageName, imageData)
+	_, err := stmt.Exec(db)
+
+	return err
+}
+
+func GetImage(imageId string) (model.UserImages, error) {
+	id := uuid.MustParse(imageId)
+	stmt := UserImages.SELECT(UserImages.ImageName, UserImages.Image).WHERE(UserImages.ID.EQ(UUID(id)))
+
+	images := []model.UserImages{}
+	err := stmt.Query(db, &images)
+
+	if len(images) != 1 {
+		return model.UserImages{}, errors.New(fmt.Sprintf("Expected 1, got %d\n", len(images)))
+	}
+
+	return images[0], err
+}