From 4473571b37dc7538e942d9364b4d0a297edea281 Mon Sep 17 00:00:00 2001
From: Michael Quigley <michael@quigley.com>
Date: Mon, 9 Dec 2024 13:44:56 -0500
Subject: [PATCH] add/delete org plumbed through to CLI (#537)

---
 cmd/zrok/adminCreateOrganization.go | 52 +++++++++++++++++++++++++++++
 cmd/zrok/adminDeleteOrganization.go | 50 +++++++++++++++++++++++++++
 controller/deleteOrganization.go    |  5 +++
 controller/store/organization.go    |  2 +-
 4 files changed, 108 insertions(+), 1 deletion(-)
 create mode 100644 cmd/zrok/adminCreateOrganization.go
 create mode 100644 cmd/zrok/adminDeleteOrganization.go

diff --git a/cmd/zrok/adminCreateOrganization.go b/cmd/zrok/adminCreateOrganization.go
new file mode 100644
index 00000000..da7b5944
--- /dev/null
+++ b/cmd/zrok/adminCreateOrganization.go
@@ -0,0 +1,52 @@
+package main
+
+import (
+	"github.com/openziti/zrok/environment"
+	"github.com/openziti/zrok/rest_client_zrok/admin"
+	"github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+)
+
+func init() {
+	adminCreateCmd.AddCommand(newAdminCreateOrganizationCommand().cmd)
+}
+
+type adminCreateOrganizationCommand struct {
+	cmd         *cobra.Command
+	description string
+}
+
+func newAdminCreateOrganizationCommand() *adminCreateOrganizationCommand {
+	cmd := &cobra.Command{
+		Use:     "organization",
+		Aliases: []string{"org"},
+		Short:   "Create a new organization",
+		Args:    cobra.NoArgs,
+	}
+	command := &adminCreateOrganizationCommand{cmd: cmd}
+	cmd.Flags().StringVarP(&command.description, "description", "d", "", "Organization description")
+	cmd.Run = command.run
+	return command
+}
+
+func (cmd *adminCreateOrganizationCommand) run(_ *cobra.Command, _ []string) {
+	env, err := environment.LoadRoot()
+	if err != nil {
+		panic(err)
+	}
+
+	zrok, err := env.Client()
+	if err != nil {
+		panic(err)
+	}
+
+	req := admin.NewCreateOrganizationParams()
+	req.Body = admin.CreateOrganizationBody{Description: cmd.description}
+
+	resp, err := zrok.Admin.CreateOrganization(req, mustGetAdminAuth())
+	if err != nil {
+		panic(err)
+	}
+
+	logrus.Infof("created new organization with token '%v'", resp.Payload.Token)
+}
diff --git a/cmd/zrok/adminDeleteOrganization.go b/cmd/zrok/adminDeleteOrganization.go
new file mode 100644
index 00000000..cb3ec6e2
--- /dev/null
+++ b/cmd/zrok/adminDeleteOrganization.go
@@ -0,0 +1,50 @@
+package main
+
+import (
+	"github.com/openziti/zrok/environment"
+	"github.com/openziti/zrok/rest_client_zrok/admin"
+	"github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+)
+
+func init() {
+	adminDeleteCmd.AddCommand(newAdminDeleteOrganizationCommand().cmd)
+}
+
+type adminDeleteOrganizationCommand struct {
+	cmd *cobra.Command
+}
+
+func newAdminDeleteOrganizationCommand() *adminDeleteOrganizationCommand {
+	cmd := &cobra.Command{
+		Use:     "organization <organizationToken>",
+		Aliases: []string{"org"},
+		Short:   "Delete an organization",
+		Args:    cobra.ExactArgs(1),
+	}
+	command := &adminDeleteOrganizationCommand{cmd: cmd}
+	cmd.Run = command.run
+	return command
+}
+
+func (cmd *adminDeleteOrganizationCommand) run(_ *cobra.Command, args []string) {
+	env, err := environment.LoadRoot()
+	if err != nil {
+		panic(err)
+	}
+
+	zrok, err := env.Client()
+	if err != nil {
+		panic(err)
+	}
+
+	req := admin.NewDeleteOrganizationParams()
+	req.Body.Token = args[0]
+
+	_, err = zrok.Admin.DeleteOrganization(req, mustGetAdminAuth())
+	if err != nil {
+		panic(err)
+	}
+
+	logrus.Infof("deleted organization with token '%v'", args[0])
+}
diff --git a/controller/deleteOrganization.go b/controller/deleteOrganization.go
index deed01b6..70df1d5c 100644
--- a/controller/deleteOrganization.go
+++ b/controller/deleteOrganization.go
@@ -38,5 +38,10 @@ func (h *deleteOrganizationHandler) Handle(params admin.DeleteOrganizationParams
 		return admin.NewDeleteOrganizationInternalServerError()
 	}
 
+	if err := trx.Commit(); err != nil {
+		logrus.Errorf("error committing transaction: %v", err)
+		return admin.NewDeleteOrganizationInternalServerError()
+	}
+
 	return admin.NewDeleteOrganizationOK()
 }
diff --git a/controller/store/organization.go b/controller/store/organization.go
index e5dc0fbf..22dd710c 100644
--- a/controller/store/organization.go
+++ b/controller/store/organization.go
@@ -25,7 +25,7 @@ func (str *Store) CreateOrganization(org *Organization, trx *sqlx.Tx) (int, erro
 
 func (str *Store) FindOrganizationByToken(token string, trx *sqlx.Tx) (*Organization, error) {
 	org := &Organization{}
-	if err := trx.QueryRowx("select * from organizations where token = $1", token).StructScan(org); err != nil {
+	if err := trx.QueryRowx("select * from organizations where token = $1 and not deleted", token).StructScan(org); err != nil {
 		return nil, errors.Wrap(err, "error selecting frontend by token")
 	}
 	return org, nil