GenerateUnicodeLocale.cpp 65 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520
  1. /*
  2. * Copyright (c) 2021-2022, Tim Flynn <trflynn89@pm.me>
  3. *
  4. * SPDX-License-Identifier: BSD-2-Clause
  5. */
  6. #include "GeneratorUtil.h"
  7. #include <AK/AllOf.h>
  8. #include <AK/CharacterTypes.h>
  9. #include <AK/Format.h>
  10. #include <AK/HashMap.h>
  11. #include <AK/JsonObject.h>
  12. #include <AK/JsonParser.h>
  13. #include <AK/JsonValue.h>
  14. #include <AK/LexicalPath.h>
  15. #include <AK/QuickSort.h>
  16. #include <AK/SourceGenerator.h>
  17. #include <AK/String.h>
  18. #include <AK/StringBuilder.h>
  19. #include <LibCore/ArgsParser.h>
  20. #include <LibCore/DirIterator.h>
  21. #include <LibCore/File.h>
  22. using StringIndexType = u32;
  23. constexpr auto s_string_index_type = "u32"sv;
  24. using DisplayPatternIndexType = u8;
  25. constexpr auto s_display_pattern_index_type = "u8"sv;
  26. using LanguageListIndexType = u8;
  27. constexpr auto s_language_list_index_type = "u8"sv;
  28. using TerritoryListIndexType = u8;
  29. constexpr auto s_territory_list_index_type = "u8"sv;
  30. using ScriptListIndexType = u8;
  31. constexpr auto s_script_list_index_type = "u8"sv;
  32. using CurrencyListIndexType = u16;
  33. constexpr auto s_currency_list_index_type = "u16"sv;
  34. using CalendarListIndexType = u8;
  35. constexpr auto s_calendar_list_index_type = "u8"sv;
  36. using DateFieldListIndexType = u8;
  37. constexpr auto s_date_field_list_index_type = "u8"sv;
  38. using KeywordListIndexType = u8;
  39. constexpr auto s_keyword_list_index_type = "u8"sv;
  40. using ListPatternIndexType = u16;
  41. constexpr auto s_list_pattern_index_type = "u16"sv;
  42. using ListPatternListIndexType = u8;
  43. constexpr auto s_list_pattern_list_index_type = "u8"sv;
  44. static String format_identifier(StringView owner, String identifier)
  45. {
  46. identifier = identifier.replace("-"sv, "_"sv, true);
  47. if (all_of(identifier, is_ascii_digit))
  48. return String::formatted("{}_{}", owner[0], identifier);
  49. if (is_ascii_lower_alpha(identifier[0]))
  50. return String::formatted("{:c}{}", to_ascii_uppercase(identifier[0]), identifier.substring_view(1));
  51. return identifier;
  52. }
  53. struct DisplayPattern {
  54. unsigned hash() const
  55. {
  56. return pair_int_hash(locale_pattern, locale_separator);
  57. }
  58. bool operator==(DisplayPattern const& other) const
  59. {
  60. return (locale_pattern == other.locale_pattern)
  61. && (locale_separator == other.locale_separator);
  62. }
  63. StringIndexType locale_pattern { 0 };
  64. StringIndexType locale_separator { 0 };
  65. };
  66. template<>
  67. struct AK::Formatter<DisplayPattern> : Formatter<FormatString> {
  68. ErrorOr<void> format(FormatBuilder& builder, DisplayPattern const& patterns)
  69. {
  70. return Formatter<FormatString>::format(builder,
  71. "{{ {}, {} }}",
  72. patterns.locale_pattern,
  73. patterns.locale_separator);
  74. }
  75. };
  76. template<>
  77. struct AK::Traits<DisplayPattern> : public GenericTraits<DisplayPattern> {
  78. static unsigned hash(DisplayPattern const& p) { return p.hash(); }
  79. };
  80. struct ListPatterns {
  81. unsigned hash() const
  82. {
  83. auto hash = pair_int_hash(type.hash(), style.hash());
  84. hash = pair_int_hash(hash, start);
  85. hash = pair_int_hash(hash, middle);
  86. hash = pair_int_hash(hash, end);
  87. hash = pair_int_hash(hash, pair);
  88. return hash;
  89. }
  90. bool operator==(ListPatterns const& other) const
  91. {
  92. return (type == other.type)
  93. && (style == other.style)
  94. && (start == other.start)
  95. && (middle == other.middle)
  96. && (end == other.end)
  97. && (pair == other.pair);
  98. }
  99. StringView type;
  100. StringView style;
  101. StringIndexType start { 0 };
  102. StringIndexType middle { 0 };
  103. StringIndexType end { 0 };
  104. StringIndexType pair { 0 };
  105. };
  106. template<>
  107. struct AK::Formatter<ListPatterns> : Formatter<FormatString> {
  108. ErrorOr<void> format(FormatBuilder& builder, ListPatterns const& patterns)
  109. {
  110. return Formatter<FormatString>::format(builder,
  111. "{{ ListPatternType::{}, ListPatternStyle::{}, {}, {}, {}, {} }}",
  112. format_identifier({}, patterns.type),
  113. format_identifier({}, patterns.style),
  114. patterns.start,
  115. patterns.middle,
  116. patterns.end,
  117. patterns.pair);
  118. }
  119. };
  120. template<>
  121. struct AK::Traits<ListPatterns> : public GenericTraits<ListPatterns> {
  122. static unsigned hash(ListPatterns const& p) { return p.hash(); }
  123. };
  124. using LanguageList = Vector<StringIndexType>;
  125. using TerritoryList = Vector<StringIndexType>;
  126. using ScriptList = Vector<StringIndexType>;
  127. using CurrencyList = Vector<StringIndexType>;
  128. using CalendarList = Vector<StringIndexType>;
  129. using DateFieldList = Vector<StringIndexType>;
  130. using KeywordList = Vector<StringIndexType>;
  131. using ListPatternList = Vector<ListPatternIndexType>;
  132. struct Locale {
  133. String language;
  134. Optional<String> territory;
  135. Optional<String> variant;
  136. DisplayPatternIndexType display_patterns { 0 };
  137. LanguageListIndexType languages { 0 };
  138. TerritoryListIndexType territories { 0 };
  139. ScriptListIndexType scripts { 0 };
  140. CurrencyListIndexType long_currencies { 0 };
  141. CurrencyListIndexType short_currencies { 0 };
  142. CurrencyListIndexType narrow_currencies { 0 };
  143. CurrencyListIndexType numeric_currencies { 0 };
  144. CalendarListIndexType calendars { 0 };
  145. DateFieldListIndexType long_date_fields { 0 };
  146. DateFieldListIndexType short_date_fields { 0 };
  147. DateFieldListIndexType narrow_date_fields { 0 };
  148. KeywordListIndexType keywords { 0 };
  149. ListPatternListIndexType list_patterns { 0 };
  150. };
  151. struct LanguageMapping {
  152. CanonicalLanguageID<StringIndexType> key {};
  153. CanonicalLanguageID<StringIndexType> alias {};
  154. };
  155. struct UnicodeLocaleData {
  156. UniqueStringStorage<StringIndexType> unique_strings;
  157. UniqueStorage<DisplayPattern, DisplayPatternIndexType> unique_display_patterns;
  158. UniqueStorage<LanguageList, LanguageListIndexType> unique_language_lists;
  159. UniqueStorage<TerritoryList, TerritoryListIndexType> unique_territory_lists;
  160. UniqueStorage<ScriptList, ScriptListIndexType> unique_script_lists;
  161. UniqueStorage<CurrencyList, CurrencyListIndexType> unique_currency_lists;
  162. UniqueStorage<CalendarList, CalendarListIndexType> unique_calendar_lists;
  163. UniqueStorage<DateFieldList, DateFieldListIndexType> unique_date_field_lists;
  164. UniqueStorage<KeywordList, KeywordListIndexType> unique_keyword_lists;
  165. UniqueStorage<ListPatterns, ListPatternIndexType> unique_list_patterns;
  166. UniqueStorage<ListPatternList, ListPatternListIndexType> unique_list_pattern_lists;
  167. HashMap<String, Locale> locales;
  168. Vector<Alias> locale_aliases;
  169. Vector<String> languages;
  170. Vector<String> territories;
  171. Vector<String> scripts;
  172. Vector<String> variants;
  173. Vector<String> currencies;
  174. Vector<String> calendars;
  175. Vector<Alias> calendar_aliases {
  176. // FIXME: Aliases should come from BCP47. See: https://unicode-org.atlassian.net/browse/CLDR-15158
  177. { "ethiopic-amete-alem"sv, "ethioaa"sv },
  178. { "gregorian"sv, "gregory"sv },
  179. };
  180. Vector<String> date_fields;
  181. Vector<Alias> date_field_aliases {
  182. // ECMA-402 and the CLDR refer to some date fields with different names. Defining these aliases
  183. // means we can remain agnostic about the naming differences elsewhere.
  184. { "dayperiod"sv, "dayPeriod"sv },
  185. { "week"sv, "weekOfYear"sv },
  186. { "zone"sv, "timeZoneName"sv },
  187. };
  188. Vector<String> keywords { "ca"sv, "nu"sv }; // FIXME: These should be parsed from BCP47. https://unicode-org.atlassian.net/browse/CLDR-15158
  189. Vector<String> list_pattern_types;
  190. Vector<String> list_pattern_styles;
  191. HashMap<String, StringIndexType> language_aliases;
  192. HashMap<String, StringIndexType> territory_aliases;
  193. HashMap<String, StringIndexType> script_aliases;
  194. HashMap<String, StringIndexType> variant_aliases;
  195. HashMap<String, StringIndexType> subdivision_aliases;
  196. Vector<LanguageMapping> complex_mappings;
  197. Vector<LanguageMapping> likely_subtags;
  198. size_t max_variant_size { 0 };
  199. };
  200. // Some parsing is expected to fail. For example, the CLDR contains language mappings
  201. // with locales such as "en-GB-oed" that are canonically invalid locale IDs.
  202. #define TRY_OR_DISCARD(expression) \
  203. ({ \
  204. auto _temporary_result = (expression); \
  205. if (_temporary_result.is_error()) \
  206. return; \
  207. _temporary_result.release_value(); \
  208. })
  209. static ErrorOr<LanguageMapping> parse_language_mapping(UnicodeLocaleData& locale_data, StringView key, StringView alias)
  210. {
  211. auto parsed_key = TRY(CanonicalLanguageID<StringIndexType>::parse(locale_data.unique_strings, key));
  212. auto parsed_alias = TRY(CanonicalLanguageID<StringIndexType>::parse(locale_data.unique_strings, alias));
  213. return LanguageMapping { move(parsed_key), move(parsed_alias) };
  214. }
  215. static ErrorOr<void> parse_core_aliases(String core_supplemental_path, UnicodeLocaleData& locale_data)
  216. {
  217. LexicalPath core_aliases_path(move(core_supplemental_path));
  218. core_aliases_path = core_aliases_path.append("aliases.json"sv);
  219. auto core_aliases_file = TRY(Core::File::open(core_aliases_path.string(), Core::OpenMode::ReadOnly));
  220. auto core_aliases = TRY(JsonValue::from_string(core_aliases_file->read_all()));
  221. auto const& supplemental_object = core_aliases.as_object().get("supplemental"sv);
  222. auto const& metadata_object = supplemental_object.as_object().get("metadata"sv);
  223. auto const& alias_object = metadata_object.as_object().get("alias"sv);
  224. auto append_aliases = [&](auto& alias_object, auto& alias_map) {
  225. alias_object.as_object().for_each_member([&](auto const& key, JsonValue const& value) {
  226. auto alias = value.as_object().get("_replacement"sv).as_string();
  227. if (key.contains('-')) {
  228. auto mapping = TRY_OR_DISCARD(parse_language_mapping(locale_data, key, alias));
  229. locale_data.max_variant_size = max(mapping.key.variants.size(), locale_data.max_variant_size);
  230. locale_data.max_variant_size = max(mapping.alias.variants.size(), locale_data.max_variant_size);
  231. locale_data.complex_mappings.append(move(mapping));
  232. } else {
  233. alias_map.set(key, locale_data.unique_strings.ensure(alias));
  234. }
  235. });
  236. };
  237. append_aliases(alias_object.as_object().get("languageAlias"sv), locale_data.language_aliases);
  238. append_aliases(alias_object.as_object().get("territoryAlias"sv), locale_data.territory_aliases);
  239. append_aliases(alias_object.as_object().get("scriptAlias"sv), locale_data.script_aliases);
  240. append_aliases(alias_object.as_object().get("variantAlias"sv), locale_data.variant_aliases);
  241. append_aliases(alias_object.as_object().get("subdivisionAlias"sv), locale_data.subdivision_aliases);
  242. return {};
  243. }
  244. static ErrorOr<void> parse_likely_subtags(String core_supplemental_path, UnicodeLocaleData& locale_data)
  245. {
  246. LexicalPath likely_subtags_path(move(core_supplemental_path));
  247. likely_subtags_path = likely_subtags_path.append("likelySubtags.json"sv);
  248. auto likely_subtags_file = TRY(Core::File::open(likely_subtags_path.string(), Core::OpenMode::ReadOnly));
  249. auto likely_subtags = TRY(JsonValue::from_string(likely_subtags_file->read_all()));
  250. auto const& supplemental_object = likely_subtags.as_object().get("supplemental"sv);
  251. auto const& likely_subtags_object = supplemental_object.as_object().get("likelySubtags"sv);
  252. likely_subtags_object.as_object().for_each_member([&](auto const& key, JsonValue const& value) {
  253. auto mapping = TRY_OR_DISCARD(parse_language_mapping(locale_data, key, value.as_string()));
  254. locale_data.max_variant_size = max(mapping.key.variants.size(), locale_data.max_variant_size);
  255. locale_data.max_variant_size = max(mapping.alias.variants.size(), locale_data.max_variant_size);
  256. locale_data.likely_subtags.append(move(mapping));
  257. });
  258. return {};
  259. }
  260. static ErrorOr<void> parse_identity(String locale_path, UnicodeLocaleData& locale_data, Locale& locale)
  261. {
  262. LexicalPath languages_path(move(locale_path)); // Note: Every JSON file defines identity data, so we can use any of them.
  263. languages_path = languages_path.append("languages.json"sv);
  264. auto languages_file = TRY(Core::File::open(languages_path.string(), Core::OpenMode::ReadOnly));
  265. auto languages = TRY(JsonValue::from_string(languages_file->read_all()));
  266. auto const& main_object = languages.as_object().get("main"sv);
  267. auto const& locale_object = main_object.as_object().get(languages_path.parent().basename());
  268. auto const& identity_object = locale_object.as_object().get("identity"sv);
  269. auto const& language_string = identity_object.as_object().get("language"sv);
  270. auto const& territory_string = identity_object.as_object().get("territory"sv);
  271. auto const& script_string = identity_object.as_object().get("script"sv);
  272. auto const& variant_string = identity_object.as_object().get("variant"sv);
  273. locale.language = language_string.as_string();
  274. if (territory_string.is_string()) {
  275. locale.territory = territory_string.as_string();
  276. if (!locale_data.territories.contains_slow(*locale.territory))
  277. locale_data.territories.append(*locale.territory);
  278. }
  279. if (script_string.is_string()) {
  280. auto script = script_string.as_string();
  281. if (!locale_data.scripts.contains_slow(script))
  282. locale_data.scripts.append(script);
  283. }
  284. if (variant_string.is_string()) {
  285. locale.variant = variant_string.as_string();
  286. if (!locale_data.variants.contains_slow(*locale.variant))
  287. locale_data.variants.append(*locale.variant);
  288. }
  289. return {};
  290. }
  291. static ErrorOr<void> parse_locale_display_patterns(String locale_path, UnicodeLocaleData& locale_data, Locale& locale)
  292. {
  293. LexicalPath locale_display_names_path(move(locale_path));
  294. locale_display_names_path = locale_display_names_path.append("localeDisplayNames.json"sv);
  295. auto locale_display_names_file = TRY(Core::File::open(locale_display_names_path.string(), Core::OpenMode::ReadOnly));
  296. auto locale_display_names = TRY(JsonValue::from_string(locale_display_names_file->read_all()));
  297. auto const& main_object = locale_display_names.as_object().get("main"sv);
  298. auto const& locale_object = main_object.as_object().get(locale_display_names_path.parent().basename());
  299. auto const& locale_display_names_object = locale_object.as_object().get("localeDisplayNames"sv);
  300. auto const& locale_display_patterns_object = locale_display_names_object.as_object().get("localeDisplayPattern"sv);
  301. auto const& locale_pattern = locale_display_patterns_object.as_object().get("localePattern"sv);
  302. auto const& locale_separator = locale_display_patterns_object.as_object().get("localeSeparator"sv);
  303. DisplayPattern patterns {};
  304. patterns.locale_pattern = locale_data.unique_strings.ensure(locale_pattern.as_string());
  305. patterns.locale_separator = locale_data.unique_strings.ensure(locale_separator.as_string());
  306. locale.display_patterns = locale_data.unique_display_patterns.ensure(move(patterns));
  307. return {};
  308. }
  309. static ErrorOr<void> preprocess_languages(String locale_path, UnicodeLocaleData& locale_data)
  310. {
  311. LexicalPath languages_path(move(locale_path));
  312. languages_path = languages_path.append("languages.json"sv);
  313. auto languages_file = TRY(Core::File::open(languages_path.string(), Core::OpenMode::ReadOnly));
  314. auto locale_languages = TRY(JsonValue::from_string(languages_file->read_all()));
  315. auto const& main_object = locale_languages.as_object().get("main"sv);
  316. auto const& locale_object = main_object.as_object().get(languages_path.parent().basename());
  317. auto const& locale_display_names_object = locale_object.as_object().get("localeDisplayNames"sv);
  318. auto const& languages_object = locale_display_names_object.as_object().get("languages"sv);
  319. languages_object.as_object().for_each_member([&](auto const& key, auto const&) {
  320. if (!key.contains("-alt-"sv) && !locale_data.languages.contains_slow(key))
  321. locale_data.languages.append(key);
  322. });
  323. return {};
  324. }
  325. static ErrorOr<void> parse_locale_languages(String locale_path, UnicodeLocaleData& locale_data, Locale& locale)
  326. {
  327. LexicalPath languages_path(move(locale_path));
  328. languages_path = languages_path.append("languages.json"sv);
  329. auto languages_file = TRY(Core::File::open(languages_path.string(), Core::OpenMode::ReadOnly));
  330. auto locale_languages = TRY(JsonValue::from_string(languages_file->read_all()));
  331. auto const& main_object = locale_languages.as_object().get("main"sv);
  332. auto const& locale_object = main_object.as_object().get(languages_path.parent().basename());
  333. auto const& locale_display_names_object = locale_object.as_object().get("localeDisplayNames"sv);
  334. auto const& languages_object = locale_display_names_object.as_object().get("languages"sv);
  335. LanguageList languages;
  336. languages.resize(locale_data.languages.size());
  337. languages_object.as_object().for_each_member([&](auto const& key, JsonValue const& value) {
  338. if (key.contains("-alt-"sv))
  339. return;
  340. auto index = locale_data.languages.find_first_index(key).value();
  341. languages[index] = locale_data.unique_strings.ensure(value.as_string());
  342. });
  343. locale.languages = locale_data.unique_language_lists.ensure(move(languages));
  344. return {};
  345. }
  346. static ErrorOr<void> parse_locale_territories(String locale_path, UnicodeLocaleData& locale_data, Locale& locale)
  347. {
  348. LexicalPath territories_path(move(locale_path));
  349. territories_path = territories_path.append("territories.json"sv);
  350. auto territories_file = TRY(Core::File::open(territories_path.string(), Core::OpenMode::ReadOnly));
  351. auto locale_territories = TRY(JsonValue::from_string(territories_file->read_all()));
  352. auto const& main_object = locale_territories.as_object().get("main"sv);
  353. auto const& locale_object = main_object.as_object().get(territories_path.parent().basename());
  354. auto const& locale_display_names_object = locale_object.as_object().get("localeDisplayNames"sv);
  355. auto const& territories_object = locale_display_names_object.as_object().get("territories"sv);
  356. TerritoryList territories;
  357. territories.resize(locale_data.territories.size());
  358. territories_object.as_object().for_each_member([&](auto const& key, JsonValue const& value) {
  359. if (auto index = locale_data.territories.find_first_index(key); index.has_value())
  360. territories[*index] = locale_data.unique_strings.ensure(value.as_string());
  361. });
  362. locale.territories = locale_data.unique_territory_lists.ensure(move(territories));
  363. return {};
  364. }
  365. static ErrorOr<void> parse_locale_scripts(String locale_path, UnicodeLocaleData& locale_data, Locale& locale)
  366. {
  367. LexicalPath scripts_path(move(locale_path));
  368. scripts_path = scripts_path.append("scripts.json"sv);
  369. auto scripts_file = TRY(Core::File::open(scripts_path.string(), Core::OpenMode::ReadOnly));
  370. auto locale_scripts = TRY(JsonValue::from_string(scripts_file->read_all()));
  371. auto const& main_object = locale_scripts.as_object().get("main"sv);
  372. auto const& locale_object = main_object.as_object().get(scripts_path.parent().basename());
  373. auto const& locale_display_names_object = locale_object.as_object().get("localeDisplayNames"sv);
  374. auto const& scripts_object = locale_display_names_object.as_object().get("scripts"sv);
  375. ScriptList scripts;
  376. scripts.resize(locale_data.scripts.size());
  377. scripts_object.as_object().for_each_member([&](auto const& key, JsonValue const& value) {
  378. if (auto index = locale_data.scripts.find_first_index(key); index.has_value())
  379. scripts[*index] = locale_data.unique_strings.ensure(value.as_string());
  380. });
  381. locale.scripts = locale_data.unique_script_lists.ensure(move(scripts));
  382. return {};
  383. }
  384. static ErrorOr<void> parse_locale_list_patterns(String misc_path, UnicodeLocaleData& locale_data, Locale& locale)
  385. {
  386. LexicalPath list_patterns_path(move(misc_path));
  387. list_patterns_path = list_patterns_path.append("listPatterns.json"sv);
  388. auto list_patterns_file = TRY(Core::File::open(list_patterns_path.string(), Core::OpenMode::ReadOnly));
  389. auto locale_list_patterns = TRY(JsonValue::from_string(list_patterns_file->read_all()));
  390. auto const& main_object = locale_list_patterns.as_object().get("main"sv);
  391. auto const& locale_object = main_object.as_object().get(list_patterns_path.parent().basename());
  392. auto const& list_patterns_object = locale_object.as_object().get("listPatterns"sv);
  393. auto list_pattern_type = [](StringView key) {
  394. if (key.contains("type-standard"sv))
  395. return "conjunction"sv;
  396. if (key.contains("type-or"sv))
  397. return "disjunction"sv;
  398. if (key.contains("type-unit"sv))
  399. return "unit"sv;
  400. VERIFY_NOT_REACHED();
  401. };
  402. auto list_pattern_style = [](StringView key) {
  403. if (key.contains("short"sv))
  404. return "short"sv;
  405. if (key.contains("narrow"sv))
  406. return "narrow"sv;
  407. return "long"sv;
  408. };
  409. ListPatternList list_patterns;
  410. list_patterns.ensure_capacity(list_patterns_object.as_object().size());
  411. list_patterns_object.as_object().for_each_member([&](auto const& key, JsonValue const& value) {
  412. auto type = list_pattern_type(key);
  413. auto style = list_pattern_style(key);
  414. auto start = locale_data.unique_strings.ensure(value.as_object().get("start"sv).as_string());
  415. auto middle = locale_data.unique_strings.ensure(value.as_object().get("middle"sv).as_string());
  416. auto end = locale_data.unique_strings.ensure(value.as_object().get("end"sv).as_string());
  417. auto pair = locale_data.unique_strings.ensure(value.as_object().get("2"sv).as_string());
  418. if (!locale_data.list_pattern_types.contains_slow(type))
  419. locale_data.list_pattern_types.append(type);
  420. if (!locale_data.list_pattern_styles.contains_slow(style))
  421. locale_data.list_pattern_styles.append(style);
  422. ListPatterns list_pattern { type, style, start, middle, end, pair };
  423. list_patterns.append(locale_data.unique_list_patterns.ensure(move(list_pattern)));
  424. });
  425. locale.list_patterns = locale_data.unique_list_pattern_lists.ensure(move(list_patterns));
  426. return {};
  427. }
  428. static ErrorOr<void> parse_locale_currencies(String numbers_path, UnicodeLocaleData& locale_data, Locale& locale)
  429. {
  430. LexicalPath currencies_path(move(numbers_path));
  431. currencies_path = currencies_path.append("currencies.json"sv);
  432. auto currencies_file = TRY(Core::File::open(currencies_path.string(), Core::OpenMode::ReadOnly));
  433. auto locale_currencies = TRY(JsonValue::from_string(currencies_file->read_all()));
  434. auto const& main_object = locale_currencies.as_object().get("main"sv);
  435. auto const& locale_object = main_object.as_object().get(currencies_path.parent().basename());
  436. auto const& locale_numbers_object = locale_object.as_object().get("numbers"sv);
  437. auto const& currencies_object = locale_numbers_object.as_object().get("currencies"sv);
  438. currencies_object.as_object().for_each_member([&](auto const& key, JsonValue const&) {
  439. if (!locale_data.currencies.contains_slow(key))
  440. locale_data.currencies.append(key);
  441. });
  442. CurrencyList long_currencies {};
  443. long_currencies.resize(locale_data.currencies.size());
  444. CurrencyList short_currencies {};
  445. short_currencies.resize(locale_data.currencies.size());
  446. CurrencyList narrow_currencies {};
  447. narrow_currencies.resize(locale_data.currencies.size());
  448. CurrencyList numeric_currencies {};
  449. numeric_currencies.resize(locale_data.currencies.size());
  450. currencies_object.as_object().for_each_member([&](auto const& key, JsonValue const& value) {
  451. auto const& long_name = value.as_object().get("displayName"sv);
  452. auto const& short_name = value.as_object().get("symbol"sv);
  453. auto const& narrow_name = value.as_object().get("symbol-alt-narrow"sv);
  454. auto const& numeric_name = value.as_object().get("displayName-count-other"sv);
  455. auto index = locale_data.currencies.find_first_index(key).value();
  456. long_currencies[index] = locale_data.unique_strings.ensure(long_name.as_string());
  457. short_currencies[index] = locale_data.unique_strings.ensure(short_name.as_string());
  458. narrow_currencies[index] = narrow_name.is_null() ? 0 : locale_data.unique_strings.ensure(narrow_name.as_string());
  459. numeric_currencies[index] = locale_data.unique_strings.ensure(numeric_name.is_null() ? long_name.as_string() : numeric_name.as_string());
  460. });
  461. locale.long_currencies = locale_data.unique_currency_lists.ensure(move(long_currencies));
  462. locale.short_currencies = locale_data.unique_currency_lists.ensure(move(short_currencies));
  463. locale.narrow_currencies = locale_data.unique_currency_lists.ensure(move(narrow_currencies));
  464. locale.numeric_currencies = locale_data.unique_currency_lists.ensure(move(numeric_currencies));
  465. return {};
  466. }
  467. static ErrorOr<void> parse_locale_calendars(String locale_path, UnicodeLocaleData& locale_data, Locale& locale)
  468. {
  469. LexicalPath locale_display_names_path(move(locale_path));
  470. locale_display_names_path = locale_display_names_path.append("localeDisplayNames.json"sv);
  471. auto locale_display_names_file = TRY(Core::File::open(locale_display_names_path.string(), Core::OpenMode::ReadOnly));
  472. auto locale_display_names = TRY(JsonValue::from_string(locale_display_names_file->read_all()));
  473. auto const& main_object = locale_display_names.as_object().get("main"sv);
  474. auto const& locale_object = main_object.as_object().get(locale_display_names_path.parent().basename());
  475. auto const& locale_display_names_object = locale_object.as_object().get("localeDisplayNames"sv);
  476. auto const& types_object = locale_display_names_object.as_object().get("types"sv);
  477. auto const& calendar_object = types_object.as_object().get("calendar"sv);
  478. calendar_object.as_object().for_each_member([&](auto const& key, auto const&) {
  479. if (!locale_data.calendars.contains_slow(key))
  480. locale_data.calendars.append(key);
  481. });
  482. CalendarList calendars;
  483. calendars.resize(locale_data.calendars.size());
  484. calendar_object.as_object().for_each_member([&](auto const& key, auto const& calendar) {
  485. auto index = locale_data.calendars.find_first_index(key).value();
  486. calendars[index] = locale_data.unique_strings.ensure(calendar.as_string());
  487. });
  488. locale.calendars = locale_data.unique_calendar_lists.ensure(move(calendars));
  489. return {};
  490. }
  491. static ErrorOr<void> parse_locale_date_fields(String dates_path, UnicodeLocaleData& locale_data, Locale& locale)
  492. {
  493. LexicalPath date_fields_path(move(dates_path));
  494. date_fields_path = date_fields_path.append("dateFields.json"sv);
  495. auto date_fields_file = TRY(Core::File::open(date_fields_path.string(), Core::OpenMode::ReadOnly));
  496. auto locale_date_fields = TRY(JsonValue::from_string(date_fields_file->read_all()));
  497. auto const& main_object = locale_date_fields.as_object().get("main"sv);
  498. auto const& locale_object = main_object.as_object().get(date_fields_path.parent().basename());
  499. auto const& dates_object = locale_object.as_object().get("dates"sv);
  500. auto const& fields_object = dates_object.as_object().get("fields"sv);
  501. auto is_sanctioned_field = [](StringView field) {
  502. // This is a copy of the units sanctioned for use within ECMA-402, with names adjusted for the names used by the CLDR.
  503. // https://tc39.es/ecma402/#table-validcodeforDateField
  504. return field.is_one_of("era"sv, "year"sv, "quarter"sv, "month"sv, "week"sv, "weekday"sv, "day"sv, "dayperiod"sv, "hour"sv, "minute"sv, "second"sv, "zone"sv);
  505. };
  506. fields_object.as_object().for_each_member([&](auto const& key, JsonValue const&) {
  507. if (!is_sanctioned_field(key))
  508. return;
  509. if (!locale_data.date_fields.contains_slow(key))
  510. locale_data.date_fields.append(key);
  511. });
  512. quick_sort(locale_data.date_fields);
  513. DateFieldList long_date_fields {};
  514. long_date_fields.resize(locale_data.date_fields.size());
  515. DateFieldList short_date_fields {};
  516. short_date_fields.resize(locale_data.date_fields.size());
  517. DateFieldList narrow_date_fields {};
  518. narrow_date_fields.resize(locale_data.date_fields.size());
  519. fields_object.as_object().for_each_member([&](auto const& key, JsonValue const& value) {
  520. if (!is_sanctioned_field(key))
  521. return;
  522. auto const& long_name = value.as_object().get("displayName"sv);
  523. auto const& short_name = fields_object.as_object().get(String::formatted("{}-short", key)).as_object().get("displayName"sv);
  524. auto const& narrow_name = fields_object.as_object().get(String::formatted("{}-narrow", key)).as_object().get("displayName"sv);
  525. auto index = locale_data.date_fields.find_first_index(key).value();
  526. long_date_fields[index] = locale_data.unique_strings.ensure(long_name.as_string());
  527. short_date_fields[index] = locale_data.unique_strings.ensure(short_name.as_string());
  528. narrow_date_fields[index] = locale_data.unique_strings.ensure(narrow_name.as_string());
  529. });
  530. locale.long_date_fields = locale_data.unique_date_field_lists.ensure(move(long_date_fields));
  531. locale.short_date_fields = locale_data.unique_date_field_lists.ensure(move(short_date_fields));
  532. locale.narrow_date_fields = locale_data.unique_date_field_lists.ensure(move(narrow_date_fields));
  533. return {};
  534. }
  535. static ErrorOr<void> parse_numeric_keywords(String locale_numbers_path, UnicodeLocaleData& locale_data, KeywordList& keywords)
  536. {
  537. static constexpr StringView key = "nu"sv;
  538. LexicalPath numbers_path(move(locale_numbers_path));
  539. numbers_path = numbers_path.append("numbers.json"sv);
  540. auto numbers_file = TRY(Core::File::open(numbers_path.string(), Core::OpenMode::ReadOnly));
  541. auto numbers = TRY(JsonValue::from_string(numbers_file->read_all()));
  542. auto const& main_object = numbers.as_object().get("main"sv);
  543. auto const& locale_object = main_object.as_object().get(numbers_path.parent().basename());
  544. auto const& locale_numbers_object = locale_object.as_object().get("numbers"sv);
  545. auto const& default_numbering_system_object = locale_numbers_object.as_object().get("defaultNumberingSystem"sv);
  546. auto const& other_numbering_systems_object = locale_numbers_object.as_object().get("otherNumberingSystems"sv);
  547. Vector<String> keyword_values {};
  548. keyword_values.append(default_numbering_system_object.as_string());
  549. other_numbering_systems_object.as_object().for_each_member([&](auto const&, JsonValue const& value) {
  550. auto keyword_value = value.as_string();
  551. if (!keyword_values.contains_slow(keyword_value))
  552. keyword_values.append(move(keyword_value));
  553. });
  554. locale_numbers_object.as_object().for_each_member([&](auto const& key, JsonValue const& value) {
  555. if (!key.starts_with("defaultNumberingSystem-alt-"sv))
  556. return;
  557. auto keyword_value = value.as_string();
  558. if (!keyword_values.contains_slow(keyword_value))
  559. keyword_values.append(move(keyword_value));
  560. });
  561. StringBuilder builder;
  562. builder.join(',', keyword_values);
  563. auto index = locale_data.keywords.find_first_index(key).value();
  564. keywords[index] = locale_data.unique_strings.ensure(builder.build());
  565. return {};
  566. }
  567. static ErrorOr<void> parse_calendar_keywords(String locale_dates_path, UnicodeLocaleData& locale_data, KeywordList& keywords)
  568. {
  569. static constexpr StringView key = "ca"sv;
  570. auto calendars_iterator = TRY(path_to_dir_iterator(locale_dates_path, {}));
  571. Vector<String> keyword_values {};
  572. while (calendars_iterator.has_next()) {
  573. auto locale_calendars_path = TRY(next_path_from_dir_iterator(calendars_iterator));
  574. LexicalPath calendars_path(move(locale_calendars_path));
  575. if (!calendars_path.basename().starts_with("ca-"sv))
  576. continue;
  577. auto calendars_file = TRY(Core::File::open(calendars_path.string(), Core::OpenMode::ReadOnly));
  578. auto calendars = TRY(JsonValue::from_string(calendars_file->read_all()));
  579. auto const& main_object = calendars.as_object().get("main"sv);
  580. auto const& locale_object = main_object.as_object().get(calendars_path.parent().basename());
  581. auto const& dates_object = locale_object.as_object().get("dates"sv);
  582. auto const& calendars_object = dates_object.as_object().get("calendars"sv);
  583. calendars_object.as_object().for_each_member([&](auto const& calendar_name, JsonValue const&) {
  584. // The generic calendar is not a supported Unicode calendar key, so skip it:
  585. // https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Global_Objects/Intl/Locale/calendar#unicode_calendar_keys
  586. if (calendar_name == "generic"sv)
  587. return;
  588. // FIXME: Similar to the calendar aliases defined in GenerateUnicodeDateTimeFormat, this
  589. // should be parsed from BCP47. https://unicode-org.atlassian.net/browse/CLDR-15158
  590. if (calendar_name == "gregorian"sv)
  591. keyword_values.append("gregory"sv);
  592. else
  593. keyword_values.append(calendar_name);
  594. });
  595. }
  596. StringBuilder builder;
  597. builder.join(',', keyword_values);
  598. auto index = locale_data.keywords.find_first_index(key).value();
  599. keywords[index] = locale_data.unique_strings.ensure(builder.build());
  600. return {};
  601. }
  602. static ErrorOr<void> parse_default_content_locales(String core_path, UnicodeLocaleData& locale_data)
  603. {
  604. LexicalPath default_content_path(move(core_path));
  605. default_content_path = default_content_path.append("defaultContent.json"sv);
  606. auto default_content_file = TRY(Core::File::open(default_content_path.string(), Core::OpenMode::ReadOnly));
  607. auto default_content = TRY(JsonValue::from_string(default_content_file->read_all()));
  608. auto const& default_content_array = default_content.as_object().get("defaultContent"sv);
  609. default_content_array.as_array().for_each([&](JsonValue const& value) {
  610. auto locale = value.as_string();
  611. StringView default_locale = locale;
  612. while (true) {
  613. if (locale_data.locales.contains(default_locale))
  614. break;
  615. auto pos = default_locale.find_last('-');
  616. if (!pos.has_value())
  617. return;
  618. default_locale = default_locale.substring_view(0, *pos);
  619. }
  620. if (default_locale != locale)
  621. locale_data.locale_aliases.append({ default_locale, move(locale) });
  622. });
  623. return {};
  624. }
  625. static ErrorOr<void> define_aliases_without_scripts(UnicodeLocaleData& locale_data)
  626. {
  627. // From ECMA-402: https://tc39.es/ecma402/#sec-internal-slots
  628. //
  629. // For locales that include a script subtag in addition to language and region, the
  630. // corresponding locale without a script subtag must also be supported.
  631. //
  632. // So we define aliases for locales that contain all three subtags, but we must also take
  633. // care to handle when the locale itself or the locale without a script subtag are an alias
  634. // by way of default-content locales.
  635. auto find_alias = [&](auto const& locale) {
  636. return locale_data.locale_aliases.find_if([&](auto const& alias) { return locale == alias.alias; });
  637. };
  638. auto append_alias_without_script = [&](auto const& locale) -> ErrorOr<void> {
  639. auto parsed_locale = TRY(CanonicalLanguageID<StringIndexType>::parse(locale_data.unique_strings, locale));
  640. if ((parsed_locale.language == 0) || (parsed_locale.script == 0) || (parsed_locale.region == 0))
  641. return {};
  642. auto locale_without_script = String::formatted("{}-{}",
  643. locale_data.unique_strings.get(parsed_locale.language),
  644. locale_data.unique_strings.get(parsed_locale.region));
  645. if (locale_data.locales.contains(locale_without_script))
  646. return {};
  647. if (find_alias(locale_without_script) != locale_data.locale_aliases.end())
  648. return {};
  649. if (auto it = find_alias(locale); it != locale_data.locale_aliases.end())
  650. locale_data.locale_aliases.append({ it->name, locale_without_script });
  651. else
  652. locale_data.locale_aliases.append({ locale, locale_without_script });
  653. return {};
  654. };
  655. for (auto const& locale : locale_data.locales)
  656. TRY(append_alias_without_script(locale.key));
  657. for (auto const& locale : locale_data.locale_aliases)
  658. TRY(append_alias_without_script(locale.alias));
  659. return {};
  660. }
  661. static ErrorOr<void> parse_all_locales(String core_path, String locale_names_path, String misc_path, String numbers_path, String dates_path, UnicodeLocaleData& locale_data)
  662. {
  663. auto identity_iterator = TRY(path_to_dir_iterator(locale_names_path));
  664. auto preprocess_iterator = TRY(path_to_dir_iterator(locale_names_path));
  665. auto locale_names_iterator = TRY(path_to_dir_iterator(move(locale_names_path)));
  666. auto misc_iterator = TRY(path_to_dir_iterator(move(misc_path)));
  667. auto numbers_iterator = TRY(path_to_dir_iterator(move(numbers_path)));
  668. auto dates_iterator = TRY(path_to_dir_iterator(move(dates_path)));
  669. LexicalPath core_supplemental_path(core_path);
  670. core_supplemental_path = core_supplemental_path.append("supplemental"sv);
  671. VERIFY(Core::File::is_directory(core_supplemental_path.string()));
  672. TRY(parse_core_aliases(core_supplemental_path.string(), locale_data));
  673. TRY(parse_likely_subtags(core_supplemental_path.string(), locale_data));
  674. auto remove_variants_from_path = [&](String path) -> ErrorOr<String> {
  675. auto parsed_locale = TRY(CanonicalLanguageID<StringIndexType>::parse(locale_data.unique_strings, LexicalPath::basename(path)));
  676. StringBuilder builder;
  677. builder.append(locale_data.unique_strings.get(parsed_locale.language));
  678. if (auto script = locale_data.unique_strings.get(parsed_locale.script); !script.is_empty())
  679. builder.appendff("-{}", script);
  680. if (auto region = locale_data.unique_strings.get(parsed_locale.region); !region.is_empty())
  681. builder.appendff("-{}", region);
  682. return builder.build();
  683. };
  684. while (identity_iterator.has_next()) {
  685. auto locale_path = TRY(next_path_from_dir_iterator(identity_iterator));
  686. auto language = TRY(remove_variants_from_path(locale_path));
  687. auto& locale = locale_data.locales.ensure(language);
  688. TRY(parse_identity(locale_path, locale_data, locale));
  689. }
  690. while (preprocess_iterator.has_next()) {
  691. auto locale_path = TRY(next_path_from_dir_iterator(preprocess_iterator));
  692. TRY(preprocess_languages(locale_path, locale_data));
  693. }
  694. quick_sort(locale_data.languages);
  695. quick_sort(locale_data.territories);
  696. quick_sort(locale_data.scripts);
  697. HashMap<String, KeywordList> keywords;
  698. auto ensure_keyword_list = [&](auto const& language) -> KeywordList& {
  699. return keywords.ensure(language, [&]() {
  700. KeywordList keywords;
  701. keywords.resize(locale_data.keywords.size());
  702. return keywords;
  703. });
  704. };
  705. while (locale_names_iterator.has_next()) {
  706. auto locale_path = TRY(next_path_from_dir_iterator(locale_names_iterator));
  707. auto language = TRY(remove_variants_from_path(locale_path));
  708. auto& locale = locale_data.locales.ensure(language);
  709. TRY(parse_locale_display_patterns(locale_path, locale_data, locale));
  710. TRY(parse_locale_languages(locale_path, locale_data, locale));
  711. TRY(parse_locale_territories(locale_path, locale_data, locale));
  712. TRY(parse_locale_scripts(locale_path, locale_data, locale));
  713. TRY(parse_locale_calendars(locale_path, locale_data, locale));
  714. }
  715. while (misc_iterator.has_next()) {
  716. auto misc_path = TRY(next_path_from_dir_iterator(misc_iterator));
  717. auto language = TRY(remove_variants_from_path(misc_path));
  718. auto& locale = locale_data.locales.ensure(language);
  719. TRY(parse_locale_list_patterns(misc_path, locale_data, locale));
  720. }
  721. while (numbers_iterator.has_next()) {
  722. auto numbers_path = TRY(next_path_from_dir_iterator(numbers_iterator));
  723. auto language = TRY(remove_variants_from_path(numbers_path));
  724. auto& locale = locale_data.locales.ensure(language);
  725. TRY(parse_locale_currencies(numbers_path, locale_data, locale));
  726. auto& keywords = ensure_keyword_list(language);
  727. TRY(parse_numeric_keywords(numbers_path, locale_data, keywords));
  728. }
  729. while (dates_iterator.has_next()) {
  730. auto dates_path = TRY(next_path_from_dir_iterator(dates_iterator));
  731. auto language = TRY(remove_variants_from_path(dates_path));
  732. auto& locale = locale_data.locales.ensure(language);
  733. TRY(parse_locale_date_fields(dates_path, locale_data, locale));
  734. auto& keywords = ensure_keyword_list(language);
  735. TRY(parse_calendar_keywords(dates_path, locale_data, keywords));
  736. }
  737. TRY(parse_default_content_locales(move(core_path), locale_data));
  738. TRY(define_aliases_without_scripts(locale_data));
  739. for (auto& list : keywords) {
  740. auto& locale = locale_data.locales.find(list.key)->value;
  741. locale.keywords = locale_data.unique_keyword_lists.ensure(move(list.value));
  742. }
  743. return {};
  744. }
  745. static void generate_unicode_locale_header(Core::File& file, UnicodeLocaleData& locale_data)
  746. {
  747. StringBuilder builder;
  748. SourceGenerator generator { builder };
  749. generator.append(R"~~~(
  750. #pragma once
  751. #include <AK/Types.h>
  752. namespace Unicode {
  753. )~~~");
  754. auto locales = locale_data.locales.keys();
  755. generate_enum(generator, format_identifier, "Locale"sv, "None"sv, locales, locale_data.locale_aliases);
  756. generate_enum(generator, format_identifier, "Language"sv, {}, locale_data.languages);
  757. generate_enum(generator, format_identifier, "Territory"sv, {}, locale_data.territories);
  758. generate_enum(generator, format_identifier, "ScriptTag"sv, {}, locale_data.scripts);
  759. generate_enum(generator, format_identifier, "Currency"sv, {}, locale_data.currencies);
  760. generate_enum(generator, format_identifier, "CalendarName"sv, {}, locale_data.calendars, locale_data.calendar_aliases);
  761. generate_enum(generator, format_identifier, "DateField"sv, {}, locale_data.date_fields, locale_data.date_field_aliases);
  762. generate_enum(generator, format_identifier, "Key"sv, {}, locale_data.keywords);
  763. generate_enum(generator, format_identifier, "Variant"sv, {}, locale_data.variants);
  764. generate_enum(generator, format_identifier, "ListPatternType"sv, {}, locale_data.list_pattern_types);
  765. generate_enum(generator, format_identifier, "ListPatternStyle"sv, {}, locale_data.list_pattern_styles);
  766. generator.append(R"~~~(
  767. }
  768. )~~~");
  769. VERIFY(file.write(generator.as_string_view()));
  770. }
  771. static void generate_unicode_locale_implementation(Core::File& file, UnicodeLocaleData& locale_data)
  772. {
  773. StringBuilder builder;
  774. SourceGenerator generator { builder };
  775. generator.set("string_index_type"sv, s_string_index_type);
  776. generator.set("locales_size"sv, String::number(locale_data.locales.size()));
  777. generator.set("territories_size", String::number(locale_data.territories.size()));
  778. generator.set("variants_size", String::number(locale_data.max_variant_size));
  779. generator.append(R"~~~(
  780. #include <AK/Array.h>
  781. #include <AK/BinarySearch.h>
  782. #include <AK/Optional.h>
  783. #include <AK/Span.h>
  784. #include <AK/StringView.h>
  785. #include <AK/Vector.h>
  786. #include <LibUnicode/Locale.h>
  787. #include <LibUnicode/UnicodeLocale.h>
  788. namespace Unicode {
  789. )~~~");
  790. locale_data.unique_strings.generate(generator);
  791. generator.append(R"~~~(
  792. struct DisplayPatternImpl {
  793. DisplayPattern to_display_pattern() const
  794. {
  795. DisplayPattern display_patterns {};
  796. display_patterns.locale_pattern = s_string_list[locale_pattern];
  797. display_patterns.locale_separator = s_string_list[locale_separator];
  798. return display_patterns;
  799. }
  800. @string_index_type@ locale_pattern { 0 };
  801. @string_index_type@ locale_separator { 0 };
  802. };
  803. struct Patterns {
  804. ListPatternType type;
  805. ListPatternStyle style;
  806. @string_index_type@ start { 0 };
  807. @string_index_type@ middle { 0 };
  808. @string_index_type@ end { 0 };
  809. @string_index_type@ pair { 0 };
  810. };
  811. )~~~");
  812. locale_data.unique_display_patterns.generate(generator, "DisplayPatternImpl"sv, "s_display_patterns"sv, 30);
  813. locale_data.unique_language_lists.generate(generator, s_string_index_type, "s_language_lists"sv);
  814. locale_data.unique_territory_lists.generate(generator, s_string_index_type, "s_territory_lists"sv);
  815. locale_data.unique_script_lists.generate(generator, s_string_index_type, "s_script_lists"sv);
  816. locale_data.unique_currency_lists.generate(generator, s_string_index_type, "s_currency_lists"sv);
  817. locale_data.unique_calendar_lists.generate(generator, s_string_index_type, "s_calendar_lists"sv);
  818. locale_data.unique_date_field_lists.generate(generator, s_string_index_type, "s_date_field_lists"sv);
  819. locale_data.unique_keyword_lists.generate(generator, s_string_index_type, "s_keyword_lists"sv);
  820. locale_data.unique_list_patterns.generate(generator, "Patterns"sv, "s_list_patterns"sv, 10);
  821. locale_data.unique_list_pattern_lists.generate(generator, s_list_pattern_index_type, "s_list_pattern_lists"sv);
  822. auto append_index = [&](auto index) {
  823. generator.append(String::formatted(", {}", index));
  824. };
  825. auto append_list_and_size = [&](auto const& list) {
  826. if (list.is_empty()) {
  827. generator.append(", {}, 0");
  828. return;
  829. }
  830. bool first = true;
  831. generator.append(", {");
  832. for (auto const& item : list) {
  833. generator.append(first ? " " : ", ");
  834. generator.append(String::number(item));
  835. first = false;
  836. }
  837. generator.append(String::formatted(" }}, {}", list.size()));
  838. };
  839. auto append_mapping = [&](auto const& keys, auto const& map, auto type, auto name, auto mapping_getter) {
  840. generator.set("type", type);
  841. generator.set("name", name);
  842. generator.set("size", String::number(keys.size()));
  843. generator.append(R"~~~(
  844. static constexpr Array<@type@, @size@> @name@ { {)~~~");
  845. bool first = true;
  846. for (auto const& key : keys) {
  847. auto const& value = map.find(key)->value;
  848. auto mapping = mapping_getter(value);
  849. generator.append(first ? " " : ", ");
  850. generator.append(String::number(mapping));
  851. first = false;
  852. }
  853. generator.append(" } };");
  854. };
  855. auto locales = locale_data.locales.keys();
  856. quick_sort(locales);
  857. append_mapping(locales, locale_data.locales, s_display_pattern_index_type, "s_locale_display_patterns"sv, [&](auto const& locale) { return locale.display_patterns; });
  858. append_mapping(locales, locale_data.locales, s_language_list_index_type, "s_languages"sv, [&](auto const& locale) { return locale.languages; });
  859. append_mapping(locales, locale_data.locales, s_territory_list_index_type, "s_territories"sv, [&](auto const& locale) { return locale.territories; });
  860. append_mapping(locales, locale_data.locales, s_script_list_index_type, "s_scripts"sv, [&](auto const& locale) { return locale.scripts; });
  861. append_mapping(locales, locale_data.locales, s_currency_list_index_type, "s_long_currencies"sv, [&](auto const& locale) { return locale.long_currencies; });
  862. append_mapping(locales, locale_data.locales, s_currency_list_index_type, "s_short_currencies"sv, [&](auto const& locale) { return locale.short_currencies; });
  863. append_mapping(locales, locale_data.locales, s_currency_list_index_type, "s_narrow_currencies"sv, [&](auto const& locale) { return locale.narrow_currencies; });
  864. append_mapping(locales, locale_data.locales, s_currency_list_index_type, "s_numeric_currencies"sv, [&](auto const& locale) { return locale.numeric_currencies; });
  865. append_mapping(locales, locale_data.locales, s_calendar_list_index_type, "s_calendars"sv, [&](auto const& locale) { return locale.calendars; });
  866. append_mapping(locales, locale_data.locales, s_date_field_list_index_type, "s_long_date_fields"sv, [&](auto const& locale) { return locale.long_date_fields; });
  867. append_mapping(locales, locale_data.locales, s_date_field_list_index_type, "s_short_date_fields"sv, [&](auto const& locale) { return locale.short_date_fields; });
  868. append_mapping(locales, locale_data.locales, s_date_field_list_index_type, "s_narrow_date_fields"sv, [&](auto const& locale) { return locale.narrow_date_fields; });
  869. append_mapping(locales, locale_data.locales, s_keyword_list_index_type, "s_keywords"sv, [&](auto const& locale) { return locale.keywords; });
  870. append_mapping(locales, locale_data.locales, s_list_pattern_list_index_type, "s_locale_list_patterns"sv, [&](auto const& locale) { return locale.list_patterns; });
  871. generator.append(R"~~~(
  872. struct CanonicalLanguageID {
  873. LanguageID to_unicode_language_id() const
  874. {
  875. LanguageID language_id {};
  876. language_id.variants.ensure_capacity(variants_size);
  877. language_id.language = s_string_list[language];
  878. if (script != 0)
  879. language_id.script = s_string_list[script];
  880. if (region != 0)
  881. language_id.region = s_string_list[region];
  882. for (size_t i = 0; i < variants_size; ++i)
  883. language_id.variants.append(s_string_list[variants[i]]);
  884. return language_id;
  885. }
  886. bool matches_variants(Vector<String> const& other_variants) const {
  887. if (variants_size == 0)
  888. return true;
  889. if (other_variants.size() != variants_size)
  890. return false;
  891. for (size_t i = 0; i < variants_size; ++i) {
  892. if (s_string_list[variants[i]] != other_variants[i])
  893. return false;
  894. }
  895. return true;
  896. };
  897. @string_index_type@ language { 0 };
  898. @string_index_type@ script { 0 };
  899. @string_index_type@ region { 0 };
  900. Array<@string_index_type@, @variants_size@> variants {};
  901. size_t variants_size { 0 };
  902. };
  903. struct LanguageMapping {
  904. CanonicalLanguageID key;
  905. CanonicalLanguageID alias;
  906. };
  907. )~~~");
  908. auto append_complex_mapping = [&](StringView name, auto& mappings) {
  909. generator.set("size", String::number(mappings.size()));
  910. generator.set("name"sv, name);
  911. generator.append(R"~~~(
  912. static constexpr Array<LanguageMapping, @size@> s_@name@ { {
  913. )~~~");
  914. quick_sort(mappings, [&](auto const& lhs, auto const& rhs) {
  915. auto const& lhs_language = locale_data.unique_strings.get(lhs.key.language);
  916. auto const& rhs_language = locale_data.unique_strings.get(rhs.key.language);
  917. // Sort the keys such that "und" language tags are at the end, as those are less specific.
  918. if (lhs_language.starts_with("und"sv) && !rhs_language.starts_with("und"sv))
  919. return false;
  920. if (!lhs_language.starts_with("und"sv) && rhs_language.starts_with("und"sv))
  921. return true;
  922. return lhs_language < rhs_language;
  923. });
  924. for (auto const& mapping : mappings) {
  925. generator.set("language"sv, String::number(mapping.key.language));
  926. generator.append(" { { @language@");
  927. append_index(mapping.key.script);
  928. append_index(mapping.key.region);
  929. append_list_and_size(mapping.key.variants);
  930. generator.set("language"sv, String::number(mapping.alias.language));
  931. generator.append(" }, { @language@");
  932. append_index(mapping.alias.script);
  933. append_index(mapping.alias.region);
  934. append_list_and_size(mapping.alias.variants);
  935. generator.append(" } },\n");
  936. }
  937. generator.append("} };\n");
  938. };
  939. append_complex_mapping("complex_alias"sv, locale_data.complex_mappings);
  940. append_complex_mapping("likely_subtags"sv, locale_data.likely_subtags);
  941. generator.append(R"~~~(
  942. static LanguageMapping const* resolve_likely_subtag(LanguageID const& language_id)
  943. {
  944. // https://unicode.org/reports/tr35/#Likely_Subtags
  945. enum class State {
  946. LanguageScriptRegion,
  947. LanguageRegion,
  948. LanguageScript,
  949. Language,
  950. UndScript,
  951. Done,
  952. };
  953. auto state = State::LanguageScriptRegion;
  954. while (state != State::Done) {
  955. LanguageID search_key;
  956. switch (state) {
  957. case State::LanguageScriptRegion:
  958. state = State::LanguageRegion;
  959. if (!language_id.script.has_value() || !language_id.region.has_value())
  960. continue;
  961. search_key.language = *language_id.language;
  962. search_key.script = *language_id.script;
  963. search_key.region = *language_id.region;
  964. break;
  965. case State::LanguageRegion:
  966. state = State::LanguageScript;
  967. if (!language_id.region.has_value())
  968. continue;
  969. search_key.language = *language_id.language;
  970. search_key.region = *language_id.region;
  971. break;
  972. case State::LanguageScript:
  973. state = State::Language;
  974. if (!language_id.script.has_value())
  975. continue;
  976. search_key.language = *language_id.language;
  977. search_key.script = *language_id.script;
  978. break;
  979. case State::Language:
  980. state = State::UndScript;
  981. search_key.language = *language_id.language;
  982. break;
  983. case State::UndScript:
  984. state = State::Done;
  985. if (!language_id.script.has_value())
  986. continue;
  987. search_key.language = "und"sv;
  988. search_key.script = *language_id.script;
  989. break;
  990. default:
  991. VERIFY_NOT_REACHED();
  992. }
  993. for (auto const& map : s_likely_subtags) {
  994. auto const& key_language = s_string_list[map.key.language];
  995. auto const& key_script = s_string_list[map.key.script];
  996. auto const& key_region = s_string_list[map.key.region];
  997. if (key_language != search_key.language)
  998. continue;
  999. if (!key_script.is_empty() || search_key.script.has_value()) {
  1000. if (key_script != search_key.script)
  1001. continue;
  1002. }
  1003. if (!key_region.is_empty() || search_key.region.has_value()) {
  1004. if (key_region != search_key.region)
  1005. continue;
  1006. }
  1007. return &map;
  1008. }
  1009. }
  1010. return nullptr;
  1011. }
  1012. )~~~");
  1013. auto append_mapping_search = [&](StringView enum_snake, StringView from_string_name, StringView collection_name, StringView unique_list) {
  1014. generator.set("enum_snake", enum_snake);
  1015. generator.set("from_string_name", from_string_name);
  1016. generator.set("collection_name", collection_name);
  1017. generator.set("unique_list", unique_list);
  1018. generator.append(R"~~~(
  1019. Optional<StringView> get_locale_@enum_snake@_mapping(StringView locale, StringView @enum_snake@)
  1020. {
  1021. auto locale_value = locale_from_string(locale);
  1022. if (!locale_value.has_value())
  1023. return {};
  1024. auto @enum_snake@_value = @from_string_name@_from_string(@enum_snake@);
  1025. if (!@enum_snake@_value.has_value())
  1026. return {};
  1027. auto locale_index = to_underlying(*locale_value) - 1; // Subtract 1 because 0 == Locale::None.
  1028. auto @enum_snake@_index = to_underlying(*@enum_snake@_value);
  1029. auto mapping_index = @collection_name@.at(locale_index);
  1030. auto const& mappings = @unique_list@.at(mapping_index);
  1031. auto @enum_snake@_string_index = mappings.at(@enum_snake@_index);
  1032. auto @enum_snake@_mapping = s_string_list.at(@enum_snake@_string_index);
  1033. if (@enum_snake@_mapping.is_empty())
  1034. return {};
  1035. return @enum_snake@_mapping;
  1036. }
  1037. )~~~");
  1038. };
  1039. auto append_from_string = [&](StringView enum_title, StringView enum_snake, auto const& values, Vector<Alias> const& aliases = {}) {
  1040. HashValueMap<String> hashes;
  1041. hashes.ensure_capacity(values.size());
  1042. for (auto const& value : values)
  1043. hashes.set(value.hash(), format_identifier(enum_title, value));
  1044. for (auto const& alias : aliases)
  1045. hashes.set(alias.alias.hash(), format_identifier(enum_title, alias.alias));
  1046. generate_value_from_string(generator, "{}_from_string"sv, enum_title, enum_snake, move(hashes));
  1047. };
  1048. auto append_alias_search = [&](StringView enum_snake, auto const& aliases) {
  1049. HashValueMap<StringIndexType> hashes;
  1050. hashes.ensure_capacity(aliases.size());
  1051. for (auto const& alias : aliases)
  1052. hashes.set(alias.key.hash(), alias.value);
  1053. ValueFromStringOptions options {};
  1054. options.return_type = "StringView"sv;
  1055. options.return_format = "s_string_list[{}]"sv;
  1056. generate_value_from_string(generator, "resolve_{}_alias"sv, s_string_index_type, enum_snake, move(hashes), options);
  1057. };
  1058. append_from_string("Locale"sv, "locale"sv, locale_data.locales.keys(), locale_data.locale_aliases);
  1059. append_from_string("Language"sv, "language"sv, locale_data.languages);
  1060. append_mapping_search("language"sv, "language"sv, "s_languages"sv, "s_language_lists"sv);
  1061. append_alias_search("language"sv, locale_data.language_aliases);
  1062. append_from_string("Territory"sv, "territory"sv, locale_data.territories);
  1063. append_mapping_search("territory"sv, "territory"sv, "s_territories"sv, "s_territory_lists"sv);
  1064. append_alias_search("territory"sv, locale_data.territory_aliases);
  1065. append_from_string("ScriptTag"sv, "script_tag"sv, locale_data.scripts);
  1066. append_mapping_search("script"sv, "script_tag"sv, "s_scripts"sv, "s_script_lists"sv);
  1067. append_alias_search("script_tag"sv, locale_data.script_aliases);
  1068. append_from_string("Currency"sv, "currency"sv, locale_data.currencies);
  1069. append_mapping_search("long_currency"sv, "currency"sv, "s_long_currencies"sv, "s_currency_lists"sv);
  1070. append_mapping_search("short_currency"sv, "currency"sv, "s_short_currencies"sv, "s_currency_lists"sv);
  1071. append_mapping_search("narrow_currency"sv, "currency"sv, "s_narrow_currencies"sv, "s_currency_lists"sv);
  1072. append_mapping_search("numeric_currency"sv, "currency"sv, "s_numeric_currencies"sv, "s_currency_lists"sv);
  1073. append_from_string("CalendarName"sv, "calendar_name"sv, locale_data.calendars, locale_data.calendar_aliases);
  1074. append_mapping_search("calendar"sv, "calendar_name"sv, "s_calendars"sv, "s_calendar_lists"sv);
  1075. append_from_string("DateField"sv, "date_field"sv, locale_data.date_fields, locale_data.date_field_aliases);
  1076. append_mapping_search("long_date_field"sv, "date_field"sv, "s_long_date_fields"sv, "s_date_field_lists"sv);
  1077. append_mapping_search("short_date_field"sv, "date_field"sv, "s_short_date_fields"sv, "s_date_field_lists"sv);
  1078. append_mapping_search("narrow_date_field"sv, "date_field"sv, "s_narrow_date_fields"sv, "s_date_field_lists"sv);
  1079. append_from_string("Key"sv, "key"sv, locale_data.keywords);
  1080. append_mapping_search("key"sv, "key"sv, "s_keywords"sv, "s_keyword_lists"sv);
  1081. append_alias_search("variant"sv, locale_data.variant_aliases);
  1082. append_alias_search("subdivision"sv, locale_data.subdivision_aliases);
  1083. append_from_string("ListPatternType"sv, "list_pattern_type"sv, locale_data.list_pattern_types);
  1084. append_from_string("ListPatternStyle"sv, "list_pattern_style"sv, locale_data.list_pattern_styles);
  1085. generator.append(R"~~~(
  1086. Optional<DisplayPattern> get_locale_display_patterns(StringView locale)
  1087. {
  1088. auto locale_value = locale_from_string(locale);
  1089. if (!locale_value.has_value())
  1090. return {};
  1091. auto locale_index = to_underlying(*locale_value) - 1; // Subtract 1 because 0 == Locale::None.
  1092. auto display_patterns_index = s_locale_display_patterns.at(locale_index);
  1093. auto const& display_patterns = s_display_patterns.at(display_patterns_index);
  1094. return display_patterns.to_display_pattern();
  1095. }
  1096. Optional<ListPatterns> get_locale_list_patterns(StringView locale, StringView list_pattern_type, StringView list_pattern_style)
  1097. {
  1098. auto locale_value = locale_from_string(locale);
  1099. if (!locale_value.has_value())
  1100. return {};
  1101. auto type_value = list_pattern_type_from_string(list_pattern_type);
  1102. if (!type_value.has_value())
  1103. return {};
  1104. auto style_value = list_pattern_style_from_string(list_pattern_style);
  1105. if (!style_value.has_value())
  1106. return {};
  1107. auto locale_index = to_underlying(*locale_value) - 1; // Subtract 1 because 0 == Locale::None.
  1108. auto list_patterns_list_index = s_locale_list_patterns.at(locale_index);
  1109. auto const& locale_list_patterns = s_list_pattern_lists.at(list_patterns_list_index);
  1110. for (auto list_patterns_index : locale_list_patterns) {
  1111. auto const& list_patterns = s_list_patterns.at(list_patterns_index);
  1112. if ((list_patterns.type == type_value) && (list_patterns.style == style_value)) {
  1113. auto const& start = s_string_list[list_patterns.start];
  1114. auto const& middle = s_string_list[list_patterns.middle];
  1115. auto const& end = s_string_list[list_patterns.end];
  1116. auto const& pair = s_string_list[list_patterns.pair];
  1117. return ListPatterns { start, middle, end, pair };
  1118. }
  1119. }
  1120. return {};
  1121. }
  1122. void resolve_complex_language_aliases(LanguageID& language_id)
  1123. {
  1124. for (auto const& map : s_complex_alias) {
  1125. auto const& key_language = s_string_list[map.key.language];
  1126. auto const& key_script = s_string_list[map.key.script];
  1127. auto const& key_region = s_string_list[map.key.region];
  1128. if ((key_language != language_id.language) && (key_language != "und"sv))
  1129. continue;
  1130. if (!key_script.is_empty() && (key_script != language_id.script))
  1131. continue;
  1132. if (!key_region.is_empty() && (key_region != language_id.region))
  1133. continue;
  1134. if (!map.key.matches_variants(language_id.variants))
  1135. continue;
  1136. auto alias = map.alias.to_unicode_language_id();
  1137. if (alias.language == "und"sv)
  1138. alias.language = move(language_id.language);
  1139. if (key_script.is_empty() && !alias.script.has_value())
  1140. alias.script = move(language_id.script);
  1141. if (key_region.is_empty() && !alias.region.has_value())
  1142. alias.region = move(language_id.region);
  1143. if (map.key.variants_size == 0 && alias.variants.is_empty())
  1144. alias.variants = move(language_id.variants);
  1145. language_id = move(alias);
  1146. break;
  1147. }
  1148. }
  1149. Optional<LanguageID> add_likely_subtags(LanguageID const& language_id)
  1150. {
  1151. // https://www.unicode.org/reports/tr35/#Likely_Subtags
  1152. auto const* likely_subtag = resolve_likely_subtag(language_id);
  1153. if (likely_subtag == nullptr)
  1154. return {};
  1155. auto maximized = language_id;
  1156. auto const& key_script = s_string_list[likely_subtag->key.script];
  1157. auto const& key_region = s_string_list[likely_subtag->key.region];
  1158. auto const& alias_language = s_string_list[likely_subtag->alias.language];
  1159. auto const& alias_script = s_string_list[likely_subtag->alias.script];
  1160. auto const& alias_region = s_string_list[likely_subtag->alias.region];
  1161. if (maximized.language == "und"sv)
  1162. maximized.language = alias_language;
  1163. if (!maximized.script.has_value() || (!key_script.is_empty() && !alias_script.is_empty()))
  1164. maximized.script = alias_script;
  1165. if (!maximized.region.has_value() || (!key_region.is_empty() && !alias_region.is_empty()))
  1166. maximized.region = alias_region;
  1167. return maximized;
  1168. }
  1169. Optional<String> resolve_most_likely_territory(LanguageID const& language_id)
  1170. {
  1171. if (auto const* likely_subtag = resolve_likely_subtag(language_id); likely_subtag != nullptr)
  1172. return s_string_list[likely_subtag->alias.region];
  1173. return {};
  1174. }
  1175. }
  1176. )~~~");
  1177. VERIFY(file.write(generator.as_string_view()));
  1178. }
  1179. ErrorOr<int> serenity_main(Main::Arguments arguments)
  1180. {
  1181. StringView generated_header_path;
  1182. StringView generated_implementation_path;
  1183. StringView core_path;
  1184. StringView locale_names_path;
  1185. StringView misc_path;
  1186. StringView numbers_path;
  1187. StringView dates_path;
  1188. Core::ArgsParser args_parser;
  1189. args_parser.add_option(generated_header_path, "Path to the Unicode locale header file to generate", "generated-header-path", 'h', "generated-header-path");
  1190. args_parser.add_option(generated_implementation_path, "Path to the Unicode locale implementation file to generate", "generated-implementation-path", 'c', "generated-implementation-path");
  1191. args_parser.add_option(core_path, "Path to cldr-core directory", "core-path", 'r', "core-path");
  1192. args_parser.add_option(locale_names_path, "Path to cldr-localenames directory", "locale-names-path", 'l', "locale-names-path");
  1193. args_parser.add_option(misc_path, "Path to cldr-misc directory", "misc-path", 'm', "misc-path");
  1194. args_parser.add_option(numbers_path, "Path to cldr-numbers directory", "numbers-path", 'n', "numbers-path");
  1195. args_parser.add_option(dates_path, "Path to cldr-dates directory", "dates-path", 'd', "dates-path");
  1196. args_parser.parse(arguments);
  1197. auto open_file = [&](StringView path) -> ErrorOr<NonnullRefPtr<Core::File>> {
  1198. if (path.is_empty()) {
  1199. args_parser.print_usage(stderr, arguments.argv[0]);
  1200. return Error::from_string_literal("Must provide all command line options"sv);
  1201. }
  1202. return Core::File::open(path, Core::OpenMode::ReadWrite);
  1203. };
  1204. auto generated_header_file = TRY(open_file(generated_header_path));
  1205. auto generated_implementation_file = TRY(open_file(generated_implementation_path));
  1206. UnicodeLocaleData locale_data;
  1207. TRY(parse_all_locales(core_path, locale_names_path, misc_path, numbers_path, dates_path, locale_data));
  1208. generate_unicode_locale_header(generated_header_file, locale_data);
  1209. generate_unicode_locale_implementation(generated_implementation_file, locale_data);
  1210. return 0;
  1211. }