mirror of
https://github.com/openziti/ziti.git
synced 2026-10-15 00:22:01 +00:00
Paul Lorenz
95 lines
2.9 KiB
Go
95 lines
2.9 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 (
|
|
"fmt"
|
|
"github.com/go-openapi/strfmt"
|
|
"github.com/michaelquigley/pfxlog"
|
|
"github.com/openziti/edge/controller/env"
|
|
"github.com/openziti/edge/controller/model"
|
|
"github.com/openziti/edge/controller/response"
|
|
"github.com/openziti/edge/rest_model"
|
|
"github.com/openziti/fabric/controller/models"
|
|
"github.com/openziti/foundation/util/stringz"
|
|
)
|
|
|
|
const EntityNameEnrollment = "enrollments"
|
|
|
|
var EnrollmentLinkFactory = NewBasicLinkFactory(EntityNameEnrollment)
|
|
|
|
func MapEnrollmentToRestEntity(ae *env.AppEnv, _ *response.RequestContext, e models.Entity) (interface{}, error) {
|
|
enrollment, ok := e.(*model.Enrollment)
|
|
|
|
if !ok {
|
|
err := fmt.Errorf("entity is not a Enrollment \"%s\"", e.GetId())
|
|
log := pfxlog.Logger()
|
|
log.Error(err)
|
|
return nil, err
|
|
}
|
|
|
|
restModel, err := MapEnrollmentToRestModel(ae, enrollment)
|
|
|
|
if err != nil {
|
|
err := fmt.Errorf("could not convert to API entity \"%s\": %s", e.GetId(), err)
|
|
log := pfxlog.Logger()
|
|
log.Error(err)
|
|
return nil, err
|
|
}
|
|
return restModel, nil
|
|
}
|
|
|
|
func MapEnrollmentToRestModel(ae *env.AppEnv, enrollment *model.Enrollment) (*rest_model.EnrollmentDetail, error) {
|
|
expiresAt := strfmt.DateTime(*enrollment.ExpiresAt)
|
|
ret := &rest_model.EnrollmentDetail{
|
|
BaseEntity: BaseEntityToRestModel(enrollment, EnrollmentLinkFactory),
|
|
EdgeRouterID: stringz.OrEmpty(enrollment.EdgeRouterId),
|
|
ExpiresAt: &expiresAt,
|
|
IdentityID: stringz.OrEmpty(enrollment.IdentityId),
|
|
Method: &enrollment.Method,
|
|
Token: &enrollment.Token,
|
|
TransitRouterID: stringz.OrEmpty(enrollment.TransitRouterId),
|
|
Username: "",
|
|
}
|
|
|
|
if enrollment.IdentityId != nil {
|
|
identity, err := ae.Handlers.Identity.Read(*enrollment.IdentityId)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
ret.Identity = ToEntityRef(identity.Name, identity, IdentityLinkFactory)
|
|
}
|
|
|
|
if enrollment.EdgeRouterId != nil {
|
|
edgeRouter, err := ae.Handlers.EdgeRouter.Read(*enrollment.EdgeRouterId)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
ret.EdgeRouter = ToEntityRef(edgeRouter.Name, edgeRouter, EdgeRouterLinkFactory)
|
|
}
|
|
|
|
if enrollment.TransitRouterId != nil {
|
|
transitRouter, err := ae.Handlers.TransitRouter.Read(*enrollment.TransitRouterId)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
ret.TransitRouter = ToEntityRef(transitRouter.Name, transitRouter, TransitRouterLinkFactory)
|
|
}
|
|
|
|
return ret, nil
|
|
}
|