Add a country list auto-generated from CLDR.

address.go

@@ -3,7 +3,10 @@
 
 package address
 
-import "regexp"
+import (
+	"regexp"
+	"sort"
+)
 
 // Address represents an address.
 type Address struct {
@@ -81,6 +84,32 @@ func (f Format) CheckPostalCode(postalCode string) bool {
 	return rx.MatchString(postalCode)
 }
 
+// CheckCountryCode checks whether the given country code is valid.
+//
+// An empty country code is considered valid.
+func CheckCountryCode(countryCode string) bool {
+	if countryCode == "" {
+		return true
+	}
+	_, ok := countries[countryCode]
+	return ok
+}
+
+// GetCountryCodes returns all known country codes.
+func GetCountryCodes() []string {
+	countryCodes := make([]string, 0, len(countries))
+	for countryCode := range countries {
+		countryCodes = append(countryCodes, countryCode)
+	}
+	sort.Strings(countryCodes)
+	return countryCodes
+}
+
+// GetCountryNames returns all known country names, keyed by country code.
+func GetCountryNames() map[string]string {
+	return countries
+}
+
 // GetFormats returns all known address formats, keyed by country code.
 //
 // The ZZ address format represents the generic fallback.

address_test.go

@@ -121,6 +121,55 @@ func TestFormat_CheckPostalCode(t *testing.T) {
 	}
 }
 
+func TestCheckCountryCode(t *testing.T) {
+	tests := []struct {
+		countryCode string
+		want        bool
+	}{
+		// Empty value.
+		{"", true},
+		// Valid country code.
+		{"FR", true},
+		// Invalid country code.
+		{"ABCD", false},
+	}
+
+	for _, tt := range tests {
+		t.Run("", func(t *testing.T) {
+			got := address.CheckCountryCode(tt.countryCode)
+			if got != tt.want {
+				t.Errorf("got %v, want %v", got, tt.want)
+			}
+		})
+	}
+}
+
+func TestGetCountryCodes(t *testing.T) {
+	countryCodes := address.GetCountryCodes()
+	for _, countryCode := range []string{"FR", "RS", "US"} {
+		if !contains(countryCodes, countryCode) {
+			t.Errorf("no %v country code found.", countryCode)
+		}
+	}
+}
+
+func TestGetCountryNames(t *testing.T) {
+	got := address.GetCountryNames()
+	want := map[string]string{
+		"FR": "France",
+		"RS": "Serbia",
+	}
+	for wantCode, wantName := range want {
+		gotName, ok := got[wantCode]
+		if !ok {
+			t.Errorf("no %v country code found.", wantCode)
+		}
+		if gotName != wantName {
+			t.Errorf("got %v, want %v", gotName, wantName)
+		}
+	}
+}
+
 func TestGetFormat(t *testing.T) {
 	// Existing format.
 	got := address.GetFormat("RS")
@@ -159,3 +208,12 @@ func TestGetFormats(t *testing.T) {
 		}
 	}
 }
+
+func contains(a []string, x string) bool {
+	for _, v := range a {
+		if v == x {
+			return true
+		}
+	}
+	return false
+}