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)
		}
	}
}