mirror of
https://github.com/kataras/iris.git
synced 2025-01-24 19:21:03 +01:00
102 lines
2.8 KiB
Go
102 lines
2.8 KiB
Go
package i18n
|
|
|
|
import (
|
|
"reflect"
|
|
"strings"
|
|
|
|
"github.com/Unknwon/i18n"
|
|
"gopkg.in/kataras/iris.v6"
|
|
)
|
|
|
|
type i18nMiddleware struct {
|
|
config Config
|
|
}
|
|
|
|
// Serve serves the request, the actual middleware's job is here
|
|
func (i *i18nMiddleware) Serve(ctx *iris.Context) {
|
|
wasByCookie := false
|
|
|
|
language := i.config.Default
|
|
if ctx.GetString(iris.TranslateLanguageContextKey) == "" {
|
|
// try to get by url parameter
|
|
language = ctx.URLParam(i.config.URLParameter)
|
|
|
|
if language == "" {
|
|
// then try to take the lang field from the cookie
|
|
language = ctx.GetCookie(iris.TranslateLanguageContextKey)
|
|
|
|
if len(language) > 0 {
|
|
wasByCookie = true
|
|
} else {
|
|
// try to get by the request headers(?)
|
|
if langHeader := ctx.RequestHeader("Accept-Language"); i18n.IsExist(langHeader) {
|
|
language = langHeader
|
|
}
|
|
}
|
|
}
|
|
// if it was not taken by the cookie, then set the cookie in order to have it
|
|
if !wasByCookie {
|
|
ctx.SetCookieKV(iris.TranslateLanguageContextKey, language)
|
|
}
|
|
if language == "" {
|
|
language = i.config.Default
|
|
}
|
|
ctx.Set(iris.TranslateLanguageContextKey, language)
|
|
}
|
|
locale := i18n.Locale{Lang: language}
|
|
|
|
ctx.Set(iris.TranslateFunctionContextKey, locale.Tr)
|
|
ctx.Next()
|
|
}
|
|
|
|
// Translate returns the translated word from a context
|
|
// the second parameter is the key of the world or line inside the .ini file
|
|
// the third parameter is the '%s' of the world or line inside the .ini file
|
|
func Translate(ctx *iris.Context, format string, args ...interface{}) string {
|
|
return ctx.Translate(format, args...)
|
|
}
|
|
|
|
// New returns a new i18n middleware
|
|
func New(c Config) iris.HandlerFunc {
|
|
if len(c.Languages) == 0 {
|
|
panic("You cannot use this middleware without set the Languages option, please try again and read the _example.")
|
|
}
|
|
i := &i18nMiddleware{config: c}
|
|
firstlanguage := ""
|
|
//load the files
|
|
for k, v := range c.Languages {
|
|
if !strings.HasSuffix(v, ".ini") {
|
|
v += ".ini"
|
|
}
|
|
err := i18n.SetMessage(k, v)
|
|
if err != nil && err != i18n.ErrLangAlreadyExist {
|
|
panic("Iris i18n Middleware: Failed to set locale file" + k + " Error:" + err.Error())
|
|
}
|
|
if firstlanguage == "" {
|
|
firstlanguage = k
|
|
}
|
|
}
|
|
// if not default language setted then set to the first of the i.options.Languages
|
|
if c.Default == "" {
|
|
c.Default = firstlanguage
|
|
}
|
|
|
|
i18n.SetDefaultLang(i.config.Default)
|
|
return i.Serve
|
|
}
|
|
|
|
// TranslatedMap returns translated map[string]interface{} from i18n structure
|
|
func TranslatedMap(sourceInterface interface{}, ctx *iris.Context) map[string]interface{} {
|
|
iType := reflect.TypeOf(sourceInterface).Elem()
|
|
result := make(map[string]interface{})
|
|
|
|
for i := 0; i < iType.NumField(); i++ {
|
|
fieldName := reflect.TypeOf(sourceInterface).Elem().Field(i).Name
|
|
fieldValue := reflect.ValueOf(sourceInterface).Elem().Field(i).String()
|
|
|
|
result[fieldName] = Translate(ctx, fieldValue)
|
|
}
|
|
|
|
return result
|
|
}
|