headless-browser: Implement dialog-related WebView callbacks

This allows us to headlessly run WPT tests which involve dialogs.
LadybirdBrowser · Oct 28, 2024 · 777eec0 · 777eec0
1 parent 6590b8a
commit 777eec0
Show file tree

Hide file tree

Showing 2 changed files with 60 additions and 0 deletions.
diff --git a/Ladybird/Headless/HeadlessWebView.cpp b/Ladybird/Headless/HeadlessWebView.cpp
@@ -35,6 +35,62 @@ HeadlessWebView::HeadlessWebView(Core::AnonymousBuffer theme, Gfx::IntSize viewp
 
         return worker_client->clone_transport();
     };
+
+    on_request_alert = [this](auto const&) {
+        m_pending_dialog = Web::Page::PendingDialog::Alert;
+    };
+
+    on_request_confirm = [this](auto const&) {
+        m_pending_dialog = Web::Page::PendingDialog::Confirm;
+    };
+
+    on_request_prompt = [this](auto const&, auto const& prompt_text) {
+        m_pending_dialog = Web::Page::PendingDialog::Prompt;
+        m_pending_prompt_text = prompt_text;
+    };
+
+    on_request_set_prompt_text = [this](auto const& prompt_text) {
+        m_pending_prompt_text = prompt_text;
+    };
+
+    on_request_accept_dialog = [this]() {
+        switch (m_pending_dialog) {
+        case Web::Page::PendingDialog::None:
+            VERIFY_NOT_REACHED();
+            break;
+        case Web::Page::PendingDialog::Alert:
+            alert_closed();
+            break;
+        case Web::Page::PendingDialog::Confirm:
+            confirm_closed(true);
+            break;
+        case Web::Page::PendingDialog::Prompt:
+            prompt_closed(move(m_pending_prompt_text));
+            break;
+        }
+
+        m_pending_dialog = Web::Page::PendingDialog::None;
+    };
+
+    on_request_dismiss_dialog = [this]() {
+        switch (m_pending_dialog) {
+        case Web::Page::PendingDialog::None:
+            VERIFY_NOT_REACHED();
+            break;
+        case Web::Page::PendingDialog::Alert:
+            alert_closed();
+            break;
+        case Web::Page::PendingDialog::Confirm:
+            confirm_closed(false);
+            break;
+        case Web::Page::PendingDialog::Prompt:
+            prompt_closed({});
+            break;
+        }
+
+        m_pending_dialog = Web::Page::PendingDialog::None;
+        m_pending_prompt_text.clear();
+    };
 }
 
 NonnullOwnPtr<HeadlessWebView> HeadlessWebView::create(Core::AnonymousBuffer theme, Gfx::IntSize window_size)

diff --git a/Ladybird/Headless/HeadlessWebView.h b/Ladybird/Headless/HeadlessWebView.h
@@ -13,6 +13,7 @@
 #include <LibCore/Promise.h>
 #include <LibGfx/Forward.h>
 #include <LibGfx/Size.h>
+#include <LibWeb/Page/Page.h>
 #include <LibWeb/PixelUnits.h>
 #include <LibWebView/ViewImplementation.h>
 
@@ -48,6 +49,9 @@ class HeadlessWebView final : public WebView::ViewImplementation {
     RefPtr<Core::Promise<RefPtr<Gfx::Bitmap>>> m_pending_screenshot;
 
     NonnullRefPtr<TestPromise> m_test_promise;
+
+    Web::Page::PendingDialog m_pending_dialog { Web::Page::PendingDialog::None };
+    Optional<String> m_pending_prompt_text;
 };
 
 }