123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832 |
- /*
- * Copyright (c) 2021, Tim Flynn <trflynn89@pm.me>
- *
- * SPDX-License-Identifier: BSD-2-Clause
- */
- #include <AK/AllOf.h>
- #include <AK/CharacterTypes.h>
- #include <AK/Format.h>
- #include <AK/HashMap.h>
- #include <AK/JsonObject.h>
- #include <AK/JsonParser.h>
- #include <AK/JsonValue.h>
- #include <AK/LexicalPath.h>
- #include <AK/QuickSort.h>
- #include <AK/SourceGenerator.h>
- #include <AK/String.h>
- #include <AK/StringBuilder.h>
- #include <LibCore/ArgsParser.h>
- #include <LibCore/DirIterator.h>
- #include <LibCore/File.h>
- struct Locale {
- String language;
- Optional<String> territory;
- Optional<String> variant;
- HashMap<String, String> languages;
- HashMap<String, String> territories;
- HashMap<String, String> scripts;
- HashMap<String, String> currencies;
- };
- struct UnicodeLocaleData {
- HashMap<String, Locale> locales;
- Vector<String> languages;
- Vector<String> territories;
- Vector<String> scripts;
- Vector<String> variants;
- Vector<String> currencies;
- HashMap<String, String> language_aliases;
- HashMap<String, String> territory_aliases;
- HashMap<String, String> script_aliases;
- HashMap<String, String> variant_aliases;
- HashMap<String, String> subdivision_aliases;
- HashMap<String, String> complex_mappings;
- HashMap<String, String> likely_subtags;
- Vector<String> likely_territory_subtags;
- };
- static void write_to_file_if_different(Core::File& file, StringView contents)
- {
- auto const current_contents = file.read_all();
- if (StringView { current_contents.bytes() } == contents)
- return;
- VERIFY(file.seek(0));
- VERIFY(file.truncate(0));
- VERIFY(file.write(contents));
- }
- static void parse_core_aliases(String core_supplemental_path, UnicodeLocaleData& locale_data)
- {
- LexicalPath core_aliases_path(move(core_supplemental_path));
- core_aliases_path = core_aliases_path.append("aliases.json"sv);
- VERIFY(Core::File::exists(core_aliases_path.string()));
- auto core_aliases_file_or_error = Core::File::open(core_aliases_path.string(), Core::OpenMode::ReadOnly);
- VERIFY(!core_aliases_file_or_error.is_error());
- auto core_aliases = JsonParser(core_aliases_file_or_error.value()->read_all()).parse();
- VERIFY(core_aliases.has_value());
- auto const& supplemental_object = core_aliases->as_object().get("supplemental"sv);
- auto const& metadata_object = supplemental_object.as_object().get("metadata"sv);
- auto const& alias_object = metadata_object.as_object().get("alias"sv);
- auto append_aliases = [&](auto& alias_object, auto& alias_map, Vector<String>* likely_subtags_list = nullptr) {
- alias_object.as_object().for_each_member([&](auto const& key, JsonValue const& value) {
- auto alias = value.as_object().get("_replacement"sv).as_string();
- if (auto aliases = alias.split(' '); likely_subtags_list && (aliases.size() > 1))
- likely_subtags_list->extend(move(aliases));
- if (key.contains('-'))
- locale_data.complex_mappings.set(key, move(alias));
- else
- alias_map.set(key, move(alias));
- });
- };
- append_aliases(alias_object.as_object().get("languageAlias"sv), locale_data.language_aliases);
- append_aliases(alias_object.as_object().get("territoryAlias"sv), locale_data.territory_aliases, &locale_data.likely_territory_subtags);
- append_aliases(alias_object.as_object().get("scriptAlias"sv), locale_data.script_aliases);
- append_aliases(alias_object.as_object().get("variantAlias"sv), locale_data.variant_aliases);
- append_aliases(alias_object.as_object().get("subdivisionAlias"sv), locale_data.subdivision_aliases);
- }
- static void parse_likely_subtags(String core_supplemental_path, UnicodeLocaleData& locale_data)
- {
- LexicalPath likely_subtags_path(move(core_supplemental_path));
- likely_subtags_path = likely_subtags_path.append("likelySubtags.json"sv);
- VERIFY(Core::File::exists(likely_subtags_path.string()));
- auto likely_subtags_file_or_error = Core::File::open(likely_subtags_path.string(), Core::OpenMode::ReadOnly);
- VERIFY(!likely_subtags_file_or_error.is_error());
- auto likely_subtags = JsonParser(likely_subtags_file_or_error.value()->read_all()).parse();
- VERIFY(likely_subtags.has_value());
- auto const& supplemental_object = likely_subtags->as_object().get("supplemental"sv);
- auto const& likely_subtags_object = supplemental_object.as_object().get("likelySubtags"sv);
- likely_subtags_object.as_object().for_each_member([&](auto const& key, JsonValue const& value) {
- auto likely_subtag = value.as_string();
- auto regions = likely_subtag.split('-');
- VERIFY(regions.size() == 3);
- // Unicode TR35 has the following footnote in section 3.2.1 Canonical Unicode Locale Identifiers
- //
- // Formally, replacement of multiple territories uses Section 4.3 Likely Subtags. However, there are a small
- // number of cases of multiple territories, so the mappings can be precomputed. This results in a faster
- // lookup with a very small subset of the likely subtags data.
- //
- // Since the likely subtags data is quite large, and resolving likely territory subtags is our only use case for
- // this data, we only generate likely subtags that contain one of the above multiple territories.
- if (locale_data.likely_territory_subtags.contains_slow(regions[2]))
- locale_data.likely_subtags.set(key, move(likely_subtag));
- });
- }
- static void parse_identity(String locale_path, UnicodeLocaleData& locale_data, Locale& locale)
- {
- LexicalPath languages_path(move(locale_path)); // Note: Every JSON file defines identity data, so we can use any of them.
- languages_path = languages_path.append("languages.json"sv);
- VERIFY(Core::File::exists(languages_path.string()));
- auto languages_file_or_error = Core::File::open(languages_path.string(), Core::OpenMode::ReadOnly);
- VERIFY(!languages_file_or_error.is_error());
- auto languages = JsonParser(languages_file_or_error.value()->read_all()).parse();
- VERIFY(languages.has_value());
- auto const& main_object = languages->as_object().get("main"sv);
- auto const& locale_object = main_object.as_object().get(languages_path.parent().basename());
- auto const& identity_object = locale_object.as_object().get("identity"sv);
- auto const& language_string = identity_object.as_object().get("language"sv);
- auto const& territory_string = identity_object.as_object().get("territory"sv);
- auto const& variant_string = identity_object.as_object().get("variant"sv);
- locale.language = language_string.as_string();
- if (!locale_data.languages.contains_slow(locale.language))
- locale_data.languages.append(locale.language);
- if (territory_string.is_string()) {
- locale.territory = territory_string.as_string();
- if (!locale_data.territories.contains_slow(*locale.territory))
- locale_data.territories.append(*locale.territory);
- }
- if (variant_string.is_string()) {
- locale.variant = variant_string.as_string();
- if (!locale_data.variants.contains_slow(*locale.variant))
- locale_data.variants.append(*locale.variant);
- }
- }
- static void parse_locale_languages(String locale_path, Locale& locale)
- {
- LexicalPath languages_path(move(locale_path));
- languages_path = languages_path.append("languages.json"sv);
- VERIFY(Core::File::exists(languages_path.string()));
- auto languages_file_or_error = Core::File::open(languages_path.string(), Core::OpenMode::ReadOnly);
- VERIFY(!languages_file_or_error.is_error());
- auto languages = JsonParser(languages_file_or_error.value()->read_all()).parse();
- VERIFY(languages.has_value());
- auto const& main_object = languages->as_object().get("main"sv);
- auto const& locale_object = main_object.as_object().get(languages_path.parent().basename());
- auto const& locale_display_names_object = locale_object.as_object().get("localeDisplayNames"sv);
- auto const& languages_object = locale_display_names_object.as_object().get("languages"sv);
- languages_object.as_object().for_each_member([&](auto const& key, JsonValue const& value) {
- locale.languages.set(key, value.as_string());
- });
- }
- static void parse_locale_territories(String locale_path, Locale& locale)
- {
- LexicalPath territories_path(move(locale_path));
- territories_path = territories_path.append("territories.json"sv);
- VERIFY(Core::File::exists(territories_path.string()));
- auto territories_file_or_error = Core::File::open(territories_path.string(), Core::OpenMode::ReadOnly);
- VERIFY(!territories_file_or_error.is_error());
- auto territories = JsonParser(territories_file_or_error.value()->read_all()).parse();
- VERIFY(territories.has_value());
- auto const& main_object = territories->as_object().get("main"sv);
- auto const& locale_object = main_object.as_object().get(territories_path.parent().basename());
- auto const& locale_display_names_object = locale_object.as_object().get("localeDisplayNames"sv);
- auto const& territories_object = locale_display_names_object.as_object().get("territories"sv);
- territories_object.as_object().for_each_member([&](auto const& key, JsonValue const& value) {
- locale.territories.set(key, value.as_string());
- });
- }
- static void parse_locale_scripts(String locale_path, UnicodeLocaleData& locale_data, Locale& locale)
- {
- LexicalPath scripts_path(move(locale_path));
- scripts_path = scripts_path.append("scripts.json"sv);
- VERIFY(Core::File::exists(scripts_path.string()));
- auto scripts_file_or_error = Core::File::open(scripts_path.string(), Core::OpenMode::ReadOnly);
- VERIFY(!scripts_file_or_error.is_error());
- auto scripts = JsonParser(scripts_file_or_error.value()->read_all()).parse();
- VERIFY(scripts.has_value());
- auto const& main_object = scripts->as_object().get("main"sv);
- auto const& locale_object = main_object.as_object().get(scripts_path.parent().basename());
- auto const& locale_display_names_object = locale_object.as_object().get("localeDisplayNames"sv);
- auto const& scripts_object = locale_display_names_object.as_object().get("scripts"sv);
- scripts_object.as_object().for_each_member([&](auto const& key, JsonValue const& value) {
- locale.scripts.set(key, value.as_string());
- if (!locale_data.scripts.contains_slow(key))
- locale_data.scripts.append(key);
- });
- }
- static void parse_locale_currencies(String numbers_path, UnicodeLocaleData& locale_data, Locale& locale)
- {
- LexicalPath currencies_path(move(numbers_path));
- currencies_path = currencies_path.append("currencies.json"sv);
- VERIFY(Core::File::exists(currencies_path.string()));
- auto currencies_file_or_error = Core::File::open(currencies_path.string(), Core::OpenMode::ReadOnly);
- VERIFY(!currencies_file_or_error.is_error());
- auto currencies = JsonParser(currencies_file_or_error.value()->read_all()).parse();
- VERIFY(currencies.has_value());
- auto const& main_object = currencies->as_object().get("main"sv);
- auto const& locale_object = main_object.as_object().get(currencies_path.parent().basename());
- auto const& locale_numbers_object = locale_object.as_object().get("numbers"sv);
- auto const& currencies_object = locale_numbers_object.as_object().get("currencies"sv);
- currencies_object.as_object().for_each_member([&](auto const& key, JsonValue const& value) {
- auto const& display_name = value.as_object().get("displayName"sv);
- locale.currencies.set(key, display_name.as_string());
- if (!locale_data.currencies.contains_slow(key))
- locale_data.currencies.append(key);
- });
- }
- static Core::DirIterator path_to_dir_iterator(String path)
- {
- LexicalPath lexical_path(move(path));
- lexical_path = lexical_path.append("main"sv);
- VERIFY(Core::File::is_directory(lexical_path.string()));
- Core::DirIterator iterator(lexical_path.string(), Core::DirIterator::SkipParentAndBaseDir);
- if (iterator.has_error()) {
- warnln("{}: {}", lexical_path.string(), iterator.error_string());
- VERIFY_NOT_REACHED();
- }
- return iterator;
- }
- static void parse_all_locales(String core_path, String locale_names_path, String numbers_path, UnicodeLocaleData& locale_data)
- {
- auto locale_names_iterator = path_to_dir_iterator(move(locale_names_path));
- auto numbers_iterator = path_to_dir_iterator(move(numbers_path));
- LexicalPath core_supplemental_path(move(core_path));
- core_supplemental_path = core_supplemental_path.append("supplemental"sv);
- VERIFY(Core::File::is_directory(core_supplemental_path.string()));
- parse_core_aliases(core_supplemental_path.string(), locale_data);
- parse_likely_subtags(core_supplemental_path.string(), locale_data);
- while (locale_names_iterator.has_next()) {
- auto locale_path = locale_names_iterator.next_full_path();
- VERIFY(Core::File::is_directory(locale_path));
- auto& locale = locale_data.locales.ensure(LexicalPath::basename(locale_path));
- parse_identity(locale_path, locale_data, locale);
- parse_locale_languages(locale_path, locale);
- parse_locale_territories(locale_path, locale);
- parse_locale_scripts(locale_path, locale_data, locale);
- }
- while (numbers_iterator.has_next()) {
- auto numbers_path = numbers_iterator.next_full_path();
- VERIFY(Core::File::is_directory(numbers_path));
- auto& locale = locale_data.locales.ensure(LexicalPath::basename(numbers_path));
- parse_locale_currencies(numbers_path, locale_data, locale);
- }
- }
- static String format_identifier(StringView owner, String identifier)
- {
- identifier.replace("-"sv, "_"sv, true);
- if (all_of(identifier, is_ascii_digit))
- return String::formatted("{}_{}", owner[0], identifier);
- return identifier.to_titlecase();
- }
- static void generate_unicode_locale_header(Core::File& file, UnicodeLocaleData& locale_data)
- {
- StringBuilder builder;
- SourceGenerator generator { builder };
- auto generate_enum = [&](StringView name, StringView default_, Vector<String>& values) {
- quick_sort(values);
- generator.set("name", name);
- generator.set("underlying", ((values.size() + !default_.is_empty()) < 256) ? "u8"sv : "u16"sv);
- generator.append(R"~~~(
- enum class @name@ : @underlying@ {)~~~");
- if (!default_.is_empty()) {
- generator.set("default", default_);
- generator.append(R"~~~(
- @default@,)~~~");
- }
- for (auto const& value : values) {
- generator.set("value", format_identifier(name, value));
- generator.append(R"~~~(
- @value@,)~~~");
- }
- generator.append(R"~~~(
- };
- )~~~");
- };
- generator.append(R"~~~(
- #pragma once
- #include <AK/Optional.h>
- #include <AK/StringView.h>
- #include <AK/Types.h>
- #include <LibUnicode/Forward.h>
- namespace Unicode {
- )~~~");
- auto locales = locale_data.locales.keys();
- generate_enum("Locale"sv, "None"sv, locales);
- generate_enum("Language"sv, {}, locale_data.languages);
- generate_enum("Territory"sv, {}, locale_data.territories);
- generate_enum("ScriptTag"sv, {}, locale_data.scripts);
- generate_enum("Currency"sv, {}, locale_data.currencies);
- generate_enum("Variant"sv, {}, locale_data.variants);
- generator.append(R"~~~(
- namespace Detail {
- Optional<Locale> locale_from_string(StringView const& locale);
- Optional<StringView> get_locale_language_mapping(StringView locale, StringView language);
- Optional<Language> language_from_string(StringView const& language);
- Optional<StringView> resolve_language_alias(StringView const& language);
- Optional<StringView> get_locale_territory_mapping(StringView locale, StringView territory);
- Optional<Territory> territory_from_string(StringView const& territory);
- Optional<StringView> resolve_territory_alias(StringView const& territory);
- Optional<StringView> get_locale_script_tag_mapping(StringView locale, StringView script_tag);
- Optional<ScriptTag> script_tag_from_string(StringView const& script_tag);
- Optional<StringView> resolve_script_tag_alias(StringView const& script_tag);
- Optional<StringView> get_locale_currency_mapping(StringView locale, StringView currency);
- Optional<Currency> currency_from_string(StringView const& currency);
- Optional<StringView> resolve_variant_alias(StringView const& variant);
- Optional<StringView> resolve_subdivision_alias(StringView const& subdivision);
- void resolve_complex_language_aliases(Unicode::LanguageID& language_id);
- Optional<String> resolve_most_likely_territory(Unicode::LanguageID const& language_id);
- }
- }
- )~~~");
- write_to_file_if_different(file, generator.as_string_view());
- }
- static void generate_unicode_locale_implementation(Core::File& file, UnicodeLocaleData& locale_data)
- {
- StringBuilder builder;
- SourceGenerator generator { builder };
- generator.set("locales_size"sv, String::number(locale_data.locales.size()));
- generator.set("territories_size", String::number(locale_data.territories.size()));
- generator.append(R"~~~(
- #include <AK/Array.h>
- #include <AK/HashMap.h>
- #include <AK/Span.h>
- #include <LibUnicode/Locale.h>
- #include <LibUnicode/UnicodeLocale.h>
- namespace Unicode {
- )~~~");
- auto format_mapping_name = [](StringView format, StringView name) {
- auto mapping_name = name.to_lowercase_string();
- mapping_name.replace("-"sv, "_"sv, true);
- return String::formatted(format, mapping_name);
- };
- auto append_mapping_list = [&](String name, auto const& keys, auto const& mappings) {
- generator.set("name", name);
- generator.set("size", String::number(keys.size()));
- generator.append(R"~~~(
- static constexpr Array<StringView, @size@> @name@ { {
- )~~~");
- constexpr size_t max_values_per_row = 10;
- size_t values_in_current_row = 0;
- for (auto const& key : keys) {
- if (values_in_current_row++ > 0)
- generator.append(" ");
- if (auto it = mappings.find(key); it != mappings.end())
- generator.set("mapping"sv, String::formatted("\"{}\"sv", it->value));
- else
- generator.set("mapping"sv, "{}"sv);
- generator.append("@mapping@,");
- if (values_in_current_row == max_values_per_row) {
- values_in_current_row = 0;
- generator.append("\n ");
- }
- }
- generator.append(R"~~~(
- } };
- )~~~");
- };
- auto append_mapping = [&](StringView name, StringView format, auto const& keys, auto get_mapping_callback) {
- Vector<String> mapping_names;
- for (auto const& locale : locale_data.locales) {
- auto mapping_name = format_mapping_name(format, locale.key);
- append_mapping_list(mapping_name, keys, get_mapping_callback(locale.value));
- mapping_names.append(move(mapping_name));
- }
- quick_sort(mapping_names);
- generator.set("name", name);
- generator.set("size", String::number(locale_data.locales.size()));
- generator.append(R"~~~(
- static constexpr Array<Span<StringView const>, @size@> @name@ { {
- )~~~");
- constexpr size_t max_values_per_row = 10;
- size_t values_in_current_row = 0;
- for (auto& mapping_name : mapping_names) {
- if (values_in_current_row++ > 0)
- generator.append(" ");
- generator.set("name", move(mapping_name));
- generator.append("@name@.span(),");
- if (values_in_current_row == max_values_per_row) {
- values_in_current_row = 0;
- generator.append("\n ");
- }
- }
- generator.append(R"~~~(
- } };
- )~~~");
- };
- append_mapping("s_languages"sv, "s_languages_{}", locale_data.languages, [](auto const& value) { return value.languages; });
- append_mapping("s_territories"sv, "s_territories_{}", locale_data.territories, [](auto const& value) { return value.territories; });
- append_mapping("s_scripts"sv, "s_scripts_{}", locale_data.scripts, [](auto const& value) { return value.scripts; });
- append_mapping("s_currencies"sv, "s_currencies_{}", locale_data.currencies, [](auto const& value) { return value.currencies; });
- generator.append(R"~~~(
- struct LanguageMapping {
- Unicode::LanguageID key;
- Unicode::LanguageID alias;
- };
- )~~~");
- auto append_complex_mapping = [&](StringView name, auto const& mappings) {
- generator.set("name"sv, name);
- generator.append(R"~~~(
- static auto const& ensure_@name@_map()
- {
- static Vector<LanguageMapping> @name@_map;
- auto append_mapping = [&](StringView key, StringView alias) {
- if (auto key_value = Unicode::parse_unicode_language_id(key); key_value.has_value()) {
- if (auto alias_value = Unicode::parse_unicode_language_id(alias); alias_value.has_value())
- @name@_map.append({ key_value.release_value(), alias_value.release_value() });
- }
- };
- )~~~");
- auto keys = mappings.keys();
- quick_sort(keys, [](auto const& lhs, auto const& rhs) {
- // Sort the keys such that "und" language tags are at the end, as those are less specific.
- if (lhs.starts_with("und"sv) && !rhs.starts_with("und"sv))
- return false;
- if (!lhs.starts_with("und"sv) && rhs.starts_with("und"sv))
- return true;
- return lhs < rhs;
- });
- for (auto const& key : keys) {
- generator.set("key"sv, key);
- generator.set("alias"sv, mappings.get(key).value());
- generator.append(R"~~~(
- append_mapping("@key@"sv, "@alias@"sv);)~~~");
- }
- generator.append(R"~~~(
- return @name@_map;
- }
- )~~~");
- };
- append_complex_mapping("complex_alias"sv, locale_data.complex_mappings);
- append_complex_mapping("likely_subtags"sv, locale_data.likely_subtags);
- generator.append(R"~~~(
- static Unicode::LanguageID const* resolve_likely_subtag(Unicode::LanguageID const& language_id)
- {
- // https://unicode.org/reports/tr35/#Likely_Subtags
- static auto const& likely_subtags_map = ensure_likely_subtags_map();
- enum class State {
- LanguageScriptRegion,
- LanguageRegion,
- LanguageScript,
- Language,
- UndScript,
- Done,
- };
- auto state = State::LanguageScriptRegion;
- while (state != State::Done) {
- Unicode::LanguageID search_key;
- switch (state) {
- case State::LanguageScriptRegion:
- state = State::LanguageRegion;
- if (!language_id.script.has_value() || !language_id.region.has_value())
- continue;
- search_key.language = language_id.language;
- search_key.script = language_id.script;
- search_key.region = language_id.region;
- break;
- case State::LanguageRegion:
- state = State::LanguageScript;
- if (!language_id.region.has_value())
- continue;
- search_key.language = language_id.language;
- search_key.region = language_id.region;
- break;
- case State::LanguageScript:
- state = State::Language;
- if (!language_id.script.has_value())
- continue;
- search_key.language = language_id.language;
- search_key.script = language_id.script;
- break;
- case State::Language:
- state = State::UndScript;
- search_key.language = language_id.language;
- break;
- case State::UndScript:
- state = State::Done;
- if (!language_id.script.has_value())
- continue;
- search_key.language = "und"sv;
- search_key.script = language_id.script;
- break;
- default:
- VERIFY_NOT_REACHED();
- }
- for (auto const& map : likely_subtags_map) {
- if (map.key.language != search_key.language)
- continue;
- if (map.key.script != search_key.script)
- continue;
- if (map.key.region != search_key.region)
- continue;
- return &map.alias;
- }
- }
- return nullptr;
- }
- namespace Detail {
- )~~~");
- auto append_mapping_search = [&](StringView enum_title, StringView enum_snake, StringView collection_name) {
- generator.set("enum_title", enum_title);
- generator.set("enum_snake", enum_snake);
- generator.set("collection_name", collection_name);
- generator.append(R"~~~(
- Optional<StringView> get_locale_@enum_snake@_mapping(StringView locale, StringView @enum_snake@)
- {
- auto locale_value = locale_from_string(locale);
- if (!locale_value.has_value())
- return {};
- auto @enum_snake@_value = @enum_snake@_from_string(@enum_snake@);
- if (!@enum_snake@_value.has_value())
- return {};
- auto locale_index = to_underlying(*locale_value) - 1; // Subtract 1 because 0 == Locale::None.
- auto @enum_snake@_index = to_underlying(*@enum_snake@_value);
- auto const& mappings = @collection_name@.at(locale_index);
- auto @enum_snake@_mapping = mappings.at(@enum_snake@_index);
- if (@enum_snake@_mapping.is_empty())
- return {};
- return @enum_snake@_mapping;
- }
- )~~~");
- };
- auto append_from_string = [&](StringView enum_title, StringView enum_snake, Vector<String> const& values) {
- generator.set("enum_title", enum_title);
- generator.set("enum_snake", enum_snake);
- generator.append(R"~~~(
- Optional<@enum_title@> @enum_snake@_from_string(StringView const& @enum_snake@)
- {
- static HashMap<StringView, @enum_title@> @enum_snake@_values { {)~~~");
- for (auto const& value : values) {
- generator.set("key"sv, value);
- generator.set("value"sv, format_identifier(enum_title, value));
- generator.append(R"~~~(
- { "@key@"sv, @enum_title@::@value@ },)~~~");
- }
- generator.append(R"~~~(
- } };
- if (auto value = @enum_snake@_values.get(@enum_snake@); value.has_value())
- return value.value();
- return {};
- }
- )~~~");
- };
- auto append_alias_search = [&](StringView enum_snake, HashMap<String, String> const& aliases) {
- generator.set("enum_snake", enum_snake);
- generator.append(R"~~~(
- Optional<StringView> resolve_@enum_snake@_alias(StringView const& @enum_snake@)
- {
- static HashMap<StringView, StringView> @enum_snake@_aliases { {
- )~~~");
- constexpr size_t max_values_per_row = 10;
- size_t values_in_current_row = 0;
- for (auto const& alias : aliases) {
- if (values_in_current_row++ > 0)
- generator.append(" ");
- generator.set("key"sv, alias.key);
- generator.set("alias"sv, alias.value);
- generator.append("{ \"@key@\"sv, \"@alias@\"sv },");
- if (values_in_current_row == max_values_per_row) {
- generator.append("\n ");
- values_in_current_row = 0;
- }
- }
- generator.append(R"~~~(
- } };
- if (auto alias = @enum_snake@_aliases.get(@enum_snake@); alias.has_value())
- return alias.value();
- return {};
- }
- )~~~");
- };
- append_from_string("Locale"sv, "locale"sv, locale_data.locales.keys());
- append_mapping_search("Language"sv, "language"sv, "s_languages"sv);
- append_from_string("Language"sv, "language"sv, locale_data.languages);
- append_alias_search("language"sv, locale_data.language_aliases);
- append_mapping_search("Territory"sv, "territory"sv, "s_territories"sv);
- append_from_string("Territory"sv, "territory"sv, locale_data.territories);
- append_alias_search("territory"sv, locale_data.territory_aliases);
- append_mapping_search("ScriptTag"sv, "script_tag"sv, "s_scripts"sv);
- append_from_string("ScriptTag"sv, "script_tag"sv, locale_data.scripts);
- append_alias_search("script_tag"sv, locale_data.script_aliases);
- append_mapping_search("Currency"sv, "currency"sv, "s_currencies"sv);
- append_from_string("Currency"sv, "currency"sv, locale_data.currencies);
- append_alias_search("variant"sv, locale_data.variant_aliases);
- append_alias_search("subdivision"sv, locale_data.subdivision_aliases);
- generator.append(R"~~~(
- void resolve_complex_language_aliases(Unicode::LanguageID& language_id)
- {
- static auto const& complex_alias_map = ensure_complex_alias_map();
- for (auto const& map : complex_alias_map) {
- if ((map.key.language != language_id.language) && (map.key.language != "und"sv))
- continue;
- if (map.key.script.has_value() && (map.key.script != language_id.script))
- continue;
- if (map.key.region.has_value() && (map.key.region != language_id.region))
- continue;
- if (!map.key.variants.is_empty() && (map.key.variants != language_id.variants))
- continue;
- auto alias = map.alias;
- if (alias.language == "und"sv)
- alias.language = move(language_id.language);
- if (!map.key.script.has_value() && !alias.script.has_value())
- alias.script = move(language_id.script);
- if (!map.key.region.has_value() && !alias.region.has_value())
- alias.region = move(language_id.region);
- if (map.key.variants.is_empty() && alias.variants.is_empty())
- alias.variants = move(language_id.variants);
- language_id = move(alias);
- break;
- }
- }
- Optional<String> resolve_most_likely_territory(Unicode::LanguageID const& language_id)
- {
- if (auto const* likely_subtag = resolve_likely_subtag(language_id); likely_subtag != nullptr)
- return likely_subtag->region;
- return {};
- }
- }
- }
- )~~~");
- write_to_file_if_different(file, generator.as_string_view());
- }
- int main(int argc, char** argv)
- {
- char const* generated_header_path = nullptr;
- char const* generated_implementation_path = nullptr;
- char const* core_path = nullptr;
- char const* locale_names_path = nullptr;
- char const* numbers_path = nullptr;
- Core::ArgsParser args_parser;
- args_parser.add_option(generated_header_path, "Path to the Unicode locale header file to generate", "generated-header-path", 'h', "generated-header-path");
- args_parser.add_option(generated_implementation_path, "Path to the Unicode locale implementation file to generate", "generated-implementation-path", 'c', "generated-implementation-path");
- args_parser.add_option(core_path, "Path to cldr-core directory", "core-path", 'r', "core-path");
- args_parser.add_option(locale_names_path, "Path to cldr-localenames directory", "locale-names-path", 'l', "locale-names-path");
- args_parser.add_option(numbers_path, "Path to cldr-numbers directory", "numbers-path", 'n', "numbers-path");
- args_parser.parse(argc, argv);
- auto open_file = [&](StringView path, StringView flags, Core::OpenMode mode = Core::OpenMode::ReadOnly) {
- if (path.is_empty()) {
- warnln("{} is required", flags);
- args_parser.print_usage(stderr, argv[0]);
- exit(1);
- }
- auto file_or_error = Core::File::open(path, mode);
- if (file_or_error.is_error()) {
- warnln("Failed to open {}: {}", path, file_or_error.release_error());
- exit(1);
- }
- return file_or_error.release_value();
- };
- auto generated_header_file = open_file(generated_header_path, "-h/--generated-header-path", Core::OpenMode::ReadWrite);
- auto generated_implementation_file = open_file(generated_implementation_path, "-c/--generated-implementation-path", Core::OpenMode::ReadWrite);
- UnicodeLocaleData locale_data;
- parse_all_locales(core_path, locale_names_path, numbers_path, locale_data);
- generate_unicode_locale_header(generated_header_file, locale_data);
- generate_unicode_locale_implementation(generated_implementation_file, locale_data);
- return 0;
- }
|