From 7ba418104b3f77ee05274c8de966d9c4ab871e53 Mon Sep 17 00:00:00 2001
From: Dennis Faust <dennis@dennisfaust.com>
Date: Thu, 8 Oct 2015 11:28:25 -0700
Subject: [PATCH] Add LanguageTags and LanguageTranslationIDs functions

Functions to return a string array of the loaded language's tags and
also the IDs of the each language's translations. These can be used
for tests and verifications purposes once the languages and translations
have been loaded.
---
 i18n/bundle/bundle.go      | 18 ++++++++++++++++++
 i18n/bundle/bundle_test.go | 28 ++++++++++++++++++++++++++++
 i18n/i18n.go               | 10 ++++++++++
 3 files changed, 56 insertions(+)

diff --git a/i18n/bundle/bundle.go b/i18n/bundle/bundle.go
index 28e65c69..38dbe3a5 100644
--- a/i18n/bundle/bundle.go
+++ b/i18n/bundle/bundle.go
@@ -134,6 +134,24 @@ func (b *Bundle) Translations() map[string]map[string]translation.Translation {
 	return b.translations
 }
 
+// LanguageTags returns the tags of all languages that that have been added.
+func (b *Bundle) LanguageTags() []string {
+	var tags []string
+	for k := range b.translations {
+		tags = append(tags, k)
+	}
+	return tags
+}
+
+// LanguageTranslationIDs returns the ids of all translations that have been added for a given language.
+func (b *Bundle) LanguageTranslationIDs(languageTag string) []string {
+	var ids []string
+	for id := range b.translations[languageTag] {
+		ids = append(ids, id)
+	}
+	return ids
+}
+
 // MustTfunc is similar to Tfunc except it panics if an error happens.
 func (b *Bundle) MustTfunc(pref string, prefs ...string) TranslateFunc {
 	tfunc, err := b.Tfunc(pref, prefs...)
diff --git a/i18n/bundle/bundle_test.go b/i18n/bundle/bundle_test.go
index cdbc3689..b9c0a059 100644
--- a/i18n/bundle/bundle_test.go
+++ b/i18n/bundle/bundle_test.go
@@ -4,6 +4,9 @@ import (
 	"fmt"
 	"testing"
 
+	"reflect"
+	"sort"
+
 	"github.com/nicksnyder/go-i18n/i18n/language"
 	"github.com/nicksnyder/go-i18n/i18n/translation"
 )
@@ -33,6 +36,31 @@ func TestMustTfunc(t *testing.T) {
 	New().MustTfunc("invalid")
 }
 
+func TestLanguageTagsAndTranslationIDs(t *testing.T) {
+	b := New()
+	translationID := "translation_id"
+	englishLanguage := languageWithTag("en-US")
+	frenchLanguage := languageWithTag("fr-FR")
+	spanishLanguage := languageWithTag("es")
+	addFakeTranslation(t, b, englishLanguage, "English"+translationID)
+	addFakeTranslation(t, b, frenchLanguage, translationID)
+	addFakeTranslation(t, b, spanishLanguage, translationID)
+
+	tags := b.LanguageTags()
+	sort.Strings(tags)
+	compareTo := []string{englishLanguage.Tag, spanishLanguage.Tag, frenchLanguage.Tag}
+	if !reflect.DeepEqual(tags, compareTo) {
+		t.Errorf("LanguageTags() = %#v; expected: %#v", tags, compareTo)
+	}
+
+	ids := b.LanguageTranslationIDs(englishLanguage.Tag)
+	sort.Strings(ids)
+	compareTo = []string{"English" + translationID}
+	if !reflect.DeepEqual(ids, compareTo) {
+		t.Errorf("LanguageTranslationIDs() = %#v; expected: %#v", ids, compareTo)
+	}
+}
+
 func TestTfuncAndLanguage(t *testing.T) {
 	b := New()
 	translationID := "translation_id"
diff --git a/i18n/i18n.go b/i18n/i18n.go
index 211a9d1b..f9684296 100644
--- a/i18n/i18n.go
+++ b/i18n/i18n.go
@@ -115,6 +115,16 @@ func AddTranslation(lang *language.Language, translations ...translation.Transla
 	defaultBundle.AddTranslation(lang, translations...)
 }
 
+// LanguageTags returns the tags of all languages that have been added.
+func LanguageTags() []string {
+	return defaultBundle.LanguageTags()
+}
+
+// LanguageTranslationIDs returns the ids of all translations that have been added for a given language.
+func LanguageTranslationIDs(languageTag string) []string {
+	return defaultBundle.LanguageTranslationIDs(languageTag)
+}
+
 // MustTfunc is similar to Tfunc except it panics if an error happens.
 func MustTfunc(languageSource string, languageSources ...string) TranslateFunc {
 	return TranslateFunc(defaultBundle.MustTfunc(languageSource, languageSources...))