i18n.go

package i18n

import (
	"reflect"
	"strings"

	"github.com/h4tr3d/iris/v6"
	"github.com/unknwon/i18n"
)

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
}