extern/zrok
1
1
Fork 0
mirror of https://github.com/openziti/zrok.git synced 2025-08-19 04:06:18 +02:00
Code Issues Packages Projects Releases Wiki Activity

skeleton bits for zrok enable

Browse Source
This commit is contained in:
Michael Quigley
2022-07-25 17:05:44 -04:00
parent 17c75b212d
commit bb57075059
15 changed files with 793 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
controller/controller.go
View File

@@ -1,8 +1,6 @@
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"
@@ -12,7 +10,6 @@ import (
"github.com/openziti-test-kitchen/zrok/rest_zrok_server/operations/identity"
"github.com/openziti-test-kitchen/zrok/rest_zrok_server/operations/metadata"
"github.com/pkg/errors"
"github.com/sirupsen/logrus"
)
var str *store.Store
@@ -32,6 +29,7 @@ func Run(cfg *Config) error {
api := operations.NewZrokAPI(swaggerSpec)
api.MetadataVersionHandler = metadata.VersionHandlerFunc(versionHandler)
api.IdentityCreateAccountHandler = identity.CreateAccountHandlerFunc(createAccountHandler)
api.IdentityEnableHandler = identity.EnableHandlerFunc(enableHandler)
server := rest_zrok_server.NewServer(api)
defer func() { _ = server.Shutdown() }()
@@ -47,47 +45,3 @@ func Run(cfg *Config) error {
func versionHandler(_ metadata.VersionParams) middleware.Responder {
return metadata.NewGetOK().WithPayload(&rest_model.Version{Version: "v0.0.0; sk3tch"})
}
func createAccountHandler(params identity.CreateAccountParams) middleware.Responder {
logrus.Infof("received account request for username '%v'", params.Body.Username)
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 {
logrus.Errorf("error generating api token: %v", err)
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 {
logrus.Errorf("error starting transaction: %v", err)
return middleware.Error(500, err.Error())
}
id, err := str.CreateAccount(a, tx)
if err != nil {
logrus.Errorf("error creating account: %v", err)
_ = tx.Rollback()
return middleware.Error(400, 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{
Token: token,
})
}
func hashPassword(raw string) string {
hash := sha512.New()
hash.Write([]byte(raw))
return hex.EncodeToString(hash.Sum(nil))
}