Refactored fastcgi middleware

middleware/fastcgi/fastcgi.go

@@ -1,6 +1,6 @@
 // Package fastcgi has middleware that acts as a FastCGI client. Requests
 // that get forwarded to FastCGI stop the middleware execution chain.
-// The most common use for this layer is to serve PHP websites via php-fpm.
+// The most common use for this package is to serve PHP websites via php-fpm.
 package fastcgi
 
 import (
@@ -17,31 +17,38 @@ import (
 func New(c middleware.Controller) (middleware.Middleware, error) {
 	root := c.Root()
 
-	var rules []fastCgi
-	for c.Next() {
-		rule := fastCgi{}
-		if !c.Args(&rule.path, &rule.address) {
-			return nil, c.ArgErr()
-		}
-		rules = append(rules, rule)
+	rules, err := parse(c)
+	if err != nil {
+		return nil, err
 	}
 
 	return func(next middleware.Handler) middleware.Handler {
-		return middleware.HandlerFunc(func(w http.ResponseWriter, r *http.Request) (int, error) {
+		return Handler{Next: next, Rules: rules, Root: root}
+	}, nil
+}
 
+// Handler is a middleware type that can handle requests as a FastCGI client.
+type Handler struct {
+	Next  middleware.Handler
+	Root  string
+	Rules []Rule
+}
+
+// ServeHTTP satisfies the middleware.Handler interface.
+func (h Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) (int, error) {
 	servedFcgi := false
-			for _, rule := range rules {
-				if middleware.Path(r.URL.Path).Matches(rule.path) {
+	for _, rule := range h.Rules {
+		if middleware.Path(r.URL.Path).Matches(rule.Path) {
 			servedFcgi = true
 
 			// Get absolute file paths
-					absPath, err := filepath.Abs(root + r.URL.Path)
+			absPath, err := filepath.Abs(h.Root + r.URL.Path)
 			if err != nil {
 				return http.StatusInternalServerError, err
 			}
 
 			// Get absolute file path to website root
-					absRootPath, err := filepath.Abs(root)
+			absRootPath, err := filepath.Abs(h.Root)
 			if err != nil {
 				return http.StatusInternalServerError, err
 			}
@@ -68,7 +75,7 @@ func New(c middleware.Controller) (middleware.Middleware, error) {
 			env["DOCUMENT_URI"] = r.URL.Path
 			env["DOCUMENT_ROOT"] = absRootPath
 
-					fcgi, err := Dial("tcp", rule.address)
+			fcgi, err := Dial("tcp", rule.Address)
 			if err != nil {
 				return http.StatusBadGateway, err
 			}
@@ -97,15 +104,27 @@ func New(c middleware.Controller) (middleware.Middleware, error) {
 	}
 
 	if !servedFcgi {
-				return next.ServeHTTP(w, r)
+		return h.Next.ServeHTTP(w, r)
 	}
 
 	return 0, nil
-		})
-	}, nil
 }
 
-type fastCgi struct {
-	path    string
-	address string
+func parse(c middleware.Controller) ([]Rule, error) {
+	var rules []Rule
+
+	for c.Next() {
+		var rule Rule
+		if !c.Args(&rule.Path, &rule.Address) {
+			return rules, c.ArgErr()
+		}
+		rules = append(rules, rule)
+	}
+
+	return rules, nil
+}
+
+// Rule represents a FastCGI handling rule.
+type Rule struct {
+	Path, Address string
 }

middleware/redirect/redirect.go

@@ -9,7 +9,7 @@ import (
 	"github.com/mholt/caddy/middleware"
 )
 
-// New instantiates a new Rewrites middleware.
+// New instantiates a new Redirect middleware.
 func New(c middleware.Controller) (middleware.Middleware, error) {
 	rules, err := parse(c)
 	if err != nil {