mounts.go

package httptools

import (
	"net/http"
	"net/url"
	"strings"
)

// Mounts is a list of handlers which are mounted at the given path.
// Mounting is a simple path prefix-based routing. The prefix will be
// stripped from the request before being passed to the associated handler.
// The original request path will available in the VarsResponseWriter
// under "OrigPath".
type Mounts map[string]http.Handler

func (m Mounts) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	vrw, ok := w.(VarsResponseWriter)
	if !ok {
		vrw = newOurResponseWriter(w)
	}
	for path, h := range m {
		path = strings.TrimRight(path, "/")
		if strings.HasPrefix(r.URL.Path, path) {
			// FIXME: This is a shallow copy
			nr := *r
			nr.URL = &url.URL{}
			*nr.URL = *r.URL

			nr.URL.Path = strings.TrimPrefix(nr.URL.Path, path)
			if _, ok := vrw.Vars()["OrigPath"]; !ok {
				vrw.Vars()["OrigPath"] = r.URL.Path
			}
			if h != nil {
				h.ServeHTTP(vrw, &nr)
			}
			return
		}
	}
	http.Error(vrw, "Not found", http.StatusNotFound)
}