summaryrefslogtreecommitdiff
path: root/ki18n/ki18n.go
blob: 8f20ad407bbdb4964bd9d0347b122d1b6ff30aa1 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90

package ki18n

import (
	"context"
	"io/fs"
	"net/http"

	"github.com/a-h/templ"
	"github.com/kataras/i18n"
)

type contextKey struct{}

const defaultLang = "en-GB"

type Strategy interface {
	Detect(r *http.Request) string
}

type cookieStrategy struct{ name string }

func (s cookieStrategy) Detect(r *http.Request) string {
	c, err := r.Cookie(s.name)
	if err != nil {
		return ""
	}
	return c.Value
}

func CookieStrategy(name string) Strategy {
	return cookieStrategy{name: name}
}

type acceptLanguageStrategy struct{}

func (acceptLanguageStrategy) Detect(r *http.Request) string {
	lang := i18n.Default.GetLocale(r).Language()
	if lang == defaultLang {
		if r.Header.Get("Accept-Language") == "" {
			return ""
		}
	}
	return lang
}

var AcceptLanguageStrategy Strategy = acceptLanguageStrategy{}

func LangMiddleware(strategies ...Strategy) func(http.Handler) http.Handler {
	return func(next http.Handler) http.Handler {
		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			lang := ""
			for _, s := range strategies {
				if l := s.Detect(r); l != "" {
					lang = l
					break
				}
			}
			if lang == "" {
				lang = defaultLang
			}
			ctx := context.WithValue(r.Context(), contextKey{}, lang)
			next.ServeHTTP(w, r.WithContext(ctx))
		})
	}
}

func Tr(ctx context.Context, format string, args ...any) templ.Component {
	lang, _ := ctx.Value(contextKey{}).(string)
	if lang == "" {
		lang = defaultLang
	}
	msg := i18n.Tr(lang, format, args...)
	if msg == "" {
		msg = format
	}
	return templ.Raw(msg)
}

func Init(fs fs.FS) error {
	loader, err := i18n.FS(fs, "*/*.yml")
	if err != nil {
		return err
	}
	inst, err := i18n.New(loader, "en-GB", "fr-FR")
	if err != nil {
		return err
	}
	i18n.Default = inst
	return nil
}