Skip to content

Commit

Permalink
commands: Fix broken server-reload on config changes
Browse files Browse the repository at this point in the history
This was accidently broken in Hugo 0.42.

Fixes #4878
  • Loading branch information
bep committed Jun 28, 2018
1 parent 1f0c4e1 commit 2955f93
Show file tree
Hide file tree
Showing 3 changed files with 26 additions and 24 deletions.
37 changes: 23 additions & 14 deletions commands/commandeer.go
Original file line number Diff line number Diff line change
Expand Up @@ -37,23 +37,27 @@ import (
"github.com/gohugoio/hugo/langs"
)

type commandeer struct {
type commandeerHugoState struct {
*deps.DepsCfg
hugo *hugolib.HugoSites
fsCreate sync.Once
}

hugo *hugolib.HugoSites
type commandeer struct {
*commandeerHugoState

// We need to reuse this on server rebuilds.
destinationFs afero.Fs

h *hugoBuilderCommon
ftch flagsToConfigHandler

visitedURLs *types.EvictingStringQueue

// We watch these for changes.
configFiles []string

doWithCommandeer func(c *commandeer) error

// We can do this only once.
fsCreate sync.Once
// We watch these for changes.
configFiles []string

// Used in cases where we get flooded with events in server mode.
debounce func(f func())
Expand All @@ -73,6 +77,7 @@ func (c *commandeer) Set(key string, value interface{}) {
}

func (c *commandeer) initFs(fs *hugofs.Fs) error {
c.destinationFs = fs.Destination
c.DepsCfg.Fs = fs

return nil
Expand All @@ -89,11 +94,12 @@ func newCommandeer(mustHaveConfigFile, running bool, h *hugoBuilderCommon, f fla
}

c := &commandeer{
h: h,
ftch: f,
doWithCommandeer: doWithCommandeer,
visitedURLs: types.NewEvictingStringQueue(10),
debounce: rebuildDebouncer,
h: h,
ftch: f,
commandeerHugoState: &commandeerHugoState{},
doWithCommandeer: doWithCommandeer,
visitedURLs: types.NewEvictingStringQueue(10),
debounce: rebuildDebouncer,
}

return c, c.loadConfig(mustHaveConfigFile, running)
Expand Down Expand Up @@ -188,8 +194,11 @@ func (c *commandeer) loadConfig(mustHaveConfigFile, running bool) error {
c.fsCreate.Do(func() {
fs := hugofs.NewFrom(sourceFs, config)

// Hugo writes the output to memory instead of the disk.
if createMemFs {
if c.destinationFs != nil {
// Need to reuse the destination on server rebuilds.
fs.Destination = c.destinationFs
} else if createMemFs {
// Hugo writes the output to memory instead of the disk.
fs.Destination = new(afero.MemMapFs)
}

Expand Down
11 changes: 2 additions & 9 deletions commands/hugo.go
Original file line number Diff line number Diff line change
Expand Up @@ -596,14 +596,6 @@ func (c *commandeer) getDirList() ([]string, error) {
return a, nil
}

func (c *commandeer) recreateAndBuildSites(watching bool) (err error) {
defer c.timeTrack(time.Now(), "Total")
if !c.h.quiet {
c.Logger.FEEDBACK.Println("Started building sites ...")
}
return c.hugo.Build(hugolib.BuildCfg{CreateSitesFromConfig: true})
}

func (c *commandeer) resetAndBuildSites() (err error) {
if !c.h.quiet {
c.Logger.FEEDBACK.Println("Started building sites ...")
Expand Down Expand Up @@ -637,9 +629,10 @@ func (c *commandeer) rebuildSites(events []fsnotify.Event) error {
}

func (c *commandeer) fullRebuild() {
c.commandeerHugoState = &commandeerHugoState{}
if err := c.loadConfig(true, true); err != nil {
jww.ERROR.Println("Failed to reload config:", err)
} else if err := c.recreateAndBuildSites(true); err != nil {
} else if err := c.buildSites(); err != nil {
jww.ERROR.Println(err)
} else if !c.h.buildWatch && !c.Cfg.GetBool("disableLiveReload") {
livereload.ForceRefresh()
Expand Down
2 changes: 1 addition & 1 deletion commands/server.go
Original file line number Diff line number Diff line change
Expand Up @@ -298,7 +298,7 @@ func (f *fileServer) createEndpoint(i int) (*http.ServeMux, string, string, erro
}
}

httpFs := afero.NewHttpFs(f.c.Fs.Destination)
httpFs := afero.NewHttpFs(f.c.destinationFs)
fs := filesOnlyFs{httpFs.Dir(absPublishDir)}

doLiveReload := !f.s.buildWatch && !f.c.Cfg.GetBool("disableLiveReload")
Expand Down

0 comments on commit 2955f93

Please sign in to comment.