feat: add account and account member commands (#108)

Author: Davidonium

internal/cli/root.go

@@ -5,6 +5,7 @@ import (
 
 	"github.com/spf13/cobra"
 
+	"github.com/qdrant/qcloud-cli/internal/cmd/account"
 	"github.com/qdrant/qcloud-cli/internal/cmd/backup"
 	"github.com/qdrant/qcloud-cli/internal/cmd/cloudprovider"
 	"github.com/qdrant/qcloud-cli/internal/cmd/cloudregion"
@@ -65,6 +66,7 @@ Documentation: https://github.com/qdrant/qcloud-cli`,
 	s.Config.BindPFlag(config.KeyAccountID, cmd.PersistentFlags().Lookup("account-id"))
 	s.Config.BindPFlag(config.KeyEndpoint, cmd.PersistentFlags().Lookup("endpoint"))
 
+	cmd.AddCommand(account.NewCommand(s))
 	cmd.AddCommand(version.NewCommand(s))
 	cmd.AddCommand(iam.NewCommand(s))
 	cmd.AddCommand(cluster.NewCommand(s))

internal/cmd/account/account.go

@@ -0,0 +1,28 @@
+package account
+
+import (
+	"github.com/spf13/cobra"
+
+	"github.com/qdrant/qcloud-cli/internal/state"
+)
+
+// NewCommand creates the account command group.
+func NewCommand(s *state.State) *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "account",
+		Short: "Manage Qdrant Cloud accounts",
+		Long: `Manage Qdrant Cloud accounts and their members.
+
+Use these commands to list, inspect, and update accounts that the current
+management key has access to. Account member commands show who belongs to the
+current account and whether they are the owner.`,
+		Args: cobra.NoArgs,
+	}
+	cmd.AddCommand(
+		newListCommand(s),
+		newDescribeCommand(s),
+		newUpdateCommand(s),
+		newMemberCommand(s),
+	)
+	return cmd
+}

internal/cmd/account/describe.go

@@ -0,0 +1,89 @@
+package account
+
+import (
+	"fmt"
+	"io"
+	"strings"
+
+	"github.com/spf13/cobra"
+
+	accountv1 "github.com/qdrant/qdrant-cloud-public-api/gen/go/qdrant/cloud/account/v1"
+
+	"github.com/qdrant/qcloud-cli/internal/cmd/base"
+	"github.com/qdrant/qcloud-cli/internal/cmd/completion"
+	"github.com/qdrant/qcloud-cli/internal/cmd/output"
+	"github.com/qdrant/qcloud-cli/internal/state"
+)
+
+func newDescribeCommand(s *state.State) *cobra.Command {
+	return base.DescribeCmd[*accountv1.Account]{
+		Use:   "describe [account-id]",
+		Short: "Describe an account",
+		Long: `Describe an account by its ID.
+
+If no account ID is provided, the current account (from --account-id, the
+active context, or the QDRANT_CLOUD_ACCOUNT_ID environment variable) is used.`,
+		Example: `# Describe the current account
+qcloud account describe
+
+# Describe a specific account
+qcloud account describe a1b2c3d4-e5f6-7890-abcd-ef1234567890
+
+# Output as JSON
+qcloud account describe --json`,
+		Args: cobra.MaximumNArgs(1),
+		Fetch: func(s *state.State, cmd *cobra.Command, args []string) (*accountv1.Account, error) {
+			ctx := cmd.Context()
+			client, err := s.Client(ctx)
+			if err != nil {
+				return nil, err
+			}
+
+			accountID, err := resolveAccountID(s, args)
+			if err != nil {
+				return nil, err
+			}
+
+			resp, err := client.Account().GetAccount(ctx, &accountv1.GetAccountRequest{
+				AccountId: accountID,
+			})
+			if err != nil {
+				return nil, fmt.Errorf("failed to get account: %w", err)
+			}
+
+			return resp.GetAccount(), nil
+		},
+		PrintText: func(_ *cobra.Command, w io.Writer, acct *accountv1.Account) error {
+			fmt.Fprintf(w, "ID:          %s\n", acct.GetId())
+			fmt.Fprintf(w, "Name:        %s\n", acct.GetName())
+			fmt.Fprintf(w, "Owner Email: %s\n", acct.GetOwnerEmail())
+			if company := acct.GetCompany(); company != nil {
+				fmt.Fprintf(w, "Company:     %s\n", company.GetName())
+				if company.Domain != nil {
+					fmt.Fprintf(w, "Domain:      %s\n", company.GetDomain())
+				}
+			}
+			if privs := acct.GetPrivileges(); len(privs) > 0 {
+				fmt.Fprintf(w, "Privileges:  %s\n", strings.Join(privs, ", "))
+			}
+			if acct.GetCreatedAt() != nil {
+				t := acct.GetCreatedAt().AsTime()
+				fmt.Fprintf(w, "Created:     %s  (%s)\n", output.HumanTime(t), output.FullDateTime(t))
+			}
+			if acct.GetLastModifiedAt() != nil {
+				t := acct.GetLastModifiedAt().AsTime()
+				fmt.Fprintf(w, "Modified:    %s  (%s)\n", output.HumanTime(t), output.FullDateTime(t))
+			}
+			return nil
+		},
+		ValidArgsFunction: completion.AccountIDCompletion(s),
+	}.CobraCommand(s)
+}
+
+// resolveAccountID returns args[0] if present, otherwise falls back to s.AccountID().
+func resolveAccountID(s *state.State, args []string) (string, error) {
+	if len(args) > 0 {
+		return args[0], nil
+	}
+	return s.AccountID()
+}

