refactor: use a dynamic model registration process so we can use build tags to exclude/include

db/db.go

@@ -4,13 +4,14 @@ import (
 	"context"
 	"fmt"
 
+	"git.lumeweb.com/LumeWeb/portal/db/models"
+
 	"github.com/redis/go-redis/v9"
 
 	"go.uber.org/zap"
 
 	"git.lumeweb.com/LumeWeb/portal/config"
 
-	"git.lumeweb.com/LumeWeb/portal/db/models"
 	"github.com/go-gorm/caches/v4"
 	"go.uber.org/fx"
 	"gorm.io/driver/mysql"
@@ -60,22 +61,7 @@ func NewDatabase(lc fx.Lifecycle, params DatabaseParams) *gorm.DB {
 
 	lc.Append(fx.Hook{
 		OnStart: func(ctx context.Context) error {
-			return db.AutoMigrate(
-				&models.APIKey{},
-				&models.Blocklist{},
-				&models.DNSLink{},
-				&models.Download{},
-				&models.EmailVerification{},
-				&models.PasswordReset{},
-				&models.Pin{},
-				&models.PublicKey{},
-				&models.Upload{},
-				&models.User{},
-				&models.S3Upload{},
-				&models.S5Challenge{},
-				&models.TusLock{},
-				&models.TusUpload{},
-			)
+			return db.AutoMigrate(models.GetModels()...)
 		},
 	})
 

db/models/api_key.go

@@ -2,6 +2,10 @@ package models
 
 import "gorm.io/gorm"
 
+func init() {
+	registerModel(&APIKey{})
+}
+
 type APIKey struct {
 	gorm.Model
 	UserID uint

db/models/blocklist.go

@@ -1,10 +1,15 @@
 package models
 
 import (
-	"gorm.io/gorm"
 	"time"
+
+	"gorm.io/gorm"
 )
 
+func init() {
+	registerModel(&Blocklist{})
+}
+
 type Blocklist struct {
 	gorm.Model
 	IP        string

db/models/dnslink.go

@@ -2,6 +2,10 @@ package models
 
 import "gorm.io/gorm"
 
+func init() {
+	registerModel(&DNSLink{})
+}
+
 type DNSLink struct {
 	gorm.Model
 	UserID   uint `gorm:"uniqueIndex:idx_user_id_upload"`

db/models/download.go

@@ -1,10 +1,15 @@
 package models
 
 import (
-	"gorm.io/gorm"
 	"time"
+
+	"gorm.io/gorm"
 )
 
+func init() {
+	registerModel(&Download{})
+}
+
 type Download struct {
 	gorm.Model
 	UserID       uint

db/models/email_verification.go

@@ -6,6 +6,10 @@ import (
 	"gorm.io/gorm"
 )
 
+func init() {
+	registerModel(&EmailVerification{})
+}
+
 type EmailVerification struct {
 	gorm.Model
 

db/models/models.go

@@ -0,0 +1,11 @@
+package models
+
+var registered []interface{}
+
+func registerModel(model interface{}) {
+	registered = append(registered, model)
+}
+
+func GetModels() []interface{} {
+	return registered
+}

db/models/password_reset.go

@@ -6,6 +6,10 @@ import (
 	"gorm.io/gorm"
 )
 
+func init() {
+	registerModel(&PasswordReset{})
+}
+
 type PasswordReset struct {
 	gorm.Model
 

db/models/pin.go

@@ -2,6 +2,10 @@ package models
 
 import "gorm.io/gorm"
 
+func init() {
+	registerModel(&Pin{})
+}
+
 type Pin struct {
 	gorm.Model
 	UploadID uint

db/models/public_key.go

@@ -2,6 +2,10 @@ package models
 
 import "gorm.io/gorm"
 
+func init() {
+	registerModel(&PublicKey{})
+}
+
 type PublicKey struct {
 	gorm.Model
 	UserID uint

db/models/s3_upload.go

@@ -2,6 +2,10 @@ package models
 
 import "gorm.io/gorm"
 
+func init() {
+	registerModel(&S3Upload{})
+}
+
 type S3Upload struct {
 	gorm.Model
 	UploadID string `gorm:"unique;not null"`

db/models/s5_challenge.go

@@ -2,6 +2,11 @@ package models
 
 import "gorm.io/gorm"
 
+func init() {
+	registerModel(&S5Challenge{})
+
+}
+
 type S5Challenge struct {
 	gorm.Model
 	Challenge string

db/models/tus_lock.go

@@ -3,11 +3,17 @@ package models
 import (
 	"context"
 	"errors"
+	"time"
+
 	"gorm.io/gorm"
 	"gorm.io/gorm/clause"
-	"time"
 )
 
+func init() {
+	registerModel(&TusLock{})
+
+}
+
 var (
 	ErrTusLockBusy = errors.New("lock is currently held by another process")
 )

db/models/tus_upload.go

@@ -2,6 +2,11 @@ package models
 
 import "gorm.io/gorm"
 
+func init() {
+	registerModel(&TusUpload{})
+
+}
+
 type TusUpload struct {
 	gorm.Model
 	Hash       []byte `gorm:"type:binary(32);uniqueIndex:idx_hash_deleted"`

db/models/upload.go

@@ -2,6 +2,10 @@ package models
 
 import "gorm.io/gorm"
 
+func init() {
+	registerModel(&Upload{})
+}
+
 type Upload struct {
 	gorm.Model
 	UserID     uint

db/models/user.go

@@ -8,6 +8,10 @@ import (
 	"gorm.io/gorm"
 )
 
+func init() {
+	registerModel(&User{})
+}
+
 type User struct {
 	gorm.Model
 	FirstName          string