/
translation_map.go
89 lines (70 loc) · 1.64 KB
/
translation_map.go
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
package i18n
import (
"bytes"
"fmt"
"io"
"golang.org/x/text/language"
)
type TranslationMap struct {
keys map[string]string
lang language.Tag
}
func NewMap(lang language.Tag, keys map[string]string) TranslationMap {
return TranslationMap{keys, lang}
}
func (t TranslationMap) Get(key string) string {
if out, ok := t.keys[key]; ok {
return out
}
return key
}
func (t TranslationMap) Has(key string) bool {
return t.keys[key] != ""
}
func (t TranslationMap) Len() int {
return len(t.keys)
}
func (t TranslationMap) Tag() language.Tag {
return t.lang
}
func (t TranslationMap) Fprint(w io.Writer, a ...interface{}) (int, error) {
var out = make([]interface{}, len(a))
for i, arg := range a {
if key, ok := arg.(string); ok {
out[i] = t.Get(key)
} else {
out[i] = arg
}
}
return fmt.Fprint(w, out...)
}
func (t TranslationMap) Fprintf(w io.Writer, key string, a ...interface{}) (int, error) {
return fmt.Fprintf(w, t.Get(key), a...)
}
func (t TranslationMap) Fprintln(w io.Writer, a ...interface{}) (int, error) {
var out = make([]interface{}, len(a))
for i, arg := range a {
if key, ok := arg.(string); ok {
out[i] = t.Get(key)
} else {
out[i] = arg
}
}
return fmt.Fprintln(w, out...)
}
func (t TranslationMap) Sprint(a ...interface{}) string {
b := new(bytes.Buffer)
t.Fprint(b, a...)
return b.String()
}
func (t TranslationMap) Sprintf(key string, a ...interface{}) string {
b := new(bytes.Buffer)
t.Fprintf(b, key, a...)
return b.String()
}
func (t TranslationMap) Sprintln(a ...interface{}) string {
b := new(bytes.Buffer)
t.Fprintln(b, a...)
return b.String()
}
var _ Translation = (*TranslationMap)(nil)