mirror of
https://github.com/hanzozt/zt.git
synced 2026-11-02 11:38:22 +00:00
Andrew MartinezandGitHub
56daae2025
* fixes #2796 generated clients for enrollment do not work - default behavior complies with OpenAPI spec, but allows for legacy PEM handling through middleware intervention - adds test for generic enrollment endpoint and for specific enrollment endpoints
127 lines
4.3 KiB
Go
127 lines
4.3 KiB
Go
/*
|
|
Copyright NetFoundry Inc.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
https://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package routes
|
|
|
|
import (
|
|
"github.com/go-openapi/runtime/middleware"
|
|
"github.com/openziti/edge-api/rest_management_api_server/operations/enrollment"
|
|
"github.com/openziti/edge-api/rest_model"
|
|
"github.com/openziti/foundation/v2/errorz"
|
|
"github.com/openziti/ziti/controller/env"
|
|
"github.com/openziti/ziti/controller/internal/permissions"
|
|
"github.com/openziti/ziti/controller/model"
|
|
"github.com/openziti/ziti/controller/response"
|
|
"time"
|
|
)
|
|
|
|
func init() {
|
|
r := NewEnrollmentRouter()
|
|
env.AddRouter(r)
|
|
}
|
|
|
|
type EnrollmentRouter struct {
|
|
BasePath string
|
|
}
|
|
|
|
func NewEnrollmentRouter() *EnrollmentRouter {
|
|
return &EnrollmentRouter{
|
|
BasePath: "/" + EntityNameEnrollment,
|
|
}
|
|
}
|
|
|
|
func (r *EnrollmentRouter) Register(ae *env.AppEnv) {
|
|
ae.ManagementApi.EnrollmentDeleteEnrollmentHandler = enrollment.DeleteEnrollmentHandlerFunc(func(params enrollment.DeleteEnrollmentParams, _ interface{}) middleware.Responder {
|
|
return ae.IsAllowed(r.Delete, params.HTTPRequest, params.ID, "", permissions.IsAdmin())
|
|
})
|
|
|
|
ae.ManagementApi.EnrollmentDetailEnrollmentHandler = enrollment.DetailEnrollmentHandlerFunc(func(params enrollment.DetailEnrollmentParams, _ interface{}) middleware.Responder {
|
|
return ae.IsAllowed(r.Detail, params.HTTPRequest, params.ID, "", permissions.IsAdmin())
|
|
})
|
|
|
|
ae.ManagementApi.EnrollmentListEnrollmentsHandler = enrollment.ListEnrollmentsHandlerFunc(func(params enrollment.ListEnrollmentsParams, _ interface{}) middleware.Responder {
|
|
return ae.IsAllowed(r.List, params.HTTPRequest, "", "", permissions.IsAdmin())
|
|
})
|
|
|
|
ae.ManagementApi.EnrollmentRefreshEnrollmentHandler = enrollment.RefreshEnrollmentHandlerFunc(func(params enrollment.RefreshEnrollmentParams, _ interface{}) middleware.Responder {
|
|
return ae.IsAllowed(func(ae *env.AppEnv, rc *response.RequestContext) {
|
|
r.Refresh(ae, rc, params)
|
|
}, params.HTTPRequest, params.ID, "", permissions.IsAdmin())
|
|
})
|
|
|
|
ae.ManagementApi.EnrollmentCreateEnrollmentHandler = enrollment.CreateEnrollmentHandlerFunc(func(params enrollment.CreateEnrollmentParams, _ interface{}) middleware.Responder {
|
|
return ae.IsAllowed(func(ae *env.AppEnv, rc *response.RequestContext) {
|
|
r.Create(ae, rc, params)
|
|
}, params.HTTPRequest, "", "", permissions.IsAdmin())
|
|
})
|
|
}
|
|
|
|
func (r *EnrollmentRouter) List(ae *env.AppEnv, rc *response.RequestContext) {
|
|
ListWithHandler[*model.Enrollment](ae, rc, ae.Managers.Enrollment, MapEnrollmentToRestEntity)
|
|
}
|
|
|
|
func (r *EnrollmentRouter) Detail(ae *env.AppEnv, rc *response.RequestContext) {
|
|
DetailWithHandler[*model.Enrollment](ae, rc, ae.Managers.Enrollment, MapEnrollmentToRestEntity)
|
|
}
|
|
|
|
func (r *EnrollmentRouter) Delete(ae *env.AppEnv, rc *response.RequestContext) {
|
|
DeleteWithHandler(rc, ae.Managers.Enrollment)
|
|
}
|
|
|
|
func (r *EnrollmentRouter) Refresh(ae *env.AppEnv, rc *response.RequestContext, params enrollment.RefreshEnrollmentParams) {
|
|
id, err := rc.GetEntityId()
|
|
|
|
if err != nil {
|
|
rc.RespondWithError(err)
|
|
return
|
|
}
|
|
|
|
if id == "" {
|
|
rc.RespondWithNotFound()
|
|
return
|
|
}
|
|
|
|
if err := ae.Managers.Enrollment.RefreshJwt(id, time.Time(*params.Refresh.ExpiresAt), rc.NewChangeContext()); err != nil {
|
|
if fe, ok := err.(*errorz.FieldError); ok {
|
|
rc.RespondWithFieldError(fe)
|
|
return
|
|
}
|
|
rc.RespondWithError(err)
|
|
return
|
|
}
|
|
|
|
rc.RespondWithEmptyOk()
|
|
}
|
|
|
|
func (r *EnrollmentRouter) Create(ae *env.AppEnv, rc *response.RequestContext, params enrollment.CreateEnrollmentParams) {
|
|
Create(rc, rc, EnrollmentLinkFactory, func() (string, error) {
|
|
return MapCreate(ae.Managers.Enrollment.Create, MapCreateEnrollmentToModel(params.Enrollment), rc)
|
|
})
|
|
|
|
}
|
|
|
|
func MapCreateEnrollmentToModel(create *rest_model.EnrollmentCreate) *model.Enrollment {
|
|
ret := &model.Enrollment{
|
|
Method: *create.Method,
|
|
IdentityId: create.IdentityID,
|
|
ExpiresAt: (*time.Time)(create.ExpiresAt),
|
|
CaId: create.CaID,
|
|
Username: create.Username,
|
|
}
|
|
|
|
return ret
|
|
}
|