fix(chat): block consumer accounts

Author: Peter Steinberger

internal/cmd/chat_dm.go

@@ -29,6 +29,9 @@ func (c *ChatDMSendCmd) Run(ctx context.Context, flags *RootFlags) error {
 	if err != nil {
 		return err
 	}
+	if err := requireWorkspaceAccount(account); err != nil {
+		return err
+	}
 
 	email := strings.TrimSpace(c.Email)
 	if email == "" {
@@ -100,6 +103,9 @@ func (c *ChatDMSpaceCmd) Run(ctx context.Context, flags *RootFlags) error {
 	if err != nil {
 		return err
 	}
+	if err := requireWorkspaceAccount(account); err != nil {
+		return err
+	}
 
 	email := strings.TrimSpace(c.Email)
 	if email == "" {

internal/cmd/chat_helpers.go

@@ -33,6 +33,28 @@ func normalizeUser(resource string) string {
 	return "users/" + user
 }
 
+func requireWorkspaceAccount(account string) error {
+	if isConsumerAccount(account) {
+		return usage("chat requires a Google Workspace account (non-gmail.com)")
+	}
+	return nil
+}
+
+func isConsumerAccount(account string) bool {
+	account = strings.TrimSpace(strings.ToLower(account))
+	at := strings.LastIndex(account, "@")
+	if at == -1 {
+		return false
+	}
+	domain := account[at+1:]
+	switch domain {
+	case "gmail.com", "googlemail.com":
+		return true
+	default:
+		return false
+	}
+}
+
 func normalizeThread(space, resource string) (string, error) {
 	thread := strings.TrimSpace(resource)
 	if thread == "" {

internal/cmd/chat_messages.go

@@ -32,6 +32,9 @@ func (c *ChatMessagesListCmd) Run(ctx context.Context, flags *RootFlags) error {
 	if err != nil {
 		return err
 	}
+	if err := requireWorkspaceAccount(account); err != nil {
+		return err
+	}
 
 	space, err := normalizeSpace(c.Space)
 	if err != nil {
@@ -140,6 +143,9 @@ func (c *ChatMessagesSendCmd) Run(ctx context.Context, flags *RootFlags) error {
 	if err != nil {
 		return err
 	}
+	if err := requireWorkspaceAccount(account); err != nil {
+		return err
+	}
 
 	space, err := normalizeSpace(c.Space)
 	if err != nil {

internal/cmd/chat_spaces.go

@@ -29,6 +29,9 @@ func (c *ChatSpacesListCmd) Run(ctx context.Context, flags *RootFlags) error {
 	if err != nil {
 		return err
 	}
+	if err := requireWorkspaceAccount(account); err != nil {
+		return err
+	}
 
 	svc, err := newChatService(ctx, account)
 	if err != nil {
@@ -103,6 +106,9 @@ func (c *ChatSpacesFindCmd) Run(ctx context.Context, flags *RootFlags) error {
 	if err != nil {
 		return err
 	}
+	if err := requireWorkspaceAccount(account); err != nil {
+		return err
+	}
 
 	displayName := strings.TrimSpace(c.DisplayName)
 	if displayName == "" {
@@ -193,6 +199,9 @@ func (c *ChatSpacesCreateCmd) Run(ctx context.Context, flags *RootFlags) error {
 	if err != nil {
 		return err
 	}
+	if err := requireWorkspaceAccount(account); err != nil {
+		return err
+	}
 
 	displayName := strings.TrimSpace(c.DisplayName)
 	if displayName == "" {

internal/cmd/chat_threads.go

@@ -27,6 +27,9 @@ func (c *ChatThreadsListCmd) Run(ctx context.Context, flags *RootFlags) error {
 	if err != nil {
 		return err
 	}
+	if err := requireWorkspaceAccount(account); err != nil {
+		return err
+	}
 
 	space, err := normalizeSpace(c.Space)
 	if err != nil {

internal/cmd/execute_chat_test.go

@@ -58,6 +58,23 @@ func TestExecute_ChatSpacesList_Text(t *testing.T) {
 	}
 }
 
+func TestExecute_ChatSpacesList_ConsumerBlocked(t *testing.T) {
+	origNew := newChatService
+	t.Cleanup(func() { newChatService = origNew })
+	newChatService = func(context.Context, string) (*chat.Service, error) {
+		t.Fatalf("unexpected chat service call")
+		return nil, nil
+	}
+
+	err := Execute([]string{"--account", "user@gmail.com", "chat", "spaces", "list"})
+	if err == nil {
+		t.Fatalf("expected error")
+	}
+	if !strings.Contains(err.Error(), "Workspace") {
+		t.Fatalf("unexpected error: %v", err)
+	}
+}
+
 func TestExecute_ChatSpacesFind_JSON(t *testing.T) {
 	origNew := newChatService
 	t.Cleanup(func() { newChatService = origNew })