Merge #345

345: Adding headers to static files r=48d90782 a=siad007

Fixes #336 
Related to #335 

Co-authored-by: Siad Ardroumli <[email protected]>
Co-authored-by: Valery Piashchynski <[email protected]>

1 files changed, 12 insertions, 0 deletions

diff --git a/service/static/service.go b/service/static/service.go
index b824e787..95b99860 100644
--- a/service/static/service.go
+++ b/service/static/service.go
@@ -36,6 +36,18 @@ func (s *Service) Init(cfg *Config, r *rrhttp.Service) (bool, error) {
 func (s *Service) middleware(f http.HandlerFunc) http.HandlerFunc {
 	// Define the http.HandlerFunc
 	return func(w http.ResponseWriter, r *http.Request) {
+		if s.cfg.Request != nil {
+			for k, v := range s.cfg.Request {
+				r.Header.Add(k, v)
+			}
+		}
+
+		if s.cfg.Response != nil {
+			for k, v := range s.cfg.Response {
+				w.Header().Set(k, v)
+			}
+		}
+
 		if !s.handleStatic(w, r) {
 			f(w, r)
 		}