mirror of
https://github.com/RGBCube/serenity
synced 2025-05-17 15:55:07 +00:00
298 lines
11 KiB
C++
298 lines
11 KiB
C++
/*
|
|
* Copyright (c) 2022, Florent Castelli <florent.castelli@gmail.com>
|
|
* Copyright (c) 2022, Sam Atkins <atkinssj@serenityos.org>
|
|
* Copyright (c) 2022, Tobias Christiansen <tobyase@serenityos.org>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include "WebDriverConnection.h"
|
|
#include "BrowserWindow.h"
|
|
#include <AK/Vector.h>
|
|
#include <LibWeb/Cookie/Cookie.h>
|
|
#include <LibWeb/Cookie/ParsedCookie.h>
|
|
#include <LibWebView/WebContentClient.h>
|
|
|
|
namespace Browser {
|
|
|
|
WebDriverConnection::WebDriverConnection(NonnullOwnPtr<Core::Stream::LocalSocket> socket, NonnullRefPtr<BrowserWindow> browser_window)
|
|
: IPC::ConnectionToServer<WebDriverSessionClientEndpoint, WebDriverSessionServerEndpoint>(*this, move(socket))
|
|
, m_browser_window(move(browser_window))
|
|
{
|
|
}
|
|
|
|
void WebDriverConnection::quit()
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: quit");
|
|
if (auto browser_window = m_browser_window.strong_ref())
|
|
browser_window->close();
|
|
}
|
|
|
|
Messages::WebDriverSessionClient::GetUrlResponse WebDriverConnection::get_url()
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: get_url");
|
|
if (auto browser_window = m_browser_window.strong_ref())
|
|
return { browser_window->active_tab().url() };
|
|
return { URL("") };
|
|
}
|
|
|
|
void WebDriverConnection::set_url(AK::URL const& url)
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: set_url {}", url);
|
|
if (auto browser_window = m_browser_window.strong_ref())
|
|
browser_window->active_tab().load(url);
|
|
}
|
|
|
|
Messages::WebDriverSessionClient::GetTitleResponse WebDriverConnection::get_title()
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: get_title");
|
|
if (auto browser_window = m_browser_window.strong_ref())
|
|
return { browser_window->active_tab().title() };
|
|
return { "" };
|
|
}
|
|
|
|
void WebDriverConnection::refresh()
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: refresh");
|
|
if (auto browser_window = m_browser_window.strong_ref())
|
|
browser_window->active_tab().reload();
|
|
}
|
|
|
|
void WebDriverConnection::back()
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: back");
|
|
if (auto browser_window = m_browser_window.strong_ref())
|
|
browser_window->active_tab().go_back();
|
|
}
|
|
|
|
void WebDriverConnection::forward()
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: forward");
|
|
if (auto browser_window = m_browser_window.strong_ref())
|
|
browser_window->active_tab().go_forward();
|
|
}
|
|
|
|
Messages::WebDriverSessionClient::GetWindowRectResponse WebDriverConnection::get_window_rect()
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: get_window_rect");
|
|
if (auto browser_window = m_browser_window.strong_ref())
|
|
return { browser_window->rect() };
|
|
return { {} };
|
|
}
|
|
|
|
void WebDriverConnection::restore_window()
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: restore_window");
|
|
if (auto browser_window = m_browser_window.strong_ref()) {
|
|
browser_window->show();
|
|
browser_window->move_to_front();
|
|
}
|
|
}
|
|
|
|
void WebDriverConnection::set_window_size(Gfx::IntSize const& size)
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: set_window_size {}", size);
|
|
if (auto browser_window = m_browser_window.strong_ref())
|
|
browser_window->resize(size);
|
|
}
|
|
|
|
void WebDriverConnection::set_window_position(Gfx::IntPoint const& position)
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: set_window_position {}", position);
|
|
if (auto browser_window = m_browser_window.strong_ref())
|
|
browser_window->move_to(position);
|
|
}
|
|
|
|
void WebDriverConnection::maximize_window()
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: maximize_window");
|
|
if (auto browser_window = m_browser_window.strong_ref())
|
|
browser_window->set_maximized(true);
|
|
}
|
|
|
|
void WebDriverConnection::minimize_window()
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: minimize_window");
|
|
if (auto browser_window = m_browser_window.strong_ref())
|
|
browser_window->set_minimized(true);
|
|
}
|
|
|
|
Messages::WebDriverSessionClient::SerializeSourceResponse WebDriverConnection::serialize_source()
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: serialize_source");
|
|
if (auto browser_window = m_browser_window.strong_ref()) {
|
|
auto& tab = browser_window->active_tab();
|
|
if (tab.webdriver_endpoints().on_serialize_source)
|
|
return { tab.webdriver_endpoints().on_serialize_source() };
|
|
}
|
|
|
|
return { {} };
|
|
}
|
|
|
|
Messages::WebDriverSessionClient::ExecuteScriptResponse WebDriverConnection::execute_script(String const& body, Vector<String> const& json_arguments, Optional<u64> const& timeout, bool async)
|
|
{
|
|
dbgln("WebDriverConnection: execute_script");
|
|
if (auto browser_window = m_browser_window.strong_ref()) {
|
|
auto& tab = browser_window->active_tab();
|
|
if (tab.webdriver_endpoints().on_execute_script) {
|
|
auto response = tab.webdriver_endpoints().on_execute_script(body, json_arguments, timeout, async);
|
|
// WebContentServer's and WebDriverSessionClient's ExecuteScriptResponse have an identical
|
|
// structure but are distinct types, so we have to convert here.
|
|
return { response.result_type(), response.json_result() };
|
|
}
|
|
}
|
|
return { {} };
|
|
}
|
|
|
|
Messages::WebDriverSessionClient::GetAllCookiesResponse WebDriverConnection::get_all_cookies()
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: get_cookies");
|
|
if (auto browser_window = m_browser_window.strong_ref()) {
|
|
if (browser_window->active_tab().on_get_cookies_entries) {
|
|
return { browser_window->active_tab().on_get_cookies_entries() };
|
|
}
|
|
}
|
|
return { {} };
|
|
}
|
|
|
|
Messages::WebDriverSessionClient::GetNamedCookieResponse WebDriverConnection::get_named_cookie(String const& name)
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: get_named_cookie {}", name);
|
|
if (auto browser_window = m_browser_window.strong_ref()) {
|
|
if (browser_window->active_tab().on_get_cookies_entries) {
|
|
for (auto cookie : browser_window->active_tab().on_get_cookies_entries()) {
|
|
if (cookie.name == name)
|
|
return { cookie };
|
|
}
|
|
return Optional<Web::Cookie::Cookie> {};
|
|
}
|
|
}
|
|
return { {} };
|
|
}
|
|
|
|
void WebDriverConnection::add_cookie(Web::Cookie::ParsedCookie const& cookie)
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: add_cookie {}", cookie.name);
|
|
if (auto browser_window = m_browser_window.strong_ref()) {
|
|
auto& tab = browser_window->active_tab();
|
|
if (tab.on_set_cookie) {
|
|
// FIXME: The spec doesn't say anything about the source
|
|
// but can we assume a cookie created through a HTTP-request to the WebDriver
|
|
// to be (source) from an HTTP-API?
|
|
tab.on_set_cookie(tab.url(), cookie, Web::Cookie::Source::Http);
|
|
}
|
|
}
|
|
}
|
|
|
|
void WebDriverConnection::update_cookie(Web::Cookie::Cookie const& cookie)
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: update_cookie {}", cookie.name);
|
|
if (auto browser_window = m_browser_window.strong_ref()) {
|
|
auto& tab = browser_window->active_tab();
|
|
if (tab.on_update_cookie) {
|
|
tab.on_update_cookie(tab.url(), cookie);
|
|
}
|
|
}
|
|
}
|
|
|
|
Messages::WebDriverSessionClient::GetDocumentElementResponse WebDriverConnection::get_document_element()
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: get_document_element");
|
|
if (auto browser_window = m_browser_window.strong_ref()) {
|
|
auto& tab = browser_window->active_tab();
|
|
if (tab.webdriver_endpoints().on_get_document_element)
|
|
return { tab.webdriver_endpoints().on_get_document_element() };
|
|
}
|
|
return { {} };
|
|
}
|
|
|
|
Messages::WebDriverSessionClient::QuerySelectorAllResponse WebDriverConnection::query_selector_all(i32 start_node_id, String const& selector)
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: query_selector_all");
|
|
if (auto browser_window = m_browser_window.strong_ref()) {
|
|
auto& tab = browser_window->active_tab();
|
|
if (tab.webdriver_endpoints().on_query_selector_all)
|
|
return { tab.webdriver_endpoints().on_query_selector_all(start_node_id, selector) };
|
|
}
|
|
return { {} };
|
|
}
|
|
|
|
Messages::WebDriverSessionClient::GetElementAttributeResponse WebDriverConnection::get_element_attribute(i32 element_id, String const& name)
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: get_element_attribute");
|
|
if (auto browser_window = m_browser_window.strong_ref()) {
|
|
auto& tab = browser_window->active_tab();
|
|
if (tab.webdriver_endpoints().on_get_element_attribute)
|
|
return { tab.webdriver_endpoints().on_get_element_attribute(element_id, name) };
|
|
}
|
|
return { {} };
|
|
}
|
|
|
|
Messages::WebDriverSessionClient::GetElementPropertyResponse WebDriverConnection::get_element_property(i32 element_id, String const& name)
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: get_element_property");
|
|
if (auto browser_window = m_browser_window.strong_ref()) {
|
|
auto& tab = browser_window->active_tab();
|
|
if (tab.webdriver_endpoints().on_get_element_property)
|
|
return { tab.webdriver_endpoints().on_get_element_property(element_id, name) };
|
|
}
|
|
return { {} };
|
|
}
|
|
|
|
Messages::WebDriverSessionClient::GetActiveDocumentsTypeResponse WebDriverConnection::get_active_documents_type()
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: get_active_documents_type");
|
|
if (auto browser_window = m_browser_window.strong_ref()) {
|
|
auto& tab = browser_window->active_tab();
|
|
if (tab.webdriver_endpoints().on_get_active_documents_type)
|
|
return { tab.webdriver_endpoints().on_get_active_documents_type() };
|
|
}
|
|
return { "" };
|
|
}
|
|
|
|
Messages::WebDriverSessionClient::GetComputedValueForElementResponse WebDriverConnection::get_computed_value_for_element(i32 element_id, String const& property_name)
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: get_computed_value_for_element");
|
|
if (auto browser_window = m_browser_window.strong_ref()) {
|
|
auto& tab = browser_window->active_tab();
|
|
if (tab.webdriver_endpoints().on_get_computed_value_for_element)
|
|
return { tab.webdriver_endpoints().on_get_computed_value_for_element(element_id, property_name) };
|
|
}
|
|
return { "" };
|
|
}
|
|
|
|
Messages::WebDriverSessionClient::GetElementTextResponse WebDriverConnection::get_element_text(i32 element_id)
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: get_element_text");
|
|
if (auto browser_window = m_browser_window.strong_ref()) {
|
|
auto& tab = browser_window->active_tab();
|
|
if (tab.webdriver_endpoints().on_get_element_text)
|
|
return { tab.webdriver_endpoints().on_get_element_text(element_id) };
|
|
}
|
|
return { "" };
|
|
}
|
|
|
|
Messages::WebDriverSessionClient::GetElementTagNameResponse WebDriverConnection::get_element_tag_name(i32 element_id)
|
|
{
|
|
dbgln("WebDriverConnection: get_computed_value_for_element");
|
|
if (auto browser_window = m_browser_window.strong_ref()) {
|
|
auto& tab = browser_window->active_tab();
|
|
if (tab.webdriver_endpoints().on_get_element_tag_name)
|
|
return { tab.webdriver_endpoints().on_get_element_tag_name(element_id) };
|
|
}
|
|
return { "" };
|
|
}
|
|
|
|
Messages::WebDriverSessionClient::TakeScreenshotResponse WebDriverConnection::take_screenshot()
|
|
{
|
|
dbgln_if(WEBDRIVER_DEBUG, "WebDriverConnection: take_screenshot");
|
|
if (auto browser_window = m_browser_window.strong_ref()) {
|
|
auto& tab = browser_window->active_tab();
|
|
if (tab.on_take_screenshot)
|
|
return { tab.on_take_screenshot() };
|
|
}
|
|
|
|
return { {} };
|
|
}
|
|
|
|
}
|