fixup! cli: Update UXState from config-ssh

greyscaled · greyscaled · commit e13657fd056c · 2020-12-02T19:44:27.000Z
diff --git a/coder-sdk/users.go b/coder-sdk/users.go
@@ -46,14 +46,6 @@ func (c Client) Me(ctx context.Context) (*User, error) {
 	return c.UserByID(ctx, Me)
 }
 
-// MyUXState updates the authenticated user's UX State.
-func (c Client) MyUXState(ctx context.Context, uxsPartial map[string]interface{}) error {
-	if err := c.requestBody(ctx, http.MethodPut, "/api/private/users/me/ux-state", uxsPartial, nil); err != nil {
-		return err
-	}
-	return nil
-}
-
 // UserByID get the details of a user by their id.
 func (c Client) UserByID(ctx context.Context, id string) (*User, error) {
 	var u User
@@ -122,6 +114,14 @@ func (c Client) UpdateUser(ctx context.Context, userID string, req UpdateUserReq
 	return c.requestBody(ctx, http.MethodPatch, "/api/private/users/"+userID, req, nil)
 }
 
+// UpdateUXState updates the authenticated user's UX State.
+func (c Client) UpdateUXState(ctx context.Context, userID string, uxsPartial map[string]interface{}) error {
+	if err := c.requestBody(ctx, http.MethodPut, "/api/private/users/"+userID+"/ux-state", uxsPartial, nil); err != nil {
+		return err
+	}
+	return nil
+}
+
 // CreateUserReq defines the request parameters for creating a new user resource.
 type CreateUserReq struct {
 	Name              string    `json:"name"`
diff --git a/internal/cmd/configssh.go b/internal/cmd/configssh.go
@@ -1,6 +1,7 @@
 package cmd
 
 import (
+	"cdr.dev/coder-cli/pkg/clog"
 	"context"
 	"fmt"
 	"io/ioutil"
@@ -135,7 +136,7 @@ func configSSH(configpath *string, remove *bool) func(cmd *cobra.Command, _ []st
 			fmt.Printf("Your private ssh key was written to \"%s\"\n", privateKeyFilepath)
 		}
 
-		writeSSHUXState(ctx, client)
+		writeSSHUXState(ctx, client, user.ID)
 		fmt.Printf("An auto-generated ssh config was written to \"%s\"\n", *configpath)
 		fmt.Println("You should now be able to ssh into your environment")
 		fmt.Printf("For example, try running\n\n\t$ ssh coder.%s\n\n", envs[0].Name)
@@ -217,11 +218,11 @@ func readStr(filename string) (string, error) {
 	return string(contents), nil
 }
 
-func writeSSHUXState(ctx context.Context, client *coder.Client) {
+func writeSSHUXState(ctx context.Context, client *coder.Client, userID string) {
 	// Update UXState that coder config-ssh has been run by the currently
 	// authenticated user
-	err := client.MyUXState(ctx, map[string]interface{}{"cliSSHConfigured": true})
+	err := client.UpdateUXState(ctx, userID, map[string]interface{}{"cliSSHConfigured": true})
 	if err != nil {
-		fmt.Println("⚠ The Coder web client may not recognize that you've configured SSH.")
+		clog.LogWarn("The Coder web client may not recognize that you've configured SSH.")
 	}
 }
