ConnectionFromClient.cpp 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719
  1. /*
  2. * Copyright (c) 2020-2022, Andreas Kling <kling@serenityos.org>
  3. * Copyright (c) 2021, Sam Atkins <atkinssj@serenityos.org>
  4. * Copyright (c) 2021-2022, Linus Groh <linusg@serenityos.org>
  5. * Copyright (c) 2022, Tobias Christiansen <tobyase@serenityos.org>
  6. * Copyright (c) 2022, Tim Flynn <trflynn89@serenityos.org>
  7. *
  8. * SPDX-License-Identifier: BSD-2-Clause
  9. */
  10. #include <AK/Debug.h>
  11. #include <AK/JsonObject.h>
  12. #include <AK/QuickSort.h>
  13. #include <LibGfx/Bitmap.h>
  14. #include <LibGfx/Font/FontDatabase.h>
  15. #include <LibGfx/SystemTheme.h>
  16. #include <LibJS/Console.h>
  17. #include <LibJS/Heap/Heap.h>
  18. #include <LibJS/Parser.h>
  19. #include <LibJS/Runtime/ConsoleObject.h>
  20. #include <LibJS/Runtime/JSONObject.h>
  21. #include <LibWeb/Bindings/MainThreadVM.h>
  22. #include <LibWeb/Cookie/ParsedCookie.h>
  23. #include <LibWeb/DOM/Document.h>
  24. #include <LibWeb/DOM/NodeList.h>
  25. #include <LibWeb/Dump.h>
  26. #include <LibWeb/Geometry/DOMRect.h>
  27. #include <LibWeb/HTML/BrowsingContext.h>
  28. #include <LibWeb/HTML/FormAssociatedElement.h>
  29. #include <LibWeb/HTML/Scripting/ClassicScript.h>
  30. #include <LibWeb/HTML/Storage.h>
  31. #include <LibWeb/HTML/Window.h>
  32. #include <LibWeb/Layout/InitialContainingBlock.h>
  33. #include <LibWeb/Loader/ContentFilter.h>
  34. #include <LibWeb/Loader/ProxyMappings.h>
  35. #include <LibWeb/Loader/ResourceLoader.h>
  36. #include <LibWeb/Painting/PaintableBox.h>
  37. #include <LibWeb/Painting/StackingContext.h>
  38. #include <LibWeb/Platform/EventLoopPlugin.h>
  39. #include <WebContent/ConnectionFromClient.h>
  40. #include <WebContent/PageHost.h>
  41. #include <WebContent/WebContentClientEndpoint.h>
  42. #include <pthread.h>
  43. namespace WebContent {
  44. ConnectionFromClient::ConnectionFromClient(NonnullOwnPtr<Core::Stream::LocalSocket> socket)
  45. : IPC::ConnectionFromClient<WebContentClientEndpoint, WebContentServerEndpoint>(*this, move(socket), 1)
  46. , m_page_host(PageHost::create(*this))
  47. {
  48. m_paint_flush_timer = Web::Platform::Timer::create_single_shot(0, [this] { flush_pending_paint_requests(); });
  49. }
  50. void ConnectionFromClient::die()
  51. {
  52. Web::Platform::EventLoopPlugin::the().quit();
  53. }
  54. Web::Page& ConnectionFromClient::page()
  55. {
  56. return m_page_host->page();
  57. }
  58. Web::Page const& ConnectionFromClient::page() const
  59. {
  60. return m_page_host->page();
  61. }
  62. void ConnectionFromClient::connect_to_webdriver(String const& webdriver_ipc_path)
  63. {
  64. // FIXME: Propagate this error back to the browser.
  65. if (auto result = m_page_host->connect_to_webdriver(webdriver_ipc_path); result.is_error())
  66. dbgln("Unable to connect to the WebDriver process: {}", result.error());
  67. }
  68. void ConnectionFromClient::update_system_theme(Core::AnonymousBuffer const& theme_buffer)
  69. {
  70. Gfx::set_system_theme(theme_buffer);
  71. auto impl = Gfx::PaletteImpl::create_with_anonymous_buffer(theme_buffer);
  72. m_page_host->set_palette_impl(*impl);
  73. }
  74. void ConnectionFromClient::update_system_fonts(String const& default_font_query, String const& fixed_width_font_query, String const& window_title_font_query)
  75. {
  76. Gfx::FontDatabase::set_default_font_query(default_font_query);
  77. Gfx::FontDatabase::set_fixed_width_font_query(fixed_width_font_query);
  78. Gfx::FontDatabase::set_window_title_font_query(window_title_font_query);
  79. }
  80. void ConnectionFromClient::update_screen_rects(Vector<Gfx::IntRect> const& rects, u32 main_screen)
  81. {
  82. m_page_host->set_screen_rects(rects, main_screen);
  83. }
  84. void ConnectionFromClient::load_url(const URL& url)
  85. {
  86. dbgln_if(SPAM_DEBUG, "handle: WebContentServer::LoadURL: url={}", url);
  87. #if defined(AK_OS_SERENITY)
  88. String process_name;
  89. if (url.host().is_empty())
  90. process_name = "WebContent";
  91. else
  92. process_name = String::formatted("WebContent: {}", url.host());
  93. pthread_setname_np(pthread_self(), process_name.characters());
  94. #endif
  95. page().load(url);
  96. }
  97. void ConnectionFromClient::load_html(String const& html, const URL& url)
  98. {
  99. dbgln_if(SPAM_DEBUG, "handle: WebContentServer::LoadHTML: html={}, url={}", html, url);
  100. page().load_html(html, url);
  101. }
  102. void ConnectionFromClient::set_viewport_rect(Gfx::IntRect const& rect)
  103. {
  104. dbgln_if(SPAM_DEBUG, "handle: WebContentServer::SetViewportRect: rect={}", rect);
  105. m_page_host->set_viewport_rect(rect);
  106. }
  107. void ConnectionFromClient::add_backing_store(i32 backing_store_id, Gfx::ShareableBitmap const& bitmap)
  108. {
  109. m_backing_stores.set(backing_store_id, *bitmap.bitmap());
  110. }
  111. void ConnectionFromClient::remove_backing_store(i32 backing_store_id)
  112. {
  113. m_backing_stores.remove(backing_store_id);
  114. m_pending_paint_requests.remove_all_matching([backing_store_id](auto& pending_repaint_request) { return pending_repaint_request.bitmap_id == backing_store_id; });
  115. }
  116. void ConnectionFromClient::paint(Gfx::IntRect const& content_rect, i32 backing_store_id)
  117. {
  118. for (auto& pending_paint : m_pending_paint_requests) {
  119. if (pending_paint.bitmap_id == backing_store_id) {
  120. pending_paint.content_rect = content_rect;
  121. return;
  122. }
  123. }
  124. auto it = m_backing_stores.find(backing_store_id);
  125. if (it == m_backing_stores.end()) {
  126. did_misbehave("Client requested paint with backing store ID");
  127. return;
  128. }
  129. auto& bitmap = *it->value;
  130. m_pending_paint_requests.append({ content_rect, bitmap, backing_store_id });
  131. m_paint_flush_timer->start();
  132. }
  133. void ConnectionFromClient::flush_pending_paint_requests()
  134. {
  135. for (auto& pending_paint : m_pending_paint_requests) {
  136. m_page_host->paint(pending_paint.content_rect, *pending_paint.bitmap);
  137. async_did_paint(pending_paint.content_rect, pending_paint.bitmap_id);
  138. }
  139. m_pending_paint_requests.clear();
  140. }
  141. void ConnectionFromClient::mouse_down(Gfx::IntPoint const& position, unsigned int button, unsigned int buttons, unsigned int modifiers)
  142. {
  143. page().handle_mousedown(position, button, buttons, modifiers);
  144. }
  145. void ConnectionFromClient::mouse_move(Gfx::IntPoint const& position, [[maybe_unused]] unsigned int button, unsigned int buttons, unsigned int modifiers)
  146. {
  147. page().handle_mousemove(position, buttons, modifiers);
  148. }
  149. void ConnectionFromClient::mouse_up(Gfx::IntPoint const& position, unsigned int button, unsigned int buttons, unsigned int modifiers)
  150. {
  151. page().handle_mouseup(position, button, buttons, modifiers);
  152. }
  153. void ConnectionFromClient::mouse_wheel(Gfx::IntPoint const& position, unsigned int button, unsigned int buttons, unsigned int modifiers, i32 wheel_delta_x, i32 wheel_delta_y)
  154. {
  155. page().handle_mousewheel(position, button, buttons, modifiers, wheel_delta_x, wheel_delta_y);
  156. }
  157. void ConnectionFromClient::doubleclick(Gfx::IntPoint const& position, unsigned int button, unsigned int buttons, unsigned int modifiers)
  158. {
  159. page().handle_doubleclick(position, button, buttons, modifiers);
  160. }
  161. void ConnectionFromClient::key_down(i32 key, unsigned int modifiers, u32 code_point)
  162. {
  163. page().handle_keydown((KeyCode)key, modifiers, code_point);
  164. }
  165. void ConnectionFromClient::key_up(i32 key, unsigned int modifiers, u32 code_point)
  166. {
  167. page().handle_keyup((KeyCode)key, modifiers, code_point);
  168. }
  169. void ConnectionFromClient::debug_request(String const& request, String const& argument)
  170. {
  171. if (request == "dump-dom-tree") {
  172. if (auto* doc = page().top_level_browsing_context().active_document())
  173. Web::dump_tree(*doc);
  174. }
  175. if (request == "dump-layout-tree") {
  176. if (auto* doc = page().top_level_browsing_context().active_document()) {
  177. if (auto* icb = doc->layout_node())
  178. Web::dump_tree(*icb);
  179. }
  180. }
  181. if (request == "dump-stacking-context-tree") {
  182. if (auto* doc = page().top_level_browsing_context().active_document()) {
  183. if (auto* icb = doc->layout_node()) {
  184. if (auto* stacking_context = icb->paint_box()->stacking_context())
  185. stacking_context->dump();
  186. }
  187. }
  188. }
  189. if (request == "dump-style-sheets") {
  190. if (auto* doc = page().top_level_browsing_context().active_document()) {
  191. for (auto& sheet : doc->style_sheets().sheets()) {
  192. Web::dump_sheet(sheet);
  193. }
  194. }
  195. }
  196. if (request == "collect-garbage") {
  197. Web::Bindings::main_thread_vm().heap().collect_garbage(JS::Heap::CollectionType::CollectGarbage, true);
  198. }
  199. if (request == "set-line-box-borders") {
  200. bool state = argument == "on";
  201. m_page_host->set_should_show_line_box_borders(state);
  202. page().top_level_browsing_context().set_needs_display(page().top_level_browsing_context().viewport_rect());
  203. }
  204. if (request == "clear-cache") {
  205. Web::ResourceLoader::the().clear_cache();
  206. }
  207. if (request == "spoof-user-agent") {
  208. Web::ResourceLoader::the().set_user_agent(argument);
  209. }
  210. if (request == "same-origin-policy") {
  211. m_page_host->page().set_same_origin_policy_enabled(argument == "on");
  212. }
  213. if (request == "scripting") {
  214. m_page_host->page().set_is_scripting_enabled(argument == "on");
  215. }
  216. if (request == "dump-local-storage") {
  217. if (auto* doc = page().top_level_browsing_context().active_document())
  218. doc->window().local_storage()->dump();
  219. }
  220. }
  221. void ConnectionFromClient::get_source()
  222. {
  223. if (auto* doc = page().top_level_browsing_context().active_document()) {
  224. async_did_get_source(doc->url(), doc->source());
  225. }
  226. }
  227. Messages::WebContentServer::SerializeSourceResponse ConnectionFromClient::serialize_source()
  228. {
  229. if (auto* doc = page().top_level_browsing_context().active_document()) {
  230. auto result = doc->serialize_fragment(Web::DOMParsing::RequireWellFormed::Yes);
  231. if (!result.is_error())
  232. return { result.release_value() };
  233. auto source = MUST(doc->serialize_fragment(Web::DOMParsing::RequireWellFormed::No));
  234. return { move(source) };
  235. }
  236. return { {} };
  237. }
  238. void ConnectionFromClient::inspect_dom_tree()
  239. {
  240. if (auto* doc = page().top_level_browsing_context().active_document()) {
  241. async_did_get_dom_tree(doc->dump_dom_tree_as_json());
  242. }
  243. }
  244. Messages::WebContentServer::InspectDomNodeResponse ConnectionFromClient::inspect_dom_node(i32 node_id, Optional<Web::CSS::Selector::PseudoElement> const& pseudo_element)
  245. {
  246. auto& top_context = page().top_level_browsing_context();
  247. top_context.for_each_in_inclusive_subtree([&](auto& ctx) {
  248. if (ctx.active_document() != nullptr) {
  249. ctx.active_document()->set_inspected_node(nullptr);
  250. }
  251. return IterationDecision::Continue;
  252. });
  253. Web::DOM::Node* node = Web::DOM::Node::from_id(node_id);
  254. // Note: Nodes without layout (aka non-visible nodes, don't have style computed)
  255. if (!node || !node->layout_node()) {
  256. return { false, "", "", "", "" };
  257. }
  258. // FIXME: Pass the pseudo-element here.
  259. node->document().set_inspected_node(node);
  260. if (node->is_element()) {
  261. auto& element = verify_cast<Web::DOM::Element>(*node);
  262. if (!element.computed_css_values())
  263. return { false, "", "", "", "" };
  264. auto serialize_json = [](Web::CSS::StyleProperties const& properties) -> String {
  265. StringBuilder builder;
  266. auto serializer = MUST(JsonObjectSerializer<>::try_create(builder));
  267. properties.for_each_property([&](auto property_id, auto& value) {
  268. MUST(serializer.add(Web::CSS::string_from_property_id(property_id), value.to_string()));
  269. });
  270. MUST(serializer.finish());
  271. return builder.to_string();
  272. };
  273. auto serialize_custom_properties_json = [](Web::DOM::Element const& element) -> String {
  274. StringBuilder builder;
  275. auto serializer = MUST(JsonObjectSerializer<>::try_create(builder));
  276. HashTable<String> seen_properties;
  277. auto const* element_to_check = &element;
  278. while (element_to_check) {
  279. for (auto const& property : element_to_check->custom_properties()) {
  280. if (!seen_properties.contains(property.key)) {
  281. seen_properties.set(property.key);
  282. MUST(serializer.add(property.key, property.value.value->to_string()));
  283. }
  284. }
  285. element_to_check = element_to_check->parent_element();
  286. }
  287. MUST(serializer.finish());
  288. return builder.to_string();
  289. };
  290. auto serialize_node_box_sizing_json = [](Web::Layout::Node const* layout_node) -> String {
  291. if (!layout_node || !layout_node->is_box()) {
  292. return "{}";
  293. }
  294. auto* box = static_cast<Web::Layout::Box const*>(layout_node);
  295. auto box_model = box->box_model();
  296. StringBuilder builder;
  297. auto serializer = MUST(JsonObjectSerializer<>::try_create(builder));
  298. MUST(serializer.add("padding_top"sv, box_model.padding.top));
  299. MUST(serializer.add("padding_right"sv, box_model.padding.right));
  300. MUST(serializer.add("padding_bottom"sv, box_model.padding.bottom));
  301. MUST(serializer.add("padding_left"sv, box_model.padding.left));
  302. MUST(serializer.add("margin_top"sv, box_model.margin.top));
  303. MUST(serializer.add("margin_right"sv, box_model.margin.right));
  304. MUST(serializer.add("margin_bottom"sv, box_model.margin.bottom));
  305. MUST(serializer.add("margin_left"sv, box_model.margin.left));
  306. MUST(serializer.add("border_top"sv, box_model.border.top));
  307. MUST(serializer.add("border_right"sv, box_model.border.right));
  308. MUST(serializer.add("border_bottom"sv, box_model.border.bottom));
  309. MUST(serializer.add("border_left"sv, box_model.border.left));
  310. if (auto* paint_box = box->paint_box()) {
  311. MUST(serializer.add("content_width"sv, paint_box->content_width()));
  312. MUST(serializer.add("content_height"sv, paint_box->content_height()));
  313. } else {
  314. MUST(serializer.add("content_width"sv, 0));
  315. MUST(serializer.add("content_height"sv, 0));
  316. }
  317. MUST(serializer.finish());
  318. return builder.to_string();
  319. };
  320. if (pseudo_element.has_value()) {
  321. auto pseudo_element_node = element.get_pseudo_element_node(pseudo_element.value());
  322. if (!pseudo_element_node)
  323. return { false, "", "", "", "" };
  324. // FIXME: Pseudo-elements only exist as Layout::Nodes, which don't have style information
  325. // in a format we can use. So, we run the StyleComputer again to get the specified
  326. // values, and have to ignore the computed values and custom properties.
  327. auto pseudo_element_style = page().focused_context().active_document()->style_computer().compute_style(element, pseudo_element);
  328. String computed_values = serialize_json(pseudo_element_style);
  329. String resolved_values = "{}";
  330. String custom_properties_json = "{}";
  331. String node_box_sizing_json = serialize_node_box_sizing_json(pseudo_element_node.ptr());
  332. return { true, computed_values, resolved_values, custom_properties_json, node_box_sizing_json };
  333. }
  334. String computed_values = serialize_json(*element.computed_css_values());
  335. String resolved_values_json = serialize_json(element.resolved_css_values());
  336. String custom_properties_json = serialize_custom_properties_json(element);
  337. String node_box_sizing_json = serialize_node_box_sizing_json(element.layout_node());
  338. return { true, computed_values, resolved_values_json, custom_properties_json, node_box_sizing_json };
  339. }
  340. return { false, "", "", "", "" };
  341. }
  342. Messages::WebContentServer::GetHoveredNodeIdResponse ConnectionFromClient::get_hovered_node_id()
  343. {
  344. if (auto* document = page().top_level_browsing_context().active_document()) {
  345. auto hovered_node = document->hovered_node();
  346. if (hovered_node)
  347. return hovered_node->id();
  348. }
  349. return (i32)0;
  350. }
  351. void ConnectionFromClient::initialize_js_console(Badge<PageHost>)
  352. {
  353. auto* document = page().top_level_browsing_context().active_document();
  354. auto realm = document->realm().make_weak_ptr();
  355. if (m_realm.ptr() == realm.ptr())
  356. return;
  357. auto& console_object = *realm->intrinsics().console_object();
  358. m_realm = realm;
  359. m_console_client = make<WebContentConsoleClient>(console_object.console(), *m_realm, *this);
  360. console_object.console().set_client(*m_console_client.ptr());
  361. }
  362. void ConnectionFromClient::js_console_input(String const& js_source)
  363. {
  364. if (m_console_client)
  365. m_console_client->handle_input(js_source);
  366. }
  367. void ConnectionFromClient::run_javascript(String const& js_source)
  368. {
  369. auto* active_document = page().top_level_browsing_context().active_document();
  370. if (!active_document)
  371. return;
  372. // This is partially based on "execute a javascript: URL request" https://html.spec.whatwg.org/multipage/browsing-the-web.html#javascript-protocol
  373. // Let settings be browsingContext's active document's relevant settings object.
  374. auto& settings = active_document->relevant_settings_object();
  375. // Let baseURL be settings's API base URL.
  376. auto base_url = settings.api_base_url();
  377. // Let script be the result of creating a classic script given scriptSource, settings, baseURL, and the default classic script fetch options.
  378. // FIXME: This doesn't pass in "default classic script fetch options"
  379. // FIXME: What should the filename be here?
  380. auto script = Web::HTML::ClassicScript::create("(client connection run_javascript)", js_source, settings, move(base_url));
  381. // Let evaluationStatus be the result of running the classic script script.
  382. auto evaluation_status = script->run();
  383. if (evaluation_status.is_error())
  384. dbgln("Exception :(");
  385. }
  386. void ConnectionFromClient::js_console_request_messages(i32 start_index)
  387. {
  388. if (m_console_client)
  389. m_console_client->send_messages(start_index);
  390. }
  391. static Optional<Web::DOM::Element&> find_element_by_id(i32 element_id)
  392. {
  393. auto* node = Web::DOM::Node::from_id(element_id);
  394. if (!node || !node->is_element())
  395. return {};
  396. return verify_cast<Web::DOM::Element>(*node);
  397. }
  398. // https://w3c.github.io/webdriver/#dfn-scrolls-into-view
  399. void ConnectionFromClient::scroll_element_into_view(i32 element_id)
  400. {
  401. auto element = find_element_by_id(element_id);
  402. if (!element.has_value())
  403. return;
  404. // 1. Let options be the following ScrollIntoViewOptions:
  405. Web::DOM::ScrollIntoViewOptions options {};
  406. // Logical scroll position "block"
  407. // "end"
  408. options.block = Web::Bindings::ScrollLogicalPosition::End;
  409. // Logical scroll position "inline"
  410. // "nearest"
  411. options.inline_ = Web::Bindings::ScrollLogicalPosition::Nearest;
  412. // 2. Run Function.[[Call]](scrollIntoView, options) with element as the this value.
  413. element->scroll_into_view(options);
  414. }
  415. // https://w3c.github.io/webdriver/#dfn-calculate-the-absolute-position
  416. static Gfx::IntPoint calculate_absolute_position_of_element(Web::Page const& page, JS::NonnullGCPtr<Web::Geometry::DOMRect> rect)
  417. {
  418. // 1. Let rect be the value returned by calling getBoundingClientRect().
  419. // 2. Let window be the associated window of current top-level browsing context.
  420. auto const* window = page.top_level_browsing_context().active_window();
  421. // 3. Let x be (scrollX of window + rect’s x coordinate).
  422. auto x = (window ? static_cast<int>(window->scroll_x()) : 0) + static_cast<int>(rect->x());
  423. // 4. Let y be (scrollY of window + rect’s y coordinate).
  424. auto y = (window ? static_cast<int>(window->scroll_y()) : 0) + static_cast<int>(rect->y());
  425. // 5. Return a pair of (x, y).
  426. return { x, y };
  427. }
  428. static Gfx::IntRect calculate_absolute_rect_of_element(Web::Page const& page, Web::DOM::Element const& element)
  429. {
  430. auto bounding_rect = element.get_bounding_client_rect();
  431. auto coordinates = calculate_absolute_position_of_element(page, bounding_rect);
  432. return Gfx::IntRect {
  433. coordinates.x(),
  434. coordinates.y(),
  435. static_cast<int>(bounding_rect->width()),
  436. static_cast<int>(bounding_rect->height())
  437. };
  438. }
  439. Messages::WebContentServer::IsElementEnabledResponse ConnectionFromClient::is_element_enabled(i32 element_id)
  440. {
  441. auto element = find_element_by_id(element_id);
  442. if (!element.has_value())
  443. return { false };
  444. auto* document = page().top_level_browsing_context().active_document();
  445. if (!document)
  446. return { false };
  447. bool enabled = !document->is_xml_document();
  448. if (enabled && is<Web::HTML::FormAssociatedElement>(*element)) {
  449. auto& form_associated_element = dynamic_cast<Web::HTML::FormAssociatedElement&>(*element);
  450. enabled = form_associated_element.enabled();
  451. }
  452. return { enabled };
  453. }
  454. Messages::WebContentServer::TakeElementScreenshotResponse ConnectionFromClient::take_element_screenshot(i32 element_id)
  455. {
  456. auto element = find_element_by_id(element_id);
  457. if (!element.has_value())
  458. return { {} };
  459. auto viewport_rect = page().top_level_browsing_context().viewport_rect();
  460. auto rect = calculate_absolute_rect_of_element(page(), *element);
  461. rect.intersect(viewport_rect);
  462. auto bitmap = Gfx::Bitmap::try_create(Gfx::BitmapFormat::BGRA8888, rect.size()).release_value_but_fixme_should_propagate_errors();
  463. m_page_host->paint(rect, *bitmap);
  464. return { bitmap->to_shareable_bitmap() };
  465. }
  466. Messages::WebContentServer::TakeDocumentScreenshotResponse ConnectionFromClient::take_document_screenshot()
  467. {
  468. auto* document = page().top_level_browsing_context().active_document();
  469. if (!document || !document->document_element())
  470. return { {} };
  471. auto bounding_rect = document->document_element()->get_bounding_client_rect();
  472. auto position = calculate_absolute_position_of_element(page(), bounding_rect);
  473. auto const& content_size = m_page_host->content_size();
  474. Gfx::IntRect rect {
  475. position.x(),
  476. position.y(),
  477. content_size.width() - position.x(),
  478. content_size.height() - position.y(),
  479. };
  480. auto bitmap = Gfx::Bitmap::try_create(Gfx::BitmapFormat::BGRA8888, rect.size()).release_value_but_fixme_should_propagate_errors();
  481. m_page_host->paint(rect, *bitmap);
  482. return { bitmap->to_shareable_bitmap() };
  483. }
  484. Messages::WebContentServer::GetSelectedTextResponse ConnectionFromClient::get_selected_text()
  485. {
  486. return page().focused_context().selected_text();
  487. }
  488. void ConnectionFromClient::select_all()
  489. {
  490. page().focused_context().select_all();
  491. page().client().page_did_change_selection();
  492. }
  493. Messages::WebContentServer::DumpLayoutTreeResponse ConnectionFromClient::dump_layout_tree()
  494. {
  495. auto* document = page().top_level_browsing_context().active_document();
  496. if (!document)
  497. return String { "(no DOM tree)" };
  498. auto* layout_root = document->layout_node();
  499. if (!layout_root)
  500. return String { "(no layout tree)" };
  501. StringBuilder builder;
  502. Web::dump_tree(builder, *layout_root);
  503. return builder.to_string();
  504. }
  505. void ConnectionFromClient::set_content_filters(Vector<String> const& filters)
  506. {
  507. for (auto& filter : filters)
  508. Web::ContentFilter::the().add_pattern(filter);
  509. }
  510. void ConnectionFromClient::set_proxy_mappings(Vector<String> const& proxies, HashMap<String, size_t> const& mappings)
  511. {
  512. auto keys = mappings.keys();
  513. quick_sort(keys, [&](auto& a, auto& b) { return a.length() < b.length(); });
  514. OrderedHashMap<String, size_t> sorted_mappings;
  515. for (auto& key : keys) {
  516. auto value = *mappings.get(key);
  517. if (value >= proxies.size())
  518. continue;
  519. sorted_mappings.set(key, value);
  520. }
  521. Web::ProxyMappings::the().set_mappings(proxies, move(sorted_mappings));
  522. }
  523. void ConnectionFromClient::set_preferred_color_scheme(Web::CSS::PreferredColorScheme const& color_scheme)
  524. {
  525. m_page_host->set_preferred_color_scheme(color_scheme);
  526. }
  527. void ConnectionFromClient::set_has_focus(bool has_focus)
  528. {
  529. m_page_host->set_has_focus(has_focus);
  530. }
  531. void ConnectionFromClient::set_is_scripting_enabled(bool is_scripting_enabled)
  532. {
  533. m_page_host->set_is_scripting_enabled(is_scripting_enabled);
  534. }
  535. void ConnectionFromClient::set_window_position(Gfx::IntPoint const& position)
  536. {
  537. m_page_host->set_window_position(position);
  538. }
  539. void ConnectionFromClient::set_window_size(Gfx::IntSize const& size)
  540. {
  541. m_page_host->set_window_size(size);
  542. }
  543. Messages::WebContentServer::GetLocalStorageEntriesResponse ConnectionFromClient::get_local_storage_entries()
  544. {
  545. auto* document = page().top_level_browsing_context().active_document();
  546. auto local_storage = document->window().local_storage();
  547. return local_storage->map();
  548. }
  549. Messages::WebContentServer::GetSessionStorageEntriesResponse ConnectionFromClient::get_session_storage_entries()
  550. {
  551. auto* document = page().top_level_browsing_context().active_document();
  552. auto session_storage = document->window().session_storage();
  553. return session_storage->map();
  554. }
  555. void ConnectionFromClient::handle_file_return(i32 error, Optional<IPC::File> const& file, i32 request_id)
  556. {
  557. auto result = m_requested_files.get(request_id);
  558. VERIFY(result.has_value());
  559. VERIFY(result.value()->on_file_request_finish);
  560. result.value()->on_file_request_finish(error != 0 ? Error::from_errno(error) : ErrorOr<i32> { file->take_fd() });
  561. m_requested_files.remove(request_id);
  562. }
  563. void ConnectionFromClient::request_file(NonnullRefPtr<Web::FileRequest>& file_request)
  564. {
  565. i32 const id = last_id++;
  566. m_requested_files.set(id, file_request);
  567. async_did_request_file(file_request->path(), id);
  568. }
  569. void ConnectionFromClient::set_system_visibility_state(bool visible)
  570. {
  571. m_page_host->page().top_level_browsing_context().set_system_visibility_state(
  572. visible
  573. ? Web::HTML::VisibilityState::Visible
  574. : Web::HTML::VisibilityState::Hidden);
  575. }
  576. Messages::WebContentServer::WebdriverExecuteScriptResponse ConnectionFromClient::webdriver_execute_script(String const& body, Vector<String> const& json_arguments, Optional<u64> const& timeout, bool async)
  577. {
  578. auto& page = m_page_host->page();
  579. auto* window = page.top_level_browsing_context().active_window();
  580. auto& vm = window->vm();
  581. auto arguments = JS::MarkedVector<JS::Value> { vm.heap() };
  582. for (auto const& argument_string : json_arguments) {
  583. // NOTE: These are assumed to be valid JSON values.
  584. auto json_value = MUST(JsonValue::from_string(argument_string));
  585. arguments.append(JS::JSONObject::parse_json_value(vm, json_value));
  586. }
  587. auto result = async
  588. ? Web::WebDriver::execute_async_script(page, body, move(arguments), timeout)
  589. : Web::WebDriver::execute_script(page, body, move(arguments), timeout);
  590. return { result.type, result.value.serialized<StringBuilder>() };
  591. }
  592. }