mirror of
https://github.com/thomiceli/opengist.git
synced 2024-12-23 13:02:39 +00:00
99 lines
2.2 KiB
Go
99 lines
2.2 KiB
Go
package db
|
|
|
|
import (
|
|
"fmt"
|
|
"math/rand"
|
|
"time"
|
|
)
|
|
|
|
type Invitation struct {
|
|
ID uint `gorm:"primaryKey"`
|
|
Code string
|
|
ExpiresAt int64
|
|
NbUsed uint
|
|
NbMax uint
|
|
}
|
|
|
|
func GetAllInvitations() ([]*Invitation, error) {
|
|
var invitations []*Invitation
|
|
dialect := db.Dialector.Name()
|
|
query := db.Model(&Invitation{})
|
|
|
|
switch dialect {
|
|
case "sqlite":
|
|
query = query.Order("(((expires_at >= strftime('%s', 'now')) AND ((nb_max <= 0) OR (nb_used < nb_max)))) DESC")
|
|
case "postgres":
|
|
query = query.Order("(((expires_at >= EXTRACT(EPOCH FROM CURRENT_TIMESTAMP)) AND ((nb_max <= 0) OR (nb_used < nb_max)))) DESC")
|
|
case "mysql":
|
|
query = query.Order("(((expires_at >= UNIX_TIMESTAMP()) AND ((nb_max <= 0) OR (nb_used < nb_max)))) DESC")
|
|
default:
|
|
return nil, fmt.Errorf("unsupported database dialect: %s", dialect)
|
|
}
|
|
|
|
err := query.Order("id ASC").Find(&invitations).Error
|
|
|
|
return invitations, err
|
|
}
|
|
|
|
func GetInvitationByID(id uint) (*Invitation, error) {
|
|
invitation := new(Invitation)
|
|
err := db.
|
|
Where("id = ?", id).
|
|
First(&invitation).Error
|
|
return invitation, err
|
|
}
|
|
|
|
func GetInvitationByCode(code string) (*Invitation, error) {
|
|
invitation := new(Invitation)
|
|
err := db.
|
|
Where("code = ?", code).
|
|
First(&invitation).Error
|
|
return invitation, err
|
|
}
|
|
|
|
func InvitationCodeExists(code string) (bool, error) {
|
|
var count int64
|
|
err := db.Model(&Invitation{}).Where("code = ?", code).Count(&count).Error
|
|
return count > 0, err
|
|
}
|
|
|
|
func (i *Invitation) Create() error {
|
|
i.Code = generateRandomCode()
|
|
return db.Create(&i).Error
|
|
}
|
|
|
|
func (i *Invitation) Update() error {
|
|
return db.Save(&i).Error
|
|
}
|
|
|
|
func (i *Invitation) Delete() error {
|
|
return db.Delete(&i).Error
|
|
}
|
|
|
|
func (i *Invitation) IsExpired() bool {
|
|
return i.ExpiresAt < time.Now().Unix()
|
|
}
|
|
|
|
func (i *Invitation) IsMaxedOut() bool {
|
|
return i.NbMax > 0 && i.NbUsed >= i.NbMax
|
|
}
|
|
|
|
func (i *Invitation) IsUsable() bool {
|
|
return !i.IsExpired() && !i.IsMaxedOut()
|
|
}
|
|
|
|
func (i *Invitation) Use() error {
|
|
i.NbUsed++
|
|
return i.Update()
|
|
}
|
|
|
|
func generateRandomCode() string {
|
|
const charset = "0123456789ABCDEF"
|
|
var seededRand = rand.New(rand.NewSource(time.Now().UnixNano()))
|
|
result := make([]byte, 16)
|
|
|
|
for i := range result {
|
|
result[i] = charset[seededRand.Intn(len(charset))]
|
|
}
|
|
return string(result)
|
|
}
|