chore: add user agent in requests with mcp version (#58)

Author: tembleking

Dockerfile

@@ -12,7 +12,7 @@ RUN nix build .#default
 
 # Final image
 # quay.io/sysdig/sysdig-mini-ubi9:1
-FROM quay.io/sysdig/sysdig-mini-ubi9@sha256:dcef7a07dc6a8655cbee5e2f3ad7822dea5a0cf4929b1b9effa39e56ce928ca0
+FROM quay.io/sysdig/sysdig-mini-ubi9@sha256:fa959bec7353ce38ed2d35190d5ec4445b3c525f1d4ad00e26afca387360a253
 
 # Copy the binary from the builder stage
 COPY --from=builder /app/result/bin/sysdig-mcp-server /usr/local/bin/sysdig-mcp-server

cmd/server/main.go

@@ -83,6 +83,7 @@ func setupLogger(logLevel string) {
 
 func setupSysdigClient(cfg *config.Config) (sysdig.ExtendedClientWithResponsesInterface, error) {
 	sysdigClient, err := sysdig.NewSysdigClient(
+		sysdig.WithVersion(Version),
 		sysdig.WithFallbackAuthentication(
 			sysdig.WithHostAndTokenFromContext(),
 			sysdig.WithFixedHostAndToken(cfg.APIHost, cfg.APIToken),

internal/infra/sysdig/client.go

@@ -3,12 +3,11 @@ package sysdig
 import (
 	"context"
 	"errors"
+	"fmt"
 	"net/http"
 	"net/url"
 )
 
-type SysdigAuthentication func(ctx context.Context, req *http.Request) error
-
 type contextKey string
 
 const (
@@ -46,7 +45,7 @@ func updateReqWithHostURL(req *http.Request, host string) error {
 	return nil
 }
 
-func WithFixedHostAndToken(host, apiToken string) SysdigAuthentication {
+func WithFixedHostAndToken(host, apiToken string) RequestEditorFn {
 	return func(ctx context.Context, req *http.Request) error {
 		if err := updateReqWithHostURL(req, host); err != nil {
 			return err
@@ -56,7 +55,7 @@ func WithFixedHostAndToken(host, apiToken string) SysdigAuthentication {
 	}
 }
 
-func WithHostAndTokenFromContext() SysdigAuthentication {
+func WithHostAndTokenFromContext() RequestEditorFn {
 	return func(ctx context.Context, req *http.Request) error {
 		if host, ok := ctx.Value(contextKeyHost).(string); ok && host != "" {
 			if err := updateReqWithHostURL(req, host); err != nil {
@@ -71,7 +70,7 @@ func WithHostAndTokenFromContext() SysdigAuthentication {
 	}
 }
 
-func WithFallbackAuthentication(auths ...SysdigAuthentication) SysdigAuthentication {
+func WithFallbackAuthentication(auths ...RequestEditorFn) RequestEditorFn {
 	return func(ctx context.Context, req *http.Request) error {
 		for _, auth := range auths {
 			if err := auth(ctx, req); err == nil {
@@ -82,10 +81,17 @@ func WithFallbackAuthentication(auths ...SysdigAuthentication) SysdigAuthenticat
 	}
 }
 
-func NewSysdigClient(requestEditors ...SysdigAuthentication) (ExtendedClientWithResponsesInterface, error) {
+func WithVersion(version string) RequestEditorFn {
+	return func(ctx context.Context, req *http.Request) error {
+		req.Header.Set("User-Agent", fmt.Sprintf("sysdig-mcp-server/%s", version))
+		return nil
+	}
+}
+
+func NewSysdigClient(requestEditors ...RequestEditorFn) (ExtendedClientWithResponsesInterface, error) {
 	editors := make([]ClientOption, len(requestEditors))
 	for i, e := range requestEditors {
-		editors[i] = WithRequestEditorFn(RequestEditorFn(e))
+		editors[i] = WithRequestEditorFn(e)
 	}
 
 	return NewClientWithResponses("", editors...)

package.nix

@@ -1,7 +1,7 @@
 { buildGoModule, versionCheckHook }:
 buildGoModule (finalAttrs: {
   pname = "sysdig-mcp-server";
-  version = "0.5.3";
+  version = "0.5.4";
   src = ./.;
   # This hash is automatically re-calculated with `just rehash-package-nix`. This is automatically called as well by `just bump`.
   vendorHash = "sha256-jf/px0p88XbfuSPMry/qZcfR0QPTF9IrPegg2CwAd6M=";