Add auth revoke command

Author: nmarcopo

CHANGELOG.md

@@ -4,7 +4,8 @@
 
 Added:
 * Adds support for pool PriorityThresholds in `v1/config/ResourcePools`
-* Adds `chronoctl auth whoami` which prints the current user
+* Adds `chronoctl auth whoami` which prints the current user 
+* Adds `chronoctl auth revoke` which revokes a session id
 
 ## v1.11.0
 

src/cmd/pkg/auth/auth.go

@@ -18,6 +18,7 @@ package auth
 import (
 	"encoding/json"
 	"fmt"
+	"io"
 	"io/ioutil"
 	"net/http"
 
@@ -51,6 +52,7 @@ func NewCommand() *cobra.Command {
 		c.newPrintAccessTokenCmd(),
 		c.newListCmd(),
 		c.newWhoAmICmd(),
+		c.newRevokeCmd(),
 	)
 	return root
 }
@@ -234,3 +236,35 @@ func (c *subcommand) newWhoAmICmd() *cobra.Command {
 	authFlags.AddFlags(cmd)
 	return cmd
 }
+
+func (c *subcommand) newRevokeCmd() *cobra.Command {
+	authFlags := client.NewClientFlags()
+	cmd := &cobra.Command{
+		Use:     "revoke",
+		GroupID: groups.Commands.ID,
+		Short:   "Revoke a session id.",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			req, err := authFlags.NewRequest(http.MethodGet, "/auth/logout", nil /* body */)
+			if err != nil {
+				return errors.WithStack(err)
+			}
+			client := http.Client{}
+			resp, err := client.Do(req)
+			if err != nil {
+				return errors.WithStack(err)
+			}
+			defer resp.Body.Close() //nolint: errcheck
+			body, err := io.ReadAll(resp.Body)
+			if err != nil {
+				return errors.WithStack(err)
+			}
+			if resp.StatusCode != http.StatusOK {
+				return errors.Errorf("%d: %s", resp.StatusCode, string(body))
+			}
+			cmd.OutOrStdout().Write(body) //nolint: errcheck
+			return nil
+		},
+	}
+	authFlags.AddFlags(cmd)
+	return cmd
+}

src/cmd/pkg/auth/auth_test.go

@@ -651,3 +651,30 @@ func TestAuthWhoAmI(t *testing.T) {
 	}
 
 }
+
+func TestAuthRevoke(t *testing.T) {
+	apiToken := "test-token"
+
+	requestReceived := false
+	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		requestReceived = true
+		require.Equal(t, http.MethodGet, r.Method)
+		require.Equal(t, "/auth/logout", r.URL.Path)
+		require.Equal(t, apiToken, r.Header.Get("API-Token"))
+
+		w.WriteHeader(http.StatusOK)
+	}))
+	t.Cleanup(ts.Close)
+
+	store := token.NewFileStore(t.TempDir())
+	c := subcommand{store: store}
+	cmd := c.newRevokeCmd()
+
+	require.NoError(t, cmd.Flags().Set("api-url", ts.URL+"/auth/logout"))
+	require.NoError(t, cmd.Flags().Set("api-token", apiToken))
+
+	stdout := &bytes.Buffer{}
+	cmd.SetOut(stdout)
+	require.NoError(t, cmd.Execute())
+	require.True(t, requestReceived)
+}