more account creation wiring

This commit is contained in:
Michael Quigley
2022-07-25 16:17:03 -04:00
parent 629d2fe0dc
commit 1d0218ad91
2 changed files with 56 additions and 5 deletions

View File

@ -1,6 +1,8 @@
package controller
import (
"crypto/sha512"
"encoding/hex"
"github.com/go-openapi/loads"
"github.com/go-openapi/runtime/middleware"
"github.com/openziti-test-kitchen/zrok/controller/store"
@ -48,11 +50,41 @@ func versionHandler(_ metadata.VersionParams) middleware.Responder {
func createAccountHandler(params identity.CreateAccountParams) middleware.Responder {
logrus.Infof("received account request for username '%v'", params.Body.Username)
apiToken, err := generateApiToken()
if params.Body == nil || params.Body.Username == "" || params.Body.Password == "" {
return middleware.Error(500, errors.Errorf("invalid username or password"))
}
token, err := generateApiToken()
if err != nil {
return middleware.Error(500, err.Error())
}
a := &store.Account{
Username: params.Body.Username,
Password: hashPassword(params.Body.Password),
Token: token,
}
tx, err := str.Begin()
if err != nil {
return middleware.Error(500, err.Error())
}
id, err := str.CreateAccount(a, tx)
if err != nil {
_ = tx.Rollback()
return middleware.Error(500, err.Error())
}
if err := tx.Commit(); err != nil {
logrus.Errorf("error comitting: %v", err)
}
logrus.Infof("account created with id = '%v'", id)
return identity.NewCreateAccountCreated().WithPayload(&rest_model.AccountResponse{
APIToken: apiToken,
APIToken: token,
})
}
func hashPassword(raw string) string {
hash := sha512.New()
hash.Write([]byte(raw))
return hex.EncodeToString(hash.Sum(nil))
}