LibUnicode: Parse and generate date, time, and date-time format patterns

Meta/Lagom/Tools/CodeGenerators/LibUnicode/GenerateUnicodeDateTimeFormat.cpp

@@ -17,18 +17,103 @@
 #include <LibCore/ArgsParser.h>
 #include <LibCore/DirIterator.h>
 #include <LibCore/File.h>
+#include <LibUnicode/DateTimeFormat.h>
 
 using StringIndexType = u16;
 constexpr auto s_string_index_type = "u16"sv;
 
+struct CalendarPattern : public Unicode::CalendarPattern {
+    StringIndexType pattern_index { 0 };
+};
+
+struct CalendarFormat {
+    CalendarPattern full_format {};
+    CalendarPattern long_format {};
+    CalendarPattern medium_format {};
+    CalendarPattern short_format {};
+};
+
+struct Calendar {
+    StringIndexType calendar { 0 };
+    CalendarFormat date_formats {};
+    CalendarFormat time_formats {};
+    CalendarFormat date_time_formats {};
+};
+
 struct Locale {
+    HashMap<String, Calendar> calendars;
 };
 
 struct UnicodeLocaleData {
     UniqueStringStorage<StringIndexType> unique_strings;
     HashMap<String, Locale> locales;
+    Vector<String> calendars;
 };
 
