Fix token rotation deprecation

humio · Sep 20, 2023 · 63dcaec · 63dcaec
1 parent e7af2c8
commit 63dcaec
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/api/users.go b/api/users.go
@@ -104,13 +104,9 @@ func (u *Users) Remove(username string) (User, error) {
 	return mutation.Result.User, err
 }
 
-func (u *Users) RotateUserApiTokenAndGet(userID string) (string, error) {
+func (u *Users) RotateToken(userID string) (string, error) {
 	var mutation struct {
-		RotateUserApiTokenMutation struct {
-			RotateUserApiToken struct {
-				Token string
-			} `graphql:"rotateUserApiToken"`
-		} `graphql:"rotateUserApiTokenAndGet(input:{id:$id})"`
+		Token string `graphql:"rotateToken(input:{id:$id})"`
 	}
 
 	variables := map[string]interface{}{
@@ -122,7 +118,7 @@ func (u *Users) RotateUserApiTokenAndGet(userID string) (string, error) {
 		return "", err
 	}
 
-	return mutation.RotateUserApiTokenMutation.RotateUserApiToken.Token, nil
+	return mutation.Token, nil
 }
 
 func userChangesetToVars(username string, changeset UserChangeSet) map[string]interface{} {

diff --git a/cmd/humioctl/users_rotate_api_token.go b/cmd/humioctl/users_rotate_api_token.go
@@ -13,7 +13,7 @@ func newUsersRotateApiTokenCmd() *cobra.Command {
 			userID := args[0]
 
 			client := NewApiClient(cmd)
-			newToken, apiErr := client.Users().RotateUserApiTokenAndGet(userID)
+			newToken, apiErr := client.Users().RotateToken(userID)
 			exitOnError(cmd, apiErr, "Error updating user")
 
 			cmd.Printf("New API Token: %s\n", newToken)