shithub: hugo

ref: 708bc78770a0b0361908f6404f57264c53252a95
dir: /commands/multilingual.go/

View raw version
package commands

import (
	"fmt"
	"sort"

	"strings"

	"github.com/spf13/cast"
	"github.com/spf13/hugo/hugolib"
	"github.com/spf13/viper"
)

func readMultilingualConfiguration() (*hugolib.HugoSites, error) {
	sites := make([]*hugolib.Site, 0)
	multilingual := viper.GetStringMap("Multilingual")
	if len(multilingual) == 0 {
		// TODO(bep) multilingo langConfigsList = append(langConfigsList, hugolib.NewLanguage("en"))
		sites = append(sites, hugolib.NewSite(hugolib.NewLanguage("en")))
	}

	if len(multilingual) > 0 {
		var err error

		languages, err := toSortedLanguages(multilingual)

		if err != nil {
			return nil, fmt.Errorf("Failed to parse multilingual config: %s", err)
		}

		for _, lang := range languages {
			sites = append(sites, hugolib.NewSite(lang))
		}

	}

	return hugolib.NewHugoSites(sites...)

}

func toSortedLanguages(l map[string]interface{}) (hugolib.Languages, error) {
	langs := make(hugolib.Languages, len(l))
	i := 0

	for lang, langConf := range l {
		langsMap, ok := langConf.(map[string]interface{})

		if !ok {
			return nil, fmt.Errorf("Language config is not a map: %v", langsMap)
		}

		language := hugolib.NewLanguage(lang)

		for k, v := range langsMap {
			loki := strings.ToLower(k)
			switch loki {
			case "title":
				language.Title = cast.ToString(v)
			case "weight":
				language.Weight = cast.ToInt(v)
			}

			// Put all into the Params map
			// TODO(bep) reconsile with the type handling etc. from other params handlers.
			language.SetParam(loki, v)
		}

		langs[i] = language
		i++
	}

	sort.Sort(langs)

	return langs, nil
}