Add an API middleware to check for updates on API calls (#1067)

Signed-off-by: lujunsan <[email protected]>

Author: lujunsan

pkg/api/server.go

@@ -32,6 +32,7 @@ import (
 	"github.com/stacklok/toolhive/pkg/container"
 	"github.com/stacklok/toolhive/pkg/logger"
 	"github.com/stacklok/toolhive/pkg/registry"
+	"github.com/stacklok/toolhive/pkg/updates"
 	"github.com/stacklok/toolhive/pkg/workloads"
 )
 
@@ -88,6 +89,28 @@ func headersMiddleware(next http.Handler) http.Handler {
 	})
 }
 
+// updateCheckMiddleware triggers update checks for API usage
+func updateCheckMiddleware() func(next http.Handler) http.Handler {
+	return func(next http.Handler) http.Handler {
+		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			go func() {
+				versionClient := updates.NewVersionClient()
+				updateChecker, err := updates.NewUpdateChecker(versionClient)
+				if err != nil {
+					logger.Warnf("unable to create update client for API: %s", err)
+					return
+				}
+
+				err = updateChecker.CheckLatestVersion()
+				if err != nil {
+					logger.Warnf("could not check for updates for API: %s", err)
+				}
+			}()
+			next.ServeHTTP(w, r)
+		})
+	}
+}
+
 // Serve starts the server on the given address and serves the API.
 // It is assumed that the caller sets up appropriate signal handling.
 // If isUnixSocket is true, address is treated as a UNIX socket path.
@@ -108,6 +131,9 @@ func Serve(
 		headersMiddleware,
 	)
 
+	// Add update check middleware
+	r.Use(updateCheckMiddleware())
+
 	// Add authentication middleware
 	authMiddleware, err := auth.GetAuthenticationMiddleware(ctx, oidcConfig, false)
 	if err != nil {