149 lignes
4.3 KiB
Go
149 lignes
4.3 KiB
Go
package languages
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"html/template"
|
|
"net/http"
|
|
"path"
|
|
"path/filepath"
|
|
|
|
"github.com/ewhal/nyaa/config"
|
|
"github.com/ewhal/nyaa/model"
|
|
"github.com/nicksnyder/go-i18n/i18n"
|
|
"github.com/nicksnyder/go-i18n/i18n/language"
|
|
)
|
|
|
|
// this interface is required to prevent a cyclic import between the languages and userService package.
|
|
type UserRetriever interface {
|
|
RetrieveCurrentUser(r *http.Request) (model.User, error)
|
|
}
|
|
|
|
var (
|
|
defaultLanguage string = config.DefaultI18nConfig.DefaultLanguage
|
|
userRetriever UserRetriever = nil
|
|
)
|
|
|
|
// Initialize the languages translation
|
|
func InitI18n(conf config.I18nConfig, retriever UserRetriever) error {
|
|
defaultLanguage = conf.DefaultLanguage
|
|
userRetriever = retriever
|
|
|
|
defaultFilepath := path.Join(conf.TranslationsDirectory, conf.DefaultLanguage+".all.json")
|
|
err := i18n.LoadTranslationFile(defaultFilepath)
|
|
if err != nil {
|
|
panic(fmt.Sprintf("failed to load default translation file '%s': %v", defaultFilepath, err))
|
|
}
|
|
|
|
paths, err := filepath.Glob(path.Join(conf.TranslationsDirectory, "*.json"))
|
|
if err != nil {
|
|
return fmt.Errorf("failed to get translation files: %v", err)
|
|
}
|
|
|
|
for _, file := range paths {
|
|
err := i18n.LoadTranslationFile(file)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to load translation file '%s': %v", file, err)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func GetDefaultLanguage() string {
|
|
return defaultLanguage
|
|
}
|
|
|
|
// When go-i18n finds a language with >0 translations, it uses it as the Tfunc
|
|
// However, if said language has a missing translation, it won't fallback to the "main" language
|
|
func TfuncAndLanguageWithFallback(language string, languages ...string) (i18n.TranslateFunc, *language.Language, error) {
|
|
// Use the last language on the args as the fallback one.
|
|
fallbackLanguage := language
|
|
if languages != nil {
|
|
fallbackLanguage = languages[len(languages)-1]
|
|
}
|
|
|
|
T, Tlang, err1 := i18n.TfuncAndLanguage(language, languages...)
|
|
fallbackT, fallbackTlang, err2 := i18n.TfuncAndLanguage(fallbackLanguage)
|
|
|
|
if err1 != nil && err2 != nil {
|
|
// fallbackT is still a valid function even with the error, it returns translationID.
|
|
return fallbackT, fallbackTlang, err2
|
|
}
|
|
|
|
return func(translationID string, args ...interface{}) string {
|
|
if translated := T(translationID, args...); translated != translationID {
|
|
return translated
|
|
}
|
|
|
|
return fallbackT(translationID, args...)
|
|
}, Tlang, nil
|
|
}
|
|
|
|
func GetAvailableLanguages() (languages map[string]string) {
|
|
languages = make(map[string]string)
|
|
var T i18n.TranslateFunc
|
|
for _, languageTag := range i18n.LanguageTags() {
|
|
T, _ = i18n.Tfunc(languageTag)
|
|
/* Translation files should have an ID with the translated language name.
|
|
If they don't, just use the languageTag */
|
|
if languageName := T("language_name"); languageName != "language_name" {
|
|
languages[languageTag] = languageName
|
|
} else {
|
|
languages[languageTag] = languageTag
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func setTranslation(tmpl *template.Template, T i18n.TranslateFunc) {
|
|
tmpl.Funcs(map[string]interface{}{
|
|
"T": func(str string, args ...interface{}) template.HTML {
|
|
return template.HTML(fmt.Sprintf(T(str), args...))
|
|
},
|
|
"Ts": func(str string, args ...interface{}) string {
|
|
return fmt.Sprintf(T(str), args...)
|
|
},
|
|
})
|
|
}
|
|
|
|
func GetDefaultTfunc() (i18n.TranslateFunc, error) {
|
|
return i18n.Tfunc(defaultLanguage)
|
|
}
|
|
|
|
func GetTfuncAndLanguageFromRequest(r *http.Request) (T i18n.TranslateFunc, Tlang *language.Language) {
|
|
defaultLanguage := GetDefaultLanguage()
|
|
|
|
userLanguage := ""
|
|
user, err := getCurrentUser(r)
|
|
if err == nil {
|
|
userLanguage = user.Language
|
|
}
|
|
|
|
cookie, err := r.Cookie("lang")
|
|
cookieLanguage := ""
|
|
if err == nil {
|
|
cookieLanguage = cookie.Value
|
|
}
|
|
|
|
// go-i18n supports the format of the Accept-Language header, thankfully.
|
|
headerLanguage := r.Header.Get("Accept-Language")
|
|
T, Tlang, _ = TfuncAndLanguageWithFallback(userLanguage, cookieLanguage, headerLanguage, defaultLanguage)
|
|
return
|
|
}
|
|
|
|
func SetTranslationFromRequest(tmpl *template.Template, r *http.Request) i18n.TranslateFunc {
|
|
r.Header.Add("Vary", "Accept-Encoding")
|
|
T, _ := GetTfuncAndLanguageFromRequest(r)
|
|
setTranslation(tmpl, T)
|
|
return T
|
|
}
|
|
|
|
func getCurrentUser(r *http.Request) (model.User, error) {
|
|
if userRetriever == nil {
|
|
return model.User{}, errors.New("failed to get current user: no user retriever set")
|
|
}
|
|
|
|
return userRetriever.RetrieveCurrentUser(r)
|
|
}
|