Fix some recently introduced error handling issues

Updates #10953
This commit is contained in:
Bjørn Erik Pedersen 2023-05-18 10:36:48 +02:00
parent 1155bbca9d
commit 834b3d7e41
No known key found for this signature in database
GPG key ID: 330E6E2BD4859D8F
4 changed files with 72 additions and 37 deletions

View file

@ -338,9 +338,6 @@ func (r *rootCommand) Run(ctx context.Context, cd *simplecobra.Commandeer, args
defer r.timeTrack(time.Now(), "Built") defer r.timeTrack(time.Now(), "Built")
} }
err := b.build() err := b.build()
if err != nil {
r.Println("Error:", err.Error())
}
return err return err
}() }()

View file

@ -153,7 +153,11 @@ func (c *hugoBuilder) getDirList() ([]string, error) {
return nil return nil
} }
watchFiles := c.hugo().PathSpec.BaseFs.WatchDirs() h, err := c.hugo()
if err != nil {
return nil, err
}
watchFiles := h.PathSpec.BaseFs.WatchDirs()
for _, fi := range watchFiles { for _, fi := range watchFiles {
if !fi.IsDir() { if !fi.IsDir() {
filenames = append(filenames, fi.Meta().Filename) filenames = append(filenames, fi.Meta().Filename)
@ -334,7 +338,11 @@ func (c *hugoBuilder) newWatcher(pollIntervalStr string, dirList ...string) (*wa
return nil, err return nil, err
} }
spec := c.hugo().Deps.SourceSpec h, err := c.hugo()
if err != nil {
return nil, err
}
spec := h.Deps.SourceSpec
for _, d := range dirList { for _, d := range dirList {
if d != "" { if d != "" {
@ -359,7 +367,7 @@ func (c *hugoBuilder) newWatcher(pollIntervalStr string, dirList ...string) (*wa
for { for {
select { select {
case evs := <-watcher.Events: case evs := <-watcher.Events:
unlock, err := c.hugo().LockBuild() unlock, err := h.LockBuild()
if err != nil { if err != nil {
c.r.logger.Errorln("Failed to acquire a build lock: %s", err) c.r.logger.Errorln("Failed to acquire a build lock: %s", err)
return return
@ -399,7 +407,11 @@ func (c *hugoBuilder) build() error {
if !c.r.quiet { if !c.r.quiet {
c.r.Println() c.r.Println()
c.hugo().PrintProcessingStats(os.Stdout) h, err := c.hugo()
if err != nil {
return err
}
h.PrintProcessingStats(os.Stdout)
c.r.Println() c.r.Println()
} }
@ -407,7 +419,11 @@ func (c *hugoBuilder) build() error {
} }
func (c *hugoBuilder) buildSites(noBuildLock bool) (err error) { func (c *hugoBuilder) buildSites(noBuildLock bool) (err error) {
return c.hugo().Build(hugolib.BuildCfg{NoBuildLock: noBuildLock}) h, err := c.hugo()
if err != nil {
return err
}
return h.Build(hugolib.BuildCfg{NoBuildLock: noBuildLock})
} }
func (c *hugoBuilder) copyStatic() (map[string]uint64, error) { func (c *hugoBuilder) copyStatic() (map[string]uint64, error) {
@ -462,7 +478,11 @@ func (c *hugoBuilder) copyStaticTo(sourceFs *filesystems.SourceFilesystem) (uint
func (c *hugoBuilder) doWithPublishDirs(f func(sourceFs *filesystems.SourceFilesystem) (uint64, error)) (map[string]uint64, error) { func (c *hugoBuilder) doWithPublishDirs(f func(sourceFs *filesystems.SourceFilesystem) (uint64, error)) (map[string]uint64, error) {
langCount := make(map[string]uint64) langCount := make(map[string]uint64)
staticFilesystems := c.hugo().BaseFs.SourceFilesystems.Static h, err := c.hugo()
if err != nil {
return nil, err
}
staticFilesystems := h.BaseFs.SourceFilesystems.Static
if len(staticFilesystems) == 0 { if len(staticFilesystems) == 0 {
c.r.logger.Infoln("No static directories found to sync") c.r.logger.Infoln("No static directories found to sync")
@ -535,16 +555,20 @@ func (c *hugoBuilder) fullBuild(noBuildLock bool) error {
} }
} }
for _, s := range c.hugo().Sites { h, err := c.hugo()
if err != nil {
return err
}
for _, s := range h.Sites {
s.ProcessingStats.Static = langCount[s.Language().Lang] s.ProcessingStats.Static = langCount[s.Language().Lang]
} }
if c.r.gc { if c.r.gc {
count, err := c.hugo().GC() count, err := h.GC()
if err != nil { if err != nil {
return err return err
} }
for _, s := range c.hugo().Sites { for _, s := range h.Sites {
// We have no way of knowing what site the garbage belonged to. // We have no way of knowing what site the garbage belonged to.
s.ProcessingStats.Cleaned = uint64(count) s.ProcessingStats.Cleaned = uint64(count)
} }
@ -691,12 +715,17 @@ func (c *hugoBuilder) handleEvents(watcher *watcher.Batcher,
dynamicEvents := []fsnotify.Event{} dynamicEvents := []fsnotify.Event{}
filtered := []fsnotify.Event{} filtered := []fsnotify.Event{}
h, err := c.hugo()
if err != nil {
c.r.logger.Errorln("Error getting the Hugo object:", err)
return
}
for _, ev := range evs { for _, ev := range evs {
if c.hugo().ShouldSkipFileChangeEvent(ev) { if h.ShouldSkipFileChangeEvent(ev) {
continue continue
} }
// Check the most specific first, i.e. files. // Check the most specific first, i.e. files.
contentMapped := c.hugo().ContentChanges.GetSymbolicLinkMappings(ev.Name) contentMapped := h.ContentChanges.GetSymbolicLinkMappings(ev.Name)
if len(contentMapped) > 0 { if len(contentMapped) > 0 {
for _, mapped := range contentMapped { for _, mapped := range contentMapped {
filtered = append(filtered, fsnotify.Event{Name: mapped, Op: ev.Op}) filtered = append(filtered, fsnotify.Event{Name: mapped, Op: ev.Op})
@ -708,7 +737,7 @@ func (c *hugoBuilder) handleEvents(watcher *watcher.Batcher,
dir, name := filepath.Split(ev.Name) dir, name := filepath.Split(ev.Name)
contentMapped = c.hugo().ContentChanges.GetSymbolicLinkMappings(dir) contentMapped = h.ContentChanges.GetSymbolicLinkMappings(dir)
if len(contentMapped) == 0 { if len(contentMapped) == 0 {
filtered = append(filtered, ev) filtered = append(filtered, ev)
@ -742,7 +771,7 @@ func (c *hugoBuilder) handleEvents(watcher *watcher.Batcher,
if istemp { if istemp {
continue continue
} }
if c.hugo().Deps.SourceSpec.IgnoreFile(ev.Name) { if h.Deps.SourceSpec.IgnoreFile(ev.Name) {
continue continue
} }
// Sometimes during rm -rf operations a '"": REMOVE' is triggered. Just ignore these // Sometimes during rm -rf operations a '"": REMOVE' is triggered. Just ignore these
@ -771,7 +800,7 @@ func (c *hugoBuilder) handleEvents(watcher *watcher.Batcher,
if err := watcher.Add(path); err != nil { if err := watcher.Add(path); err != nil {
return err return err
} }
} else if !staticSyncer.isStatic(path) { } else if !staticSyncer.isStatic(h, path) {
// Hugo's rebuilding logic is entirely file based. When you drop a new folder into // Hugo's rebuilding logic is entirely file based. When you drop a new folder into
// /content on OSX, the above logic will handle future watching of those files, // /content on OSX, the above logic will handle future watching of those files,
// but the initial CREATE is lost. // but the initial CREATE is lost.
@ -788,7 +817,7 @@ func (c *hugoBuilder) handleEvents(watcher *watcher.Batcher,
} }
} }
if staticSyncer.isStatic(ev.Name) { if staticSyncer.isStatic(h, ev.Name) {
staticEvents = append(staticEvents, ev) staticEvents = append(staticEvents, ev)
} else { } else {
dynamicEvents = append(dynamicEvents, ev) dynamicEvents = append(dynamicEvents, ev)
@ -818,7 +847,11 @@ func (c *hugoBuilder) handleEvents(watcher *watcher.Batcher,
// force refresh when more than one file // force refresh when more than one file
if !c.errState.wasErr() && len(staticEvents) == 1 { if !c.errState.wasErr() && len(staticEvents) == 1 {
ev := staticEvents[0] ev := staticEvents[0]
h := c.hugo() h, err := c.hugo()
if err != nil {
c.r.logger.Errorln("Error getting the Hugo object:", err)
return
}
path := h.BaseFs.SourceFilesystems.MakeStaticPathRelative(ev.Name) path := h.BaseFs.SourceFilesystems.MakeStaticPathRelative(ev.Name)
path = h.RelURL(helpers.ToSlashTrimLeading(path), false) path = h.RelURL(helpers.ToSlashTrimLeading(path), false)
@ -831,7 +864,7 @@ func (c *hugoBuilder) handleEvents(watcher *watcher.Batcher,
if len(dynamicEvents) > 0 { if len(dynamicEvents) > 0 {
partitionedEvents := partitionDynamicEvents( partitionedEvents := partitionDynamicEvents(
c.hugo().BaseFs.SourceFilesystems, h.BaseFs.SourceFilesystems,
dynamicEvents) dynamicEvents)
onePageName := pickOneWriteOrCreatePath(partitionedEvents.ContentEvents) onePageName := pickOneWriteOrCreatePath(partitionedEvents.ContentEvents)
@ -857,7 +890,7 @@ func (c *hugoBuilder) handleEvents(watcher *watcher.Batcher,
// Nothing has changed. // Nothing has changed.
return return
} else if len(changed) == 1 { } else if len(changed) == 1 {
pathToRefresh := c.hugo().PathSpec.RelURL(helpers.ToSlashTrimLeading(changed[0]), false) pathToRefresh := h.PathSpec.RelURL(helpers.ToSlashTrimLeading(changed[0]), false)
livereload.RefreshPath(pathToRefresh) livereload.RefreshPath(pathToRefresh)
} else { } else {
livereload.ForceRefresh() livereload.ForceRefresh()
@ -872,7 +905,7 @@ func (c *hugoBuilder) handleEvents(watcher *watcher.Batcher,
if navigate { if navigate {
if onePageName != "" { if onePageName != "" {
p = c.hugo().GetContentPage(onePageName) p = h.GetContentPage(onePageName)
} }
} }
@ -886,10 +919,10 @@ func (c *hugoBuilder) handleEvents(watcher *watcher.Batcher,
} }
} }
func (c *hugoBuilder) hugo() *hugolib.HugoSites { func (c *hugoBuilder) hugo() (*hugolib.HugoSites, error) {
h, err := c.r.HugFromConfig(c.conf()) h, err := c.r.HugFromConfig(c.conf())
if err != nil { if err != nil {
panic(err) return nil, err
} }
if c.s != nil { if c.s != nil {
// A running server, register the media types. // A running server, register the media types.
@ -897,7 +930,7 @@ func (c *hugoBuilder) hugo() *hugolib.HugoSites {
s.RegisterMediaTypes() s.RegisterMediaTypes()
} }
} }
return h return h, nil
} }
func (c *hugoBuilder) hugoTry() *hugolib.HugoSites { func (c *hugoBuilder) hugoTry() *hugolib.HugoSites {
@ -953,7 +986,10 @@ func (c *hugoBuilder) rebuildSites(events []fsnotify.Event) error {
} }
c.errState.setBuildErr(nil) c.errState.setBuildErr(nil)
visited := c.visitedURLs.PeekAllSet() visited := c.visitedURLs.PeekAllSet()
h := c.hugo() h, err := c.hugo()
if err != nil {
return err
}
if c.fastRenderMode { if c.fastRenderMode {
// Make sure we always render the home pages // Make sure we always render the home pages
for _, l := range c.conf().configs.Languages { for _, l := range c.conf().configs.Languages {

View file

@ -234,7 +234,7 @@ func (f *fileServer) createEndpoint(i int) (*http.ServeMux, net.Listener, string
// First check the error state // First check the error state
err := f.c.getErrorWithContext() err := f.c.getErrorWithContext()
if err != nil { if err != nil {
f.c.errState.setWasErr(false) f.c.errState.setWasErr(true)
w.WriteHeader(500) w.WriteHeader(500)
r, err := f.errorTemplate(err) r, err := f.errorTemplate(err)
if err != nil { if err != nil {
@ -242,8 +242,8 @@ func (f *fileServer) createEndpoint(i int) (*http.ServeMux, net.Listener, string
} }
port = 1313 port = 1313
if !f.c.errState.isPaused() { if lrport := conf.configs.GetFirstLanguageConfig().BaseURLLiveReload().Port(); lrport != 0 {
port = conf.configs.Base.Internal.LiveReloadPort port = lrport
} }
lr := *u lr := *u
lr.Host = fmt.Sprintf("%s:%d", lr.Hostname(), port) lr.Host = fmt.Sprintf("%s:%d", lr.Hostname(), port)
@ -432,9 +432,6 @@ func (c *serverCommand) Run(ctx context.Context, cd *simplecobra.Commandeer, arg
err := func() error { err := func() error {
defer c.r.timeTrack(time.Now(), "Built") defer c.r.timeTrack(time.Now(), "Built")
err := c.build() err := c.build()
if err != nil {
c.r.Println("Error:", err.Error())
}
return err return err
}() }()
if err != nil { if err != nil {
@ -703,8 +700,12 @@ func (c *serverCommand) partialReRender(urls ...string) error {
visited[url] = true visited[url] = true
} }
h, err := c.hugo()
if err != nil {
return err
}
// Note: We do not set NoBuildLock as the file lock is not acquired at this stage. // Note: We do not set NoBuildLock as the file lock is not acquired at this stage.
return c.hugo().Build(hugolib.BuildCfg{NoBuildLock: false, RecentlyVisited: visited, PartialReRender: true, ErrRecovery: c.errState.wasErr()}) return h.Build(hugolib.BuildCfg{NoBuildLock: false, RecentlyVisited: visited, PartialReRender: true, ErrRecovery: c.errState.wasErr()})
} }
func (c *serverCommand) serve() error { func (c *serverCommand) serve() error {
@ -877,8 +878,8 @@ type staticSyncer struct {
c *hugoBuilder c *hugoBuilder
} }
func (s *staticSyncer) isStatic(filename string) bool { func (s *staticSyncer) isStatic(h *hugolib.HugoSites, filename string) bool {
return s.c.hugo().BaseFs.SourceFilesystems.IsStatic(filename) return h.BaseFs.SourceFilesystems.IsStatic(filename)
} }
func (s *staticSyncer) syncsStaticEvents(staticEvents []fsnotify.Event) error { func (s *staticSyncer) syncsStaticEvents(staticEvents []fsnotify.Event) error {
@ -1013,7 +1014,7 @@ func injectLiveReloadScript(src io.Reader, baseURL url.URL) string {
func partitionDynamicEvents(sourceFs *filesystems.SourceFilesystems, events []fsnotify.Event) (de dynamicEvents) { func partitionDynamicEvents(sourceFs *filesystems.SourceFilesystems, events []fsnotify.Event) (de dynamicEvents) {
for _, e := range events { for _, e := range events {
if sourceFs.IsAsset(e.Name) { if !sourceFs.IsContent(e.Name) {
de.AssetEvents = append(de.AssetEvents, e) de.AssetEvents = append(de.AssetEvents, e)
} else { } else {
de.ContentEvents = append(de.ContentEvents, e) de.ContentEvents = append(de.ContentEvents, e)

View file

@ -21,8 +21,9 @@ import (
) )
func main() { func main() {
log.SetFlags(0)
err := commands.Execute(os.Args[1:]) err := commands.Execute(os.Args[1:])
if err != nil { if err != nil {
log.Fatal(err) log.Fatalf("Error: %s", err)
} }
} }