Add version to headers

1 files changed, 16 insertions, 0 deletions

diff --git a/app/middleware/server_header.go b/app/middleware/server_header.go
new file mode 100644
index 0000000..7ccbbda
--- /dev/null
+++ b/app/middleware/server_header.go
@@ -0,0 +1,16 @@
+package middleware
+
+import (
+        "fmt"
+
+        "github.com/labstack/echo/v4"
+)
+
+func AddServerHeader(version string) echo.MiddlewareFunc {
+        return func(next echo.HandlerFunc) echo.HandlerFunc {
+                return func(c echo.Context) error {
+                        c.Response().Header().Add("Server", fmt.Sprintf("cloud-identity-broker/%s", version))
+                        return next(c)
+                }
+        }
+}