locales.go

// Pipe - A small and beautiful blogging platform written in golang.
// Copyright (C) 2017-2018, b3log.org
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

// Package i18n includes internationalization related manipulations.
package i18n

import (
	"encoding/json"
	"fmt"
	"io/ioutil"
	"os"
	"path/filepath"
	"sort"
	"strings"

	"github.com/b3log/pipe/log"
	"github.com/b3log/pipe/util"
	"github.com/b3log/pipe/model"
)

// Logger
var logger = log.NewLogger(os.Stdout)

type locale struct {
	Name     string
	Langs    map[string]interface{}
	TimeZone string
}

var locales = map[string]locale{}

// Load loads i18n message configurations.
func Load() {
	f, _ := os.Open(filepath.ToSlash(filepath.Join(model.Conf.StaticRoot, "i18n")))
	names, _ := f.Readdirnames(-1)
	f.Close()

	for _, name := range names {
		if !util.IsLetter(rune(name[0])) || !strings.HasSuffix(name, ".json") {
			continue
		}

		loc := name[:strings.LastIndex(name, ".")]
		load(loc)
	}

	logger.Tracef("loaded [%d] language configuration files", len(locales))
}

func load(localeStr string) {
	bytes, err := ioutil.ReadFile(filepath.ToSlash(filepath.Join(model.Conf.StaticRoot, "i18n/"+localeStr+".json")))
	if nil != err {
		logger.Fatal("reads i18n configurations fialed: " + err.Error())
	}

	l := locale{Name: localeStr}

	err = json.Unmarshal(bytes, &l.Langs)
	if nil != err {
		logger.Fatal("parses i18n configurations failed: " + err.Error())
	}

	locales[localeStr] = l
}

// GetMessagef gets a message with the specified locale, key and arguments
func GetMessagef(locale, key string, a ...interface{}) string {
	msg := GetMessage(locale, key)

	return fmt.Sprintf(msg, a...)
}

// GetMessage gets a message with the specified locale and key.
func GetMessage(locale, key string) string {
	return locales[locale].Langs[key].(string)
}

// GetMessages gets all messages with the specified locale.
func GetMessages(locale string) map[string]interface{} {
	return locales[locale].Langs
}

// GetLocalesNames gets names of all locales. Returns ["zh_CN", "en_US"] for example.
func GetLocalesNames() []string {
	var ret []string

	for name := range locales {
		ret = append(ret, name)
	}

	sort.Strings(ret)

	return ret
}