status shows user handle

Author: gerep

client/auth.go

@@ -20,6 +20,10 @@ type LoginResponse struct {
 	RefreshToken string `json:"refresh_token"`
 }
 
+type CurrentUserResponse struct {
+	Handle string `json:"handle"`
+}
+
 func FetchAccessToken() (*LoginResponse, error) {
 	api_url := viper.GetString("api_url")
 	client := &http.Client{}
@@ -82,6 +86,21 @@ func LoginWithCode(code string) (*LoginResponse, error) {
 	return &creds, nil
 }
 
+func FetchCurrentUser() (*CurrentUserResponse, error) {
+	body, err := fetchWithAuth("GET", "/v1/users/me")
+	if err != nil {
+		return nil, err
+	}
+
+	var user CurrentUserResponse
+	err = json.Unmarshal(body, &user)
+	if err != nil {
+		return nil, err
+	}
+
+	return &user, nil
+}
+
 func fetchWithAuth(method string, url string) ([]byte, error) {
 	body, code, err := fetchWithAuthAndPayload(method, url, []byte{})
 	if err != nil {

cmd/status.go

@@ -36,8 +36,13 @@ func checkAuthStatus() {
 		return
 	}
 
+	user, err := api.FetchCurrentUser()
+	if err == nil && user != nil && user.Handle != "" {
+		fmt.Printf("Logged in as @%s\n", user.Handle)
+		return
+	}
+
 	fmt.Println("Logged in")
-	// TODO: Consider adding user data endpoint to show email/username
 }
 
 func checkVersionStatus(cmd *cobra.Command) {