Add support for extra header

Signed-off-by: Bryan Frimin <bryan@getprobo.com>

pkg/probod/api_config.go

@@ -20,7 +20,8 @@ type (
 	}
 
 	apiConfig struct {
-		Addr string     `json:"addr"`
-		Cors corsConfig `json:"cors"`
+		Addr              string            `json:"addr"`
+		Cors              corsConfig        `json:"cors"`
+		ExtraHeaderFields map[string]string `json:"extra-header-fields"`
 	}
 )

pkg/probod/probod.go

@@ -224,6 +224,7 @@ func (impl *Implm) Run(
 	serverHandler, err := server.NewServer(
 		server.Config{
 			AllowedOrigins:    impl.cfg.Api.Cors.AllowedOrigins,
+			ExtraHeaderFields: impl.cfg.Api.ExtraHeaderFields,
 			Probo:             proboService,
 			Usrmgr:            usrmgrService,
 			ConnectorRegistry: defaultConnectorRegistry,

pkg/server/server.go

@@ -34,6 +34,7 @@ import (
 // Config holds the configuration for the server
 type Config struct {
 	AllowedOrigins    []string
+	ExtraHeaderFields map[string]string
 	Probo             *probo.Service
 	Usrmgr            *usrmgr.Service
 	Auth              console_v1.AuthConfig
@@ -45,9 +46,10 @@ type Config struct {
 
 // Server represents the main server that handles both API and frontend requests
 type Server struct {
-	apiServer *api.Server
-	webServer *web.Server
-	router    *chi.Mux
+	apiServer         *api.Server
+	webServer         *web.Server
+	router            *chi.Mux
+	extraHeaderFields map[string]string
 }
 
 // NewServer creates a new server instance
@@ -77,9 +79,10 @@ func NewServer(cfg Config) (*Server, error) {
 	router := chi.NewRouter()
 
 	server := &Server{
-		apiServer: apiServer,
-		webServer: webServer,
-		router:    router,
+		apiServer:         apiServer,
+		webServer:         webServer,
+		router:            router,
+		extraHeaderFields: cfg.ExtraHeaderFields,
 	}
 
 	// Set up routes
@@ -106,5 +109,9 @@ func (s *Server) setupRoutes() {
 
 // ServeHTTP implements the http.Handler interface
 func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+	for key, value := range s.extraHeaderFields {
+		w.Header().Set(key, value)
+	}
+
 	s.router.ServeHTTP(w, r)
 }