org.go

package org

import (
	"context"
	"fmt"

	"github.com/rilldata/rill/cli/pkg/cmdutil"
	adminv1 "github.com/rilldata/rill/proto/gen/rill/admin/v1"
	"github.com/spf13/cobra"
)

func OrgCmd(ch *cmdutil.Helper) *cobra.Command {
	orgCmd := &cobra.Command{
		Use:               "org",
		Short:             "Manage organisations",
		PersistentPreRunE: cmdutil.CheckAuth(ch),
	}

	orgCmd.AddCommand(CreateCmd(ch))
	orgCmd.AddCommand(EditCmd(ch))
	orgCmd.AddCommand(SwitchCmd(ch))
	orgCmd.AddCommand(ListCmd(ch))
	orgCmd.AddCommand(DeleteCmd(ch))
	orgCmd.AddCommand(RenameCmd(ch))

	return orgCmd
}

func orgNames(ctx context.Context, ch *cmdutil.Helper) ([]string, error) {
	c, err := ch.Client()
	if err != nil {
		return nil, err
	}

	resp, err := c.ListOrganizations(ctx, &adminv1.ListOrganizationsRequest{})
	if err != nil {
		return nil, err
	}

	if len(resp.Organizations) == 0 {
		return nil, fmt.Errorf("you are not a member of any orgs")
	}

	var orgNames []string
	for _, org := range resp.Organizations {
		orgNames = append(orgNames, org.Name)
	}

	return orgNames, nil
}