prompt 1,2,3

internal/db/db.go

@@ -0,0 +1,57 @@
+package db
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+	"time"
+
+	"trustcontact/internal/config"
+
+	"gorm.io/driver/postgres"
+	"gorm.io/driver/sqlite"
+	"gorm.io/gorm"
+	gormlogger "gorm.io/gorm/logger"
+)
+
+func Open(cfg *config.Config) (*gorm.DB, error) {
+	if cfg == nil {
+		return nil, fmt.Errorf("config is nil")
+	}
+
+	gormCfg := &gorm.Config{Logger: newLogger(cfg.Env)}
+
+	switch cfg.DBDriver {
+	case config.DBDriverSQLite:
+		if err := ensureSQLiteDir(cfg.SQLitePath); err != nil {
+			return nil, fmt.Errorf("prepare sqlite dir: %w", err)
+		}
+		return gorm.Open(sqlite.Open(cfg.SQLitePath), gormCfg)
+	case config.DBDriverPostgres:
+		return gorm.Open(postgres.Open(cfg.PostgresDSN), gormCfg)
+	default:
+		return nil, fmt.Errorf("unsupported db driver: %s", cfg.DBDriver)
+	}
+}
+
+func newLogger(env string) gormlogger.Interface {
+	level := gormlogger.Warn
+	if env == config.EnvDevelop {
+		level = gormlogger.Info
+	}
+
+	return gormlogger.Default.LogMode(level)
+}
+
+func ensureSQLiteDir(sqlitePath string) error {
+	dir := filepath.Dir(sqlitePath)
+	if dir == "." || dir == "" {
+		return nil
+	}
+
+	return os.MkdirAll(dir, 0o755)
+}
+
+func nowUTC() time.Time {
+	return time.Now().UTC()
+}

internal/db/migrate.go

@@ -0,0 +1,15 @@
+package db
+
+import (
+	"trustcontact/internal/models"
+
+	"gorm.io/gorm"
+)
+
+func Migrate(database *gorm.DB) error {
+	return database.AutoMigrate(
+		&models.User{},
+		&models.EmailVerificationToken{},
+		&models.PasswordResetToken{},
+	)
+}

internal/db/seed.go

@@ -0,0 +1,76 @@
+package db
+
+import (
+	"fmt"
+
+	"trustcontact/internal/auth"
+	"trustcontact/internal/models"
+
+	"gorm.io/gorm"
+	"gorm.io/gorm/clause"
+)
+
+func Seed(database *gorm.DB) error {
+	passwordHash, err := auth.HashPassword("password")
+	if err != nil {
+		return fmt.Errorf("hash seed password: %w", err)
+	}
+
+	seedUsers := []models.User{
+		{
+			Email:         "admin@example.com",
+			Role:          models.RoleAdmin,
+			EmailVerified: true,
+			PasswordHash:  passwordHash,
+		},
+		{
+			Email:         "user@example.com",
+			Role:          models.RoleUser,
+			EmailVerified: true,
+			PasswordHash:  passwordHash,
+		},
+		{
+			Email:         "demo1@example.com",
+			Role:          models.RoleUser,
+			EmailVerified: true,
+			PasswordHash:  passwordHash,
+		},
+		{
+			Email:         "demo2@example.com",
+			Role:          models.RoleUser,
+			EmailVerified: true,
+			PasswordHash:  passwordHash,
+		},
+		{
+			Email:         "demo3@example.com",
+			Role:          models.RoleUser,
+			EmailVerified: true,
+			PasswordHash:  passwordHash,
+		},
+	}
+
+	for _, user := range seedUsers {
+		if err := upsertUser(database, user); err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+func upsertUser(database *gorm.DB, user models.User) error {
+	result := database.Clauses(clause.OnConflict{
+		Columns: []clause.Column{{Name: "email"}},
+		DoUpdates: clause.AssignmentColumns([]string{
+			"role",
+			"email_verified",
+			"password_hash",
+			"updated_at",
+		}),
+	}).Create(&user)
+	if result.Error != nil {
+		return fmt.Errorf("seed user %s: %w", user.Email, result.Error)
+	}
+
+	return nil
+}