existing language detection reimplemented

icrowley · Dec 23, 2014 · 64a824f · 64a824f
1 parent 170d5bc
commit 64a824f
Show file tree

Hide file tree

Showing 13 changed files with 23 additions and 27 deletions.
diff --git a/fake.go b/fake.go
@@ -17,17 +17,25 @@ var r = rand.New(rand.NewSource(time.Now().UnixNano()))
 var lang = "en"
 var useExternalData = false
 var enFallback = false
+var availLangs = GetLangs()
 
 var (
 	ErrInsufficientParams = fmt.Errorf("Insufficient params to lookup the samples")
 	ErrNoLanguageFn       = func(lang string) error { return fmt.Errorf("The language passed (%s) is not available", lang) }
 	ErrNoSamplesFn        = func(lang string) error { return fmt.Errorf("No samples found for language: %s", lang) }
 )
 
-func SetLang(newLang string) error {
-	availLangs := []string{"en", "br", "ca", "da", "de", "fi", "fr", "mx", "nl", "se", "sn", "uk", "us", "ja", "ar", "cn", "cs",
-		"ga", "it", "kr", "nb", "ph", "th", "vn"}
+func GetLangs() []string {
+	var langs []string
+	for k, v := range data {
+		if v.isDir && k != "/" && k != "/data" {
+			langs = append(langs, strings.Replace(k, "/data", "", 1))
+		}
+	}
+	return langs
+}
 
+func SetLang(newLang string) error {
 	found := false
 	for _, l := range availLangs {
 		if newLang == l {

diff --git a/test/addresses_test.go b/test/addresses_test.go
@@ -7,8 +7,7 @@ import (
 )
 
 func TestAddresses(t *testing.T) {
-	langs := []string{"en", "ru"}
-	for _, lang := range langs {
+	for _, lang := range fake.GetLangs() {
 		fake.SetLang(lang)
 
 		v := fake.Continent()

diff --git a/test/credit_cards_test.go b/test/credit_cards_test.go
@@ -7,8 +7,7 @@ import (
 )
 
 func TestCreditCards(t *testing.T) {
-	langs := []string{"en", "ru"}
-	for _, lang := range langs {
+	for _, lang := range fake.GetLangs() {
 		fake.SetLang(lang)
 
 		v := fake.CreditCardType()

diff --git a/test/currencies_test.go b/test/currencies_test.go
@@ -7,8 +7,7 @@ import (
 )
 
 func TestCurrencies(t *testing.T) {
-	langs := []string{"en", "ru"}
-	for _, lang := range langs {
+	for _, lang := range fake.GetLangs() {
 		fake.SetLang(lang)
 
 		v := fake.Currency()

diff --git a/test/dates_test.go b/test/dates_test.go
@@ -8,8 +8,7 @@ import (
 )
 
 func TestDates(t *testing.T) {
-	langs := []string{"en", "ru"}
-	for _, lang := range langs {
+	for _, lang := range fake.GetLangs() {
 		fake.SetLang(lang)
 
 		v := fake.WeekDay()

diff --git a/test/general_test.go b/test/general_test.go
@@ -7,8 +7,7 @@ import (
 )
 
 func TestGeneral(t *testing.T) {
-	langs := []string{"en", "ru"}
-	for _, lang := range langs {
+	for _, lang := range fake.GetLangs() {
 		fake.SetLang(lang)
 
 		v := fake.Password(4, 10, true, true, true)

diff --git a/test/geo_test.go b/test/geo_test.go
@@ -7,8 +7,7 @@ import (
 )
 
 func TestGeo(t *testing.T) {
-	langs := []string{"en", "ru"}
-	for _, lang := range langs {
+	for _, lang := range fake.GetLangs() {
 		fake.SetLang(lang)
 
 		f := fake.Latitute()

diff --git a/test/internet_test.go b/test/internet_test.go
@@ -7,8 +7,7 @@ import (
 )
 
 func TestInternet(t *testing.T) {
-	langs := []string{"en", "ru"}
-	for _, lang := range langs {
+	for _, lang := range fake.GetLangs() {
 		fake.SetLang(lang)
 
 		v := fake.UserName()

diff --git a/test/jobs_test.go b/test/jobs_test.go
@@ -7,8 +7,7 @@ import (
 )
 
 func TestJobs(t *testing.T) {
-	langs := []string{"en", "ru"}
-	for _, lang := range langs {
+	for _, lang := range fake.GetLangs() {
 		fake.SetLang(lang)
 
 		v := fake.Company()

diff --git a/test/lorem_ipsum_test.go b/test/lorem_ipsum_test.go
@@ -7,8 +7,7 @@ import (
 )
 
 func TestLoremIpsum(t *testing.T) {
-	langs := []string{"en", "ru"}
-	for _, lang := range langs {
+	for _, lang := range fake.GetLangs() {
 		fake.SetLang(lang)
 
 		v := fake.Character()

diff --git a/test/names_test.go b/test/names_test.go
@@ -7,8 +7,7 @@ import (
 )
 
 func TestNames(t *testing.T) {
-	langs := []string{"en", "ru"}
-	for _, lang := range langs {
+	for _, lang := range fake.GetLangs() {
 		fake.SetLang(lang)
 
 		v := fake.MaleFirstName()

diff --git a/test/personal_test.go b/test/personal_test.go
@@ -7,8 +7,7 @@ import (
 )
 
 func TestPersonal(t *testing.T) {
-	langs := []string{"en", "ru"}
-	for _, lang := range langs {
+	for _, lang := range fake.GetLangs() {
 		fake.SetLang(lang)
 
 		v := fake.Gender()

diff --git a/test/products_test.go b/test/products_test.go
@@ -7,8 +7,7 @@ import (
 )
 
 func TestProducts(t *testing.T) {
-	langs := []string{"en", "ru"}
-	for _, lang := range langs {
+	for _, lang := range fake.GetLangs() {
 		fake.SetLang(lang)
 
 		v := fake.Brand()