githaven-fork/modules/translation/i18n/i18n.go

// Copyright 2022 The Gitea Authors. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.

package i18n

import (
	"errors"
	"fmt"
	"os"
	"reflect"
	"sync"
	"time"

	"code.gitea.io/gitea/modules/log"
	"code.gitea.io/gitea/modules/setting"

	"gopkg.in/ini.v1"
)

var (
	ErrLocaleAlreadyExist = errors.New("lang already exists")

	DefaultLocales = NewLocaleStore(true)
)

type locale struct {
	// This mutex will be set if we have live-reload enabled (e.g. dev mode)
	reloadMu *sync.RWMutex

	store    *LocaleStore
	langName string

	idxToMsgMap map[int]string // the map idx is generated by store's trKeyToIdxMap

	sourceFileName      string
	sourceFileInfo      os.FileInfo
	lastReloadCheckTime time.Time
}

type LocaleStore struct {
	// This mutex will be set if we have live-reload enabled (e.g. dev mode)
	reloadMu *sync.RWMutex

	langNames []string
	langDescs []string
	localeMap map[string]*locale

	// this needs to be locked when live-reloading
	trKeyToIdxMap map[string]int

	defaultLang string
}

func NewLocaleStore(isProd bool) *LocaleStore {
	store := &LocaleStore{localeMap: make(map[string]*locale), trKeyToIdxMap: make(map[string]int)}
	if !isProd {
		store.reloadMu = &sync.RWMutex{}
	}
	return store
}

// AddLocaleByIni adds locale by ini into the store
// if source is a string, then the file is loaded. In dev mode, this file will be checked for live-reloading
// if source is a []byte, then the content is used
// Note: this is not concurrent safe
func (store *LocaleStore) AddLocaleByIni(langName, langDesc string, source interface{}) error {
	if _, ok := store.localeMap[langName]; ok {
		return ErrLocaleAlreadyExist
	}

	l := &locale{store: store, langName: langName}
	if store.reloadMu != nil {
		l.reloadMu = &sync.RWMutex{}
		l.reloadMu.Lock() // Arguably this is not necessary as AddLocaleByIni isn't concurrent safe - but for consistency we do this
		defer l.reloadMu.Unlock()
	}

	if fileName, ok := source.(string); ok {
		l.sourceFileName = fileName
		l.sourceFileInfo, _ = os.Stat(fileName) // live-reload only works for regular files. the error can be ignored
	}

	var err error
	l.idxToMsgMap, err = store.readIniToIdxToMsgMap(source)
	if err != nil {
		return err
	}

	store.langNames = append(store.langNames, langName)
	store.langDescs = append(store.langDescs, langDesc)

	store.localeMap[l.langName] = l

	return nil
}

// readIniToIdxToMsgMap will read a provided ini and creates an idxToMsgMap
func (store *LocaleStore) readIniToIdxToMsgMap(source interface{}) (map[int]string, error) {
	iniFile, err := ini.LoadSources(ini.LoadOptions{
		IgnoreInlineComment:         true,
		UnescapeValueCommentSymbols: true,
	}, source)
	if err != nil {
		return nil, fmt.Errorf("unable to load ini: %w", err)
	}
	iniFile.BlockMode = false

	idxToMsgMap := make(map[int]string)

	if store.reloadMu != nil {
		store.reloadMu.Lock()
		defer store.reloadMu.Unlock()
	}

	for _, section := range iniFile.Sections() {
		for _, key := range section.Keys() {

			var trKey string
			if section.Name() == "" || section.Name() == "DEFAULT" {
				trKey = key.Name()
			} else {
				trKey = section.Name() + "." + key.Name()
			}

			// Instead of storing the key strings in multiple different maps we compute a idx which will act as numeric code for key
			// This reduces the size of the locale idxToMsgMaps
			idx, ok := store.trKeyToIdxMap[trKey]
			if !ok {
				idx = len(store.trKeyToIdxMap)
				store.trKeyToIdxMap[trKey] = idx
			}
			idxToMsgMap[idx] = key.Value()
		}
	}
	iniFile = nil
	return idxToMsgMap, nil
}

func (store *LocaleStore) idxForTrKey(trKey string) (int, bool) {
	if store.reloadMu != nil {
		store.reloadMu.RLock()
		defer store.reloadMu.RUnlock()
	}
	idx, ok := store.trKeyToIdxMap[trKey]
	return idx, ok
}

// HasLang reports if a language is available in the store
func (store *LocaleStore) HasLang(langName string) bool {
	_, ok := store.localeMap[langName]
	return ok
}

// ListLangNameDesc reports if a language available in the store
func (store *LocaleStore) ListLangNameDesc() (names, desc []string) {
	return store.langNames, store.langDescs
}

