diff options
Diffstat (limited to 'vendor/_nuts/github.com/go-martini/martini/static.go')
| -rw-r--r-- | vendor/_nuts/github.com/go-martini/martini/static.go | 135 | 
1 files changed, 135 insertions, 0 deletions
| diff --git a/vendor/_nuts/github.com/go-martini/martini/static.go b/vendor/_nuts/github.com/go-martini/martini/static.go new file mode 100644 index 0000000..51af6cf --- /dev/null +++ b/vendor/_nuts/github.com/go-martini/martini/static.go @@ -0,0 +1,135 @@ +package martini + +import ( +	"log" +	"net/http" +	"net/url" +	"path" +	"path/filepath" +	"strings" +) + +// StaticOptions is a struct for specifying configuration options for the martini.Static middleware. +type StaticOptions struct { +	// Prefix is the optional prefix used to serve the static directory content +	Prefix string +	// SkipLogging will disable [Static] log messages when a static file is served. +	SkipLogging bool +	// IndexFile defines which file to serve as index if it exists. +	IndexFile string +	// Expires defines which user-defined function to use for producing a HTTP Expires Header +	// https://developers.google.com/speed/docs/insights/LeverageBrowserCaching +	Expires func() string +	// Fallback defines a default URL to serve when the requested resource was +	// not found. +	Fallback string +	// Exclude defines a pattern for URLs this handler should never process. +	Exclude string +} + +func prepareStaticOptions(options []StaticOptions) StaticOptions { +	var opt StaticOptions +	if len(options) > 0 { +		opt = options[0] +	} + +	// Defaults +	if len(opt.IndexFile) == 0 { +		opt.IndexFile = "index.html" +	} +	// Normalize the prefix if provided +	if opt.Prefix != "" { +		// Ensure we have a leading '/' +		if opt.Prefix[0] != '/' { +			opt.Prefix = "/" + opt.Prefix +		} +		// Remove any trailing '/' +		opt.Prefix = strings.TrimRight(opt.Prefix, "/") +	} +	return opt +} + +// Static returns a middleware handler that serves static files in the given directory. +func Static(directory string, staticOpt ...StaticOptions) Handler { +	if !filepath.IsAbs(directory) { +		directory = filepath.Join(Root, directory) +	} +	dir := http.Dir(directory) +	opt := prepareStaticOptions(staticOpt) + +	return func(res http.ResponseWriter, req *http.Request, log *log.Logger) { +		if req.Method != "GET" && req.Method != "HEAD" { +			return +		} +		if opt.Exclude != "" && strings.HasPrefix(req.URL.Path, opt.Exclude) { +			return +		} +		file := req.URL.Path +		// if we have a prefix, filter requests by stripping the prefix +		if opt.Prefix != "" { +			if !strings.HasPrefix(file, opt.Prefix) { +				return +			} +			file = file[len(opt.Prefix):] +			if file != "" && file[0] != '/' { +				return +			} +		} +		f, err := dir.Open(file) +		if err != nil { +			// try any fallback before giving up +			if opt.Fallback != "" { +				file = opt.Fallback // so that logging stays true +				f, err = dir.Open(opt.Fallback) +			} + +			if err != nil { +				// discard the error? +				return +			} +		} +		defer f.Close() + +		fi, err := f.Stat() +		if err != nil { +			return +		} + +		// try to serve index file +		if fi.IsDir() { +			// redirect if missing trailing slash +			if !strings.HasSuffix(req.URL.Path, "/") { +				dest := url.URL{ +					Path:     req.URL.Path + "/", +					RawQuery: req.URL.RawQuery, +					Fragment: req.URL.Fragment, +				} +				http.Redirect(res, req, dest.String(), http.StatusFound) +				return +			} + +			file = path.Join(file, opt.IndexFile) +			f, err = dir.Open(file) +			if err != nil { +				return +			} +			defer f.Close() + +			fi, err = f.Stat() +			if err != nil || fi.IsDir() { +				return +			} +		} + +		if !opt.SkipLogging { +			log.Println("[Static] Serving " + file) +		} + +		// Add an Expires header to the static content +		if opt.Expires != nil { +			res.Header().Set("Expires", opt.Expires()) +		} + +		http.ServeContent(res, req, file, fi.ModTime(), f) +	} +} | 
