zrok/controller/listMemberships.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})
}