70 lines
1.8 KiB
Go
70 lines
1.8 KiB
Go
package sql
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"time"
|
|
|
|
"github.com/authorizerdev/authorizer/server/db/models"
|
|
"github.com/google/uuid"
|
|
"gorm.io/gorm/clause"
|
|
)
|
|
|
|
// UpsertOTP to add or update otp
|
|
func (p *provider) UpsertOTP(ctx context.Context, otp *models.OTP) (*models.OTP, error) {
|
|
if otp.ID == "" {
|
|
otp.ID = uuid.New().String()
|
|
}
|
|
// check if email or phone number is present
|
|
if otp.Email == "" && otp.PhoneNumber == "" {
|
|
return nil, errors.New("email or phone_number is required")
|
|
}
|
|
uniqueField := models.FieldNameEmail
|
|
if otp.Email == "" && otp.PhoneNumber != "" {
|
|
uniqueField = models.FieldNamePhoneNumber
|
|
}
|
|
otp.Key = otp.ID
|
|
otp.CreatedAt = time.Now().Unix()
|
|
otp.UpdatedAt = time.Now().Unix()
|
|
res := p.db.Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: uniqueField}},
|
|
DoUpdates: clause.AssignmentColumns([]string{"otp", "expires_at", "updated_at"}),
|
|
}).Create(&otp)
|
|
if res.Error != nil {
|
|
return nil, res.Error
|
|
}
|
|
|
|
return otp, nil
|
|
}
|
|
|
|
// GetOTPByEmail to get otp for a given email address
|
|
func (p *provider) GetOTPByEmail(ctx context.Context, emailAddress string) (*models.OTP, error) {
|
|
var otp models.OTP
|
|
result := p.db.Where("email = ?", emailAddress).First(&otp)
|
|
if result.Error != nil {
|
|
return nil, result.Error
|
|
}
|
|
return &otp, nil
|
|
}
|
|
|
|
// GetOTPByPhoneNumber to get otp for a given phone number
|
|
func (p *provider) GetOTPByPhoneNumber(ctx context.Context, phoneNumber string) (*models.OTP, error) {
|
|
var otp models.OTP
|
|
result := p.db.Where("phone_number = ?", phoneNumber).First(&otp)
|
|
if result.Error != nil {
|
|
return nil, result.Error
|
|
}
|
|
return &otp, nil
|
|
}
|
|
|
|
// DeleteOTP to delete otp
|
|
func (p *provider) DeleteOTP(ctx context.Context, otp *models.OTP) error {
|
|
result := p.db.Delete(&models.OTP{
|
|
ID: otp.ID,
|
|
})
|
|
if result.Error != nil {
|
|
return result.Error
|
|
}
|
|
return nil
|
|
}
|