feat: implement EmailExists, PubkeyExists, CreateAccount,AddPubkeyToAccount, LoginPassword, LoginPubkey

This commit is contained in:
Derrick Hammer 2024-01-16 13:30:36 -05:00
parent 1bbedecad9
commit 9ad8d70f09
Signed by: pcfreak30
GPG Key ID: C997C339BE476FF2
3 changed files with 126 additions and 1 deletions

View File

@ -1,6 +1,10 @@
package account package account
import "git.lumeweb.com/LumeWeb/portal/interfaces" import (
"git.lumeweb.com/LumeWeb/portal/db/models"
"git.lumeweb.com/LumeWeb/portal/interfaces"
"golang.org/x/crypto/bcrypt"
)
var ( var (
_ interfaces.AccountService = (*AccountServiceImpl)(nil) _ interfaces.AccountService = (*AccountServiceImpl)(nil)
@ -13,3 +17,89 @@ type AccountServiceImpl struct {
func NewAccountService(portal interfaces.Portal) interfaces.AccountService { func NewAccountService(portal interfaces.Portal) interfaces.AccountService {
return &AccountServiceImpl{portal: portal} 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
}

27
account/jwt.go Normal file
View File

@ -0,0 +1,27 @@
package account
import (
"crypto/ed25519"
"github.com/golang-jwt/jwt/v5"
"time"
)
func generateToken(privateKey ed25519.PrivateKey, userID uint) (string, error) {
// Define the claims
claims := jwt.MapClaims{
"iss": "portal",
"sub": userID,
"exp": time.Now().Add(time.Hour * 24).Unix(),
}
// Create the token
token := jwt.NewWithClaims(jwt.SigningMethodEdDSA, claims)
// Sign the token with the Ed25519 private key
tokenString, err := token.SignedString(privateKey)
if err != nil {
return "", err
}
return tokenString, nil
}

View File

@ -1,4 +1,12 @@
package interfaces package interfaces
import "git.lumeweb.com/LumeWeb/portal/db/models"
type AccountService interface { type AccountService interface {
EmailExists(email string) (bool, models.User)
PubkeyExists(pubkey string) (bool, models.PublicKey)
CreateAccount(email string, password string) (*models.User, error)
AddPubkeyToAccount(user models.User, pubkey string) error
LoginPassword(email string, password string) (string, error)
LoginPubkey(pubkey string) (string, error)
} }