|
@@ -1,782 +1,203 @@
|
|
|
/*
|
|
|
* Copyright (c) 2022, Dex♪ <dexes.ttp@gmail.com>
|
|
|
+ * Copyright (c) 2023, Tim Flynn <trflynn89@serenityos.org>
|
|
|
*
|
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
|
*/
|
|
|
|
|
|
-#include <AK/Assertions.h>
|
|
|
-#include <AK/ByteBuffer.h>
|
|
|
-#include <AK/Format.h>
|
|
|
-#include <AK/HashTable.h>
|
|
|
+#include <AK/Platform.h>
|
|
|
+
|
|
|
+#if !defined(AK_OS_SERENITY)
|
|
|
+# define AK_DONT_REPLACE_STD
|
|
|
+#endif
|
|
|
+
|
|
|
+#include <AK/Badge.h>
|
|
|
+#include <AK/DeprecatedString.h>
|
|
|
#include <AK/LexicalPath.h>
|
|
|
#include <AK/NonnullOwnPtr.h>
|
|
|
-#include <AK/StringBuilder.h>
|
|
|
-#include <AK/Types.h>
|
|
|
+#include <AK/URL.h>
|
|
|
+#include <AK/Vector.h>
|
|
|
#include <LibCore/ArgsParser.h>
|
|
|
-#include <LibCore/ConfigFile.h>
|
|
|
#include <LibCore/DeprecatedFile.h>
|
|
|
#include <LibCore/EventLoop.h>
|
|
|
-#include <LibCore/System.h>
|
|
|
-#include <LibCore/SystemServerTakeover.h>
|
|
|
#include <LibCore/Timer.h>
|
|
|
-#include <LibGemini/GeminiRequest.h>
|
|
|
-#include <LibGemini/GeminiResponse.h>
|
|
|
-#include <LibGemini/Job.h>
|
|
|
#include <LibGfx/Bitmap.h>
|
|
|
#include <LibGfx/Font/FontDatabase.h>
|
|
|
-#include <LibGfx/ImageDecoder.h>
|
|
|
#include <LibGfx/PNGWriter.h>
|
|
|
+#include <LibGfx/Point.h>
|
|
|
#include <LibGfx/Rect.h>
|
|
|
-#include <LibHTTP/HttpRequest.h>
|
|
|
-#include <LibHTTP/HttpResponse.h>
|
|
|
-#include <LibHTTP/HttpsJob.h>
|
|
|
-#include <LibHTTP/Job.h>
|
|
|
-#include <LibMain/Main.h>
|
|
|
+#include <LibGfx/ShareableBitmap.h>
|
|
|
+#include <LibGfx/Size.h>
|
|
|
+#include <LibGfx/StandardCursor.h>
|
|
|
+#include <LibGfx/SystemTheme.h>
|
|
|
+#include <LibWeb/Cookie/Cookie.h>
|
|
|
#include <LibWeb/Cookie/ParsedCookie.h>
|
|
|
-#include <LibWeb/DOM/Document.h>
|
|
|
-#include <LibWeb/HTML/BrowsingContext.h>
|
|
|
-#include <LibWeb/Layout/Viewport.h>
|
|
|
-#include <LibWeb/Loader/ResourceLoader.h>
|
|
|
-#include <LibWeb/Page/Page.h>
|
|
|
-#include <LibWeb/Painting/PaintableBox.h>
|
|
|
-#include <LibWeb/Platform/EventLoopPluginSerenity.h>
|
|
|
-#include <LibWeb/Platform/FontPluginSerenity.h>
|
|
|
-#include <LibWeb/Platform/ImageCodecPlugin.h>
|
|
|
-#include <LibWeb/WebSockets/WebSocket.h>
|
|
|
-#include <LibWebSocket/ConnectionInfo.h>
|
|
|
-#include <LibWebSocket/Message.h>
|
|
|
-#include <LibWebSocket/WebSocket.h>
|
|
|
-#include <WebContent/WebDriverConnection.h>
|
|
|
-
|
|
|
-class HeadlessBrowserPageClient final : public Web::PageClient {
|
|
|
-public:
|
|
|
- static NonnullOwnPtr<HeadlessBrowserPageClient> create()
|
|
|
- {
|
|
|
- return adopt_own(*new HeadlessBrowserPageClient());
|
|
|
- }
|
|
|
-
|
|
|
- virtual Web::Page& page() override { return *m_page; }
|
|
|
- virtual Web::Page const& page() const override { return *m_page; }
|
|
|
-
|
|
|
- Web::Layout::Viewport* layout_root()
|
|
|
- {
|
|
|
- auto* document = page().top_level_browsing_context().active_document();
|
|
|
- if (!document)
|
|
|
- return nullptr;
|
|
|
- return document->layout_node();
|
|
|
- }
|
|
|
-
|
|
|
- void load(AK::URL const& url)
|
|
|
- {
|
|
|
- page().load(url);
|
|
|
- }
|
|
|
-
|
|
|
- virtual void paint(Web::DevicePixelRect const& content_rect, Gfx::Bitmap& target) override
|
|
|
- {
|
|
|
- Gfx::Painter painter(target);
|
|
|
-
|
|
|
- if (auto* document = page().top_level_browsing_context().active_document())
|
|
|
- document->update_layout();
|
|
|
-
|
|
|
- painter.fill_rect({ {}, content_rect.size().to_type<int>() }, palette().base());
|
|
|
-
|
|
|
- auto* layout_root = this->layout_root();
|
|
|
- if (!layout_root) {
|
|
|
- return;
|
|
|
- }
|
|
|
-
|
|
|
- Web::PaintContext context(painter, palette(), device_pixels_per_css_pixel());
|
|
|
- context.set_should_show_line_box_borders(false);
|
|
|
- context.set_device_viewport_rect(content_rect);
|
|
|
- context.set_has_focus(true);
|
|
|
- layout_root->paint_all_phases(context);
|
|
|
- }
|
|
|
-
|
|
|
- void setup_palette(Core::AnonymousBuffer theme_buffer)
|
|
|
- {
|
|
|
- m_palette_impl = Gfx::PaletteImpl::create_with_anonymous_buffer(theme_buffer);
|
|
|
- }
|
|
|
-
|
|
|
- void set_viewport_rect(Gfx::IntRect viewport_rect)
|
|
|
- {
|
|
|
- page().top_level_browsing_context().set_viewport_rect(page().device_to_css_rect(viewport_rect.to_type<Web::DevicePixels>()));
|
|
|
- }
|
|
|
-
|
|
|
- void set_screen_rect(Web::DevicePixelRect screen_rect)
|
|
|
- {
|
|
|
- m_screen_rect = screen_rect;
|
|
|
- }
|
|
|
-
|
|
|
- ErrorOr<void> connect_to_webdriver(StringView webdriver_ipc_path)
|
|
|
- {
|
|
|
- VERIFY(!m_webdriver);
|
|
|
- m_webdriver = TRY(WebContent::WebDriverConnection::connect(*this, webdriver_ipc_path));
|
|
|
- return {};
|
|
|
- }
|
|
|
-
|
|
|
- // ^Web::PageClient
|
|
|
- virtual bool is_connection_open() const override
|
|
|
- {
|
|
|
- if (m_webdriver)
|
|
|
- return m_webdriver->is_open();
|
|
|
- return true;
|
|
|
- }
|
|
|
-
|
|
|
- virtual Gfx::Palette palette() const override
|
|
|
- {
|
|
|
- return Gfx::Palette(*m_palette_impl);
|
|
|
- }
|
|
|
-
|
|
|
- virtual Web::DevicePixelRect screen_rect() const override
|
|
|
- {
|
|
|
- return m_screen_rect;
|
|
|
- }
|
|
|
-
|
|
|
- virtual float device_pixels_per_css_pixel() const override
|
|
|
- {
|
|
|
- return 1.0f;
|
|
|
- }
|
|
|
-
|
|
|
- virtual Web::CSS::PreferredColorScheme preferred_color_scheme() const override
|
|
|
- {
|
|
|
- return m_preferred_color_scheme;
|
|
|
- }
|
|
|
-
|
|
|
- virtual void page_did_change_title(DeprecatedString const&) override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual void page_did_start_loading(AK::URL const&, bool) override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual void page_did_finish_loading(AK::URL const&) override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual void page_did_change_selection() override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual void page_did_request_cursor_change(Gfx::StandardCursor) override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual void page_did_request_context_menu(Web::CSSPixelPoint) override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual void page_did_request_link_context_menu(Web::CSSPixelPoint, AK::URL const&, DeprecatedString const&, unsigned) override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual void page_did_request_image_context_menu(Web::CSSPixelPoint, AK::URL const&, DeprecatedString const&, unsigned, Gfx::Bitmap const*) override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual void page_did_click_link(AK::URL const&, DeprecatedString const&, unsigned) override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual void page_did_middle_click_link(AK::URL const&, DeprecatedString const&, unsigned) override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual void page_did_enter_tooltip_area(Web::CSSPixelPoint, DeprecatedString const&) override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual void page_did_leave_tooltip_area() override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual void page_did_hover_link(AK::URL const&) override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual void page_did_unhover_link() override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual void page_did_invalidate(Web::CSSPixelRect const&) override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual void page_did_change_favicon(Gfx::Bitmap const&) override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual void page_did_layout() override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual void page_did_request_scroll_into_view(Web::CSSPixelRect const&) override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual void page_did_request_alert(DeprecatedString const&) override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual void page_did_request_confirm(DeprecatedString const&) override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual void page_did_request_prompt(DeprecatedString const&, DeprecatedString const&) override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual DeprecatedString page_did_request_cookie(AK::URL const&, Web::Cookie::Source) override
|
|
|
- {
|
|
|
- return DeprecatedString::empty();
|
|
|
- }
|
|
|
-
|
|
|
- virtual void page_did_set_cookie(AK::URL const&, Web::Cookie::ParsedCookie const&, Web::Cookie::Source) override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- void request_file(Web::FileRequest request) override
|
|
|
- {
|
|
|
- auto file = Core::System::open(request.path(), O_RDONLY);
|
|
|
- request.on_file_request_finish(move(file));
|
|
|
- }
|
|
|
-
|
|
|
-private:
|
|
|
- HeadlessBrowserPageClient()
|
|
|
- : m_page(make<Web::Page>(*this))
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- NonnullOwnPtr<Web::Page> m_page;
|
|
|
-
|
|
|
- RefPtr<Gfx::PaletteImpl> m_palette_impl;
|
|
|
- Web::DevicePixelRect m_screen_rect { 0, 0, 800, 600 };
|
|
|
- Web::CSS::PreferredColorScheme m_preferred_color_scheme { Web::CSS::PreferredColorScheme::Auto };
|
|
|
+#include <LibWeb/Loader/FrameLoader.h>
|
|
|
+#include <LibWebView/ViewImplementation.h>
|
|
|
+#include <LibWebView/WebContentClient.h>
|
|
|
|
|
|
- RefPtr<WebContent::WebDriverConnection> m_webdriver;
|
|
|
-};
|
|
|
+#if !defined(AK_OS_SERENITY)
|
|
|
+# include <Ladybird/HelperProcess.h>
|
|
|
+# include <QCoreApplication>
|
|
|
+#endif
|
|
|
|
|
|
-class ImageCodecPluginHeadless : public Web::Platform::ImageCodecPlugin {
|
|
|
+class HeadlessWebContentView final : public WebView::ViewImplementation {
|
|
|
public:
|
|
|
- ImageCodecPluginHeadless() = default;
|
|
|
- virtual ~ImageCodecPluginHeadless() override = default;
|
|
|
-
|
|
|
- virtual Optional<Web::Platform::DecodedImage> decode_image(ReadonlyBytes data) override
|
|
|
- {
|
|
|
- auto decoder = Gfx::ImageDecoder::try_create_for_raw_bytes(data);
|
|
|
-
|
|
|
- if (!decoder)
|
|
|
- return Web::Platform::DecodedImage { false, 0, Vector<Web::Platform::Frame> {} };
|
|
|
-
|
|
|
- if (!decoder->frame_count())
|
|
|
- return Web::Platform::DecodedImage { false, 0, Vector<Web::Platform::Frame> {} };
|
|
|
-
|
|
|
- Vector<Web::Platform::Frame> frames;
|
|
|
- for (size_t i = 0; i < decoder->frame_count(); ++i) {
|
|
|
- auto frame_or_error = decoder->frame(i);
|
|
|
- if (frame_or_error.is_error()) {
|
|
|
- frames.append({ {}, 0 });
|
|
|
- } else {
|
|
|
- auto frame = frame_or_error.release_value();
|
|
|
- frames.append({ move(frame.image), static_cast<size_t>(frame.duration) });
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- return Web::Platform::DecodedImage {
|
|
|
- decoder->is_animated(),
|
|
|
- static_cast<u32>(decoder->loop_count()),
|
|
|
- frames,
|
|
|
- };
|
|
|
- }
|
|
|
-};
|
|
|
-
|
|
|
-static HashTable<RefPtr<Web::ResourceLoaderConnectorRequest>> s_all_requests;
|
|
|
-
|
|
|
-class HeadlessRequestServer : public Web::ResourceLoaderConnector {
|
|
|
-public:
|
|
|
- class HTTPHeadlessRequest
|
|
|
- : public Web::ResourceLoaderConnectorRequest
|
|
|
- , public Weakable<HTTPHeadlessRequest> {
|
|
|
- public:
|
|
|
- static ErrorOr<NonnullRefPtr<HTTPHeadlessRequest>> create(DeprecatedString const& method, AK::URL const& url, HashMap<DeprecatedString, DeprecatedString> const& request_headers, ReadonlyBytes request_body, Core::ProxyData const&)
|
|
|
- {
|
|
|
- auto stream_backing_buffer = TRY(ByteBuffer::create_uninitialized(1 * MiB));
|
|
|
- auto underlying_socket = TRY(Core::TCPSocket::connect(url.host(), url.port().value_or(80)));
|
|
|
- TRY(underlying_socket->set_blocking(false));
|
|
|
- auto socket = TRY(Core::BufferedSocket<Core::TCPSocket>::create(move(underlying_socket)));
|
|
|
-
|
|
|
- HTTP::HttpRequest request;
|
|
|
- if (method.equals_ignoring_ascii_case("head"sv))
|
|
|
- request.set_method(HTTP::HttpRequest::HEAD);
|
|
|
- else if (method.equals_ignoring_ascii_case("get"sv))
|
|
|
- request.set_method(HTTP::HttpRequest::GET);
|
|
|
- else if (method.equals_ignoring_ascii_case("post"sv))
|
|
|
- request.set_method(HTTP::HttpRequest::POST);
|
|
|
- else
|
|
|
- request.set_method(HTTP::HttpRequest::Invalid);
|
|
|
- request.set_url(move(url));
|
|
|
- request.set_headers(request_headers);
|
|
|
- request.set_body(TRY(ByteBuffer::copy(request_body)));
|
|
|
-
|
|
|
- return adopt_ref(*new HTTPHeadlessRequest(move(request), move(socket), move(stream_backing_buffer)));
|
|
|
- }
|
|
|
-
|
|
|
- virtual ~HTTPHeadlessRequest() override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual void set_should_buffer_all_input(bool) override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual bool stop() override
|
|
|
- {
|
|
|
- return false;
|
|
|
- }
|
|
|
-
|
|
|
- virtual void stream_into(Stream&) override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- private:
|
|
|
- HTTPHeadlessRequest(HTTP::HttpRequest&& request, NonnullOwnPtr<Core::BufferedSocketBase> socket, ByteBuffer&& stream_backing_buffer)
|
|
|
- : m_stream_backing_buffer(move(stream_backing_buffer))
|
|
|
- , m_output_stream(try_make<FixedMemoryStream>(m_stream_backing_buffer.bytes()).release_value_but_fixme_should_propagate_errors())
|
|
|
- , m_socket(move(socket))
|
|
|
- , m_job(HTTP::Job::construct(move(request), *m_output_stream))
|
|
|
- {
|
|
|
- m_job->on_headers_received = [weak_this = make_weak_ptr()](auto& response_headers, auto response_code) {
|
|
|
- if (auto strong_this = weak_this.strong_ref()) {
|
|
|
- strong_this->m_response_code = response_code;
|
|
|
- for (auto& header : response_headers) {
|
|
|
- strong_this->m_response_headers.set(header.key, header.value);
|
|
|
- }
|
|
|
- }
|
|
|
- };
|
|
|
- m_job->on_finish = [weak_this = make_weak_ptr()](bool success) {
|
|
|
- Core::deferred_invoke([weak_this, success] {
|
|
|
- if (auto strong_this = weak_this.strong_ref()) {
|
|
|
- ReadonlyBytes response_bytes { strong_this->m_output_stream->bytes().data(), strong_this->m_output_stream->offset() };
|
|
|
- auto response_buffer = ByteBuffer::copy(response_bytes).release_value_but_fixme_should_propagate_errors();
|
|
|
- strong_this->on_buffered_request_finish(success, strong_this->m_output_stream->offset(), strong_this->m_response_headers, strong_this->m_response_code, response_buffer);
|
|
|
- }
|
|
|
- });
|
|
|
- };
|
|
|
- m_job->start(*m_socket);
|
|
|
- }
|
|
|
-
|
|
|
- Optional<u32> m_response_code;
|
|
|
- ByteBuffer m_stream_backing_buffer;
|
|
|
- NonnullOwnPtr<FixedMemoryStream> m_output_stream;
|
|
|
- NonnullOwnPtr<Core::BufferedSocketBase> m_socket;
|
|
|
- NonnullRefPtr<HTTP::Job> m_job;
|
|
|
- HashMap<DeprecatedString, DeprecatedString, CaseInsensitiveStringTraits> m_response_headers;
|
|
|
- };
|
|
|
-
|
|
|
- class HTTPSHeadlessRequest
|
|
|
- : public Web::ResourceLoaderConnectorRequest
|
|
|
- , public Weakable<HTTPSHeadlessRequest> {
|
|
|
- public:
|
|
|
- static ErrorOr<NonnullRefPtr<HTTPSHeadlessRequest>> create(DeprecatedString const& method, AK::URL const& url, HashMap<DeprecatedString, DeprecatedString> const& request_headers, ReadonlyBytes request_body, Core::ProxyData const&)
|
|
|
- {
|
|
|
- auto stream_backing_buffer = TRY(ByteBuffer::create_uninitialized(1 * MiB));
|
|
|
- auto underlying_socket = TRY(TLS::TLSv12::connect(url.host(), url.port().value_or(443)));
|
|
|
- TRY(underlying_socket->set_blocking(false));
|
|
|
- auto socket = TRY(Core::BufferedSocket<TLS::TLSv12>::create(move(underlying_socket)));
|
|
|
-
|
|
|
- HTTP::HttpRequest request;
|
|
|
- if (method.equals_ignoring_ascii_case("head"sv))
|
|
|
- request.set_method(HTTP::HttpRequest::HEAD);
|
|
|
- else if (method.equals_ignoring_ascii_case("get"sv))
|
|
|
- request.set_method(HTTP::HttpRequest::GET);
|
|
|
- else if (method.equals_ignoring_ascii_case("post"sv))
|
|
|
- request.set_method(HTTP::HttpRequest::POST);
|
|
|
- else
|
|
|
- request.set_method(HTTP::HttpRequest::Invalid);
|
|
|
- request.set_url(move(url));
|
|
|
- request.set_headers(request_headers);
|
|
|
- request.set_body(TRY(ByteBuffer::copy(request_body)));
|
|
|
-
|
|
|
- return adopt_ref(*new HTTPSHeadlessRequest(move(request), move(socket), move(stream_backing_buffer)));
|
|
|
- }
|
|
|
-
|
|
|
- virtual ~HTTPSHeadlessRequest() override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual void set_should_buffer_all_input(bool) override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual bool stop() override
|
|
|
- {
|
|
|
- return false;
|
|
|
- }
|
|
|
-
|
|
|
- virtual void stream_into(Stream&) override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- private:
|
|
|
- HTTPSHeadlessRequest(HTTP::HttpRequest&& request, NonnullOwnPtr<Core::BufferedSocketBase> socket, ByteBuffer&& stream_backing_buffer)
|
|
|
- : m_stream_backing_buffer(move(stream_backing_buffer))
|
|
|
- , m_output_stream(try_make<FixedMemoryStream>(m_stream_backing_buffer.bytes()).release_value_but_fixme_should_propagate_errors())
|
|
|
- , m_socket(move(socket))
|
|
|
- , m_job(HTTP::HttpsJob::construct(move(request), *m_output_stream))
|
|
|
- {
|
|
|
- m_job->on_headers_received = [weak_this = make_weak_ptr()](auto& response_headers, auto response_code) {
|
|
|
- if (auto strong_this = weak_this.strong_ref()) {
|
|
|
- strong_this->m_response_code = response_code;
|
|
|
- for (auto& header : response_headers) {
|
|
|
- strong_this->m_response_headers.set(header.key, header.value);
|
|
|
- }
|
|
|
- }
|
|
|
- };
|
|
|
- m_job->on_finish = [weak_this = make_weak_ptr()](bool success) {
|
|
|
- Core::deferred_invoke([weak_this, success] {
|
|
|
- if (auto strong_this = weak_this.strong_ref()) {
|
|
|
- ReadonlyBytes response_bytes { strong_this->m_output_stream->bytes().data(), strong_this->m_output_stream->offset() };
|
|
|
- auto response_buffer = ByteBuffer::copy(response_bytes).release_value_but_fixme_should_propagate_errors();
|
|
|
- strong_this->on_buffered_request_finish(success, strong_this->m_output_stream->offset(), strong_this->m_response_headers, strong_this->m_response_code, response_buffer);
|
|
|
- }
|
|
|
- });
|
|
|
- };
|
|
|
- m_job->start(*m_socket);
|
|
|
- }
|
|
|
-
|
|
|
- Optional<u32> m_response_code;
|
|
|
- ByteBuffer m_stream_backing_buffer;
|
|
|
- NonnullOwnPtr<FixedMemoryStream> m_output_stream;
|
|
|
- NonnullOwnPtr<Core::BufferedSocketBase> m_socket;
|
|
|
- NonnullRefPtr<HTTP::HttpsJob> m_job;
|
|
|
- HashMap<DeprecatedString, DeprecatedString, CaseInsensitiveStringTraits> m_response_headers;
|
|
|
- };
|
|
|
-
|
|
|
- class GeminiHeadlessRequest
|
|
|
- : public Web::ResourceLoaderConnectorRequest
|
|
|
- , public Weakable<GeminiHeadlessRequest> {
|
|
|
- public:
|
|
|
- static ErrorOr<NonnullRefPtr<GeminiHeadlessRequest>> create(DeprecatedString const&, AK::URL const& url, HashMap<DeprecatedString, DeprecatedString> const&, ReadonlyBytes, Core::ProxyData const&)
|
|
|
- {
|
|
|
- auto stream_backing_buffer = TRY(ByteBuffer::create_uninitialized(1 * MiB));
|
|
|
- auto underlying_socket = TRY(Core::TCPSocket::connect(url.host(), url.port().value_or(80)));
|
|
|
- TRY(underlying_socket->set_blocking(false));
|
|
|
- auto socket = TRY(Core::BufferedSocket<Core::TCPSocket>::create(move(underlying_socket)));
|
|
|
-
|
|
|
- Gemini::GeminiRequest request;
|
|
|
- request.set_url(url);
|
|
|
-
|
|
|
- return adopt_ref(*new GeminiHeadlessRequest(move(request), move(socket), move(stream_backing_buffer)));
|
|
|
- }
|
|
|
-
|
|
|
- virtual ~GeminiHeadlessRequest() override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual void set_should_buffer_all_input(bool) override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual bool stop() override
|
|
|
- {
|
|
|
- return false;
|
|
|
- }
|
|
|
-
|
|
|
- virtual void stream_into(Stream&) override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- private:
|
|
|
- GeminiHeadlessRequest(Gemini::GeminiRequest&& request, NonnullOwnPtr<Core::BufferedSocketBase> socket, ByteBuffer&& stream_backing_buffer)
|
|
|
- : m_stream_backing_buffer(move(stream_backing_buffer))
|
|
|
- , m_output_stream(try_make<FixedMemoryStream>(m_stream_backing_buffer.bytes()).release_value_but_fixme_should_propagate_errors())
|
|
|
- , m_socket(move(socket))
|
|
|
- , m_job(Gemini::Job::construct(move(request), *m_output_stream))
|
|
|
- {
|
|
|
- m_job->on_headers_received = [weak_this = make_weak_ptr()](auto& response_headers, auto response_code) {
|
|
|
- if (auto strong_this = weak_this.strong_ref()) {
|
|
|
- strong_this->m_response_code = response_code;
|
|
|
- for (auto& header : response_headers) {
|
|
|
- strong_this->m_response_headers.set(header.key, header.value);
|
|
|
- }
|
|
|
- }
|
|
|
- };
|
|
|
- m_job->on_finish = [weak_this = make_weak_ptr()](bool success) {
|
|
|
- Core::deferred_invoke([weak_this, success] {
|
|
|
- if (auto strong_this = weak_this.strong_ref()) {
|
|
|
- ReadonlyBytes response_bytes { strong_this->m_output_stream->bytes().data(), strong_this->m_output_stream->offset() };
|
|
|
- auto response_buffer = ByteBuffer::copy(response_bytes).release_value_but_fixme_should_propagate_errors();
|
|
|
- strong_this->on_buffered_request_finish(success, strong_this->m_output_stream->offset(), strong_this->m_response_headers, strong_this->m_response_code, response_buffer);
|
|
|
- }
|
|
|
- });
|
|
|
- };
|
|
|
- m_job->start(*m_socket);
|
|
|
- }
|
|
|
-
|
|
|
- Optional<u32> m_response_code;
|
|
|
- ByteBuffer m_stream_backing_buffer;
|
|
|
- NonnullOwnPtr<FixedMemoryStream> m_output_stream;
|
|
|
- NonnullOwnPtr<Core::BufferedSocketBase> m_socket;
|
|
|
- NonnullRefPtr<Gemini::Job> m_job;
|
|
|
- HashMap<DeprecatedString, DeprecatedString, CaseInsensitiveStringTraits> m_response_headers;
|
|
|
- };
|
|
|
-
|
|
|
- static NonnullRefPtr<HeadlessRequestServer> create()
|
|
|
- {
|
|
|
- return adopt_ref(*new HeadlessRequestServer());
|
|
|
- }
|
|
|
-
|
|
|
- virtual ~HeadlessRequestServer() override { }
|
|
|
-
|
|
|
- virtual void prefetch_dns(AK::URL const&) override { }
|
|
|
- virtual void preconnect(AK::URL const&) override { }
|
|
|
-
|
|
|
- virtual RefPtr<Web::ResourceLoaderConnectorRequest> start_request(DeprecatedString const& method, AK::URL const& url, HashMap<DeprecatedString, DeprecatedString> const& request_headers, ReadonlyBytes request_body, Core::ProxyData const& proxy) override
|
|
|
+ static ErrorOr<NonnullOwnPtr<HeadlessWebContentView>> create(Core::AnonymousBuffer theme, Gfx::IntSize const& window_size, StringView web_driver_ipc_path)
|
|
|
{
|
|
|
- RefPtr<Web::ResourceLoaderConnectorRequest> request;
|
|
|
- if (url.scheme().equals_ignoring_ascii_case("http"sv)) {
|
|
|
- auto request_or_error = HTTPHeadlessRequest::create(method, url, request_headers, request_body, proxy);
|
|
|
- if (request_or_error.is_error())
|
|
|
- return {};
|
|
|
- request = request_or_error.release_value();
|
|
|
- }
|
|
|
- if (url.scheme().equals_ignoring_ascii_case("https"sv)) {
|
|
|
- auto request_or_error = HTTPSHeadlessRequest::create(method, url, request_headers, request_body, proxy);
|
|
|
- if (request_or_error.is_error())
|
|
|
- return {};
|
|
|
- request = request_or_error.release_value();
|
|
|
- }
|
|
|
- if (url.scheme().equals_ignoring_ascii_case("gemini"sv)) {
|
|
|
- auto request_or_error = GeminiHeadlessRequest::create(method, url, request_headers, request_body, proxy);
|
|
|
- if (request_or_error.is_error())
|
|
|
- return {};
|
|
|
- request = request_or_error.release_value();
|
|
|
- }
|
|
|
- if (request)
|
|
|
- s_all_requests.set(request);
|
|
|
- return request;
|
|
|
- }
|
|
|
-
|
|
|
-private:
|
|
|
- HeadlessRequestServer() { }
|
|
|
-};
|
|
|
+ auto view = TRY(adopt_nonnull_own_or_enomem(new (nothrow) HeadlessWebContentView()));
|
|
|
|
|
|
-class HeadlessWebSocketClientManager : public Web::WebSockets::WebSocketClientManager {
|
|
|
-public:
|
|
|
- class HeadlessWebSocket
|
|
|
- : public Web::WebSockets::WebSocketClientSocket
|
|
|
- , public Weakable<HeadlessWebSocket> {
|
|
|
- public:
|
|
|
- static NonnullRefPtr<HeadlessWebSocket> create(NonnullRefPtr<WebSocket::WebSocket> underlying_socket)
|
|
|
- {
|
|
|
- return adopt_ref(*new HeadlessWebSocket(move(underlying_socket)));
|
|
|
- }
|
|
|
-
|
|
|
- virtual ~HeadlessWebSocket() override
|
|
|
- {
|
|
|
- }
|
|
|
-
|
|
|
- virtual Web::WebSockets::WebSocket::ReadyState ready_state() override
|
|
|
- {
|
|
|
- switch (m_websocket->ready_state()) {
|
|
|
- case WebSocket::ReadyState::Connecting:
|
|
|
- return Web::WebSockets::WebSocket::ReadyState::Connecting;
|
|
|
- case WebSocket::ReadyState::Open:
|
|
|
- return Web::WebSockets::WebSocket::ReadyState::Open;
|
|
|
- case WebSocket::ReadyState::Closing:
|
|
|
- return Web::WebSockets::WebSocket::ReadyState::Closing;
|
|
|
- case WebSocket::ReadyState::Closed:
|
|
|
- return Web::WebSockets::WebSocket::ReadyState::Closed;
|
|
|
- }
|
|
|
- VERIFY_NOT_REACHED();
|
|
|
- }
|
|
|
-
|
|
|
- virtual DeprecatedString subprotocol_in_use() override
|
|
|
- {
|
|
|
- return m_websocket->subprotocol_in_use();
|
|
|
- }
|
|
|
-
|
|
|
- virtual void send(ByteBuffer binary_or_text_message, bool is_text) override
|
|
|
- {
|
|
|
- m_websocket->send(WebSocket::Message(binary_or_text_message, is_text));
|
|
|
- }
|
|
|
-
|
|
|
- virtual void send(StringView message) override
|
|
|
- {
|
|
|
- m_websocket->send(WebSocket::Message(message));
|
|
|
- }
|
|
|
+#if defined(AK_OS_SERENITY)
|
|
|
+ view->m_client_state.client = TRY(WebView::WebContentClient::try_create(*view));
|
|
|
|
|
|
- virtual void close(u16 code, DeprecatedString reason) override
|
|
|
- {
|
|
|
- m_websocket->close(code, reason);
|
|
|
- }
|
|
|
+ if (!web_driver_ipc_path.is_empty())
|
|
|
+ view->client().async_connect_to_webdriver(web_driver_ipc_path);
|
|
|
+#else
|
|
|
+ auto candidate_web_content_paths = TRY(get_paths_for_helper_process("WebContent"sv));
|
|
|
+ view->m_client_state.client = TRY(view->launch_web_content_process(candidate_web_content_paths, web_driver_ipc_path));
|
|
|
+#endif
|
|
|
|
|
|
- private:
|
|
|
- HeadlessWebSocket(NonnullRefPtr<WebSocket::WebSocket> underlying_socket)
|
|
|
- : m_websocket(move(underlying_socket))
|
|
|
- {
|
|
|
- m_websocket->on_open = [weak_this = make_weak_ptr()] {
|
|
|
- if (auto strong_this = weak_this.strong_ref())
|
|
|
- if (strong_this->on_open)
|
|
|
- strong_this->on_open();
|
|
|
- };
|
|
|
- m_websocket->on_message = [weak_this = make_weak_ptr()](auto message) {
|
|
|
- if (auto strong_this = weak_this.strong_ref()) {
|
|
|
- if (strong_this->on_message) {
|
|
|
- strong_this->on_message(Web::WebSockets::WebSocketClientSocket::Message {
|
|
|
- .data = move(message.data()),
|
|
|
- .is_text = message.is_text(),
|
|
|
- });
|
|
|
- }
|
|
|
- }
|
|
|
- };
|
|
|
- m_websocket->on_error = [weak_this = make_weak_ptr()](auto error) {
|
|
|
- if (auto strong_this = weak_this.strong_ref()) {
|
|
|
- if (strong_this->on_error) {
|
|
|
- switch (error) {
|
|
|
- case WebSocket::WebSocket::Error::CouldNotEstablishConnection:
|
|
|
- strong_this->on_error(Web::WebSockets::WebSocketClientSocket::Error::CouldNotEstablishConnection);
|
|
|
- return;
|
|
|
- case WebSocket::WebSocket::Error::ConnectionUpgradeFailed:
|
|
|
- strong_this->on_error(Web::WebSockets::WebSocketClientSocket::Error::ConnectionUpgradeFailed);
|
|
|
- return;
|
|
|
- case WebSocket::WebSocket::Error::ServerClosedSocket:
|
|
|
- strong_this->on_error(Web::WebSockets::WebSocketClientSocket::Error::ServerClosedSocket);
|
|
|
- return;
|
|
|
- }
|
|
|
- VERIFY_NOT_REACHED();
|
|
|
- }
|
|
|
- }
|
|
|
- };
|
|
|
- m_websocket->on_close = [weak_this = make_weak_ptr()](u16 code, DeprecatedString reason, bool was_clean) {
|
|
|
- if (auto strong_this = weak_this.strong_ref())
|
|
|
- if (strong_this->on_close)
|
|
|
- strong_this->on_close(code, move(reason), was_clean);
|
|
|
- };
|
|
|
- }
|
|
|
+ view->client().async_update_system_theme(move(theme));
|
|
|
+ view->client().async_update_system_fonts(Gfx::FontDatabase::default_font_query(), Gfx::FontDatabase::fixed_width_font_query(), Gfx::FontDatabase::window_title_font_query());
|
|
|
|
|
|
- NonnullRefPtr<WebSocket::WebSocket> m_websocket;
|
|
|
- };
|
|
|
+ view->client().async_set_viewport_rect({ { 0, 0 }, window_size });
|
|
|
+ view->client().async_set_window_size(window_size);
|
|
|
|
|
|
- static NonnullRefPtr<HeadlessWebSocketClientManager> create()
|
|
|
- {
|
|
|
- return adopt_ref(*new HeadlessWebSocketClientManager());
|
|
|
+ return view;
|
|
|
}
|
|
|
|
|
|
- virtual ~HeadlessWebSocketClientManager() override { }
|
|
|
-
|
|
|
- virtual RefPtr<Web::WebSockets::WebSocketClientSocket> connect(AK::URL const& url, DeprecatedString const& origin, Vector<DeprecatedString> const& protocols) override
|
|
|
+ RefPtr<Gfx::Bitmap> take_screenshot()
|
|
|
{
|
|
|
- WebSocket::ConnectionInfo connection_info(url);
|
|
|
- connection_info.set_origin(origin);
|
|
|
- connection_info.set_protocols(protocols);
|
|
|
-
|
|
|
- auto connection = HeadlessWebSocket::create(WebSocket::WebSocket::create(move(connection_info)));
|
|
|
- return connection;
|
|
|
+ return client().take_document_screenshot().bitmap();
|
|
|
}
|
|
|
|
|
|
private:
|
|
|
- HeadlessWebSocketClientManager() { }
|
|
|
+ HeadlessWebContentView() = default;
|
|
|
+
|
|
|
+ void notify_server_did_layout(Badge<WebView::WebContentClient>, Gfx::IntSize) override { }
|
|
|
+ void notify_server_did_paint(Badge<WebView::WebContentClient>, i32) override { }
|
|
|
+ void notify_server_did_invalidate_content_rect(Badge<WebView::WebContentClient>, Gfx::IntRect const&) override { }
|
|
|
+ void notify_server_did_change_selection(Badge<WebView::WebContentClient>) override { }
|
|
|
+ void notify_server_did_request_cursor_change(Badge<WebView::WebContentClient>, Gfx::StandardCursor) override { }
|
|
|
+ void notify_server_did_change_title(Badge<WebView::WebContentClient>, DeprecatedString const&) override { }
|
|
|
+ void notify_server_did_request_scroll(Badge<WebView::WebContentClient>, i32, i32) override { }
|
|
|
+ void notify_server_did_request_scroll_to(Badge<WebView::WebContentClient>, Gfx::IntPoint) override { }
|
|
|
+ void notify_server_did_request_scroll_into_view(Badge<WebView::WebContentClient>, Gfx::IntRect const&) override { }
|
|
|
+ void notify_server_did_enter_tooltip_area(Badge<WebView::WebContentClient>, Gfx::IntPoint, DeprecatedString const&) override { }
|
|
|
+ void notify_server_did_leave_tooltip_area(Badge<WebView::WebContentClient>) override { }
|
|
|
+ void notify_server_did_hover_link(Badge<WebView::WebContentClient>, const URL&) override { }
|
|
|
+ void notify_server_did_unhover_link(Badge<WebView::WebContentClient>) override { }
|
|
|
+ void notify_server_did_click_link(Badge<WebView::WebContentClient>, const URL&, DeprecatedString const&, unsigned) override { }
|
|
|
+ void notify_server_did_middle_click_link(Badge<WebView::WebContentClient>, const URL&, DeprecatedString const&, unsigned) override { }
|
|
|
+ void notify_server_did_start_loading(Badge<WebView::WebContentClient>, const URL&, bool) override { }
|
|
|
+ void notify_server_did_finish_loading(Badge<WebView::WebContentClient>, const URL&) override { }
|
|
|
+ void notify_server_did_request_navigate_back(Badge<WebView::WebContentClient>) override { }
|
|
|
+ void notify_server_did_request_navigate_forward(Badge<WebView::WebContentClient>) override { }
|
|
|
+ void notify_server_did_request_refresh(Badge<WebView::WebContentClient>) override { }
|
|
|
+ void notify_server_did_request_context_menu(Badge<WebView::WebContentClient>, Gfx::IntPoint) override { }
|
|
|
+ void notify_server_did_request_link_context_menu(Badge<WebView::WebContentClient>, Gfx::IntPoint, const URL&, DeprecatedString const&, unsigned) override { }
|
|
|
+ void notify_server_did_request_image_context_menu(Badge<WebView::WebContentClient>, Gfx::IntPoint, const URL&, DeprecatedString const&, unsigned, Gfx::ShareableBitmap const&) override { }
|
|
|
+ void notify_server_did_request_alert(Badge<WebView::WebContentClient>, DeprecatedString const&) override { }
|
|
|
+ void notify_server_did_request_confirm(Badge<WebView::WebContentClient>, DeprecatedString const&) override { }
|
|
|
+ void notify_server_did_request_prompt(Badge<WebView::WebContentClient>, DeprecatedString const&, DeprecatedString const&) override { }
|
|
|
+ void notify_server_did_request_set_prompt_text(Badge<WebView::WebContentClient>, DeprecatedString const&) override { }
|
|
|
+ void notify_server_did_request_accept_dialog(Badge<WebView::WebContentClient>) override { }
|
|
|
+ void notify_server_did_request_dismiss_dialog(Badge<WebView::WebContentClient>) override { }
|
|
|
+ void notify_server_did_get_source(const URL&, DeprecatedString const&) override { }
|
|
|
+ void notify_server_did_get_dom_tree(DeprecatedString const&) override { }
|
|
|
+ void notify_server_did_get_dom_node_properties(i32, DeprecatedString const&, DeprecatedString const&, DeprecatedString const&, DeprecatedString const&) override { }
|
|
|
+ void notify_server_did_get_accessibility_tree(DeprecatedString const&) override { }
|
|
|
+ void notify_server_did_output_js_console_message(i32) override { }
|
|
|
+ void notify_server_did_get_js_console_messages(i32, Vector<DeprecatedString> const&, Vector<DeprecatedString> const&) override { }
|
|
|
+ void notify_server_did_change_favicon(Gfx::Bitmap const&) override { }
|
|
|
+ Vector<Web::Cookie::Cookie> notify_server_did_request_all_cookies(Badge<WebView::WebContentClient>, URL const&) override { return {}; }
|
|
|
+ Optional<Web::Cookie::Cookie> notify_server_did_request_named_cookie(Badge<WebView::WebContentClient>, URL const&, DeprecatedString const&) override { return {}; }
|
|
|
+ DeprecatedString notify_server_did_request_cookie(Badge<WebView::WebContentClient>, const URL&, Web::Cookie::Source) override { return {}; }
|
|
|
+ void notify_server_did_set_cookie(Badge<WebView::WebContentClient>, const URL&, Web::Cookie::ParsedCookie const&, Web::Cookie::Source) override { }
|
|
|
+ void notify_server_did_update_cookie(Badge<WebView::WebContentClient>, Web::Cookie::Cookie const&) override { }
|
|
|
+ void notify_server_did_close_browsing_context(Badge<WebView::WebContentClient>) override { }
|
|
|
+ void notify_server_did_update_resource_count(i32) override { }
|
|
|
+ void notify_server_did_request_restore_window() override { }
|
|
|
+ Gfx::IntPoint notify_server_did_request_reposition_window(Gfx::IntPoint) override { return {}; }
|
|
|
+ Gfx::IntSize notify_server_did_request_resize_window(Gfx::IntSize) override { return {}; }
|
|
|
+ Gfx::IntRect notify_server_did_request_maximize_window() override { return {}; }
|
|
|
+ Gfx::IntRect notify_server_did_request_minimize_window() override { return {}; }
|
|
|
+ Gfx::IntRect notify_server_did_request_fullscreen_window() override { return {}; }
|
|
|
+ void notify_server_did_request_file(Badge<WebView::WebContentClient>, DeprecatedString const&, i32) override { }
|
|
|
+ void notify_server_did_finish_handling_input_event(bool) override { }
|
|
|
+ void update_zoom() override { }
|
|
|
+ void create_client() override { }
|
|
|
};
|
|
|
|
|
|
-static void load_page_for_screenshot_and_exit(HeadlessBrowserPageClient& page_client, int take_screenshot_after)
|
|
|
+static ErrorOr<NonnullRefPtr<Core::Timer>> load_page_for_screenshot_and_exit(Core::EventLoop& event_loop, HeadlessWebContentView& view, int screenshot_timeout)
|
|
|
{
|
|
|
- dbgln("Taking screenshot after {} seconds", take_screenshot_after);
|
|
|
-
|
|
|
- auto timer = Core::Timer::create_single_shot(
|
|
|
- take_screenshot_after * 1000,
|
|
|
- [&]() {
|
|
|
- // FIXME: Allow passing the output path as argument
|
|
|
- DeprecatedString output_file_path = "output.png";
|
|
|
- dbgln("Saving to {}", output_file_path);
|
|
|
-
|
|
|
- if (Core::DeprecatedFile::exists(output_file_path))
|
|
|
- MUST(Core::DeprecatedFile::remove(output_file_path, Core::DeprecatedFile::RecursionMode::Disallowed));
|
|
|
+ // FIXME: Allow passing the output path as an argument.
|
|
|
+ static constexpr auto output_file_path = "output.png"sv;
|
|
|
|
|
|
- auto output_file = MUST(Core::File::open(output_file_path, Core::File::OpenMode::Write));
|
|
|
+ if (Core::DeprecatedFile::exists(output_file_path))
|
|
|
+ TRY(Core::DeprecatedFile::remove(output_file_path, Core::DeprecatedFile::RecursionMode::Disallowed));
|
|
|
|
|
|
- auto output_rect = page_client.screen_rect();
|
|
|
- auto output_bitmap = MUST(Gfx::Bitmap::create(Gfx::BitmapFormat::BGRx8888, output_rect.size().to_type<int>()));
|
|
|
+ outln("Taking screenshot after {} seconds", screenshot_timeout);
|
|
|
|
|
|
- page_client.paint(output_rect, output_bitmap);
|
|
|
+ auto timer = TRY(Core::Timer::create_single_shot(
|
|
|
+ screenshot_timeout * 1000,
|
|
|
+ [&]() {
|
|
|
+ if (auto screenshot = view.take_screenshot()) {
|
|
|
+ outln("Saving screenshot to {}", output_file_path);
|
|
|
|
|
|
- auto image_buffer = MUST(Gfx::PNGWriter::encode(output_bitmap));
|
|
|
- MUST(output_file->write(image_buffer.bytes()));
|
|
|
+ auto output_file = MUST(Core::File::open(output_file_path, Core::File::OpenMode::Write));
|
|
|
+ auto image_buffer = MUST(Gfx::PNGWriter::encode(*screenshot));
|
|
|
+ MUST(output_file->write(image_buffer.bytes()));
|
|
|
+ } else {
|
|
|
+ warnln("No screenshot available");
|
|
|
+ }
|
|
|
|
|
|
- exit(0);
|
|
|
- }).release_value_but_fixme_should_propagate_errors();
|
|
|
+ event_loop.quit(0);
|
|
|
+ }));
|
|
|
|
|
|
timer->start();
|
|
|
+ return timer;
|
|
|
}
|
|
|
|
|
|
ErrorOr<int> serenity_main(Main::Arguments arguments)
|
|
|
{
|
|
|
- int take_screenshot_after = 1;
|
|
|
+#if !defined(AK_OS_SERENITY)
|
|
|
+ QCoreApplication app(arguments.argc, arguments.argv);
|
|
|
+#endif
|
|
|
+ Core::EventLoop event_loop;
|
|
|
+
|
|
|
+ int screenshot_timeout = 1;
|
|
|
StringView url;
|
|
|
- StringView resources_folder;
|
|
|
- StringView error_page_url;
|
|
|
- StringView ca_certs_path;
|
|
|
- StringView webdriver_ipc_path;
|
|
|
+ auto resources_folder = "/res"sv;
|
|
|
+ StringView web_driver_ipc_path;
|
|
|
|
|
|
- Core::EventLoop event_loop;
|
|
|
Core::ArgsParser args_parser;
|
|
|
args_parser.set_general_help("This utility runs the Browser in headless mode.");
|
|
|
- args_parser.add_option(take_screenshot_after, "Take a screenshot after [n] seconds (default: 1)", "screenshot", 's', "n");
|
|
|
+ args_parser.add_option(screenshot_timeout, "Take a screenshot after [n] seconds (default: 1)", "screenshot", 's', "n");
|
|
|
args_parser.add_option(resources_folder, "Path of the base resources folder (defaults to /res)", "resources", 'r', "resources-root-path");
|
|
|
- args_parser.add_option(error_page_url, "URL for the error page (defaults to file:///res/html/error.html)", "error-page", 'e', "error-page-url");
|
|
|
- args_parser.add_option(ca_certs_path, "The bundled ca certificates file", "certs", 'c', "ca-certs-path");
|
|
|
- args_parser.add_option(webdriver_ipc_path, "Path to the WebDriver IPC socket", "webdriver-ipc-path", 0, "path");
|
|
|
+ args_parser.add_option(web_driver_ipc_path, "Path to the WebDriver IPC socket", "webdriver-ipc-path", 0, "path");
|
|
|
args_parser.add_positional_argument(url, "URL to open", "url", Core::ArgsParser::Required::Yes);
|
|
|
args_parser.parse(arguments);
|
|
|
|
|
|
- Web::Platform::EventLoopPlugin::install(*new Web::Platform::EventLoopPluginSerenity);
|
|
|
- Web::Platform::FontPlugin::install(*new Web::Platform::FontPluginSerenity);
|
|
|
- Web::Platform::ImageCodecPlugin::install(*new ImageCodecPluginHeadless);
|
|
|
- Web::ResourceLoader::initialize(HeadlessRequestServer::create());
|
|
|
- Web::WebSockets::WebSocketClientManager::initialize(HeadlessWebSocketClientManager::create());
|
|
|
-
|
|
|
- if (!resources_folder.is_empty()) {
|
|
|
- Web::FrameLoader::set_default_favicon_path(LexicalPath::join(resources_folder, "icons/16x16/app-browser.png"sv).string());
|
|
|
- Gfx::FontDatabase::set_default_fonts_lookup_path(LexicalPath::join(resources_folder, "fonts"sv).string());
|
|
|
- }
|
|
|
- if (!ca_certs_path.is_empty()) {
|
|
|
- auto config_result = Core::ConfigFile::open(ca_certs_path);
|
|
|
- if (config_result.is_error()) {
|
|
|
- dbgln("Failed to load CA Certificates: {}", config_result.error());
|
|
|
- } else {
|
|
|
- auto config = config_result.release_value();
|
|
|
- DefaultRootCACertificates::the().reload_certificates(config);
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
Gfx::FontDatabase::set_default_font_query("Katica 10 400 0");
|
|
|
Gfx::FontDatabase::set_window_title_font_query("Katica 10 700 0");
|
|
|
Gfx::FontDatabase::set_fixed_width_font_query("Csilla 10 400 0");
|
|
|
|
|
|
- if (!error_page_url.is_empty())
|
|
|
- Web::FrameLoader::set_error_page_url(error_page_url);
|
|
|
-
|
|
|
- auto page_client = HeadlessBrowserPageClient::create();
|
|
|
+ auto fonts_path = LexicalPath::join(resources_folder, "fonts"sv);
|
|
|
+ Gfx::FontDatabase::set_default_fonts_lookup_path(fonts_path.string());
|
|
|
|
|
|
- if (!resources_folder.is_empty()) {
|
|
|
- auto system_theme = TRY(Gfx::load_system_theme(LexicalPath::join(resources_folder, "themes/Default.ini"sv).string()));
|
|
|
- page_client->setup_palette(system_theme);
|
|
|
- } else {
|
|
|
- auto system_theme = TRY(Gfx::load_system_theme("/res/themes/Default.ini"));
|
|
|
- page_client->setup_palette(system_theme);
|
|
|
- }
|
|
|
+ auto theme_path = LexicalPath::join(resources_folder, "themes"sv, "Default.ini"sv);
|
|
|
+ auto theme = TRY(Gfx::load_system_theme(theme_path.string()));
|
|
|
|
|
|
- dbgln("Loading {}", url);
|
|
|
- page_client->load(AK::URL(url));
|
|
|
+ // FIXME: Allow passing the window size as an argument.
|
|
|
+ static constexpr Gfx::IntSize window_size { 800, 600 };
|
|
|
|
|
|
- // FIXME: Allow passing these values as arguments
|
|
|
- page_client->set_viewport_rect({ 0, 0, 800, 600 });
|
|
|
- page_client->set_screen_rect({ 0, 0, 800, 600 });
|
|
|
+ auto view = TRY(HeadlessWebContentView::create(move(theme), window_size, web_driver_ipc_path));
|
|
|
+ view->load(URL { url });
|
|
|
|
|
|
- if (!webdriver_ipc_path.is_empty())
|
|
|
- TRY(page_client->connect_to_webdriver(webdriver_ipc_path));
|
|
|
- else
|
|
|
- load_page_for_screenshot_and_exit(*page_client, take_screenshot_after);
|
|
|
+ RefPtr<Core::Timer> timer;
|
|
|
+ if (web_driver_ipc_path.is_empty())
|
|
|
+ timer = TRY(load_page_for_screenshot_and_exit(event_loop, *view, screenshot_timeout));
|
|
|
|
|
|
return event_loop.exec();
|
|
|
}
|