79 lines
1.7 KiB
Go
79 lines
1.7 KiB
Go
package models
|
|
|
|
import (
|
|
"GoMembership/internal/utils"
|
|
"GoMembership/pkg/errors"
|
|
"GoMembership/pkg/logger"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type Verification struct {
|
|
gorm.Model
|
|
VerifiedAt *time.Time `json:"verified_at"`
|
|
VerificationToken string `json:"token"`
|
|
UserID uint `json:"user_id"`
|
|
Type string
|
|
}
|
|
|
|
func (v *Verification) Create(db *gorm.DB) error {
|
|
if err := db.Create(v).Error; err != nil {
|
|
return err
|
|
}
|
|
logger.Info.Printf("verification created: %#v", v)
|
|
// Preload all associations to return the fully populated object
|
|
return db.First(v, v.ID).Error // Refresh the verification object with all associations
|
|
}
|
|
|
|
func (v *Verification) Update(db *gorm.DB) error {
|
|
|
|
return db.Transaction(func(tx *gorm.DB) error {
|
|
// Check if the user exists in the database
|
|
var existingVerification Verification
|
|
|
|
logger.Info.Printf("updating verification: %#v", v)
|
|
if err := tx.First(&existingVerification, v.ID).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := tx.Model(&existingVerification).Updates(v).Error; err != nil {
|
|
return err
|
|
}
|
|
return tx.First(v, v.ID).Error
|
|
})
|
|
|
|
}
|
|
|
|
func (v *Verification) Delete(db *gorm.DB) error {
|
|
return db.Delete(&v).Error
|
|
}
|
|
|
|
func (v *Verification) Validate() error {
|
|
if v.VerifiedAt != nil {
|
|
return errors.ErrAlreadyVerified
|
|
}
|
|
|
|
t := time.Now()
|
|
v.VerifiedAt = &t
|
|
return nil
|
|
}
|
|
|
|
func CreateVerification(verificationType string) (*Verification, error) {
|
|
token, err := GenerateVerificationToken()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
v := Verification{
|
|
UserID: 0,
|
|
VerificationToken: token,
|
|
Type: verificationType,
|
|
}
|
|
|
|
return &v, nil
|
|
}
|
|
|
|
func GenerateVerificationToken() (string, error) {
|
|
return utils.GenerateRandomString(32)
|
|
}
|