Permalink
Browse files

Add noHttpCache to hugo server

Fixes #3897
  • Loading branch information...
bep committed Sep 22, 2017
1 parent c0370e0 commit 0b34af216154367af7f53ce93d44e6b3d58c3f34
Showing with 13 additions and 1 deletion.
  1. +13 −1 commands/server.go
View
@@ -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"))

0 comments on commit 0b34af2

Please sign in to comment.