chore(cli): support deleting tokens by id

ethanndickson · ethanndickson · commit 294916478c73 · 2025-02-03T06:53:07.000Z
diff --git a/cli/testdata/coder_tokens_remove_--help.golden b/cli/testdata/coder_tokens_remove_--help.golden
@@ -1,11 +1,15 @@
 coder v0.0.0-devel
 
 USAGE:
-  coder tokens remove <name>
+  coder tokens remove [flags] <name|id|token>
 
   Delete a token
 
   Aliases: delete, rm
 
+OPTIONS:
+  -y, --yes bool
+          Bypass prompts.
+
 ———
 Run `coder --help` for a list of global options.
diff --git a/cli/tokens.go b/cli/tokens.go
@@ -3,13 +3,15 @@ package cli
 import (
 	"fmt"
 	"os"
+	"strings"
 	"time"
 
 	"golang.org/x/exp/slices"
 	"golang.org/x/xerrors"
 
 	"github.com/coder/coder/v2/cli/cliui"
 	"github.com/coder/coder/v2/codersdk"
+	"github.com/coder/pretty"
 	"github.com/coder/serpent"
 )
 
@@ -223,17 +225,52 @@ func (r *RootCmd) listTokens() *serpent.Command {
 func (r *RootCmd) removeToken() *serpent.Command {
 	client := new(codersdk.Client)
 	cmd := &serpent.Command{
-		Use:     "remove <name>",
+		Use:     "remove <name|id|token>",
 		Aliases: []string{"delete"},
 		Short:   "Delete a token",
 		Middleware: serpent.Chain(
 			serpent.RequireNArgs(1),
 			r.InitClient(client),
 		),
+		Options: serpent.OptionSet{
+			cliui.SkipPromptOption(),
+		},
 		Handler: func(inv *serpent.Invocation) error {
 			token, err := client.APIKeyByName(inv.Context(), codersdk.Me, inv.Args[0])
 			if err != nil {
-				return xerrors.Errorf("fetch api key by name %s: %w", inv.Args[0], err)
+				// If it's a token, we need to extract the ID
+				maybeID := strings.Split(inv.Args[0], "-")[0]
+				token, err = client.APIKeyByID(inv.Context(), codersdk.Me, maybeID)
+				if err != nil {
+					return xerrors.Errorf("fetch api key by name or id: %w", err)
+				}
+			}
+
+			var prompt string
+			if token.TokenName == "" {
+				prompt = fmt.Sprintf("Are you sure you want to delete the token with ID %s?\n  ",
+					pretty.Sprint(cliui.DefaultStyles.Code, token.ID),
+				)
+			} else {
+				prompt = fmt.Sprintf("Are you sure you want to delete the token with the name %s? (ID: %s)\n  ",
+					pretty.Sprint(cliui.DefaultStyles.Code, token.TokenName),
+					pretty.Sprint(cliui.DefaultStyles.Code, token.ID),
+				)
+			}
+
+			if !token.LastUsed.IsZero() {
+				prompt = fmt.Sprintf("%sIt was last used on %s.", prompt, pretty.Sprint(cliui.DefaultStyles.Code, token.LastUsed.String()))
+			} else {
+				prompt = fmt.Sprintf("%sIt has never been used.", prompt)
+			}
+
+			_, err = cliui.Prompt(inv, cliui.PromptOptions{
+				Text:      prompt,
+				IsConfirm: true,
+				Default:   cliui.ConfirmYes,
+			})
+			if err != nil {
+				return err
 			}
 
 			err = client.DeleteAPIKey(inv.Context(), codersdk.Me, token.ID)
diff --git a/cli/tokens_test.go b/cli/tokens_test.go
@@ -93,7 +93,7 @@ func TestTokens(t *testing.T) {
 	require.Contains(t, res, secondTokenID)
 
 	// Test creating a token for third user from second user's (non-admin) session
-	inv, root = clitest.New(t, "tokens", "create", "--name", "token-two", "--user", thirdUser.ID.String())
+	inv, root = clitest.New(t, "tokens", "create", "--name", "failed-token", "--user", thirdUser.ID.String())
 	clitest.SetupConfig(t, secondUserClient, root)
 	buf = new(bytes.Buffer)
 	inv.Stdout = buf
@@ -113,7 +113,41 @@ func TestTokens(t *testing.T) {
 	require.Len(t, tokens, 1)
 	require.Equal(t, id, tokens[0].ID)
 
-	inv, root = clitest.New(t, "tokens", "rm", "token-one")
+	// Delete by name
+	inv, root = clitest.New(t, "tokens", "rm", "token-one", "--yes")
+	clitest.SetupConfig(t, client, root)
+	buf = new(bytes.Buffer)
+	inv.Stdout = buf
+	err = inv.WithContext(ctx).Run()
+	require.NoError(t, err)
+	res = buf.String()
+	require.NotEmpty(t, res)
+	require.Contains(t, res, "deleted")
+
+	// Delete by ID
+	inv, root = clitest.New(t, "tokens", "rm", secondTokenID, "--yes")
+	clitest.SetupConfig(t, client, root)
+	buf = new(bytes.Buffer)
+	inv.Stdout = buf
+	err = inv.WithContext(ctx).Run()
+	require.NoError(t, err)
+	res = buf.String()
+	require.NotEmpty(t, res)
+	require.Contains(t, res, "deleted")
+
+	// Create third token
+	inv, root = clitest.New(t, "tokens", "create", "--name", "token-three")
+	clitest.SetupConfig(t, client, root)
+	buf = new(bytes.Buffer)
+	inv.Stdout = buf
+	err = inv.WithContext(ctx).Run()
+	require.NoError(t, err)
+	res = buf.String()
+	require.NotEmpty(t, res)
+	fourthToken := res
+
+	// Delete by token
+	inv, root = clitest.New(t, "tokens", "rm", fourthToken, "--yes")
 	clitest.SetupConfig(t, client, root)
 	buf = new(bytes.Buffer)
 	inv.Stdout = buf
diff --git a/docs/reference/cli/tokens_remove.md b/docs/reference/cli/tokens_remove.md
