Add noHttpCache to hugo server

Fixes #3897
This commit is contained in:
Bjørn Erik Pedersen 2017-09-22 17:13:21 +02:00
parent c0370e0ee3
commit 0b34af2161

View file

@ -39,6 +39,7 @@ var (
serverInterface string
serverPort int
serverWatch bool
noHttpCache bool
)
var serverCmd = &cobra.Command{
@ -86,6 +87,7 @@ func init() {
serverCmd.Flags().IntVarP(&serverPort, "port", "p", 1313, "port on which the server will listen")
serverCmd.Flags().StringVarP(&serverInterface, "bind", "", "127.0.0.1", "interface to which the server will bind")
serverCmd.Flags().BoolVarP(&serverWatch, "watch", "w", true, "watch filesystem for changes and recreate as needed")
serverCmd.Flags().BoolVar(&noHttpCache, "noHttpCache", false, "prevent HTTP caching")
serverCmd.Flags().BoolVarP(&serverAppend, "appendPort", "", true, "append port to baseURL")
serverCmd.Flags().BoolVar(&disableLiveReload, "disableLiveReload", false, "watch without enabling live browser reload on rebuild")
serverCmd.Flags().BoolVar(&navigateToChanged, "navigateToChanged", false, "navigate to changed content file on live browser reload")
@ -205,7 +207,17 @@ func (c *commandeer) serve(port int) {
httpFs := afero.NewHttpFs(c.Fs.Destination)
fs := filesOnlyFs{httpFs.Dir(c.PathSpec().AbsPathify(c.Cfg.GetString("publishDir")))}
fileserver := http.FileServer(fs)
decorate := func(h http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
if noHttpCache {
w.Header().Set("Cache-Control", " no-store, no-cache, must-revalidate, max-age=0")
w.Header().Set("Pragma", "no-cache")
}
h.ServeHTTP(w, r)
})
}
fileserver := decorate(http.FileServer(fs))
// We're only interested in the path
u, err := url.Parse(c.Cfg.GetString("baseURL"))