upd: change CountryCode class into a Country class

[gigi.git] / src / org / cacert / gigi / dbObjects / Country.java

diff --git a/src/org/cacert/gigi/dbObjects/Country.java b/src/org/cacert/gigi/dbObjects/Country.java
new file mode 100644 (file)
index 0000000..977b148
--- /dev/null
+++ b/src/org/cacert/gigi/dbObjects/Country.java
@@ -0,0 +1,114 @@
+package org.cacert.gigi.dbObjects;
+
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.LinkedList;
+import java.util.List;
+import java.util.Map;
+import java.util.Random;
+
+import org.cacert.gigi.GigiApiException;
+import org.cacert.gigi.database.GigiPreparedStatement;
+import org.cacert.gigi.database.GigiResultSet;
+import org.cacert.gigi.output.template.SprintfCommand;
+
+public class Country {
+
+    public enum CountryCodeType {
+        CODE_2_CHARS(2), //
+        CODE_3_CHARS(3); //
+
+        private final int len;
+
+        private CountryCodeType(int len) {
+            this.len = len;
+        }
+
+        public int getLen() {
+            return len;
+        }
+    }
+
+    private final int id;
+
+    private final String country;
+
+    private final String countryCode2;
+
+    private final String countryCode3;
+
+    private static final List<Country> countries;
+
+    private static final Map<String, Country> byString;
+    static {
+        LinkedList<Country> cs = new LinkedList<>();
+        HashMap<String, Country> ccd = new HashMap<>();
+        try (GigiPreparedStatement ps = new GigiPreparedStatement("SELECT `id`, `english` as country, `code2`, `code3` FROM `countryIsoCode`", true)) {
+            GigiResultSet rs = ps.executeQuery();
+            while (rs.next()) {
+                Country e = new Country(rs);
+                ccd.put(e.countryCode2, e);
+                ccd.put(e.countryCode3, e);
+                cs.add(e);
+            }
+        }
+        countries = Collections.unmodifiableList(new ArrayList<>(cs));
+        byString = Collections.unmodifiableMap(ccd);
+    }
+
+    private Country(GigiResultSet rs) {
+        this.id = rs.getInt("id");
+        this.country = rs.getString("country");
+        this.countryCode2 = rs.getString("code2");
+        this.countryCode3 = rs.getString("code3");
+    }
+
+    public int getId() {
+        return id;
+    }
+
+    public String getName() {
+        return country;
+    }
+
+    public String getCode() {
+        return countryCode2;
+    }
+
+    public String getCode(CountryCodeType type) {
+        switch (type) {
+        case CODE_2_CHARS:
+            return countryCode2;
+        case CODE_3_CHARS:
+            return countryCode3;
+        default:
+            throw new IllegalArgumentException("Enum switch was non-exhaustive");
+        }
+    }
+
+    public static List<Country> getCountries() {
+        return countries;
+    }
+
+    public static void checkCountryCode(String countrycode, CountryCodeType cType) throws GigiApiException {
+        getCountryByCode(countrycode, cType);
+    }
+
+    public static Country getCountryByCode(String countrycode, CountryCodeType cType) throws GigiApiException {
+        if (countrycode.length() != cType.getLen()) {
+            throw new GigiApiException(SprintfCommand.createSimple("Country code length does not have the required length of {0} characters", Integer.toString(cType.getLen())));
+        }
+        Country i = byString.get(countrycode);
+        if (i == null) {
+            throw new GigiApiException("Country Code was wrong.");
+        }
+        return i;
+    }
+
+    public static Country getRandomCountry() {
+        List<Country> cc = Country.getCountries();
+        int rnd = new Random().nextInt(cc.size());
+        return cc.get(rnd);
+    }
+}