implemented patching frontend descriptions (#834)

This commit is contained in:
Michael Quigley
2025-02-05 10:48:02 -05:00
parent 2a49705a48
commit 5228e25fb7
2 changed files with 67 additions and 0 deletions

View File

@@ -87,6 +87,7 @@ func Run(inCfg *config.Config) error {
api.ShareShareHandler = newShareHandler() api.ShareShareHandler = newShareHandler()
api.ShareUnaccessHandler = newUnaccessHandler() api.ShareUnaccessHandler = newUnaccessHandler()
api.ShareUnshareHandler = newUnshareHandler() api.ShareUnshareHandler = newUnshareHandler()
api.ShareUpdateAccessHandler = newUpdateAccessHandler()
api.ShareUpdateShareHandler = newUpdateShareHandler() api.ShareUpdateShareHandler = newUpdateShareHandler()
if err := controllerStartup(); err != nil { if err := controllerStartup(); err != nil {

View 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()
}