shithub: hugo

Download patch

ref: d953e39e63deac608e7f1442a6c0d5b2fe56180e
parent: 9798a9567028b364f10027002eb0026767356828
author: Bjørn Erik Pedersen <[email protected]>
date: Mon Aug 8 06:12:39 EDT 2016

Add parent as owner to Site

And pull up common member vars like Tmpl and Multilinguage.

Or: the final multilingual TODO-fixes.

See #2309

--- a/hugolib/handler_meta.go
+++ b/hugolib/handler_meta.go
@@ -74,7 +74,7 @@
 			return
 		}
 
-		results <- h.PageConvert(p, s.Tmpl)
+		results <- h.PageConvert(p, s.owner.tmpl)
 	}
 }
 
--- a/hugolib/hugo_sites.go
+++ b/hugolib/hugo_sites.go
@@ -35,7 +35,10 @@
 type HugoSites struct {
 	Sites []*Site
 
-	Multilingual *Multilingual
+	tmpl    tpl.Template
+	runMode runmode
+
+	multilingual *Multilingual
 }
 
 // NewHugoSites creates a new collection of sites given the input sites, building
@@ -47,7 +50,12 @@
 		return nil, err
 	}
 
-	return &HugoSites{Multilingual: langConfig, Sites: sites}, nil
+	h := &HugoSites{multilingual: langConfig, Sites: sites}
+
+	for _, s := range sites {
+		s.owner = h
+	}
+	return h, nil
 }
 
 // NewHugoSitesFromConfiguration creates HugoSites from the global Viper config.
@@ -92,7 +100,6 @@
 }
 
 func (h *HugoSites) reCreateFromConfig() error {
-	oldSite := h.Sites[0]
 	sites, err := createSitesFromConfig()
 
 	if err != nil {
@@ -106,13 +113,13 @@
 	}
 
 	h.Sites = sites
-	h.Multilingual = langConfig
 
-	for _, s := range h.Sites {
-		// TODO(bep) ml Tmpl
-		s.Tmpl = oldSite.Tmpl
+	for _, s := range sites {
+		s.owner = h
 	}
 
+	h.multilingual = langConfig
+
 	return nil
 }
 
@@ -157,6 +164,8 @@
 		}
 	}
 
+	h.runMode.Watching = config.Watching
+
 	// We should probably refactor the Site and pull up most of the logic from there to here,
 	// but that seems like a daunting task.
 	// So for now, if there are more than one site (language),
@@ -163,12 +172,6 @@
 	// we pre-process the first one, then configure all the sites based on that.
 	firstSite := h.Sites[0]
 
-	for _, s := range h.Sites {
-		// TODO(bep) ml
-		s.Multilingual = h.Multilingual
-		s.RunMode.Watching = config.Watching
-	}
-
 	if err := firstSite.preProcess(config); err != nil {
 		return err
 	}
@@ -178,8 +181,6 @@
 	if len(h.Sites) > 1 {
 		// Initialize the rest
 		for _, site := range h.Sites[1:] {
-			// TODO(bep) ml Tmpl
-			site.Tmpl = firstSite.Tmpl
 			site.initializeSiteInfo()
 		}
 	}
@@ -231,11 +232,7 @@
 		return errors.New("Rebuild does not support 'ResetState'. Use Build.")
 	}
 
-	for _, s := range h.Sites {
-		// TODO(bep) ml
-		s.Multilingual = h.Multilingual
-		s.RunMode.Watching = config.Watching
-	}
+	h.runMode.Watching = config.Watching
 
 	firstSite := h.Sites[0]
 
