114 lines
2.6 KiB
Go
114 lines
2.6 KiB
Go
package services
|
|
|
|
import (
|
|
"code.smartsheep.studio/hydrogen/passport/pkg/database"
|
|
"code.smartsheep.studio/hydrogen/passport/pkg/models"
|
|
"code.smartsheep.studio/hydrogen/passport/pkg/security"
|
|
"fmt"
|
|
"github.com/google/uuid"
|
|
"github.com/samber/lo"
|
|
"gorm.io/gorm"
|
|
"time"
|
|
)
|
|
|
|
func GetAccount(id uint) (models.Account, error) {
|
|
var account models.Account
|
|
if err := database.C.Where(models.Account{
|
|
BaseModel: models.BaseModel{ID: id},
|
|
}).First(&account).Error; err != nil {
|
|
return account, err
|
|
}
|
|
|
|
return account, nil
|
|
}
|
|
|
|
func LookupAccount(id string) (models.Account, error) {
|
|
var account models.Account
|
|
if err := database.C.Where(models.Account{Name: id}).First(&account).Error; err == nil {
|
|
return account, nil
|
|
}
|
|
|
|
var contact models.AccountContact
|
|
if err := database.C.Where(models.AccountContact{Content: id}).First(&contact).Error; err == nil {
|
|
if err := database.C.
|
|
Where(models.Account{
|
|
BaseModel: models.BaseModel{ID: contact.AccountID},
|
|
}).First(&account).Error; err == nil {
|
|
return account, err
|
|
}
|
|
}
|
|
|
|
return account, fmt.Errorf("account was not found")
|
|
}
|
|
|
|
func CreateAccount(name, nick, email, password string) (models.Account, error) {
|
|
user := models.Account{
|
|
Name: name,
|
|
Nick: nick,
|
|
State: models.PendingAccountState,
|
|
Profile: models.AccountProfile{
|
|
Experience: 100,
|
|
},
|
|
Factors: []models.AuthFactor{
|
|
{
|
|
Type: models.PasswordAuthFactor,
|
|
Secret: security.HashPassword(password),
|
|
},
|
|
{
|
|
Type: models.EmailPasswordFactor,
|
|
Secret: uuid.NewString()[:8],
|
|
},
|
|
},
|
|
Contacts: []models.AccountContact{
|
|
{
|
|
Type: models.EmailAccountContact,
|
|
Content: email,
|
|
IsPrimary: true,
|
|
VerifiedAt: nil,
|
|
},
|
|
},
|
|
PowerLevel: 0,
|
|
ConfirmedAt: nil,
|
|
}
|
|
|
|
if err := database.C.Create(&user).Error; err != nil {
|
|
return user, err
|
|
}
|
|
|
|
if tk, err := NewMagicToken(models.ConfirmMagicToken, &user, nil); err != nil {
|
|
return user, err
|
|
} else if err := NotifyMagicToken(tk); err != nil {
|
|
return user, err
|
|
}
|
|
|
|
return user, nil
|
|
}
|
|
|
|
func ConfirmAccount(code string) error {
|
|
token, err := ValidateMagicToken(code, models.ConfirmMagicToken)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var user models.Account
|
|
if err := database.C.Where(&models.Account{
|
|
BaseModel: models.BaseModel{ID: *token.AssignTo},
|
|
}).First(&user).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
return database.C.Transaction(func(tx *gorm.DB) error {
|
|
user.ConfirmedAt = lo.ToPtr(time.Now())
|
|
user.PowerLevel += 5
|
|
|
|
if err := database.C.Delete(&token).Error; err != nil {
|
|
return err
|
|
}
|
|
if err := database.C.Save(&user).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
})
|
|
}
|