Brendan LE GLAUNEC
5849898283
Unit tests functional and coverage back to 100% Add more routes to dictionary, add more credentials, add default port 5554, rename cameradar logs ENV variable, improve unit test readability, remove tmp file
114 lines
2.8 KiB
Go
114 lines
2.8 KiB
Go
package ut
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/go-playground/locales"
|
|
)
|
|
|
|
// UniversalTranslator holds all locale & translation data
|
|
type UniversalTranslator struct {
|
|
translators map[string]Translator
|
|
fallback Translator
|
|
}
|
|
|
|
// New returns a new UniversalTranslator instance set with
|
|
// the fallback locale and locales it should support
|
|
func New(fallback locales.Translator, supportedLocales ...locales.Translator) *UniversalTranslator {
|
|
|
|
t := &UniversalTranslator{
|
|
translators: make(map[string]Translator),
|
|
}
|
|
|
|
for _, v := range supportedLocales {
|
|
|
|
trans := newTranslator(v)
|
|
t.translators[strings.ToLower(trans.Locale())] = trans
|
|
|
|
if fallback.Locale() == v.Locale() {
|
|
t.fallback = trans
|
|
}
|
|
}
|
|
|
|
if t.fallback == nil && fallback != nil {
|
|
t.fallback = newTranslator(fallback)
|
|
}
|
|
|
|
return t
|
|
}
|
|
|
|
// FindTranslator trys to find a Translator based on an array of locales
|
|
// and returns the first one it can find, otherwise returns the
|
|
// fallback translator.
|
|
func (t *UniversalTranslator) FindTranslator(locales ...string) (trans Translator, found bool) {
|
|
|
|
for _, locale := range locales {
|
|
|
|
if trans, found = t.translators[strings.ToLower(locale)]; found {
|
|
return
|
|
}
|
|
}
|
|
|
|
return t.fallback, false
|
|
}
|
|
|
|
// GetTranslator returns the specified translator for the given locale,
|
|
// or fallback if not found
|
|
func (t *UniversalTranslator) GetTranslator(locale string) (trans Translator, found bool) {
|
|
|
|
if trans, found = t.translators[strings.ToLower(locale)]; found {
|
|
return
|
|
}
|
|
|
|
return t.fallback, false
|
|
}
|
|
|
|
// GetFallback returns the fallback locale
|
|
func (t *UniversalTranslator) GetFallback() Translator {
|
|
return t.fallback
|
|
}
|
|
|
|
// AddTranslator adds the supplied translator, if it already exists the override param
|
|
// will be checked and if false an error will be returned, otherwise the translator will be
|
|
// overridden; if the fallback matches the supplied translator it will be overridden as well
|
|
// NOTE: this is normally only used when translator is embedded within a library
|
|
func (t *UniversalTranslator) AddTranslator(translator locales.Translator, override bool) error {
|
|
|
|
lc := strings.ToLower(translator.Locale())
|
|
_, ok := t.translators[lc]
|
|
if ok && !override {
|
|
return &ErrExistingTranslator{locale: translator.Locale()}
|
|
}
|
|
|
|
trans := newTranslator(translator)
|
|
|
|
if t.fallback.Locale() == translator.Locale() {
|
|
|
|
// because it's optional to have a fallback, I don't impose that limitation
|
|
// don't know why you wouldn't but...
|
|
if !override {
|
|
return &ErrExistingTranslator{locale: translator.Locale()}
|
|
}
|
|
|
|
t.fallback = trans
|
|
}
|
|
|
|
t.translators[lc] = trans
|
|
|
|
return nil
|
|
}
|
|
|
|
// VerifyTranslations runs through all locales and identifies any issues
|
|
// eg. missing plural rules for a locale
|
|
func (t *UniversalTranslator) VerifyTranslations() (err error) {
|
|
|
|
for _, trans := range t.translators {
|
|
err = trans.VerifyTranslations()
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
|
|
return
|
|
}
|