124 lines
2.4 KiB
Go
124 lines
2.4 KiB
Go
package i18n
|
|
|
|
import (
|
|
"context"
|
|
"io/fs"
|
|
"net/http"
|
|
"sync"
|
|
|
|
"github.com/pkg/errors"
|
|
"golang.org/x/text/language"
|
|
"golang.org/x/text/message"
|
|
"golang.org/x/text/message/catalog"
|
|
|
|
"github.com/derfenix/goi18n/internal"
|
|
)
|
|
|
|
var defaultLanguage = language.Russian
|
|
|
|
var (
|
|
initOnce sync.Once
|
|
builder *catalog.Builder
|
|
|
|
supportedLanguages []language.Tag
|
|
supportedLanguagesMap = map[string]struct{}{}
|
|
)
|
|
|
|
func NewExternalLoader(baseURL string, header http.Header) *internal.ExternalLoader {
|
|
return internal.NewExternalLoader(baseURL, header)
|
|
}
|
|
|
|
func NewExternalLoaderWithClient(baseURL string, header http.Header, client *http.Client) *internal.ExternalLoader {
|
|
return internal.NewExternalLoaderWithClient(baseURL, header, client)
|
|
}
|
|
|
|
type Translatable interface {
|
|
Translate(ctx context.Context) string
|
|
}
|
|
|
|
func TryTranslate(ctx context.Context, obj interface{}) (string, bool) {
|
|
if translatable, ok := obj.(Translatable); ok {
|
|
return translatable.Translate(ctx), true
|
|
}
|
|
|
|
return "", false
|
|
}
|
|
|
|
func Init(fs fs.ReadDirFS) error {
|
|
var err error
|
|
|
|
initOnce.Do(func() {
|
|
if builder != nil {
|
|
return
|
|
}
|
|
|
|
initCatalog, err := internal.InitBuilder(fs)
|
|
if err != nil {
|
|
err = errors.Wrap(err, "init catalog")
|
|
|
|
return
|
|
}
|
|
|
|
builder = initCatalog
|
|
|
|
// Fill the local cache
|
|
GetLanguages()
|
|
})
|
|
|
|
return err
|
|
}
|
|
|
|
func GetPrinter(lang language.Tag) *message.Printer {
|
|
_ = GetLanguages()
|
|
|
|
base, _ := lang.Base()
|
|
if _, ok := supportedLanguagesMap[lang.String()]; !ok {
|
|
if _, ok = supportedLanguagesMap[base.String()]; !ok {
|
|
lang = defaultLanguage
|
|
}
|
|
}
|
|
|
|
p := message.NewPrinter(lang, message.Catalog(builder))
|
|
|
|
return p
|
|
}
|
|
|
|
func GetLanguages() []language.Tag {
|
|
if supportedLanguages == nil {
|
|
if builder == nil {
|
|
return nil
|
|
}
|
|
|
|
supportedLanguages = builder.Languages()
|
|
|
|
for idx := range supportedLanguages {
|
|
supportedLanguagesMap[supportedLanguages[idx].String()] = struct{}{}
|
|
|
|
base, _ := supportedLanguages[idx].Base()
|
|
supportedLanguagesMap[base.String()] = struct{}{}
|
|
}
|
|
}
|
|
|
|
return supportedLanguages
|
|
}
|
|
|
|
func RefreshTranslations() error {
|
|
if builder == nil {
|
|
return nil
|
|
}
|
|
|
|
if err := internal.RefreshTranslations(builder); err != nil {
|
|
return errors.WithMessage(err, "refresh translations")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func SetExternalBuilder(b func(builder *catalog.Builder) error) {
|
|
internal.SetExtendBuilder(b)
|
|
}
|
|
|
|
func SetExternalLoader(loader internal.Loader) {
|
|
internal.SetExternalLoader(loader)
|
|
}
|