mirror of
https://github.com/openziti/zrok.git
synced 2024-12-22 23:02:52 +01:00
36 lines
1.2 KiB
Go
36 lines
1.2 KiB
Go
|
package controller
|
||
|
|
||
|
import (
|
||
|
"github.com/go-openapi/runtime/middleware"
|
||
|
"github.com/openziti/zrok/rest_model_zrok"
|
||
|
"github.com/openziti/zrok/rest_server_zrok/operations/metadata"
|
||
|
"github.com/sirupsen/logrus"
|
||
|
)
|
||
|
|
||
|
type listMembershipsHandler struct{}
|
||
|
|
||
|
func newListMembershipsHandler() *listMembershipsHandler {
|
||
|
return &listMembershipsHandler{}
|
||
|
}
|
||
|
|
||
|
func (h *listMembershipsHandler) Handle(_ metadata.ListMembershipsParams, principal *rest_model_zrok.Principal) middleware.Responder {
|
||
|
trx, err := str.Begin()
|
||
|
if err != nil {
|
||
|
logrus.Errorf("error startin transaction: %v", err)
|
||
|
return metadata.NewListMembershipsInternalServerError()
|
||
|
}
|
||
|
defer func() { _ = trx.Rollback() }()
|
||
|
|
||
|
oms, err := str.FindOrganizationsForAccount(int(principal.ID), trx)
|
||
|
if err != nil {
|
||
|
logrus.Errorf("error finding organizations for account '%v': %v", principal.Email, err)
|
||
|
return metadata.NewListMembershipsInternalServerError()
|
||
|
}
|
||
|
|
||
|
var out []*metadata.ListMembershipsOKBodyMembershipsItems0
|
||
|
for _, om := range oms {
|
||
|
out = append(out, &metadata.ListMembershipsOKBodyMembershipsItems0{Token: om.Token, Description: om.Description, Admin: om.Admin})
|
||
|
}
|
||
|
return metadata.NewListMembershipsOK().WithPayload(&metadata.ListMembershipsOKBody{Memberships: out})
|
||
|
}
|