publicResourceAllocator (#99)

controller/share.go

@@ -2,8 +2,10 @@ package controller
 
 import (
 	"github.com/go-openapi/runtime/middleware"
+	"github.com/openziti-test-kitchen/zrok/controller/store"
 	"github.com/openziti-test-kitchen/zrok/rest_model_zrok"
 	"github.com/openziti-test-kitchen/zrok/rest_server_zrok/operations/service"
+	"github.com/sirupsen/logrus"
 )
 
 type shareHandler struct{}
@@ -13,10 +15,83 @@ func newShareHandler() *shareHandler {
 }
 
 func (h *shareHandler) Handle(params service.ShareParams, principal *rest_model_zrok.Principal) middleware.Responder {
-	switch params.Body.ShareMode {
-	case "public":
-		return newSharePublicHandler().Handle(params, principal)
-	default:
+	tx, err := str.Begin()
+	if err != nil {
+		logrus.Errorf("error starting transaction: %v", err)
 		return service.NewShareInternalServerError()
 	}
+	defer func() { _ = tx.Rollback() }()
+
+	envZId := params.Body.ZID
+	envId := 0
+	if envs, err := str.FindEnvironmentsForAccount(int(principal.ID), tx); err == nil {
+		found := false
+		for _, env := range envs {
+			if env.ZId == envZId {
+				logrus.Debugf("found identity '%v' for user '%v'", envZId, principal.Email)
+				envId = env.Id
+				found = true
+				break
+			}
+		}
+		if !found {
+			logrus.Errorf("environment '%v' not found for user '%v'", envZId, principal.Email)
+			return service.NewShareUnauthorized()
+		}
+	} else {
+		logrus.Errorf("error finding environments for account '%v'", principal.Email)
+		return service.NewShareInternalServerError()
+	}
+
+	edge, err := edgeClient()
+	if err != nil {
+		logrus.Error(err)
+		return service.NewShareInternalServerError()
+	}
+	svcName, err := createServiceName()
+	if err != nil {
+		logrus.Error(err)
+		return service.NewShareInternalServerError()
+	}
+
+	var svcZId string
+	var frontendEndpoints []string
+	switch params.Body.ShareMode {
+	case "public":
+		svcZId, frontendEndpoints, err = newPublicResourceAllocator().Allocate(envZId, svcName, params, edge)
+		if err != nil {
+			logrus.Error(err)
+			return service.NewShareInternalServerError()
+		}
+
+	default:
+		logrus.Errorf("unknown share mode '%v", params.Body.ShareMode)
+		return service.NewShareInternalServerError()
+	}
+
+	logrus.Debugf("allocated service '%v'", svcName)
+
+	sid, err := str.CreateService(envId, &store.Service{
+		ZId:                  svcZId,
+		Name:                 svcName,
+		ShareMode:            "public",
+		BackendMode:          "proxy",
+		FrontendEndpoint:     &frontendEndpoints[0],
+		BackendProxyEndpoint: &params.Body.BackendProxyEndpoint,
+	}, tx)
+	if err != nil {
+		logrus.Errorf("error creating service record: %v", err)
+		_ = tx.Rollback()
+		return service.NewShareInternalServerError()
+	}
+	if err := tx.Commit(); err != nil {
+		logrus.Errorf("error committing service record: %v", err)
+		return service.NewShareInternalServerError()
+	}
+	logrus.Infof("recorded service '%v' with id '%v' for '%v'", svcName, sid, principal.Email)
+
+	return service.NewShareCreated().WithPayload(&rest_model_zrok.ShareResponse{
+		FrontendProxyEndpoint: frontendEndpoints[0],
+		SvcName:               svcName,
+	})
 }