Merge pull request #682 from weingart/mime_fix

mime: Use a map, and error on duplicate extensions.

caddy/setup/mime.go

@@ -20,8 +20,8 @@ func Mime(c *Controller) (middleware.Middleware, error) {
 	}, nil
 }
 
-func mimeParse(c *Controller) ([]mime.Config, error) {
-	var configs []mime.Config
+func mimeParse(c *Controller) (mime.Config, error) {
+	configs := mime.Config{}
 
 	for c.Next() {
 		// At least one extension is required
@@ -29,22 +29,22 @@ func mimeParse(c *Controller) ([]mime.Config, error) {
 		args := c.RemainingArgs()
 		switch len(args) {
 		case 2:
-			if err := validateExt(args[0]); err != nil {
+			if err := validateExt(configs, args[0]); err != nil {
 				return configs, err
 			}
-			configs = append(configs, mime.Config{Ext: args[0], ContentType: args[1]})
+			configs[args[0]] = args[1]
 		case 1:
 			return configs, c.ArgErr()
 		case 0:
 			for c.NextBlock() {
 				ext := c.Val()
-				if err := validateExt(ext); err != nil {
+				if err := validateExt(configs, ext); err != nil {
 					return configs, err
 				}
 				if !c.NextArg() {
 					return configs, c.ArgErr()
 				}
-				configs = append(configs, mime.Config{Ext: ext, ContentType: c.Val()})
+				configs[ext] = c.Val()
 			}
 		}
 
@@ -54,9 +54,12 @@ func mimeParse(c *Controller) ([]mime.Config, error) {
 }
 
 // validateExt checks for valid file name extension.
-func validateExt(ext string) error {
+func validateExt(configs mime.Config, ext string) error {
 	if !strings.HasPrefix(ext, ".") {
 		return fmt.Errorf(`mime: invalid extension "%v" (must start with dot)`, ext)
 	}
+	if _, ok := configs[ext]; ok {
+		return fmt.Errorf(`mime: duplicate extension "%v" found`, ext)
+	}
 	return nil
 }

caddy/setup/mime_test.go

@@ -42,6 +42,11 @@ func TestMime(t *testing.T) {
 		 .html text/html
 		 .txt text/plain
 		} `, false},
+		{`mime {
+		 .foo text/foo
+		 .bar text/bar
+		 .foo text/foobar
+		} `, true},
 		{`mime { .html text/html } `, false},
 		{`mime { .html
 		} `, true},

middleware/mime/mime.go

@@ -2,40 +2,31 @@ package mime
 
 import (
 	"net/http"
-	"path/filepath"
+	"path"
 
 	"github.com/mholt/caddy/middleware"
 )
 
-// Config represent a mime config.
-type Config struct {
-	Ext         string
-	ContentType string
-}
-
-// SetContent sets the Content-Type header of the request if the request path
-// is supported.
-func (c Config) SetContent(w http.ResponseWriter, r *http.Request) bool {
-	ext := filepath.Ext(r.URL.Path)
-	if ext != c.Ext {
-		return false
-	}
-	w.Header().Set("Content-Type", c.ContentType)
-	return true
-}
+// Config represent a mime config.  Map from extension to mime-type.
+// Note, this should be safe with concurrent read access, as this is
+// not modified concurrently.
+type Config map[string]string
 
 // Mime sets Content-Type header of requests based on configurations.
 type Mime struct {
 	Next    middleware.Handler
-	Configs []Config
+	Configs Config
 }
 
 // ServeHTTP implements the middleware.Handler interface.
 func (e Mime) ServeHTTP(w http.ResponseWriter, r *http.Request) (int, error) {
-	for _, c := range e.Configs {
-		if ok := c.SetContent(w, r); ok {
-			break
-		}
+
+	// Get a clean /-path, grab the extension
+	ext := path.Ext(path.Clean(r.URL.Path))
+
+	if contentType, ok := e.Configs[ext]; ok {
+		w.Header().Set("Content-Type", contentType)
 	}
+
 	return e.Next.ServeHTTP(w, r)
 }

middleware/mime/mime_test.go

@@ -11,18 +11,13 @@ import (
 
 func TestMimeHandler(t *testing.T) {
 
-	mimes := map[string]string{
+	mimes := Config{
 		".html": "text/html",
 		".txt":  "text/plain",
 		".swf":  "application/x-shockwave-flash",
 	}
 
-	var configs []Config
-	for ext, contentType := range mimes {
-		configs = append(configs, Config{Ext: ext, ContentType: contentType})
-	}
-
-	m := Mime{Configs: configs}
+	m := Mime{Configs: mimes}
 
 	w := httptest.NewRecorder()
 	exts := []string{