Merge pull request #27 from spiral/feature/arguments

Feature/arguments

1 files changed, 10 insertions, 1 deletions

diff --git a/service/static/service.go b/service/static/service.go
index 2324dcd1..add242e4 100644
--- a/service/static/service.go
+++ b/service/static/service.go
@@ -56,7 +56,16 @@ func (s *Service) Serve() error { return nil }
 func (s *Service) Stop() {}
 
 // middleware must return true if request/response pair is handled within the middleware.
-func (s *Service) middleware(w http.ResponseWriter, r *http.Request) bool {
+func (s *Service) middleware(f http.HandlerFunc) http.HandlerFunc {
+	// Define the http.HandlerFunc
+	return func(w http.ResponseWriter, r *http.Request) {
+		if !s.handleStatic(w, r) {
+			f(w, r)
+		}
+	}
+}
+
+func (s *Service) handleStatic(w http.ResponseWriter, r *http.Request) bool {
 	fPath := r.URL.Path
 
 	if !strings.HasPrefix(fPath, "/") {