// SetDefaultLang sets default language as a fallback
func (store *LocaleStore) SetDefaultLang(lang string) {
	store.defaultLang = lang
}

// Tr translates content to target language. fall back to default language.
func (store *LocaleStore) Tr(lang, trKey string, trArgs ...interface{}) string {
	l, ok := store.localeMap[lang]
	if !ok {
		l, ok = store.localeMap[store.defaultLang]
	}

	if ok {
		return l.Tr(trKey, trArgs...)
	}
	return trKey
}

// reloadIfNeeded will check if the locale needs to be reloaded
// this function will assume that the l.reloadMu has been RLocked if it already exists
func (l *locale) reloadIfNeeded() {
	if l.reloadMu == nil {
		return
	}

	now := time.Now()
	if now.Sub(l.lastReloadCheckTime) < time.Second || l.sourceFileInfo == nil || l.sourceFileName == "" {
		return
	}

	l.reloadMu.RUnlock()
	l.reloadMu.Lock() // (NOTE: a pre-emption can occur between these two locks so we need to recheck)
	defer l.reloadMu.RLock()
	defer l.reloadMu.Unlock()

	if now.Sub(l.lastReloadCheckTime) < time.Second || l.sourceFileInfo == nil || l.sourceFileName == "" {
		return
	}

	l.lastReloadCheckTime = now
	sourceFileInfo, err := os.Stat(l.sourceFileName)
	if err != nil || sourceFileInfo.ModTime().Equal(l.sourceFileInfo.ModTime()) {
		return
	}

	idxToMsgMap, err := l.store.readIniToIdxToMsgMap(l.sourceFileName)
	if err == nil {
		l.idxToMsgMap = idxToMsgMap
	} else {
		log.Error("Unable to live-reload the locale file %q, err: %v", l.sourceFileName, err)
	}

	// We will set the sourceFileInfo to this file to prevent repeated attempts to re-load this broken file
	l.sourceFileInfo = sourceFileInfo
}

// Tr translates content to locale language. fall back to default language.
func (l *locale) Tr(trKey string, trArgs ...interface{}) string {
	if l.reloadMu != nil {
		l.reloadMu.RLock()
		defer l.reloadMu.RUnlock()
		l.reloadIfNeeded()
	}

	msg, _ := l.tryTr(trKey, trArgs...)
	return msg
}

func (l *locale) tryTr(trKey string, trArgs ...interface{}) (msg string, found bool) {
	trMsg := trKey

	// convert the provided trKey to a common idx from the store
	idx, ok := l.store.idxForTrKey(trKey)

	if ok {
		if msg, found = l.idxToMsgMap[idx]; found {
			trMsg = msg // use the translation that we have found
		} else if l.langName != l.store.defaultLang {
			// No translation available in our current language... fallback to the default language

			// Attempt to get the default language from the locale store
			if def, ok := l.store.localeMap[l.store.defaultLang]; ok {

				if def.reloadMu != nil {
					def.reloadMu.RLock()
					def.reloadIfNeeded()
				}
				if msg, found = def.idxToMsgMap[idx]; found {
					trMsg = msg // use the translation that we have found
				}
				if def.reloadMu != nil {
					def.reloadMu.RUnlock()
				}
			}
		}
	}

	if !found && !setting.IsProd {
		log.Error("missing i18n translation key: %q", trKey)
	}

	if len(trArgs) == 0 {
		return trMsg, found
	}

	fmtArgs := make([]interface{}, 0, len(trArgs))
	for _, arg := range trArgs {
		val := reflect.ValueOf(arg)
		if val.Kind() == reflect.Slice {
			// Previously, we would accept Tr(lang, key, a, [b, c], d, [e, f]) as Sprintf(msg, a, b, c, d, e, f)
			// but this is an unstable behavior.
			//
			// So we restrict the accepted arguments to either:
			//
			// 1. Tr(lang, key, [slice-items]) as Sprintf(msg, items...)
			// 2. Tr(lang, key, args...) as Sprintf(msg, args...)
			if len(trArgs) == 1 {
				for i := 0; i < val.Len(); i++ {
					fmtArgs = append(fmtArgs, val.Index(i).Interface())
				}
			} else {
				log.Error("the args for i18n shouldn't contain uncertain slices, key=%q, args=%v", trKey, trArgs)
				break
			}
		} else {
			fmtArgs = append(fmtArgs, arg)
		}
	}

	return fmt.Sprintf(trMsg, fmtArgs...), found
}

// ResetDefaultLocales resets the current default locales
// NOTE: this is not synchronized
func ResetDefaultLocales(isProd bool) {
	DefaultLocales = NewLocaleStore(isProd)
}

// Tr use default locales to translate content to target language.
func Tr(lang, trKey string, trArgs ...interface{}) string {
	return DefaultLocales.Tr(lang, trKey, trArgs...)
}