chore: add cli command to fetch group sync settings as json

coder · Emyrk · Sep 17, 2024 · Sep 16, 2024 · Sep 16, 2024 · Sep 16, 2024
commit 6a59ee11636821a83e3b143d24cd5e5ce4ea5be6
diff --git a/cli/organization.go b/cli/organization.go
@@ -26,6 +26,7 @@ func (r *RootCmd) organizations() *serpent.Command {
 			r.createOrganization(),
 			r.organizationMembers(orgContext),
 			r.organizationRoles(orgContext),
+			r.organizationSettings(orgContext),
 		},
 	}
 

diff --git a/cli/organizationsettings.go b/cli/organizationsettings.go
@@ -0,0 +1,84 @@
+package cli
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"strings"
+
+	"github.com/coder/coder/v2/codersdk"
+	"github.com/coder/serpent"
+)
+
+func (r *RootCmd) organizationSettings(orgContext *OrganizationContext) *serpent.Command {
+	cmd := &serpent.Command{
+		Use:     "settings",
+		Short:   "Manage organization settings.",
+		Aliases: []string{"setting"},
+		Handler: func(inv *serpent.Invocation) error {
+			return inv.Command.HelpHandler(inv)
+		},
+		Hidden: true,
+		Children: []*serpent.Command{
+			r.printOrganizationSetting(orgContext),
+		},
+	}
+	return cmd
+}
+
+func (r *RootCmd) printOrganizationSetting(orgContext *OrganizationContext) *serpent.Command {
+	client := new(codersdk.Client)
+	cmd := &serpent.Command{
+		Use:   "show <groupsync | rolesync>",
+		Short: "Outputs specified organization setting.",
+		Long: FormatExamples(
+			Example{
+				Description: "Output group sync settings.",
+				Command:     "coder organization settings show groupsync",
+			},
+		),
+		Options: []serpent.Option{},
+		Middleware: serpent.Chain(
+			serpent.RequireNArgs(1),
+			r.InitClient(client),
+		),
+		Handler: func(inv *serpent.Invocation) error {
+			ctx := inv.Context()
+			org, err := orgContext.Selected(inv, client)
+			if err != nil {
+				return err
+			}
+
+			var setting any
+			switch strings.ToLower(inv.Args[0]) {
+			case "groupsync", "group-sync":
+				setting, err = client.GroupIDPSyncSettings(ctx, org.ID.String())
+			case "rolesync", "role-sync":
+				// TODO: Implement role sync settings.
+				return fmt.Errorf("role sync settings are not implemented")
+			default:
+				_, _ = fmt.Fprintln(inv.Stderr, "Valid organization settings are: 'groupsync', 'rolesync'")
+				return fmt.Errorf("unknown organization setting %s", inv.Args[0])
+			}
+
+			if err != nil {
+				return fmt.Errorf("failed to get organization setting %s: %w", inv.Args[0], err)
+			}
+
+			settingJson, err := json.Marshal(setting)
+			if err != nil {
+				return fmt.Errorf("failed to marshal organization setting %s: %w", inv.Args[0], err)
+			}
+
+			var dst bytes.Buffer
+			err = json.Indent(&dst, settingJson, "", "\t")
+			if err != nil {
+				return fmt.Errorf("failed to indent organization setting as json %s: %w", inv.Args[0], err)
+			}
+
+			_, err = fmt.Fprintln(inv.Stdout, dst.String())
+			return err
+		},
+	}
+	return cmd
+}
diff --git a/cli/root.go b/cli/root.go
@@ -29,6 +29,7 @@ import (
 	"golang.org/x/mod/semver"
 	"golang.org/x/xerrors"
 
+	"github.com/coder/coder/v2/coderd/database/db2sdk"
 	"github.com/coder/pretty"
 
 	"github.com/coder/coder/v2/buildinfo"
@@ -657,7 +658,10 @@ func (o *OrganizationContext) Selected(inv *serpent.Invocation, client *codersdk
 	}
 
 	// No org selected, and we are more than 1? Return an error.
-	return codersdk.Organization{}, xerrors.Errorf("Must select an organization with --org=<org_name>.")
+	validOrgs := db2sdk.List(orgs, func(org codersdk.Organization) string {
+		return fmt.Sprintf("%q", org.Name)
+	})
+	return codersdk.Organization{}, xerrors.Errorf("Must select an organization with --org=<org_name>. Choose from: %s", strings.Join(validOrgs, ", "))
 }
 
 func splitNamedWorkspace(identifier string) (owner string, workspaceName string, err error) {