list public frontends for account implementation (#996)

controller/listPublicFrontendsForAccount.go

@@ -0,0 +1,52 @@
+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 listPublicFrontendsForAccountHandler struct{}
+
+func newListPublicFrontendsForAccountHandler() *listPublicFrontendsForAccountHandler {
+	return &listPublicFrontendsForAccountHandler{}
+}
+
+func (h *listPublicFrontendsForAccountHandler) Handle(_ metadata.ListPublicFrontendsForAccountParams, principal *rest_model_zrok.Principal) middleware.Responder {
+	trx, err := str.Begin()
+	if err != nil {
+		logrus.Errorf("error starting transaction: %v", err)
+		return metadata.NewListPublicFrontendsForAccountInternalServerError()
+	}
+	defer trx.Rollback()
+
+	var publicFrontends []*metadata.ListPublicFrontendsForAccountOKBodyPublicFrontendsItems0
+
+	openFes, err := str.FindOpenPublicFrontends(trx)
+	if err != nil {
+		logrus.Errorf("error finding open public frontends: %v", err)
+		return metadata.NewListPublicFrontendsForAccountInternalServerError()
+	}
+	for _, openFe := range openFes {
+		publicFrontends = append(publicFrontends, &metadata.ListPublicFrontendsForAccountOKBodyPublicFrontendsItems0{
+			PublicName:  *openFe.PublicName,
+			URLTemplate: *openFe.UrlTemplate,
+		})
+	}
+
+	closedFes, err := str.FindClosedPublicFrontendsGrantedToAccount(int(principal.ID), trx)
+	if err != nil {
+		logrus.Errorf("error finding closed public frontends: %v", err)
+		return metadata.NewListPublicFrontendsForAccountInternalServerError()
+	}
+	for _, closedFe := range closedFes {
+		publicFrontends = append(publicFrontends, &metadata.ListPublicFrontendsForAccountOKBodyPublicFrontendsItems0{
+			PublicName:  *closedFe.PublicName,
+			URLTemplate: *closedFe.UrlTemplate,
+		})
+	}
+
+	payload := &metadata.ListPublicFrontendsForAccountOKBody{PublicFrontends: publicFrontends}
+	return metadata.NewListPublicFrontendsForAccountOK().WithPayload(payload)
+}