mirror of
https://github.com/openziti/zrok.git
synced 2025-08-19 04:06:18 +02:00
implemented patching frontend descriptions (#834)
This commit is contained in:
@@ -87,6 +87,7 @@ func Run(inCfg *config.Config) error {
|
||||
api.ShareShareHandler = newShareHandler()
|
||||
api.ShareUnaccessHandler = newUnaccessHandler()
|
||||
api.ShareUnshareHandler = newUnshareHandler()
|
||||
api.ShareUpdateAccessHandler = newUpdateAccessHandler()
|
||||
api.ShareUpdateShareHandler = newUpdateShareHandler()
|
||||
|
||||
if err := controllerStartup(); err != nil {
|
||||
|
66
controller/updateAccess.go
Normal file
66
controller/updateAccess.go
Normal file
@@ -0,0 +1,66 @@
|
||||
package controller
|
||||
|
||||
import (
|
||||
"github.com/go-openapi/runtime/middleware"
|
||||
"github.com/openziti/zrok/rest_model_zrok"
|
||||
"github.com/openziti/zrok/rest_server_zrok/operations/share"
|
||||
"github.com/sirupsen/logrus"
|
||||
)
|
||||
|
||||
type updateAccessHandler struct{}
|
||||
|
||||
func newUpdateAccessHandler() *updateAccessHandler {
|
||||
return &updateAccessHandler{}
|
||||
}
|
||||
|
||||
func (h *updateAccessHandler) Handle(params share.UpdateAccessParams, principal *rest_model_zrok.Principal) middleware.Responder {
|
||||
feToken := params.Body.FrontendToken
|
||||
desc := params.Body.Description
|
||||
|
||||
trx, err := str.Begin()
|
||||
if err != nil {
|
||||
logrus.Errorf("error starting transaction: %v", err)
|
||||
return share.NewUpdateAccessInternalServerError()
|
||||
}
|
||||
defer func() { _ = trx.Rollback() }()
|
||||
|
||||
fe, err := str.FindFrontendWithToken(feToken, trx)
|
||||
if err != nil {
|
||||
logrus.Errorf("error finding frontend with token '%v': %v", feToken, err)
|
||||
return share.NewUpdateAccessNotFound()
|
||||
}
|
||||
|
||||
envs, err := str.FindEnvironmentsForAccount(int(principal.ID), trx)
|
||||
if err != nil {
|
||||
logrus.Errorf("error finding environments for account '%v': %v", principal.Email, err)
|
||||
}
|
||||
|
||||
envMatched := false
|
||||
for _, env := range envs {
|
||||
if fe.EnvironmentId != nil && env.Id == *fe.EnvironmentId {
|
||||
envMatched = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !envMatched {
|
||||
logrus.Errorf("account '%v' does not own frontend '%v'", principal.Email, feToken)
|
||||
return share.NewUpdateAccessNotFound()
|
||||
}
|
||||
|
||||
if desc != "" {
|
||||
fe.Description = &desc
|
||||
} else {
|
||||
fe.Description = nil
|
||||
}
|
||||
if err := str.UpdateFrontend(fe, trx); err != nil {
|
||||
logrus.Errorf("error updating frontend '%v': %v", feToken, err)
|
||||
return share.NewUpdateAccessInternalServerError()
|
||||
}
|
||||
|
||||
if err := trx.Commit(); err != nil {
|
||||
logrus.Errorf("error committing transaction for frontend '%v': %v", feToken, err)
|
||||
return share.NewUpdateAccessInternalServerError()
|
||||
}
|
||||
|
||||
return share.NewUpdateAccessOK()
|
||||
}
|
Reference in New Issue
Block a user