Send CLI user-agent and print messages

Author: gschier

api.go

@@ -2,6 +2,7 @@ package yaakcli
 
 import (
 	"encoding/json"
+	"fmt"
 	"github.com/pterm/pterm"
 	"io"
 	"net/http"
@@ -30,6 +31,7 @@ func SendAPIRequest(r *http.Request) []byte {
 	}
 
 	r.Header.Set("X-Yaak-Session", token)
+	r.Header.Set("User-Agent", fmt.Sprintf("YaakCli/%s (%s)", CLIVersion, GetUAPlatform()))
 
 	client := &http.Client{Timeout: 10 * time.Second}
 	resp, err := client.Do(r)
@@ -52,6 +54,11 @@ func SendAPIRequest(r *http.Request) []byte {
 		os.Exit(1)
 	}
 
+	message := resp.Header.Get("X-Cli-Message")
+	if message != "" {
+		pterm.Info.Printf(message)
+	}
+
 	return body
 }
 

cmd_root.go

@@ -5,15 +5,18 @@ import (
 	"os"
 )
 
-func rootCmd(version string) *cobra.Command {
+var CLIVersion string
+
+func rootCmd(v string) *cobra.Command {
+	CLIVersion = v
 	var fVersion bool
 	cmd := &cobra.Command{
 		Use:   "yaakcli",
 		Short: "Develop plugins for Yaak",
 		Long:  "Generate, build, and debug plugins for Yaak, the most intuitive desktop API client",
 		Run: func(cmd *cobra.Command, args []string) {
 			if fVersion {
-				println(version)
+				println(CLIVersion)
 				os.Exit(0)
 			}
 

util.go

@@ -3,6 +3,7 @@ package yaakcli
 import (
 	"os"
 	"path"
+	"runtime"
 	"strings"
 )
 
@@ -39,3 +40,16 @@ func prodStagingDevStr(prod, staging, dev string) string {
 		return prod
 	}
 }
+
+func GetUAPlatform() string {
+	switch runtime.GOOS {
+	case "windows":
+		return "Win"
+	case "darwin":
+		return "Mac"
+	case "linux":
+		return "Linux"
+	default:
+		return "Unknown"
+	}
+}