106 lines
2.5 KiB
Go
106 lines
2.5 KiB
Go
package account
|
|
|
|
import (
|
|
"git.lumeweb.com/LumeWeb/portal/db/models"
|
|
"git.lumeweb.com/LumeWeb/portal/interfaces"
|
|
"golang.org/x/crypto/bcrypt"
|
|
)
|
|
|
|
var (
|
|
_ interfaces.AccountService = (*AccountServiceImpl)(nil)
|
|
)
|
|
|
|
type AccountServiceImpl struct {
|
|
portal interfaces.Portal
|
|
}
|
|
|
|
func NewAccountService(portal interfaces.Portal) interfaces.AccountService {
|
|
return &AccountServiceImpl{portal: portal}
|
|
}
|
|
|
|
func (s AccountServiceImpl) EmailExists(email string) (bool, models.User) {
|
|
var user models.User
|
|
|
|
result := s.portal.Database().Model(&models.Upload{}).Where(&models.User{Email: email}).First(&user)
|
|
|
|
return result.RowsAffected > 0, user
|
|
}
|
|
func (s AccountServiceImpl) PubkeyExists(pubkey string) (bool, models.PublicKey) {
|
|
var model models.PublicKey
|
|
|
|
result := s.portal.Database().Model(&models.PublicKey{}).Where(&models.PublicKey{Key: pubkey}).First(&model)
|
|
|
|
return result.RowsAffected > 0, model
|
|
}
|
|
func (s AccountServiceImpl) CreateAccount(email string, password string) (*models.User, error) {
|
|
var user models.User
|
|
|
|
bytes, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
user.Email = email
|
|
user.PasswordHash = string(bytes)
|
|
|
|
result := s.portal.Database().Create(&user)
|
|
|
|
if result.Error != nil {
|
|
return nil, result.Error
|
|
}
|
|
|
|
return &user, nil
|
|
}
|
|
func (s AccountServiceImpl) AddPubkeyToAccount(user models.User, pubkey string) error {
|
|
var model models.PublicKey
|
|
|
|
model.Key = pubkey
|
|
model.UserID = user.ID
|
|
|
|
result := s.portal.Database().Create(&model)
|
|
|
|
if result.Error != nil {
|
|
return result.Error
|
|
}
|
|
|
|
return nil
|
|
}
|
|
func (s AccountServiceImpl) LoginPassword(email string, password string) (string, error) {
|
|
var user models.User
|
|
|
|
result := s.portal.Database().Model(&models.User{}).Where(&models.User{Email: email}).First(&user)
|
|
|
|
if result.RowsAffected == 0 || result.Error != nil {
|
|
return "", result.Error
|
|
}
|
|
|
|
err := bcrypt.CompareHashAndPassword([]byte(user.PasswordHash), []byte(password))
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
token, err := generateToken(s.portal.Identity(), user.ID)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return token, nil
|
|
}
|
|
|
|
func (s AccountServiceImpl) LoginPubkey(pubkey string) (string, error) {
|
|
var model models.PublicKey
|
|
|
|
result := s.portal.Database().Model(&models.PublicKey{}).Where(&models.PublicKey{Key: pubkey}).First(&model)
|
|
|
|
if result.RowsAffected == 0 || result.Error != nil {
|
|
return "", result.Error
|
|
}
|
|
|
|
token, err := generateToken(s.portal.Identity(), model.UserID)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return token, nil
|
|
}
|