internal/cmd/account/describe_test.go

@@ -0,0 +1,108 @@
+package account_test
+
+import (
+	"encoding/json"
+	"fmt"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	"google.golang.org/protobuf/types/known/timestamppb"
+
+	accountv1 "github.com/qdrant/qdrant-cloud-public-api/gen/go/qdrant/cloud/account/v1"
+
+	"github.com/qdrant/qcloud-cli/internal/testutil"
+)
+
+func TestAccountDescribe_TableOutput(t *testing.T) {
+	env := testutil.NewTestEnv(t)
+
+	now := time.Now()
+	env.AccountServer.GetAccountCalls.Returns(&accountv1.GetAccountResponse{
+		Account: &accountv1.Account{
+			Id:         "acct-001",
+			Name:       "Production",
+			OwnerEmail: "owner@example.com",
+			Company: &accountv1.Company{
+				Name:   "Acme Corp",
+				Domain: new("acme.com"),
+			},
+			Privileges: []string{"premium"},
+			CreatedAt:  timestamppb.New(now.Add(-48 * time.Hour)),
+		},
+	}, nil)
+
+	stdout, _, err := testutil.Exec(t, env, "account", "describe", "acct-001")
+	require.NoError(t, err)
+	assert.Contains(t, stdout, "acct-001")
+	assert.Contains(t, stdout, "Production")
+	assert.Contains(t, stdout, "owner@example.com")
+	assert.Contains(t, stdout, "Acme Corp")
+	assert.Contains(t, stdout, "acme.com")
+	assert.Contains(t, stdout, "premium")
+
+	req, ok := env.AccountServer.GetAccountCalls.Last()
+	require.True(t, ok)
+	assert.Equal(t, "acct-001", req.GetAccountId())
+}
+
+func TestAccountDescribe_DefaultsToCurrentAccount(t *testing.T) {
+	env := testutil.NewTestEnv(t, testutil.WithAccountID("default-acct"))
+
+	env.AccountServer.GetAccountCalls.Returns(&accountv1.GetAccountResponse{
+		Account: &accountv1.Account{
+			Id:   "default-acct",
+			Name: "Default",
+		},
+	}, nil)
+
+	stdout, _, err := testutil.Exec(t, env, "account", "describe")
+	require.NoError(t, err)
+	assert.Contains(t, stdout, "default-acct")
+
+	req, ok := env.AccountServer.GetAccountCalls.Last()
+	require.True(t, ok)
+	assert.Equal(t, "default-acct", req.GetAccountId())
+}
+
+func TestAccountDescribe_JSONOutput(t *testing.T) {
+	env := testutil.NewTestEnv(t)
+
+	env.AccountServer.GetAccountCalls.Returns(&accountv1.GetAccountResponse{
+		Account: &accountv1.Account{
+			Id:         "acct-json",
+			Name:       "JSON Account",
+			OwnerEmail: "json@example.com",
+		},
+	}, nil)
+
+	stdout, _, err := testutil.Exec(t, env, "account", "describe", "acct-json", "--json")
+	require.NoError(t, err)
+
+	var result struct {
+		ID         string `json:"id"`
+		Name       string `json:"name"`
+		OwnerEmail string `json:"ownerEmail"`
+	}
+	require.NoError(t, json.Unmarshal([]byte(stdout), &result))
+	assert.Equal(t, "acct-json", result.ID)
+	assert.Equal(t, "JSON Account", result.Name)
+	assert.Equal(t, "json@example.com", result.OwnerEmail)
+}
+
+func TestAccountDescribe_BackendError(t *testing.T) {
+	env := testutil.NewTestEnv(t)
+
+	env.AccountServer.GetAccountCalls.Returns(nil, fmt.Errorf("not found"))
+
+	_, _, err := testutil.Exec(t, env, "account", "describe", "acct-bad")
+	require.Error(t, err)
+}
+
+func TestAccountDescribe_TooManyArgs(t *testing.T) {
+	env := testutil.NewTestEnv(t)
+
+	_, _, err := testutil.Exec(t, env, "account", "describe", "arg1", "arg2")
+	require.Error(t, err)
+}

internal/cmd/account/list.go

