Add GORM models and database initialization

Author: dustywusty

go.mod

@@ -3,6 +3,9 @@ module tinychess
 go 1.22.2
 
 require (
-	github.com/google/uuid v1.5.0 // indirect
-	github.com/notnil/chess v1.10.0 // indirect
+    gorm.io/driver/postgres v1.5.4
+    gorm.io/gorm v1.25.5
+    github.com/google/uuid v1.5.0
+    github.com/notnil/chess v1.10.0
 )
+

internal/storage/db.go

@@ -0,0 +1,18 @@
+package storage
+
+import (
+	"gorm.io/driver/postgres"
+	"gorm.io/gorm"
+)
+
+// New initializes the database connection and performs migrations.
+func New(dsn string) (*gorm.DB, error) {
+	db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
+	if err != nil {
+		return nil, err
+	}
+	if err := db.AutoMigrate(&Game{}, &GameSession{}, &UserSession{}, &Move{}); err != nil {
+		return nil, err
+	}
+	return db, nil
+}

internal/storage/models.go

@@ -0,0 +1,48 @@
+package storage
+
+import (
+	"time"
+
+	"github.com/google/uuid"
+)
+
+// Game represents a chess game.
+type Game struct {
+	ID        uuid.UUID `gorm:"type:uuid;default:gen_random_uuid();primaryKey"`
+	FEN       string
+	PGN       string
+	CreatedAt time.Time
+	UpdatedAt time.Time
+	Sessions  []GameSession
+	Moves     []Move
+}
+
+// GameSession represents an instance of a game session.
+type GameSession struct {
+	ID        uuid.UUID `gorm:"type:uuid;default:gen_random_uuid();primaryKey"`
+	GameID    uuid.UUID `gorm:"type:uuid;index"`
+	Game      Game      `gorm:"constraint:OnDelete:CASCADE;"`
+	CreatedAt time.Time
+	UpdatedAt time.Time
+	Users     []UserSession
+}
+
+// UserSession links a user to a game session.
+type UserSession struct {
+	ID            uuid.UUID `gorm:"type:uuid;default:gen_random_uuid();primaryKey"`
+	GameID        uuid.UUID `gorm:"type:uuid;index"`
+	GameSessionID uuid.UUID `gorm:"type:uuid;index"`
+	Color         string
+	Role          string
+	CreatedAt     time.Time
+	UpdatedAt     time.Time
+}
+
+// Move stores a single move in a game.
+type Move struct {
+	ID        uuid.UUID `gorm:"type:uuid;default:gen_random_uuid();primaryKey"`
+	GameID    uuid.UUID `gorm:"type:uuid;index"`
+	Number    int
+	UCI       string
+	CreatedAt time.Time
+}

main.go

@@ -4,10 +4,12 @@ import (
 	"flag"
 	"log"
 	"net/http"
+	"os"
 
 	"tinychess/internal/game"
 	"tinychess/internal/handlers"
 	"tinychess/internal/logging"
+	"tinychess/internal/storage"
 	"tinychess/internal/templates"
 )
 
@@ -18,6 +20,12 @@ func main() {
 
 	templates.SetCommit(commit)
 
+	if dsn := os.Getenv("DATABASE_URL"); dsn != "" {
+		if _, err := storage.New(dsn); err != nil {
+			log.Fatalf("failed to initialize database: %v", err)
+		}
+	}
+
 	// Initialize game hub
 	hub := game.NewHub()