mirror of
https://github.com/openziti/zrok.git
synced 2025-02-16 10:19:18 +01:00
more account creation wiring
This commit is contained in:
parent
629d2fe0dc
commit
1d0218ad91
@ -3,6 +3,7 @@ package main
|
|||||||
import (
|
import (
|
||||||
"github.com/openziti-test-kitchen/zrok/rest_model"
|
"github.com/openziti-test-kitchen/zrok/rest_model"
|
||||||
"github.com/openziti-test-kitchen/zrok/rest_zrok_client/identity"
|
"github.com/openziti-test-kitchen/zrok/rest_zrok_client/identity"
|
||||||
|
"github.com/openziti/foundation/v2/term"
|
||||||
"github.com/sirupsen/logrus"
|
"github.com/sirupsen/logrus"
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
)
|
)
|
||||||
@ -18,18 +19,36 @@ var createCmd = &cobra.Command{
|
|||||||
}
|
}
|
||||||
|
|
||||||
var createAccountCmd = &cobra.Command{
|
var createAccountCmd = &cobra.Command{
|
||||||
Use: "account <username>",
|
Use: "account",
|
||||||
Short: "create new zrok account",
|
Short: "create new zrok account",
|
||||||
Run: func(_ *cobra.Command, args []string) {
|
Run: func(_ *cobra.Command, _ []string) {
|
||||||
|
username, err := term.Prompt("New Username: ")
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
password, err := term.PromptPassword("New Password: ", false)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
confirm, err := term.PromptPassword("Confirm Password: ", false)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
if confirm != password {
|
||||||
|
panic("confirmed password mismatch")
|
||||||
|
}
|
||||||
|
|
||||||
zrok := newZrokClient()
|
zrok := newZrokClient()
|
||||||
req := identity.NewCreateAccountParams()
|
req := identity.NewCreateAccountParams()
|
||||||
req.Body = &rest_model.AccountRequest{
|
req.Body = &rest_model.AccountRequest{
|
||||||
Username: args[0],
|
Username: username,
|
||||||
|
Password: password,
|
||||||
}
|
}
|
||||||
resp, err := zrok.Identity.CreateAccount(req)
|
resp, err := zrok.Identity.CreateAccount(req)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic(err)
|
panic(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
logrus.Infof("api token = '%v'", resp.Payload.APIToken)
|
logrus.Infof("api token = '%v'", resp.Payload.APIToken)
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
@ -1,6 +1,8 @@
|
|||||||
package controller
|
package controller
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"crypto/sha512"
|
||||||
|
"encoding/hex"
|
||||||
"github.com/go-openapi/loads"
|
"github.com/go-openapi/loads"
|
||||||
"github.com/go-openapi/runtime/middleware"
|
"github.com/go-openapi/runtime/middleware"
|
||||||
"github.com/openziti-test-kitchen/zrok/controller/store"
|
"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 {
|
func createAccountHandler(params identity.CreateAccountParams) middleware.Responder {
|
||||||
logrus.Infof("received account request for username '%v'", params.Body.Username)
|
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 {
|
if err != nil {
|
||||||
return middleware.Error(500, err.Error())
|
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{
|
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))
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user