This commit is contained in:
Darko Luketic 2017-12-30 13:49:23 +01:00
parent eec3f7da65
commit a8a4165664

13
i18n.go
View File

@ -1,12 +1,14 @@
package i18n
import (
"log"
"github.com/nicksnyder/go-i18n/i18n/bundle"
"net/http"
"context"
"log"
"net/http"
"github.com/nicksnyder/go-i18n/i18n/bundle"
)
// Config is the configuration struct of the middleware
type Config struct {
DefaultLanguage string
Files []string // files to load
@ -16,10 +18,12 @@ type Config struct {
bundle *bundle.Bundle
}
// I18nMiddleware is the middleware that encapsulates the config
type I18nMiddleware struct {
config Config
}
// New creates a new Middleware. It requires a Config parameter.
func New(c Config) *I18nMiddleware {
if c.DefaultLanguage == "" {
log.Fatal("i18n: No default language set")
@ -49,6 +53,7 @@ func New(c Config) *I18nMiddleware {
return &I18nMiddleware{config: c}
}
// Middleware is a http.Handler compatible middleware
func (i *I18nMiddleware) Middleware(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
bycookie := false
@ -66,6 +71,7 @@ func (i *I18nMiddleware) Middleware(next http.Handler) http.Handler {
if !bycookie {
http.SetCookie(w, &http.Cookie{HttpOnly: true, Name: "lang", Value: lang})
}
ctx0 := context.WithValue(r.Context(), "i18nlang", lang)
ctx1 := context.WithValue(ctx0, "i18nrlang", rlang)
ctx2 := context.WithValue(ctx1, "i18ndlang", i.config.DefaultLanguage)
@ -74,6 +80,7 @@ func (i *I18nMiddleware) Middleware(next http.Handler) http.Handler {
})
}
// MiddlewareFunc is a http.HandlerFunc compatible middleware
func (i *I18nMiddleware) MiddlewareFunc(next http.HandlerFunc) http.HandlerFunc {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
bycookie := false