diff options
author | Christopher Speller <crspeller@gmail.com> | 2016-05-12 23:56:07 -0400 |
---|---|---|
committer | Christopher Speller <crspeller@gmail.com> | 2016-05-12 23:56:07 -0400 |
commit | 38ee83e45b4de7edf89bf9f0ef629eb4c6ad0fa8 (patch) | |
tree | a4fde09672192b97d453ad605b030bd5a10c5a45 /vendor/github.com/nicksnyder/go-i18n/goi18n/merge.go | |
parent | 84d2482ddbff9564c9ad75b2d30af66e3ddfd44d (diff) | |
download | chat-38ee83e45b4de7edf89bf9f0ef629eb4c6ad0fa8.tar.gz chat-38ee83e45b4de7edf89bf9f0ef629eb4c6ad0fa8.tar.bz2 chat-38ee83e45b4de7edf89bf9f0ef629eb4c6ad0fa8.zip |
Moving to glide
Diffstat (limited to 'vendor/github.com/nicksnyder/go-i18n/goi18n/merge.go')
-rw-r--r-- | vendor/github.com/nicksnyder/go-i18n/goi18n/merge.go | 127 |
1 files changed, 127 insertions, 0 deletions
diff --git a/vendor/github.com/nicksnyder/go-i18n/goi18n/merge.go b/vendor/github.com/nicksnyder/go-i18n/goi18n/merge.go new file mode 100644 index 000000000..1317fe958 --- /dev/null +++ b/vendor/github.com/nicksnyder/go-i18n/goi18n/merge.go @@ -0,0 +1,127 @@ +package main + +import ( + "encoding/json" + "fmt" + "gopkg.in/yaml.v2" + "io/ioutil" + "path/filepath" + "reflect" + "sort" + + "github.com/nicksnyder/go-i18n/i18n/bundle" + "github.com/nicksnyder/go-i18n/i18n/language" + "github.com/nicksnyder/go-i18n/i18n/translation" +) + +type mergeCommand struct { + translationFiles []string + sourceLanguageTag string + outdir string + format string +} + +func (mc *mergeCommand) execute() error { + if len(mc.translationFiles) < 1 { + return fmt.Errorf("need at least one translation file to parse") + } + + if lang := language.Parse(mc.sourceLanguageTag); lang == nil { + return fmt.Errorf("invalid source locale: %s", mc.sourceLanguageTag) + } + + marshal, err := newMarshalFunc(mc.format) + if err != nil { + return err + } + + bundle := bundle.New() + for _, tf := range mc.translationFiles { + if err := bundle.LoadTranslationFile(tf); err != nil { + return fmt.Errorf("failed to load translation file %s because %s\n", tf, err) + } + } + + translations := bundle.Translations() + sourceLanguageTag := language.NormalizeTag(mc.sourceLanguageTag) + sourceTranslations := translations[sourceLanguageTag] + if sourceTranslations == nil { + return fmt.Errorf("no translations found for source locale %s", sourceLanguageTag) + } + for translationID, src := range sourceTranslations { + for _, localeTranslations := range translations { + if dst := localeTranslations[translationID]; dst == nil || reflect.TypeOf(src) != reflect.TypeOf(dst) { + localeTranslations[translationID] = src.UntranslatedCopy() + } + } + } + + for localeID, localeTranslations := range translations { + lang := language.MustParse(localeID)[0] + all := filter(localeTranslations, func(t translation.Translation) translation.Translation { + return t.Normalize(lang) + }) + if err := mc.writeFile("all", all, localeID, marshal); err != nil { + return err + } + + untranslated := filter(localeTranslations, func(t translation.Translation) translation.Translation { + if t.Incomplete(lang) { + return t.Normalize(lang).Backfill(sourceTranslations[t.ID()]) + } + return nil + }) + if err := mc.writeFile("untranslated", untranslated, localeID, marshal); err != nil { + return err + } + } + return nil +} + +type marshalFunc func(interface{}) ([]byte, error) + +func (mc *mergeCommand) writeFile(label string, translations []translation.Translation, localeID string, marshal marshalFunc) error { + sort.Sort(translation.SortableByID(translations)) + buf, err := marshal(marshalInterface(translations)) + if err != nil { + return fmt.Errorf("failed to marshal %s strings to %s because %s", localeID, mc.format, err) + } + filename := filepath.Join(mc.outdir, fmt.Sprintf("%s.%s.%s", localeID, label, mc.format)) + if err := ioutil.WriteFile(filename, buf, 0666); err != nil { + return fmt.Errorf("failed to write %s because %s", filename, err) + } + return nil +} + +func filter(translations map[string]translation.Translation, filter func(translation.Translation) translation.Translation) []translation.Translation { + filtered := make([]translation.Translation, 0, len(translations)) + for _, translation := range translations { + if t := filter(translation); t != nil { + filtered = append(filtered, t) + } + } + return filtered + +} + +func newMarshalFunc(format string) (marshalFunc, error) { + switch format { + case "json": + return func(v interface{}) ([]byte, error) { + return json.MarshalIndent(v, "", " ") + }, nil + case "yaml": + return func(v interface{}) ([]byte, error) { + return yaml.Marshal(v) + }, nil + } + return nil, fmt.Errorf("unsupported format: %s\n", format) +} + +func marshalInterface(translations []translation.Translation) []interface{} { + mi := make([]interface{}, len(translations)) + for i, translation := range translations { + mi[i] = translation.MarshalInterface() + } + return mi +} |