1 files changed, 90 insertions, 0 deletions
diff --git a/src/libcalamares/locale/Lookup.cpp b/src/libcalamares/locale/Lookup.cpp
new file mode 100644
index 000000000..d38d417d4
--- /dev/null
+++ b/src/libcalamares/locale/Lookup.cpp
@@ -0,0 +1,90 @@
+/* === This file is part of Calamares - <https://github.com/calamares> ===
+ *
+ *   Copyright 2019, Adriaan de Groot <groot@kde.org>
+ *
+ *   Calamares is free software: you can redistribute it and/or modify
+ *   it under the terms of the GNU General Public License as published by
+ *   the Free Software Foundation, either version 3 of the License, or
+ *   (at your option) any later version.
+ *
+ *   Calamares is distributed in the hope that it will be useful,
+ *   but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ *   GNU General Public License for more details.
+ *
+ *   You should have received a copy of the GNU General Public License
+ *   along with Calamares. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "Lookup.h"
+
+#include "CountryData_p.cpp"
+
+namespace Calamares
+{
+
+struct TwoChar
+{
+    TwoChar( const QString& code )
+        : cc1(0)
+        , cc2(0)
+    {
+        if ( code.length() == 2 )
+        {
+            cc1 = code[0].toLatin1();
+            cc2 = code[1].toLatin1();
+        }
+    }
+    
+    char cc1;
+    char cc2;
+};
+
+static const CountryData* lookup( TwoChar c )
+{
+    if ( !c.cc1 )
+        return nullptr;
+    
+    const CountryData* p = std::find_if(country_data_table, country_data_table + country_data_size,
+        [c=c]( const CountryData& d ){ return (d.cc1 == c.cc1) && (d.cc2 == c.cc2); }
+    );
+    if ( p == country_data_table + country_data_size )
+        return nullptr;
+    return p;
+}
+    
+QLocale::Country countryForCode(const QString& code)
+{
+    const CountryData* p = lookup( TwoChar( code ) );
+    return p ? p->c : QLocale::Country::AnyCountry;
+}
+
+QLocale::Language languageForCountry(const QString& code)
+{
+    const CountryData* p = lookup( TwoChar( code ) );
+    return p ? p->l : QLocale::Language::AnyLanguage;
+}
+
+QPair<QLocale::Country, QLocale::Language> countryData(const QString& code)
+{
+    const CountryData* p = lookup( TwoChar( code ) );
+    return p ? qMakePair( p->c, p->l ) : qMakePair( QLocale::Country::AnyCountry, QLocale::Language::AnyLanguage );
+}
+
+QLocale countryLocale(const QString& code)
+{
+    auto p = countryData( code );
+    return QLocale( p.second, p.first );
+}
+
+QLocale::Language languageForCountry(QLocale::Country country)
+{
+    const CountryData* p = std::find_if(country_data_table, country_data_table + country_data_size,
+        [c=country]( const CountryData& d ){ return d.c == c; }
+    );
+    if ( p == country_data_table + country_data_size )
+        return QLocale::Language::AnyLanguage;
+    return p->l;
+}
+
+}  // namespace