108 lines
3.0 KiB
Go
108 lines
3.0 KiB
Go
package i18n
|
|
|
|
import (
|
|
"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
|
|
FilesBytes map[string][]byte // or slices of []bytes with the embedded file data
|
|
Debug bool
|
|
URLParam string
|
|
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")
|
|
}
|
|
if len(c.Files) == 0 && len(c.FilesBytes) == 0 {
|
|
log.Fatal("i18n: You need to supply either Config.Files and|or Config.FileBytes for language files to be loaded|parsed")
|
|
}
|
|
|
|
b := bundle.New()
|
|
c.bundle = b
|
|
|
|
for _, file := range c.Files {
|
|
if e := b.LoadTranslationFile(file); e != nil {
|
|
log.Fatal("i18n:", e.Error())
|
|
}
|
|
}
|
|
for s, by := range c.FilesBytes {
|
|
if e := b.ParseTranslationFileBytes(s, by); e != nil {
|
|
log.Fatal("i18n:", e.Error())
|
|
}
|
|
}
|
|
|
|
if c.Debug {
|
|
log.Println("i18n: Loaded languages")
|
|
log.Println(b.Translations())
|
|
}
|
|
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
|
|
lang := r.URL.Query().Get(i.config.URLParam)
|
|
rlang := r.Header.Get("Accept-Language")
|
|
if lang == "" {
|
|
lc, e := r.Cookie("lang")
|
|
if e != nil {
|
|
lang = ""
|
|
} else {
|
|
lang = lc.Value
|
|
bycookie = true
|
|
}
|
|
}
|
|
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)
|
|
ctx3 := context.WithValue(ctx2, "i18nTfunc", i.config.bundle.MustTfunc(lang, rlang, i.config.DefaultLanguage))
|
|
next.ServeHTTP(w, r.WithContext(ctx3))
|
|
})
|
|
}
|
|
|
|
// 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
|
|
lang := r.URL.Query().Get(i.config.URLParam)
|
|
rlang := r.Header.Get("Accept-Language")
|
|
if lang == "" {
|
|
lc, e := r.Cookie("lang")
|
|
if e != nil {
|
|
lang = ""
|
|
} else {
|
|
lang = lc.Value
|
|
bycookie = true
|
|
}
|
|
}
|
|
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)
|
|
ctx3 := context.WithValue(ctx2, "i18nTfunc", i.config.bundle.MustTfunc(lang, rlang, i.config.DefaultLanguage))
|
|
next.ServeHTTP(w, r.WithContext(ctx3))
|
|
})
|
|
}
|