@@ -0,0 +1,52 @@
+package account
+
+import (
+	"io"
+
+	"github.com/spf13/cobra"
+
+	accountv1 "github.com/qdrant/qdrant-cloud-public-api/gen/go/qdrant/cloud/account/v1"
+
+	"github.com/qdrant/qcloud-cli/internal/cmd/base"
+	"github.com/qdrant/qcloud-cli/internal/cmd/output"
+	"github.com/qdrant/qcloud-cli/internal/state"
+)
+
+func newListCommand(s *state.State) *cobra.Command {
+	return base.ListCmd[*accountv1.ListAccountsResponse]{
+		Use:   "list",
+		Short: "List accounts",
+		Long: `List all accounts associated with the authenticated management key.
+
+Returns every account the current API key has access to. No account ID is
+required because the server resolves accounts from the caller's credentials.`,
+		Example: `# List all accessible accounts
+qcloud account list
+
+# Output as JSON
+qcloud account list --json`,
+		Fetch: func(s *state.State, cmd *cobra.Command) (*accountv1.ListAccountsResponse, error) {
+			ctx := cmd.Context()
+			client, err := s.Client(ctx)
+			if err != nil {
+				return nil, err
+			}
+
+			return client.Account().ListAccounts(ctx, &accountv1.ListAccountsRequest{})
+		},
+		PrintText: func(_ *cobra.Command, w io.Writer, resp *accountv1.ListAccountsResponse) error {
+			t := output.NewTable[*accountv1.Account](w)
+			t.AddField("ID", func(v *accountv1.Account) string { return v.GetId() })
+			t.AddField("NAME", func(v *accountv1.Account) string { return v.GetName() })
+			t.AddField("OWNER EMAIL", func(v *accountv1.Account) string { return v.GetOwnerEmail() })
+			t.AddField("CREATED", func(v *accountv1.Account) string {
+				if v.GetCreatedAt() != nil {
+					return output.HumanTime(v.GetCreatedAt().AsTime())
+				}
+				return ""
+			})
+			t.Write(resp.GetItems())
+			return nil
+		},
+	}.CobraCommand(s)
+}

internal/cmd/account/list_test.go

@@ -0,0 +1,86 @@
+package account_test
+
+import (
+	"encoding/json"
+	"fmt"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	"google.golang.org/protobuf/types/known/timestamppb"
+
+	accountv1 "github.com/qdrant/qdrant-cloud-public-api/gen/go/qdrant/cloud/account/v1"
+
+	"github.com/qdrant/qcloud-cli/internal/testutil"
+)
+
+func TestAccountList_TableOutput(t *testing.T) {
+	env := testutil.NewTestEnv(t)
+
+	env.AccountServer.ListAccountsCalls.Returns(&accountv1.ListAccountsResponse{
+		Items: []*accountv1.Account{
+			{
+				Id:         "acct-001",
+				Name:       "Production",
+				OwnerEmail: "owner@example.com",
+				CreatedAt:  timestamppb.New(time.Now().Add(-24 * time.Hour)),
+			},
+		},
+	}, nil)
+
+	stdout, _, err := testutil.Exec(t, env, "account", "list")
+	require.NoError(t, err)
+	assert.Contains(t, stdout, "ID")
+	assert.Contains(t, stdout, "NAME")
+	assert.Contains(t, stdout, "OWNER EMAIL")
+	assert.Contains(t, stdout, "CREATED")
+	assert.Contains(t, stdout, "acct-001")
+	assert.Contains(t, stdout, "Production")
+	assert.Contains(t, stdout, "owner@example.com")
+	assert.Contains(t, stdout, "ago")
+}
+
+func TestAccountList_JSONOutput(t *testing.T) {
+	env := testutil.NewTestEnv(t)
+
+	env.AccountServer.ListAccountsCalls.Returns(&accountv1.ListAccountsResponse{
+		Items: []*accountv1.Account{
+			{Id: "acct-json", Name: "Test Account"},
+		},
+	}, nil)
+
+	stdout, _, err := testutil.Exec(t, env, "account", "list", "--json")
+	require.NoError(t, err)
+
+	var result struct {
+		Items []struct {
+			ID   string `json:"id"`
+			Name string `json:"name"`
+		} `json:"items"`
+	}
+	require.NoError(t, json.Unmarshal([]byte(stdout), &result))
+	require.Len(t, result.Items, 1)
+	assert.Equal(t, "acct-json", result.Items[0].ID)
+	assert.Equal(t, "Test Account", result.Items[0].Name)
+}
+
+func TestAccountList_BackendError(t *testing.T) {
+	env := testutil.NewTestEnv(t)
+
+	env.AccountServer.ListAccountsCalls.Returns(nil, fmt.Errorf("service unavailable"))
+
+	_, _, err := testutil.Exec(t, env, "account", "list")
+	require.Error(t, err)
+}
+
+func TestAccountList_Empty(t *testing.T) {
+	env := testutil.NewTestEnv(t)
+
+	env.AccountServer.ListAccountsCalls.Returns(&accountv1.ListAccountsResponse{}, nil)
+
+	stdout, _, err := testutil.Exec(t, env, "account", "list")
+	require.NoError(t, err)
+	assert.Contains(t, stdout, "ID")
+	assert.Contains(t, stdout, "NAME")
+}