zrok/controller/listFrontends.go

package controller

import (
	"github.com/go-openapi/runtime/middleware"
	"github.com/michaelquigley/df/dl"
	"github.com/openziti/zrok/v2/rest_model_zrok"
	"github.com/openziti/zrok/v2/rest_server_zrok/operations/admin"
)

type listFrontendsHandler struct{}

func newListFrontendsHandler() *listFrontendsHandler {
	return &listFrontendsHandler{}
}

func (h *listFrontendsHandler) Handle(params admin.ListFrontendsParams, principal *rest_model_zrok.Principal) middleware.Responder {
	if !principal.Admin {
		dl.Error("invalid admin principal")
		return admin.NewListFrontendsUnauthorized()
	}

	trx, err := str.Begin()
	if err != nil {
		dl.Errorf("error starting transaction: %v", err)
		return admin.NewListFrontendsInternalServerError()
	}
	defer func() { _ = trx.Rollback() }()

	sfes, err := str.FindPublicFrontends(trx)
	if err != nil {
		dl.Errorf("error finding public frontends: %v", err)
		return admin.NewListFrontendsInternalServerError()
	}

	var frontends []*admin.ListFrontendsOKBodyItems0
	for _, sfe := range sfes {
		frontend := &admin.ListFrontendsOKBodyItems0{
			FrontendToken:  sfe.Token,
			ZID:            sfe.ZId,
			CreatedAt:      sfe.CreatedAt.UnixMilli(),
			UpdatedAt:      sfe.UpdatedAt.UnixMilli(),
			Dynamic:        sfe.Dynamic,
			PermissionMode: string(sfe.PermissionMode),
		}
		if sfe.UrlTemplate != nil {
			frontend.URLTemplate = *sfe.UrlTemplate
		}
		if sfe.PublicName != nil {
			frontend.PublicName = *sfe.PublicName
		}
		frontends = append(frontends, frontend)
	}
	return admin.NewListFrontendsOK().WithPayload(frontends)
}