@@ -300,7 +297,7 @@
 // Render the cross-site artifacts.
 func (h *HugoSites) render() error {
 
-	if !h.Multilingual.enabled() {
+	if !h.multilingual.enabled() {
 		return nil
 	}
 
@@ -353,7 +350,7 @@
 
 	if len(h.Sites) > 1 {
 		pages := h.Sites[0].AllPages
-		allTranslations := pagesToTranslationsMap(h.Multilingual, pages)
+		allTranslations := pagesToTranslationsMap(h.multilingual, pages)
 		assignTranslationsToPages(allTranslations, pages)
 	}
 }
@@ -374,7 +371,7 @@
 		go func(pages <-chan *Page, wg *sync.WaitGroup) {
 			defer wg.Done()
 			for p := range pages {
-				if err := handleShortcodes(p, s.Tmpl); err != nil {
+				if err := handleShortcodes(p, s.owner.tmpl); err != nil {
 					jww.ERROR.Printf("Failed to handle shortcodes for page %s: %s", p.BaseFileName(), err)
 				}
 
--- a/hugolib/multilingual.go
+++ b/hugolib/multilingual.go
@@ -72,7 +72,10 @@
 }
 
 func (s *Site) multilingualEnabled() bool {
-	return s.Multilingual != nil && s.Multilingual.enabled()
+	if s.owner == nil {
+		return false
+	}
+	return s.owner.multilingual != nil && s.owner.multilingual.enabled()
 }
 
 func toSortedLanguages(l map[string]interface{}) (helpers.Languages, error) {
--- a/hugolib/planner.go
+++ b/hugolib/planner.go
@@ -32,9 +32,9 @@
 		} else {
 			fmt.Fprintf(out, " (renderer: n/a)")
 		}
-		if s.Tmpl != nil {
+		if s.owner.tmpl != nil {
 			for _, l := range p.layouts() {
-				fmt.Fprintf(out, " (layout: %s, exists: %t)", l, s.Tmpl.Lookup(l) != nil)
+				fmt.Fprintf(out, " (layout: %s, exists: %t)", l, s.owner.tmpl.Lookup(l) != nil)
 			}
 		}
 		fmt.Fprintf(out, "\n")
--- a/hugolib/site.go
+++ b/hugolib/site.go
@@ -76,11 +76,11 @@
 //
 // 5. The entire collection of files is written to disk.
 type Site struct {
+	owner          *HugoSites
 	Pages          Pages
 	AllPages       Pages
 	rawAllPages    Pages
 	Files          []*source.File
-	Tmpl           tpl.Template
 	Taxonomies     TaxonomyList
 	Source         source.Input
 	Sections       Taxonomy
@@ -89,19 +89,16 @@
 	timer          *nitro.B
 	targets        targetList
 	targetListInit sync.Once
-	RunMode        runmode
-	// TODO(bep ml remove
-	Multilingual *Multilingual
-	draftCount   int
-	futureCount  int
-	expiredCount int
-	Data         map[string]interface{}
-	Language     *helpers.Language
+	draftCount     int
+	futureCount    int
+	expiredCount   int
+	Data           map[string]interface{}
+	Language       *helpers.Language
 }
 
 // reset returns a new Site prepared for rebuild.
 func (s *Site) reset() *Site {
-	return &Site{Language: s.Language, Multilingual: s.Multilingual}
+	return &Site{Language: s.Language}
 }
 
 // newSite creates a new site in the given language.
@@ -423,7 +420,7 @@
 }
 
 func (s *Site) running() bool {
-	return s.RunMode.Watching
+	return s.owner.runMode.Watching
 }
 
 func init() {
@@ -478,7 +475,7 @@
 
 	if len(tmplChanged) > 0 {
 		s.prepTemplates(nil)
-		s.Tmpl.PrintErrors()
+		s.owner.tmpl.PrintErrors()
 		s.timerStep("template prep")
 	}
 
@@ -602,10 +599,10 @@
 }
 
 func (s *Site) loadTemplates() {
-	s.Tmpl = tpl.InitializeT()
-	s.Tmpl.LoadTemplates(s.absLayoutDir())
+	s.owner.tmpl = tpl.InitializeT()
+	s.owner.tmpl.LoadTemplates(s.absLayoutDir())
 	if s.hasTheme() {
-		s.Tmpl.LoadTemplatesWithPrefix(s.absThemeDir()+"/layouts", "theme")
+		s.owner.tmpl.LoadTemplatesWithPrefix(s.absThemeDir()+"/layouts", "theme")
 	}
 }
 
@@ -613,12 +610,12 @@
 	s.loadTemplates()
 
 	if withTemplate != nil {
-		if err := withTemplate(s.Tmpl); err != nil {
+		if err := withTemplate(s.owner.tmpl); err != nil {
 			return err
 		}
 	}
 
-	s.Tmpl.MarkReady()
+	s.owner.tmpl.MarkReady()
 
 	return nil
 }
@@ -724,7 +721,7 @@
 		return
 	}
 	s.prepTemplates(config.withTemplate)
-	s.Tmpl.PrintErrors()
+	s.owner.tmpl.PrintErrors()
 	s.timerStep("initialize & template prep")
 
 	if err = s.readDataFromSourceFS(); err != nil {
@@ -856,8 +853,8 @@
 		languages helpers.Languages
 	)
 
-	if s.Multilingual != nil {
-		languages = s.Multilingual.Languages
+	if s.owner != nil && s.owner.multilingual != nil {
+		languages = s.owner.multilingual.Languages
 	}
 
 	params := lang.Params()
@@ -872,16 +869,21 @@
 		languagePrefix = "/" + lang.Lang
 	}
 
+	var multilingual *Multilingual
+
+	if s.owner != nil {
+		multilingual = s.owner.multilingual
+	}
+
 	s.Info = SiteInfo{
-		BaseURL:         template.URL(helpers.SanitizeURLKeepTrailingSlash(viper.GetString("BaseURL"))),
-		Title:           lang.GetString("Title"),
-		Author:          lang.GetStringMap("author"),
-		Social:          lang.GetStringMapString("social"),
-		LanguageCode:    lang.GetString("languagecode"),
-		Copyright:       lang.GetString("copyright"),
-		DisqusShortname: lang.GetString("DisqusShortname"),
-		// TODO(bep) ml consolidate the below (make into methods etc.)
-		multilingual:          s.Multilingual,
+		BaseURL:               template.URL(helpers.SanitizeURLKeepTrailingSlash(viper.GetString("BaseURL"))),
+		Title:                 lang.GetString("Title"),
+		Author:                lang.GetStringMap("author"),
+		Social:                lang.GetStringMapString("social"),
+		LanguageCode:          lang.GetString("languagecode"),
+		Copyright:             lang.GetString("copyright"),
+		DisqusShortname:       lang.GetString("DisqusShortname"),
+		multilingual:          multilingual,
 		Language:              lang,
 		LanguagePrefix:        languagePrefix,
 		Languages:             languages,
@@ -1420,8 +1422,8 @@
 		}
 	}
 
-	if s.Multilingual.enabled() {
-		mainLang := s.Multilingual.DefaultLang.Lang
+	if s.owner.multilingual.enabled() {
+		mainLang := s.owner.multilingual.DefaultLang.Lang
 		mainLangURL := helpers.AbsURL(mainLang, false)
 		jww.DEBUG.Printf("Write redirect to main language %s: %s", mainLang, mainLangURL)
 		if err := s.publishDestAlias(s.languageAliasTarget(), "/", mainLangURL); err != nil {
@@ -1450,7 +1452,7 @@
 		var layouts []string
 		if !p.IsRenderable() {
 			self := "__" + p.TargetPath()
-			_, err := s.Tmpl.GetClone().New(self).Parse(string(p.Content))
+			_, err := s.owner.tmpl.GetClone().New(self).Parse(string(p.Content))
 			if err != nil {
 				results <- err
 				continue
@@ -2154,7 +2156,7 @@
 
 func (s *Site) findFirstLayout(layouts ...string) (string, bool) {
 	for _, layout := range layouts {
-		if s.Tmpl.Lookup(layout) != nil {
+		if s.owner.tmpl.Lookup(layout) != nil {
 			return layout, true
 		}
 	}
@@ -2164,7 +2166,7 @@
 func (s *Site) renderThing(d interface{}, layout string, w io.Writer) error {
 
 	// If the template doesn't exist, then return, but leave the Writer open
-	if templ := s.Tmpl.Lookup(layout); templ != nil {
+	if templ := s.owner.tmpl.Lookup(layout); templ != nil {
 		return templ.Execute(w, d)
 	}
 	return fmt.Errorf("Layout not found: %s", layout)