+static void parse_date_time_pattern(CalendarPattern& format, String pattern, UnicodeLocaleData& locale_data)
+{
+    // FIXME: This is very incomplete. Similar to NumberFormat, the pattern string will need to be
+    //        parsed to fill in the CalendarPattern struct, and modified to be useable at runtime.
+    //        For now, this is enough to implement the DateTimeFormat constructor.
+    //
+    // https://unicode.org/reports/tr35/tr35-dates.html#Date_Field_Symbol_Table
+    format.pattern_index = locale_data.unique_strings.ensure(move(pattern));
+}
+
+static ErrorOr<void> parse_calendars(String locale_calendars_path, UnicodeLocaleData& locale_data, Locale& locale)
+{
+    LexicalPath calendars_path(move(locale_calendars_path));
+    if (!calendars_path.basename().starts_with("ca-"sv))
+        return {};
+
+    auto calendars_file = TRY(Core::File::open(calendars_path.string(), Core::OpenMode::ReadOnly));
+    auto calendars = TRY(JsonValue::from_string(calendars_file->read_all()));
+
+    auto const& main_object = calendars.as_object().get("main"sv);
+    auto const& locale_object = main_object.as_object().get(calendars_path.parent().basename());
+    auto const& dates_object = locale_object.as_object().get("dates"sv);
+    auto const& calendars_object = dates_object.as_object().get("calendars"sv);
+
+    auto ensure_calendar = [&](auto const& calendar) -> Calendar& {
+        return locale.calendars.ensure(calendar, [&]() {
+            auto calendar_index = locale_data.unique_strings.ensure(calendar);
+            return Calendar { .calendar = calendar_index };
+        });
+    };
+
+    auto parse_patterns = [&](auto& formats, auto const& patterns_object) {
+        auto full_format = patterns_object.get("full"sv);
+        parse_date_time_pattern(formats.full_format, full_format.as_string(), locale_data);
+
+        auto long_format = patterns_object.get("long"sv);
+        parse_date_time_pattern(formats.long_format, long_format.as_string(), locale_data);
+
+        auto medium_format = patterns_object.get("medium"sv);
+        parse_date_time_pattern(formats.medium_format, medium_format.as_string(), locale_data);
+
+        auto short_format = patterns_object.get("short"sv);
+        parse_date_time_pattern(formats.short_format, short_format.as_string(), locale_data);
+    };
+
+    calendars_object.as_object().for_each_member([&](auto const& calendar_name, JsonValue const& value) {
+        auto& calendar = ensure_calendar(calendar_name);
+
+        if (!locale_data.calendars.contains_slow(calendar_name))
+            locale_data.calendars.append(calendar_name);
+
+        auto const& date_formats_object = value.as_object().get("dateFormats"sv);
+        parse_patterns(calendar.date_formats, date_formats_object.as_object());
+
+        auto const& time_formats_object = value.as_object().get("timeFormats"sv);
+        parse_patterns(calendar.time_formats, time_formats_object.as_object());
+
+        auto const& date_time_formats_object = value.as_object().get("dateTimeFormats"sv);
+        parse_patterns(calendar.date_time_formats, date_time_formats_object.as_object());
+    });
+
+    return {};
+}
+
 static ErrorOr<void> parse_all_locales(String dates_path, UnicodeLocaleData& locale_data)
 {
     auto dates_iterator = TRY(path_to_dir_iterator(move(dates_path)));
@@ -48,15 +133,26 @@ static ErrorOr<void> parse_all_locales(String dates_path, UnicodeLocaleData& loc
 
     while (dates_iterator.has_next()) {
         auto dates_path = TRY(next_path_from_dir_iterator(dates_iterator));
-        auto language = TRY(remove_variants_from_path(dates_path));
+        auto calendars_iterator = TRY(path_to_dir_iterator(dates_path, {}));
 
-        [[maybe_unused]] auto& locale = locale_data.locales.ensure(language);
+        auto language = TRY(remove_variants_from_path(dates_path));
+        auto& locale = locale_data.locales.ensure(language);
+
+        while (calendars_iterator.has_next()) {
+            auto calendars_path = TRY(next_path_from_dir_iterator(calendars_iterator));
+            TRY(parse_calendars(move(calendars_path), locale_data, locale));
+        }
     }
 
     return {};
 }
 
-static void generate_unicode_locale_header(Core::File& file, UnicodeLocaleData&)
+static String format_identifier(StringView, StringView identifier)
+{
+    return identifier.to_titlecase_string();
+}
+
+static void generate_unicode_locale_header(Core::File& file, UnicodeLocaleData& locale_data)
 {
     StringBuilder builder;
     SourceGenerator generator { builder };
@@ -64,10 +160,25 @@ static void generate_unicode_locale_header(Core::File& file, UnicodeLocaleData&)
     generator.append(R"~~~(
 #pragma once
 
+#include <AK/Optional.h>
 #include <AK/StringView.h>
 #include <LibUnicode/Forward.h>
 
 namespace Unicode {
+)~~~");
+
+    generate_enum(generator, format_identifier, "Calendar"sv, {}, locale_data.calendars);
+
+    generator.append(R"~~~(
+namespace Detail {
+
+Optional<Calendar> calendar_from_string(StringView calendar);
+Optional<Unicode::CalendarFormat> get_calendar_date_format(StringView locale, StringView calendar);
+Optional<Unicode::CalendarFormat> get_calendar_time_format(StringView locale, StringView calendar);
+Optional<Unicode::CalendarFormat> get_calendar_date_time_format(StringView locale, StringView calendar);
+
+}
+
 }
 )~~~");
 
@@ -82,6 +193,9 @@ static void generate_unicode_locale_implementation(Core::File& file, UnicodeLoca
 
     generator.append(R"~~~(
 #include <AK/Array.h>
+#include <AK/BinarySearch.h>
+#include <LibUnicode/DateTimeFormat.h>
+#include <LibUnicode/Locale.h>
 #include <LibUnicode/UnicodeDateTimeFormat.h>
 
 namespace Unicode::Detail {
@@ -90,6 +204,139 @@ namespace Unicode::Detail {
     locale_data.unique_strings.generate(generator);
 
     generator.append(R"~~~(
+struct CalendarPattern {
+    Unicode::CalendarPattern to_unicode_calendar_pattern() const {
+        Unicode::CalendarPattern calendar_pattern {};
+        calendar_pattern.pattern = s_string_list[pattern];
+        return calendar_pattern;
+    }
+
+    @string_index_type@ pattern { 0 };
+};
+
+struct CalendarFormat {
+    Unicode::CalendarFormat to_unicode_calendar_format() const {
+        Unicode::CalendarFormat calendar_format {};
+
+        calendar_format.full_format = full_format.to_unicode_calendar_pattern();
+        calendar_format.long_format = long_format.to_unicode_calendar_pattern();
+        calendar_format.medium_format = medium_format.to_unicode_calendar_pattern();
+        calendar_format.short_format = short_format.to_unicode_calendar_pattern();
+
+        return calendar_format;
+    }
+
+    CalendarPattern full_format {};
+    CalendarPattern long_format {};
+    CalendarPattern medium_format {};
+    CalendarPattern short_format {};
+};
+
+struct CalendarData {
+    @string_index_type@ calendar { 0 };
+    CalendarFormat date_formats {};
+    CalendarFormat time_formats {};
+    CalendarFormat date_time_formats {};
+};
+)~~~");
+
+    auto append_calendar_pattern = [&](auto const& calendar_pattern) {
+        generator.set("pattern"sv, String::number(calendar_pattern.pattern_index));
+        generator.append("{ @pattern@ },");
+    };
+
+    auto append_calendar_format = [&](auto const& calendar_format) {
+        generator.append("{ ");
+        append_calendar_pattern(calendar_format.full_format);
+        generator.append(" ");
+        append_calendar_pattern(calendar_format.long_format);
+        generator.append(" ");
+        append_calendar_pattern(calendar_format.medium_format);
+        generator.append(" ");
+        append_calendar_pattern(calendar_format.short_format);
+        generator.append(" },");
+    };
+
+    auto append_calendars = [&](String name, auto const& calendars) {
+        generator.set("name", name);
+        generator.set("size", String::number(calendars.size()));
+
+        generator.append(R"~~~(
+static constexpr Array<CalendarData, @size@> @name@ { {)~~~");
+
+        for (auto const& calendar_key : locale_data.calendars) {
+            auto const& calendar = calendars.find(calendar_key)->value;
+
+            generator.set("calendar"sv, String::number(calendar.calendar));
+            generator.append(R"~~~(
+    { @calendar@, )~~~");
+
+            append_calendar_format(calendar.date_formats);
+            generator.append(" ");
+            append_calendar_format(calendar.time_formats);
+            generator.append(" ");
+            append_calendar_format(calendar.date_time_formats);
+            generator.append(" },");
+        }
+
+        generator.append(R"~~~(
+} };
+)~~~");
+    };
+
+    generate_mapping(generator, locale_data.locales, "CalendarData"sv, "s_calendars"sv, "s_calendars_{}", [&](auto const& name, auto const& value) { append_calendars(name, value.calendars); });
+
+    auto append_from_string = [&](StringView enum_title, StringView enum_snake, auto const& values) {
+        HashValueMap<String> hashes;
+        hashes.ensure_capacity(values.size());
+
+        for (auto const& value : values)
+            hashes.set(value.hash(), format_identifier(enum_title, value));
+
+        generate_value_from_string(generator, "{}_from_string"sv, enum_title, enum_snake, move(hashes));
+    };
+
+    append_from_string("Calendar"sv, "calendar"sv, locale_data.calendars);
+
+    generator.append(R"~~~(
+static CalendarData const* find_calendar_data(StringView locale, StringView calendar)
+{
+    auto locale_value = locale_from_string(locale);
+    if (!locale_value.has_value())
+        return nullptr;
+
+    auto calendar_value = calendar_from_string(calendar);
+    if (!calendar_value.has_value())
+        return nullptr;
+
+    auto locale_index = to_underlying(*locale_value) - 1; // Subtract 1 because 0 == Locale::None.
+    auto calendar_index = to_underlying(*calendar_value);
+
+    auto const& calendars = s_calendars.at(locale_index);
+    return &calendars[calendar_index];
+}
+
+Optional<Unicode::CalendarFormat> get_calendar_date_format(StringView locale, StringView calendar)
+{
+    if (auto const* data = find_calendar_data(locale, calendar); data != nullptr)
+        return data->date_formats.to_unicode_calendar_format();
+    return {};
+}
+
+Optional<Unicode::CalendarFormat> get_calendar_time_format(StringView locale, StringView calendar)
+{
+    if (auto const* data = find_calendar_data(locale, calendar); data != nullptr)
+        return data->time_formats.to_unicode_calendar_format();
+    return {};
+}
+
+Optional<Unicode::CalendarFormat> get_calendar_date_time_format(StringView locale, StringView calendar)
+{
+    if (auto const* data = find_calendar_data(locale, calendar); data != nullptr)
+        return data->date_time_formats.to_unicode_calendar_format();
+    return {};
+}
+
 }
 )~~~");
 

Meta/Lagom/Tools/CodeGenerators/LibUnicode/GeneratorUtil.h

@@ -137,10 +137,11 @@ struct CanonicalLanguageID {
     Vector<StringIndexType> variants {};
 };
 
-inline ErrorOr<Core::DirIterator> path_to_dir_iterator(String path)
+inline ErrorOr<Core::DirIterator> path_to_dir_iterator(String path, StringView subpath = "main"sv)
 {
     LexicalPath lexical_path(move(path));
-    lexical_path = lexical_path.append("main"sv);
+    if (!subpath.is_empty())
+        lexical_path = lexical_path.append(subpath);
 
     Core::DirIterator iterator(lexical_path.string(), Core::DirIterator::SkipParentAndBaseDir);
     if (iterator.has_error())

Userland/Libraries/LibUnicode/CMakeLists.txt

@@ -4,6 +4,7 @@ SET(SOURCES
     ${UNICODE_DATA_SOURCES}
     CharacterTypes.cpp
     CurrencyCode.cpp
+    DateTimeFormat.cpp
     Locale.cpp
     NumberFormat.cpp
 )

Userland/Libraries/LibUnicode/DateTimeFormat.cpp

@@ -0,0 +1,66 @@
+/*
+ * Copyright (c) 2021, Tim Flynn <trflynn89@pm.me>
+ *
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
+#include <LibUnicode/DateTimeFormat.h>
+
+#if ENABLE_UNICODE_DATA
+#    include <LibUnicode/UnicodeDateTimeFormat.h>
+#endif
+
+namespace Unicode {
+
+CalendarPatternStyle calendar_pattern_style_from_string(StringView style)
+{
+    if (style == "narrow"sv)
+        return CalendarPatternStyle::Narrow;
+    if (style == "short"sv)
+        return CalendarPatternStyle::Short;
+    if (style == "long"sv)
+        return CalendarPatternStyle::Long;
+    if (style == "numeric"sv)
+        return CalendarPatternStyle::Numeric;
+    if (style == "2-digit"sv)
+        return CalendarPatternStyle::TwoDigit;
+    VERIFY_NOT_REACHED();
+}
+
+StringView calendar_pattern_style_to_string(CalendarPatternStyle style)
+{
+    switch (style) {
+    case CalendarPatternStyle::Narrow:
+        return "narrow"sv;
+    case CalendarPatternStyle::Short:
+        return "short"sv;
+    case CalendarPatternStyle::Long:
+        return "long"sv;
+    case CalendarPatternStyle::Numeric:
+        return "Numeric"sv;
+    case CalendarPatternStyle::TwoDigit:
+        return "2-digit"sv;
+    default:
+        VERIFY_NOT_REACHED();
+    }
+}
+
+Optional<CalendarFormat> get_calendar_format([[maybe_unused]] StringView locale, [[maybe_unused]] StringView calendar, [[maybe_unused]] CalendarFormatType type)
+{
+#if ENABLE_UNICODE_DATA
+    switch (type) {
+    case CalendarFormatType::Date:
+        return Detail::get_calendar_date_format(locale, calendar);
+    case CalendarFormatType::Time:
+        return Detail::get_calendar_time_format(locale, calendar);
+    case CalendarFormatType::DateTime:
+        return Detail::get_calendar_date_time_format(locale, calendar);
+    default:
+        VERIFY_NOT_REACHED();
+    }
+#else
+    return {};
+#endif
+}
+
+}

Userland/Libraries/LibUnicode/DateTimeFormat.h

@@ -0,0 +1,60 @@
+/*
+ * Copyright (c) 2021, Tim Flynn <trflynn89@pm.me>
+ *
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
+#pragma once
+
+#include <AK/Optional.h>
+#include <AK/String.h>
+#include <AK/StringView.h>
+#include <AK/Types.h>
+#include <LibUnicode/Forward.h>
+
+namespace Unicode {
+
+enum class CalendarPatternStyle : u8 {
+    Narrow,
+    Short,
+    Long,
+    Numeric,
+    TwoDigit,
+};
+
+struct CalendarPattern {
+    String pattern {};
+    Optional<String> pattern12 {};
+
+    // https://unicode.org/reports/tr35/tr35-dates.html#Calendar_Fields
+    Optional<CalendarPatternStyle> era {};
+    Optional<CalendarPatternStyle> year {};
+    Optional<CalendarPatternStyle> month {};
+    Optional<CalendarPatternStyle> weekday {};
+    Optional<CalendarPatternStyle> day {};
+    Optional<CalendarPatternStyle> day_period {};
+    Optional<CalendarPatternStyle> hour {};
+    Optional<CalendarPatternStyle> minute {};
+    Optional<CalendarPatternStyle> second {};
+    Optional<u8> fractional_second_digits {};
+    Optional<CalendarPatternStyle> time_zone_name {};
+};
+
+enum class CalendarFormatType : u8 {
+    Date,
+    Time,
+    DateTime,
+};
+
+struct CalendarFormat {
+    CalendarPattern full_format {};
+    CalendarPattern long_format {};
+    CalendarPattern medium_format {};
+    CalendarPattern short_format {};
+};
+
+CalendarPatternStyle calendar_pattern_style_from_string(StringView style);
+StringView calendar_pattern_style_to_string(CalendarPatternStyle style);
+Optional<CalendarFormat> get_calendar_format(StringView locale, StringView calendar, CalendarFormatType type);
+
+}

Userland/Libraries/LibUnicode/Forward.h

@@ -10,6 +10,8 @@
 
 namespace Unicode {
 
+enum class CalendarFormatType : u8;
+enum class CalendarPatternStyle : u8;
 enum class CompactNumberFormatType : u8;
 enum class Condition : u8;
 enum class GeneralCategory : u8;
@@ -24,6 +26,8 @@ enum class Style : u8;
 enum class Territory : u8;
 enum class WordBreakProperty : u8;
 
+struct CalendarFormat;
+struct CalendarPattern;
 struct CurrencyCode;
 struct Keyword;
 struct LanguageID;