feat: add user-agent (#2)

Author: algirdasci

.goreleaser.yaml

@@ -8,6 +8,11 @@ builds:
     flags:
       - -mod=vendor
       - -v
+    ldflags:
+      - -X github.com/hostinger/api-cli/utils/version.Build={{ .ShortCommit }}
+      - -X github.com/hostinger/api-cli/utils/version.Major={{ .Major }}
+      - -X github.com/hostinger/api-cli/utils/version.Minor={{ .Minor }}
+      - -X github.com/hostinger/api-cli/utils/version.Patch={{ .Patch }}
     goos:
       - windows
       - darwin

api/request.go

@@ -1,10 +1,14 @@
 package api
 
 import (
+	"context"
+	"fmt"
 	"github.com/hostinger/api-cli/client"
+	"github.com/hostinger/api-cli/utils"
 	"github.com/oapi-codegen/oapi-codegen/v2/pkg/securityprovider"
 	"github.com/spf13/viper"
 	"log"
+	"net/http"
 )
 
 func Request() client.ClientWithResponsesInterface {
@@ -18,10 +22,19 @@ func Request() client.ClientWithResponsesInterface {
 		apiUrl = "https://developers.hostinger.com"
 	}
 
-	c, err := client.NewClientWithResponses(apiUrl, client.WithRequestEditorFn(bearerToken.Intercept))
+	c, err := client.NewClientWithResponses(
+		apiUrl,
+		client.WithRequestEditorFn(bearerToken.Intercept),
+		client.WithRequestEditorFn(addUserAgent),
+	)
 	if err != nil {
 		log.Fatal(err)
 	}
 
 	return c
 }
+
+func addUserAgent(ctx context.Context, req *http.Request) error {
+	req.Header.Set("User-Agent", fmt.Sprintf("hostinger-cli/%s", utils.CLIVersion.String(false)))
+	return nil
+}

cmd/root.go

@@ -44,6 +44,7 @@ func init() {
 	})
 
 	RootCmd.AddCommand(vps.GroupCmd)
+	RootCmd.AddCommand(VersionCmd)
 }
 
 func initConfig() {

cmd/version.go

@@ -0,0 +1,15 @@
+package cmd
+
+import (
+	"fmt"
+	"github.com/hostinger/api-cli/utils"
+	"github.com/spf13/cobra"
+)
+
+var VersionCmd = &cobra.Command{
+	Use:   "version",
+	Short: "Show the current version",
+	Run: func(cmd *cobra.Command, args []string) {
+		fmt.Println(utils.CLIVersion.String(true))
+	},
+}

main.go

@@ -1,6 +1,8 @@
 package main
 
-import "github.com/hostinger/api-cli/cmd"
+import (
+	"github.com/hostinger/api-cli/cmd"
+)
 
 func main() {
 	cmd.Execute()

utils/version.go

@@ -0,0 +1,50 @@
+package utils
+
+import (
+	"bytes"
+	"fmt"
+	"strconv"
+)
+
+type Version struct {
+	Major, Minor, Patch int
+	Build               string
+}
+
+var (
+	Build string
+	Major string
+	Minor string
+	Patch string
+
+	CLIVersion Version
+)
+
+func init() {
+	if Build != "" {
+		CLIVersion.Build = Build
+	}
+	if Major != "" {
+		i, _ := strconv.Atoi(Major)
+		CLIVersion.Major = i
+	}
+	if Minor != "" {
+		i, _ := strconv.Atoi(Minor)
+		CLIVersion.Minor = i
+	}
+	if Patch != "" {
+		i, _ := strconv.Atoi(Patch)
+		CLIVersion.Patch = i
+	}
+}
+
+func (v Version) String(withBuild bool) string {
+	var buffer bytes.Buffer
+	if withBuild {
+		buffer.WriteString(fmt.Sprintf("%d.%d.%d (Build: %s)", v.Major, v.Minor, v.Patch, v.Build))
+	} else {
+		buffer.WriteString(fmt.Sprintf("%d.%d.%d", v.Major, v.Minor, v.Patch))
+	}
+
+	return buffer.String()
+}