add organization member handler (#537)

controller/addOrganizationMember.go

@@ -0,0 +1,47 @@
+package controller
+
+import (
+	"github.com/go-openapi/runtime/middleware"
+	"github.com/openziti/zrok/rest_model_zrok"
+	"github.com/openziti/zrok/rest_server_zrok/operations/admin"
+	"github.com/sirupsen/logrus"
+)
+
+type addOrganizationMemberHandler struct{}
+
+func newAddOrganizationMemberHandler() *addOrganizationMemberHandler {
+	return &addOrganizationMemberHandler{}
+}
+
+func (h *addOrganizationMemberHandler) Handle(params admin.AddOrganizationMemberParams, principal *rest_model_zrok.Principal) middleware.Responder {
+	if !principal.Admin {
+		logrus.Error("invalid admin principal")
+		return admin.NewAddOrganizationMemberUnauthorized()
+	}
+
+	trx, err := str.Begin()
+	if err != nil {
+		logrus.Errorf("error starting transaction: %v", err)
+		return admin.NewAddOrganizationMemberInternalServerError()
+	}
+	defer func() { _ = trx.Rollback() }()
+
+	acct, err := str.FindAccountWithEmail(params.Body.Email, trx)
+	if err != nil {
+		logrus.Errorf("error finding account with email address '%v': %v", params.Body.Email, err)
+		return admin.NewAddOrganizationMemberNotFound()
+	}
+
+	org, err := str.FindOrganizationByToken(params.Body.Token, trx)
+	if err != nil {
+		logrus.Errorf("error finding organization '%v': %v", params.Body.Token, err)
+		return admin.NewAddOrganizationMemberNotFound()
+	}
+
+	if err := str.AddAccountToOrganization(acct.Id, org.Id, trx); err != nil {
+		logrus.Errorf("error adding account '%v' to organization '%v': %v", acct.Id, org.Id, err)
+		return admin.NewAddOrganizationMemberInternalServerError()
+	}
+
+	return admin.NewAddOrganizationMemberCreated()
+}