ViewImplementation.cpp 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421
  1. /*
  2. * Copyright (c) 2023, Linus Groh <linusg@serenityos.org>
  3. *
  4. * SPDX-License-Identifier: BSD-2-Clause
  5. */
  6. #include <AK/Error.h>
  7. #include <AK/LexicalPath.h>
  8. #include <AK/String.h>
  9. #include <LibCore/DateTime.h>
  10. #include <LibCore/StandardPaths.h>
  11. #include <LibGfx/ImageFormats/PNGWriter.h>
  12. #include <LibWeb/Infra/Strings.h>
  13. #include <LibWebView/ViewImplementation.h>
  14. namespace WebView {
  15. ViewImplementation::ViewImplementation()
  16. {
  17. m_backing_store_shrink_timer = Core::Timer::create_single_shot(3000, [this] {
  18. resize_backing_stores_if_needed(WindowResizeInProgress::No);
  19. }).release_value_but_fixme_should_propagate_errors();
  20. m_repeated_crash_timer = Core::Timer::create_single_shot(1000, [this] {
  21. // Reset the "crashing a lot" counter after 1 second in case we just
  22. // happen to be visiting crashy websites a lot.
  23. this->m_crash_count = 0;
  24. }).release_value_but_fixme_should_propagate_errors();
  25. on_request_file = [this](auto const& path, auto request_id) {
  26. auto file = Core::File::open(path, Core::File::OpenMode::Read);
  27. if (file.is_error())
  28. client().async_handle_file_return(file.error().code(), {}, request_id);
  29. else
  30. client().async_handle_file_return(0, IPC::File(*file.value()), request_id);
  31. };
  32. }
  33. WebContentClient& ViewImplementation::client()
  34. {
  35. VERIFY(m_client_state.client);
  36. return *m_client_state.client;
  37. }
  38. WebContentClient const& ViewImplementation::client() const
  39. {
  40. VERIFY(m_client_state.client);
  41. return *m_client_state.client;
  42. }
  43. void ViewImplementation::server_did_paint(Badge<WebContentClient>, i32 bitmap_id, Gfx::IntSize size)
  44. {
  45. if (m_client_state.back_bitmap.id != bitmap_id)
  46. return;
  47. m_client_state.has_usable_bitmap = true;
  48. m_client_state.back_bitmap.pending_paints--;
  49. m_client_state.back_bitmap.last_painted_size = size;
  50. swap(m_client_state.back_bitmap, m_client_state.front_bitmap);
  51. // We don't need the backup bitmap anymore, so drop it.
  52. m_backup_bitmap = nullptr;
  53. if (on_ready_to_paint)
  54. on_ready_to_paint();
  55. if (m_client_state.got_repaint_requests_while_painting) {
  56. m_client_state.got_repaint_requests_while_painting = false;
  57. request_repaint();
  58. }
  59. }
  60. void ViewImplementation::server_did_invalidate_content_rect(Badge<WebContentClient>, Gfx::IntRect)
  61. {
  62. request_repaint();
  63. }
  64. void ViewImplementation::server_did_change_selection(Badge<WebContentClient>)
  65. {
  66. request_repaint();
  67. }
  68. void ViewImplementation::load(AK::URL const& url)
  69. {
  70. m_url = url;
  71. client().async_load_url(url);
  72. }
  73. void ViewImplementation::load_html(StringView html)
  74. {
  75. client().async_load_html(html);
  76. }
  77. void ViewImplementation::load_empty_document()
  78. {
  79. load_html(""sv);
  80. }
  81. void ViewImplementation::zoom_in()
  82. {
  83. if (m_zoom_level >= ZOOM_MAX_LEVEL)
  84. return;
  85. m_zoom_level += ZOOM_STEP;
  86. update_zoom();
  87. }
  88. void ViewImplementation::zoom_out()
  89. {
  90. if (m_zoom_level <= ZOOM_MIN_LEVEL)
  91. return;
  92. m_zoom_level -= ZOOM_STEP;
  93. update_zoom();
  94. }
  95. void ViewImplementation::reset_zoom()
  96. {
  97. m_zoom_level = 1.0f;
  98. update_zoom();
  99. }
  100. void ViewImplementation::set_preferred_color_scheme(Web::CSS::PreferredColorScheme color_scheme)
  101. {
  102. client().async_set_preferred_color_scheme(color_scheme);
  103. }
  104. DeprecatedString ViewImplementation::selected_text()
  105. {
  106. return client().get_selected_text();
  107. }
  108. Optional<String> ViewImplementation::selected_text_with_whitespace_collapsed()
  109. {
  110. auto selected_text = MUST(Web::Infra::strip_and_collapse_whitespace(this->selected_text()));
  111. if (selected_text.is_empty())
  112. return OptionalNone {};
  113. return selected_text;
  114. }
  115. void ViewImplementation::select_all()
  116. {
  117. client().async_select_all();
  118. }
  119. void ViewImplementation::get_source()
  120. {
  121. client().async_get_source();
  122. }
  123. void ViewImplementation::inspect_dom_tree()
  124. {
  125. client().async_inspect_dom_tree();
  126. }
  127. void ViewImplementation::inspect_accessibility_tree()
  128. {
  129. client().async_inspect_accessibility_tree();
  130. }
  131. ErrorOr<ViewImplementation::DOMNodeProperties> ViewImplementation::inspect_dom_node(i32 node_id, Optional<Web::CSS::Selector::PseudoElement> pseudo_element)
  132. {
  133. auto response = client().inspect_dom_node(node_id, pseudo_element);
  134. if (!response.has_style())
  135. return Error::from_string_view("Inspected node returned no style"sv);
  136. return DOMNodeProperties {
  137. .computed_style_json = TRY(String::from_deprecated_string(response.take_computed_style())),
  138. .resolved_style_json = TRY(String::from_deprecated_string(response.take_resolved_style())),
  139. .custom_properties_json = TRY(String::from_deprecated_string(response.take_custom_properties())),
  140. .node_box_sizing_json = TRY(String::from_deprecated_string(response.take_node_box_sizing())),
  141. .aria_properties_state_json = TRY(String::from_deprecated_string(response.take_aria_properties_state())),
  142. };
  143. }
  144. void ViewImplementation::clear_inspected_dom_node()
  145. {
  146. client().inspect_dom_node(0, {});
  147. }
  148. i32 ViewImplementation::get_hovered_node_id()
  149. {
  150. return client().get_hovered_node_id();
  151. }
  152. void ViewImplementation::set_dom_node_text(i32 node_id, String text)
  153. {
  154. client().async_set_dom_node_text(node_id, move(text));
  155. }
  156. Optional<i32> ViewImplementation::set_dom_node_tag(i32 node_id, String name)
  157. {
  158. return client().set_dom_node_tag(node_id, move(name));
  159. }
  160. void ViewImplementation::add_dom_node_attributes(i32 node_id, Vector<Attribute> attributes)
  161. {
  162. client().async_add_dom_node_attributes(node_id, move(attributes));
  163. }
  164. void ViewImplementation::replace_dom_node_attribute(i32 node_id, String name, Vector<Attribute> replacement_attributes)
  165. {
  166. client().async_replace_dom_node_attribute(node_id, move(name), move(replacement_attributes));
  167. }
  168. void ViewImplementation::remove_dom_node(i32 node_id)
  169. {
  170. client().async_remove_dom_node(node_id);
  171. }
  172. Optional<String> ViewImplementation::get_dom_node_html(i32 node_id)
  173. {
  174. return client().get_dom_node_html(node_id);
  175. }
  176. void ViewImplementation::debug_request(DeprecatedString const& request, DeprecatedString const& argument)
  177. {
  178. client().async_debug_request(request, argument);
  179. }
  180. void ViewImplementation::run_javascript(StringView js_source)
  181. {
  182. client().async_run_javascript(js_source);
  183. }
  184. void ViewImplementation::js_console_input(DeprecatedString const& js_source)
  185. {
  186. client().async_js_console_input(js_source);
  187. }
  188. void ViewImplementation::js_console_request_messages(i32 start_index)
  189. {
  190. client().async_js_console_request_messages(start_index);
  191. }
  192. void ViewImplementation::alert_closed()
  193. {
  194. client().async_alert_closed();
  195. }
  196. void ViewImplementation::confirm_closed(bool accepted)
  197. {
  198. client().async_confirm_closed(accepted);
  199. }
  200. void ViewImplementation::prompt_closed(Optional<String> response)
  201. {
  202. client().async_prompt_closed(move(response));
  203. }
  204. void ViewImplementation::color_picker_closed(Optional<Color> picked_color)
  205. {
  206. client().async_color_picker_closed(picked_color);
  207. }
  208. void ViewImplementation::toggle_media_play_state()
  209. {
  210. client().async_toggle_media_play_state();
  211. }
  212. void ViewImplementation::toggle_media_mute_state()
  213. {
  214. client().async_toggle_media_mute_state();
  215. }
  216. void ViewImplementation::toggle_media_loop_state()
  217. {
  218. client().async_toggle_media_loop_state();
  219. }
  220. void ViewImplementation::toggle_media_controls_state()
  221. {
  222. client().async_toggle_media_controls_state();
  223. }
  224. void ViewImplementation::handle_resize()
  225. {
  226. resize_backing_stores_if_needed(WindowResizeInProgress::Yes);
  227. m_backing_store_shrink_timer->restart();
  228. }
  229. void ViewImplementation::resize_backing_stores_if_needed(WindowResizeInProgress window_resize_in_progress)
  230. {
  231. if (m_client_state.has_usable_bitmap) {
  232. // NOTE: We keep the outgoing front bitmap as a backup so we have something to paint until we get a new one.
  233. m_backup_bitmap = m_client_state.front_bitmap.bitmap;
  234. m_backup_bitmap_size = m_client_state.front_bitmap.last_painted_size;
  235. }
  236. m_client_state.has_usable_bitmap = false;
  237. auto viewport_rect = this->viewport_rect();
  238. if (viewport_rect.is_empty())
  239. return;
  240. Gfx::IntSize minimum_needed_size;
  241. if (window_resize_in_progress == WindowResizeInProgress::Yes) {
  242. // Pad the minimum needed size so that we don't have to keep reallocating backing stores while the window is being resized.
  243. minimum_needed_size = { viewport_rect.width() + 256, viewport_rect.height() + 256 };
  244. } else {
  245. // If we're not in the middle of a resize, we can shrink the backing store size to match the viewport size.
  246. minimum_needed_size = viewport_rect.size();
  247. m_client_state.front_bitmap = {};
  248. m_client_state.back_bitmap = {};
  249. }
  250. auto reallocate_backing_store_if_needed = [&](SharedBitmap& backing_store) {
  251. if (!backing_store.bitmap || !backing_store.bitmap->size().contains(minimum_needed_size)) {
  252. if (auto new_bitmap_or_error = Gfx::Bitmap::create_shareable(Gfx::BitmapFormat::BGRA8888, minimum_needed_size); !new_bitmap_or_error.is_error()) {
  253. if (backing_store.bitmap)
  254. client().async_remove_backing_store(backing_store.id);
  255. backing_store.pending_paints = 0;
  256. backing_store.bitmap = new_bitmap_or_error.release_value();
  257. backing_store.id = m_client_state.next_bitmap_id++;
  258. client().async_add_backing_store(backing_store.id, backing_store.bitmap->to_shareable_bitmap());
  259. }
  260. backing_store.last_painted_size = viewport_rect.size();
  261. }
  262. };
  263. reallocate_backing_store_if_needed(m_client_state.front_bitmap);
  264. reallocate_backing_store_if_needed(m_client_state.back_bitmap);
  265. request_repaint();
  266. }
  267. void ViewImplementation::request_repaint()
  268. {
  269. // If this widget was instantiated but not yet added to a window,
  270. // it won't have a back bitmap yet, so we can just skip repaint requests.
  271. if (!m_client_state.back_bitmap.bitmap)
  272. return;
  273. // Don't request a repaint until pending paint requests have finished.
  274. if (m_client_state.back_bitmap.pending_paints) {
  275. m_client_state.got_repaint_requests_while_painting = true;
  276. return;
  277. }
  278. m_client_state.back_bitmap.pending_paints++;
  279. client().async_paint(viewport_rect(), m_client_state.back_bitmap.id);
  280. }
  281. void ViewImplementation::handle_web_content_process_crash()
  282. {
  283. dbgln("WebContent process crashed!");
  284. ++m_crash_count;
  285. constexpr size_t max_reasonable_crash_count = 5U;
  286. if (m_crash_count >= max_reasonable_crash_count) {
  287. dbgln("WebContent has crashed {} times in quick succession! Not restarting...", m_crash_count);
  288. m_repeated_crash_timer->stop();
  289. return;
  290. }
  291. m_repeated_crash_timer->restart();
  292. create_client();
  293. VERIFY(m_client_state.client);
  294. // Don't keep a stale backup bitmap around.
  295. m_backup_bitmap = nullptr;
  296. handle_resize();
  297. StringBuilder builder;
  298. builder.append("<html><head><title>Crashed: "sv);
  299. builder.append(escape_html_entities(m_url.to_deprecated_string()));
  300. builder.append("</title></head><body>"sv);
  301. builder.append("<h1>Web page crashed"sv);
  302. if (!m_url.host().has<Empty>()) {
  303. builder.appendff(" on {}", escape_html_entities(m_url.serialized_host().release_value_but_fixme_should_propagate_errors()));
  304. }
  305. builder.append("</h1>"sv);
  306. auto escaped_url = escape_html_entities(m_url.to_deprecated_string());
  307. builder.appendff("The web page <a href=\"{}\">{}</a> has crashed.<br><br>You can reload the page to try again.", escaped_url, escaped_url);
  308. builder.append("</body></html>"sv);
  309. load_html(builder.to_deprecated_string());
  310. }
  311. ErrorOr<void> ViewImplementation::take_screenshot(ScreenshotType type)
  312. {
  313. Gfx::ShareableBitmap bitmap;
  314. switch (type) {
  315. case ScreenshotType::Visible:
  316. if (auto* visible_bitmap = m_client_state.has_usable_bitmap ? m_client_state.front_bitmap.bitmap.ptr() : m_backup_bitmap.ptr())
  317. bitmap = visible_bitmap->to_shareable_bitmap();
  318. break;
  319. case ScreenshotType::Full:
  320. bitmap = client().take_document_screenshot();
  321. break;
  322. }
  323. if (!bitmap.is_valid())
  324. return Error::from_string_view("Failed to take a screenshot of the current tab"sv);
  325. LexicalPath path { Core::StandardPaths::downloads_directory() };
  326. path = path.append(TRY(Core::DateTime::now().to_string("screenshot-%Y-%m-%d-%H-%M-%S.png"sv)));
  327. auto encoded = TRY(Gfx::PNGWriter::encode(*bitmap.bitmap()));
  328. auto screenshot_file = TRY(Core::File::open(path.string(), Core::File::OpenMode::Write));
  329. TRY(screenshot_file->write_until_depleted(encoded));
  330. return {};
  331. }
  332. void ViewImplementation::set_user_style_sheet(String source)
  333. {
  334. client().async_set_user_style(move(source));
  335. }
  336. void ViewImplementation::use_native_user_style_sheet()
  337. {
  338. extern StringView native_stylesheet_source;
  339. set_user_style_sheet(MUST(String::from_utf8(native_stylesheet_source)));
  340. }
  341. void ViewImplementation::enable_inspector_prototype()
  342. {
  343. client().async_enable_inspector_prototype();
  344. }
  345. }