WebContentConsoleClient.cpp 7.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197
  1. /*
  2. * Copyright (c) 2021, Brandon Scott <xeon.productions@gmail.com>
  3. * Copyright (c) 2020, Hunter Salyer <thefalsehonesty@gmail.com>
  4. * Copyright (c) 2021, Sam Atkins <atkinssj@serenityos.org>
  5. *
  6. * SPDX-License-Identifier: BSD-2-Clause
  7. */
  8. #include "WebContentConsoleClient.h"
  9. #include <LibJS/Interpreter.h>
  10. #include <LibJS/MarkupGenerator.h>
  11. #include <LibJS/Script.h>
  12. #include <LibWeb/Bindings/WindowObject.h>
  13. #include <WebContent/ConsoleGlobalObject.h>
  14. namespace WebContent {
  15. WebContentConsoleClient::WebContentConsoleClient(JS::Console& console, WeakPtr<JS::Interpreter> interpreter, ClientConnection& client)
  16. : ConsoleClient(console)
  17. , m_client(client)
  18. , m_interpreter(interpreter)
  19. {
  20. JS::DeferGC defer_gc(m_interpreter->heap());
  21. auto console_global_object = m_interpreter->heap().allocate_without_global_object<ConsoleGlobalObject>(static_cast<Web::Bindings::WindowObject&>(m_interpreter->global_object()));
  22. console_global_object->initialize_global_object();
  23. m_console_global_object = JS::make_handle(console_global_object);
  24. }
  25. void WebContentConsoleClient::handle_input(String const& js_source)
  26. {
  27. auto script_or_error = JS::Script::parse(js_source, m_interpreter->realm(), "");
  28. StringBuilder output_html;
  29. auto result = JS::ThrowCompletionOr<JS::Value> { JS::js_undefined() };
  30. if (script_or_error.is_error()) {
  31. auto error = script_or_error.error()[0];
  32. auto hint = error.source_location_hint(js_source);
  33. if (!hint.is_empty())
  34. output_html.append(String::formatted("<pre>{}</pre>", escape_html_entities(hint)));
  35. result = m_interpreter->vm().throw_completion<JS::SyntaxError>(*m_console_global_object.cell(), error.to_string());
  36. } else {
  37. // FIXME: This is not the correct way to do this, we probably want to have
  38. // multiple execution contexts we switch between.
  39. auto& global_object_before = m_interpreter->realm().global_object();
  40. VERIFY(is<Web::Bindings::WindowObject>(global_object_before));
  41. auto& this_value_before = m_interpreter->realm().global_environment().global_this_value();
  42. m_interpreter->realm().set_global_object(*m_console_global_object.cell(), &global_object_before);
  43. result = m_interpreter->run(script_or_error.value());
  44. m_interpreter->realm().set_global_object(global_object_before, &this_value_before);
  45. }
  46. if (result.is_error()) {
  47. m_interpreter->vm().clear_exception();
  48. output_html.append("Uncaught exception: ");
  49. auto error = *result.throw_completion().value();
  50. if (error.is_object())
  51. output_html.append(JS::MarkupGenerator::html_from_error(error.as_object()));
  52. else
  53. output_html.append(JS::MarkupGenerator::html_from_value(error));
  54. print_html(output_html.string_view());
  55. return;
  56. }
  57. print_html(JS::MarkupGenerator::html_from_value(result.value()));
  58. }
  59. void WebContentConsoleClient::print_html(String const& line)
  60. {
  61. m_message_log.append({ .type = ConsoleOutput::Type::HTML, .data = line });
  62. m_client.async_did_output_js_console_message(m_message_log.size() - 1);
  63. }
  64. void WebContentConsoleClient::clear_output()
  65. {
  66. m_message_log.append({ .type = ConsoleOutput::Type::Clear, .data = "" });
  67. m_client.async_did_output_js_console_message(m_message_log.size() - 1);
  68. }
  69. void WebContentConsoleClient::begin_group(String const& label, bool start_expanded)
  70. {
  71. m_message_log.append({ .type = start_expanded ? ConsoleOutput::Type::BeginGroup : ConsoleOutput::Type::BeginGroupCollapsed, .data = label });
  72. m_client.async_did_output_js_console_message(m_message_log.size() - 1);
  73. }
  74. void WebContentConsoleClient::end_group()
  75. {
  76. m_message_log.append({ .type = ConsoleOutput::Type::EndGroup, .data = "" });
  77. m_client.async_did_output_js_console_message(m_message_log.size() - 1);
  78. }
  79. void WebContentConsoleClient::send_messages(i32 start_index)
  80. {
  81. // FIXME: Cap the number of messages we send at once?
  82. auto messages_to_send = m_message_log.size() - start_index;
  83. if (messages_to_send < 1) {
  84. // When the console is first created, it requests any messages that happened before
  85. // then, by requesting with start_index=0. If we don't have any messages at all, that
  86. // is still a valid request, and we can just ignore it.
  87. if (start_index != 0)
  88. m_client.did_misbehave("Requested non-existent console message index.");
  89. return;
  90. }
  91. // FIXME: Replace with a single Vector of message structs
  92. Vector<String> message_types;
  93. Vector<String> messages;
  94. message_types.ensure_capacity(messages_to_send);
  95. messages.ensure_capacity(messages_to_send);
  96. for (size_t i = start_index; i < m_message_log.size(); i++) {
  97. auto& message = m_message_log[i];
  98. switch (message.type) {
  99. case ConsoleOutput::Type::HTML:
  100. message_types.append("html");
  101. break;
  102. case ConsoleOutput::Type::Clear:
  103. message_types.append("clear");
  104. break;
  105. case ConsoleOutput::Type::BeginGroup:
  106. message_types.append("group");
  107. break;
  108. case ConsoleOutput::Type::BeginGroupCollapsed:
  109. message_types.append("groupCollapsed");
  110. break;
  111. case ConsoleOutput::Type::EndGroup:
  112. message_types.append("groupEnd");
  113. break;
  114. }
  115. messages.append(message.data);
  116. }
  117. m_client.async_did_get_js_console_messages(start_index, message_types, messages);
  118. }
  119. void WebContentConsoleClient::clear()
  120. {
  121. clear_output();
  122. }
  123. // 2.3. Printer(logLevel, args[, options]), https://console.spec.whatwg.org/#printer
  124. JS::ThrowCompletionOr<JS::Value> WebContentConsoleClient::printer(JS::Console::LogLevel log_level, PrinterArguments arguments)
  125. {
  126. if (log_level == JS::Console::LogLevel::Trace) {
  127. auto trace = arguments.get<JS::Console::Trace>();
  128. StringBuilder html;
  129. if (!trace.label.is_empty())
  130. html.appendff("<span class='title'>{}</span><br>", escape_html_entities(trace.label));
  131. html.append("<span class='trace'>");
  132. for (auto& function_name : trace.stack)
  133. html.appendff("-> {}<br>", escape_html_entities(function_name));
  134. html.append("</span>");
  135. print_html(html.string_view());
  136. return JS::js_undefined();
  137. }
  138. if (log_level == JS::Console::LogLevel::Group || log_level == JS::Console::LogLevel::GroupCollapsed) {
  139. auto group = arguments.get<JS::Console::Group>();
  140. begin_group(group.label, log_level == JS::Console::LogLevel::Group);
  141. return JS::js_undefined();
  142. }
  143. auto output = String::join(" ", arguments.get<Vector<JS::Value>>());
  144. m_console.output_debug_message(log_level, output);
  145. StringBuilder html;
  146. switch (log_level) {
  147. case JS::Console::LogLevel::Debug:
  148. html.append("<span class=\"debug\">(d) ");
  149. break;
  150. case JS::Console::LogLevel::Error:
  151. html.append("<span class=\"error\">(e) ");
  152. break;
  153. case JS::Console::LogLevel::Info:
  154. html.append("<span class=\"info\">(i) ");
  155. break;
  156. case JS::Console::LogLevel::Log:
  157. html.append("<span class=\"log\"> ");
  158. break;
  159. case JS::Console::LogLevel::Warn:
  160. case JS::Console::LogLevel::CountReset:
  161. html.append("<span class=\"warn\">(w) ");
  162. break;
  163. default:
  164. html.append("<span>");
  165. break;
  166. }
  167. html.append(escape_html_entities(output));
  168. html.append("</span>");
  169. print_html(html.string_view());
  170. return JS::js_undefined();
  171. }
  172. }