User service for all user related functions

This commit is contained in:
Michael Bumann
2022-01-19 21:16:20 +01:00
parent 818dd78c6f
commit 7bd74a69d8
5 changed files with 79 additions and 68 deletions

View File

@@ -2,12 +2,9 @@ package service
import (
"context"
"database/sql"
"fmt"
"math/rand"
"github.com/getAlby/lndhub.go/db/models"
"github.com/getAlby/lndhub.go/lib/security"
"github.com/getAlby/lndhub.go/lib/tokens"
"github.com/labstack/gommon/random"
"github.com/lightningnetwork/lnd/lnrpc"
@@ -62,42 +59,3 @@ func (svc *LndhubService) GenerateToken(login, password, inRefreshToken string)
}
return accessToken, refreshToken, nil
}
func (svc *LndhubService) CreateUser() (user *models.User, err error) {
user = &models.User{}
// generate user login/password (TODO: allow the user to choose a login/password?)
user.Login = randStringBytes(8)
password := randStringBytes(15)
// we only store the hashed password but return the initial plain text password in the HTTP response
hashedPassword := security.HashPassword(password)
user.Password = hashedPassword
// Create user and the user's accounts
// We use double-entry bookkeeping so we use 4 accounts: incoming, current, outgoing and fees
// Wrapping this in a transaction in case something fails
err = svc.DB.RunInTx(context.TODO(), &sql.TxOptions{}, func(ctx context.Context, tx bun.Tx) error {
if _, err := tx.NewInsert().Model(user).Exec(ctx); err != nil {
return err
}
accountTypes := []string{"incoming", "current", "outgoing", "fees"}
for _, accountType := range accountTypes {
account := models.Account{UserID: user.ID, Type: accountType}
if _, err := tx.NewInsert().Model(&account).Exec(ctx); err != nil {
return err
}
}
return nil
})
//return actual password in the response, not the hashed one
user.Password = password
return user, err
}
func randStringBytes(n int) string {
b := make([]byte, n)
for i := range b {
b[i] = alphaNumBytes[rand.Intn(len(alphaNumBytes))]
}
return string(b)
}