pkg/models/orm.go

@@ -7,27 +7,26 @@ import (
 )
 
 type ID interface {
-	AnthroveUserID
+	AnthroveUserID | AnthroveSourceID | AnthrovePostID
-	AnthroveSourceID
-	AnthrovePostID
 }
 
-type BaseModel struct {
+type BaseModel[T ID] struct {
-	ID        string `gorm:"primaryKey"`
+	ID        T `gorm:"primaryKey"`
 	CreatedAt time.Time
 	UpdatedAt time.Time
 	DeletedAt gorm.DeletedAt `gorm:"index"`
 }
 
-func (base *BaseModel) BeforeCreate(db *gorm.DB) error {
+func (base *BaseModel[T]) BeforeCreate(db *gorm.DB) error {
+	var defaultVar T
 
-	if base.ID == "" {
+	if base.ID == defaultVar {
 		id, err := gonanoid.New(25)
 		if err != nil {
 			return err
 		}
 
-		base.ID = id
+		base.ID = T(id)
 	}
 
 	return nil