GenerateCSSTransformFunctions.cpp 4.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137
  1. /*
  2. * Copyright (c) 2022, Sam Atkins <atkinssj@serenityos.org>
  3. *
  4. * SPDX-License-Identifier: BSD-2-Clause
  5. */
  6. #include "GeneratorUtil.h"
  7. #include <AK/SourceGenerator.h>
  8. #include <AK/StringBuilder.h>
  9. #include <LibCore/ArgsParser.h>
  10. #include <LibMain/Main.h>
  11. ErrorOr<void> generate_header_file(JsonObject& transforms_data, Core::Stream::File& file);
  12. ErrorOr<void> generate_implementation_file(JsonObject& transforms_data, Core::Stream::File& file);
  13. ErrorOr<int> serenity_main(Main::Arguments arguments)
  14. {
  15. StringView generated_header_path;
  16. StringView generated_implementation_path;
  17. StringView identifiers_json_path;
  18. Core::ArgsParser args_parser;
  19. args_parser.add_option(generated_header_path, "Path to the TransformFunctions header file to generate", "generated-header-path", 'h', "generated-header-path");
  20. args_parser.add_option(generated_implementation_path, "Path to the TransformFunctions implementation file to generate", "generated-implementation-path", 'c', "generated-implementation-path");
  21. args_parser.add_option(identifiers_json_path, "Path to the JSON file to read from", "json-path", 'j', "json-path");
  22. args_parser.parse(arguments);
  23. auto json = TRY(read_entire_file_as_json(identifiers_json_path));
  24. VERIFY(json.is_object());
  25. auto transforms_data = json.as_object();
  26. auto generated_header_file = TRY(Core::Stream::File::open(generated_header_path, Core::Stream::OpenMode::Write));
  27. auto generated_implementation_file = TRY(Core::Stream::File::open(generated_implementation_path, Core::Stream::OpenMode::Write));
  28. TRY(generate_header_file(transforms_data, *generated_header_file));
  29. TRY(generate_implementation_file(transforms_data, *generated_implementation_file));
  30. return 0;
  31. }
  32. static String title_casify_transform_function(StringView input)
  33. {
  34. // Transform function names look like `fooBar`, so we just have to make the first character uppercase.
  35. StringBuilder builder;
  36. builder.append(toupper(input[0]));
  37. builder.append(input.substring_view(1));
  38. return builder.to_string();
  39. }
  40. ErrorOr<void> generate_header_file(JsonObject& transforms_data, Core::Stream::File& file)
  41. {
  42. StringBuilder builder;
  43. SourceGenerator generator { builder };
  44. generator.append(R"~~~(
  45. #pragma once
  46. #include <AK/Optional.h>
  47. #include <AK/StringView.h>
  48. namespace Web::CSS {
  49. )~~~");
  50. generator.appendln("enum class TransformFunction {");
  51. transforms_data.for_each_member([&](auto& name, auto&) {
  52. auto member_generator = generator.fork();
  53. member_generator.set("name:titlecase", title_casify_transform_function(name));
  54. member_generator.appendln(" @name:titlecase@,");
  55. });
  56. generator.appendln("};");
  57. generator.appendln("Optional<TransformFunction> transform_function_from_string(StringView);");
  58. generator.appendln("StringView to_string(TransformFunction);");
  59. generator.appendln("\n}");
  60. TRY(file.write(generator.as_string_view().bytes()));
  61. return {};
  62. }
  63. ErrorOr<void> generate_implementation_file(JsonObject& transforms_data, Core::Stream::File& file)
  64. {
  65. StringBuilder builder;
  66. SourceGenerator generator { builder };
  67. generator.append(R"~~~(
  68. #include <LibWeb/CSS/TransformFunctions.h>
  69. #include <AK/Assertions.h>
  70. namespace Web::CSS {
  71. )~~~");
  72. generator.append(R"~~~(
  73. Optional<TransformFunction> transform_function_from_string(StringView name)
  74. {
  75. )~~~");
  76. transforms_data.for_each_member([&](auto& name, auto&) {
  77. auto member_generator = generator.fork();
  78. member_generator.set("name", name);
  79. member_generator.set("name:titlecase", title_casify_transform_function(name));
  80. member_generator.append(R"~~~(
  81. if (name.equals_ignoring_case("@name@"sv))
  82. return TransformFunction::@name:titlecase@;
  83. )~~~");
  84. });
  85. generator.append(R"~~~(
  86. return {};
  87. }
  88. )~~~");
  89. generator.append(R"~~~(
  90. StringView to_string(TransformFunction transform_function)
  91. {
  92. switch (transform_function) {
  93. )~~~");
  94. transforms_data.for_each_member([&](auto& name, auto&) {
  95. auto member_generator = generator.fork();
  96. member_generator.set("name", name);
  97. member_generator.set("name:titlecase", title_casify_transform_function(name));
  98. member_generator.append(R"~~~(
  99. case TransformFunction::@name:titlecase@:
  100. return "@name@"sv;
  101. )~~~");
  102. });
  103. generator.append(R"~~~(
  104. default:
  105. VERIFY_NOT_REACHED();
  106. }
  107. }
  108. )~~~");
  109. generator.appendln("\n}");
  110. TRY(file.write(generator.as_string_view().bytes()));
  111. return {};
  112. }