blob: dd52267c07841200b32907fd22161a664ed2ba6c [file] [log] [blame]
/*
* Copyright (C) 2017 Igalia S.L.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
* THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
* THE POSSIBILITY OF SUCH DAMAGE.
*/
#include "config.h"
#include "Session.h"
#include "CommandResult.h"
#include "SessionHost.h"
#include "WebDriverAtoms.h"
#include <wtf/CryptographicallyRandomNumber.h>
#include <wtf/HashSet.h>
#include <wtf/HexNumber.h>
#include <wtf/NeverDestroyed.h>
namespace WebDriver {
// https://w3c.github.io/webdriver/webdriver-spec.html#dfn-session-script-timeout
static const Seconds defaultScriptTimeout = 30_s;
// https://w3c.github.io/webdriver/webdriver-spec.html#dfn-session-page-load-timeout
static const Seconds defaultPageLoadTimeout = 300_s;
// https://w3c.github.io/webdriver/webdriver-spec.html#dfn-session-implicit-wait-timeout
static const Seconds defaultImplicitWaitTimeout = 0_s;
const String& Session::webElementIdentifier()
{
// The web element identifier is a constant defined by the spec in Section 11 Elements.
// https://www.w3.org/TR/webdriver/#elements
static NeverDestroyed<String> webElementID { "element-6066-11e4-a52e-4f735466cecf"_s };
return webElementID;
}
Session::Session(std::unique_ptr<SessionHost>&& host)
: m_host(WTFMove(host))
, m_scriptTimeout(defaultScriptTimeout)
, m_pageLoadTimeout(defaultPageLoadTimeout)
, m_implicitWaitTimeout(defaultImplicitWaitTimeout)
{
if (capabilities().timeouts)
setTimeouts(capabilities().timeouts.value(), [](CommandResult&&) { });
}
Session::~Session()
{
}
const String& Session::id() const
{
return m_host->sessionID();
}
const Capabilities& Session::capabilities() const
{
return m_host->capabilities();
}
bool Session::isConnected() const
{
return m_host->isConnected();
}
static Optional<String> firstWindowHandleInResult(JSON::Value& result)
{
RefPtr<JSON::Array> handles;
if (result.asArray(handles) && handles->length()) {
auto handleValue = handles->get(0);
String handle;
if (handleValue->asString(handle))
return handle;
}
return WTF::nullopt;
}
void Session::closeAllToplevelBrowsingContexts(const String& toplevelBrowsingContext, Function<void (CommandResult&&)>&& completionHandler)
{
closeTopLevelBrowsingContext(toplevelBrowsingContext, [this, protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
if (auto handle = firstWindowHandleInResult(*result.result())) {
closeAllToplevelBrowsingContexts(handle.value(), WTFMove(completionHandler));
return;
}
completionHandler(CommandResult::success());
});
}
void Session::close(Function<void (CommandResult&&)>&& completionHandler)
{
m_toplevelBrowsingContext = WTF::nullopt;
getWindowHandles([this, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
if (auto handle = firstWindowHandleInResult(*result.result())) {
closeAllToplevelBrowsingContexts(handle.value(), WTFMove(completionHandler));
return;
}
completionHandler(CommandResult::success());
});
}
void Session::getTimeouts(Function<void (CommandResult&&)>&& completionHandler)
{
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setInteger("script"_s, m_scriptTimeout.millisecondsAs<int>());
parameters->setInteger("pageLoad"_s, m_pageLoadTimeout.millisecondsAs<int>());
parameters->setInteger("implicit"_s, m_implicitWaitTimeout.millisecondsAs<int>());
completionHandler(CommandResult::success(WTFMove(parameters)));
}
void Session::setTimeouts(const Timeouts& timeouts, Function<void (CommandResult&&)>&& completionHandler)
{
if (timeouts.script)
m_scriptTimeout = timeouts.script.value();
if (timeouts.pageLoad)
m_pageLoadTimeout = timeouts.pageLoad.value();
if (timeouts.implicit)
m_implicitWaitTimeout = timeouts.implicit.value();
completionHandler(CommandResult::success());
}
void Session::switchToTopLevelBrowsingContext(Optional<String> toplevelBrowsingContext)
{
m_toplevelBrowsingContext = toplevelBrowsingContext;
m_currentBrowsingContext = WTF::nullopt;
}
void Session::switchToBrowsingContext(Optional<String> browsingContext)
{
// Automation sends empty strings for main frame.
if (!browsingContext || browsingContext.value().isEmpty())
m_currentBrowsingContext = WTF::nullopt;
else
m_currentBrowsingContext = browsingContext;
}
Optional<String> Session::pageLoadStrategyString() const
{
if (!capabilities().pageLoadStrategy)
return WTF::nullopt;
switch (capabilities().pageLoadStrategy.value()) {
case PageLoadStrategy::None:
return String("None");
case PageLoadStrategy::Normal:
return String("Normal");
case PageLoadStrategy::Eager:
return String("Eager");
}
return WTF::nullopt;
}
void Session::createTopLevelBrowsingContext(Function<void (CommandResult&&)>&& completionHandler)
{
ASSERT(!m_toplevelBrowsingContext);
m_host->sendCommandToBackend("createBrowsingContext"_s, nullptr, [this, protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) mutable {
if (response.isError || !response.responseObject) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
String handle;
if (!response.responseObject->getString("handle"_s, handle)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
switchToTopLevelBrowsingContext(handle);
completionHandler(CommandResult::success());
});
}
void Session::handleUserPrompts(Function<void (CommandResult&&)>&& completionHandler)
{
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
m_host->sendCommandToBackend("isShowingJavaScriptDialog"_s, WTFMove(parameters), [this, protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) mutable {
if (response.isError || !response.responseObject) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
bool isShowingJavaScriptDialog;
if (!response.responseObject->getBoolean("result", isShowingJavaScriptDialog)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
if (!isShowingJavaScriptDialog) {
completionHandler(CommandResult::success());
return;
}
handleUnexpectedAlertOpen(WTFMove(completionHandler));
});
}
void Session::handleUnexpectedAlertOpen(Function<void (CommandResult&&)>&& completionHandler)
{
switch (capabilities().unhandledPromptBehavior.valueOr(UnhandledPromptBehavior::DismissAndNotify)) {
case UnhandledPromptBehavior::Dismiss:
dismissAlert(WTFMove(completionHandler));
break;
case UnhandledPromptBehavior::Accept:
acceptAlert(WTFMove(completionHandler));
break;
case UnhandledPromptBehavior::DismissAndNotify:
dismissAndNotifyAlert(WTFMove(completionHandler));
break;
case UnhandledPromptBehavior::AcceptAndNotify:
acceptAndNotifyAlert(WTFMove(completionHandler));
break;
case UnhandledPromptBehavior::Ignore:
reportUnexpectedAlertOpen(WTFMove(completionHandler));
break;
}
}
void Session::dismissAndNotifyAlert(Function<void (CommandResult&&)>&& completionHandler)
{
reportUnexpectedAlertOpen([this, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
dismissAlert([errorResult = WTFMove(result), completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
completionHandler(WTFMove(errorResult));
});
});
}
void Session::acceptAndNotifyAlert(Function<void (CommandResult&&)>&& completionHandler)
{
reportUnexpectedAlertOpen([this, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
acceptAlert([errorResult = WTFMove(result), completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
completionHandler(WTFMove(errorResult));
});
});
}
void Session::reportUnexpectedAlertOpen(Function<void (CommandResult&&)>&& completionHandler)
{
getAlertText([completionHandler = WTFMove(completionHandler)](CommandResult&& result) {
Optional<String> alertText;
if (!result.isError()) {
String valueString;
if (result.result()->asString(valueString))
alertText = valueString;
}
auto errorResult = CommandResult::fail(CommandResult::ErrorCode::UnexpectedAlertOpen);
if (alertText) {
RefPtr<JSON::Object> additonalData = JSON::Object::create();
additonalData->setString("text"_s, alertText.value());
errorResult.setAdditionalErrorData(WTFMove(additonalData));
}
completionHandler(WTFMove(errorResult));
});
}
void Session::go(const String& url, Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, url, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("handle"_s, m_toplevelBrowsingContext.value());
parameters->setString("url"_s, url);
parameters->setInteger("pageLoadTimeout"_s, m_pageLoadTimeout.millisecondsAs<int>());
if (auto pageLoadStrategy = pageLoadStrategyString())
parameters->setString("pageLoadStrategy"_s, pageLoadStrategy.value());
m_host->sendCommandToBackend("navigateBrowsingContext"_s, WTFMove(parameters), [this, protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
switchToBrowsingContext(WTF::nullopt);
completionHandler(CommandResult::success());
});
});
}
void Session::getCurrentURL(Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("handle"_s, m_toplevelBrowsingContext.value());
m_host->sendCommandToBackend("getBrowsingContext"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError || !response.responseObject) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
RefPtr<JSON::Object> browsingContext;
if (!response.responseObject->getObject("context", browsingContext)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
String url;
if (!browsingContext->getString("url", url)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
completionHandler(CommandResult::success(JSON::Value::create(url)));
});
});
}
void Session::back(Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("handle"_s, m_toplevelBrowsingContext.value());
parameters->setInteger("pageLoadTimeout"_s, m_pageLoadTimeout.millisecondsAs<int>());
if (auto pageLoadStrategy = pageLoadStrategyString())
parameters->setString("pageLoadStrategy"_s, pageLoadStrategy.value());
m_host->sendCommandToBackend("goBackInBrowsingContext"_s, WTFMove(parameters), [this, protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
switchToBrowsingContext(WTF::nullopt);
completionHandler(CommandResult::success());
});
});
}
void Session::forward(Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("handle"_s, m_toplevelBrowsingContext.value());
parameters->setInteger("pageLoadTimeout"_s, m_pageLoadTimeout.millisecondsAs<int>());
if (auto pageLoadStrategy = pageLoadStrategyString())
parameters->setString("pageLoadStrategy"_s, pageLoadStrategy.value());
m_host->sendCommandToBackend("goForwardInBrowsingContext"_s, WTFMove(parameters), [this, protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
switchToBrowsingContext(WTF::nullopt);
completionHandler(CommandResult::success());
});
});
}
void Session::refresh(Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("handle"_s, m_toplevelBrowsingContext.value());
parameters->setInteger("pageLoadTimeout"_s, m_pageLoadTimeout.millisecondsAs<int>());
if (auto pageLoadStrategy = pageLoadStrategyString())
parameters->setString("pageLoadStrategy"_s, pageLoadStrategy.value());
m_host->sendCommandToBackend("reloadBrowsingContext"_s, WTFMove(parameters), [this, protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
switchToBrowsingContext(WTF::nullopt);
completionHandler(CommandResult::success());
});
});
}
void Session::getTitle(Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
parameters->setString("function"_s, "function() { return document.title; }"_s);
parameters->setArray("arguments"_s, JSON::Array::create());
m_host->sendCommandToBackend("evaluateJavaScriptFunction"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError || !response.responseObject) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
String valueString;
if (!response.responseObject->getString("result"_s, valueString)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
RefPtr<JSON::Value> resultValue;
if (!JSON::Value::parseJSON(valueString, resultValue)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
completionHandler(CommandResult::success(WTFMove(resultValue)));
});
});
}
void Session::getWindowHandle(Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("handle"_s, m_toplevelBrowsingContext.value());
m_host->sendCommandToBackend("getBrowsingContext"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError || !response.responseObject) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
RefPtr<JSON::Object> browsingContext;
if (!response.responseObject->getObject("context"_s, browsingContext)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
String handle;
if (!browsingContext->getString("handle"_s, handle)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
completionHandler(CommandResult::success(JSON::Value::create(handle)));
});
}
void Session::closeTopLevelBrowsingContext(const String& toplevelBrowsingContext, Function<void (CommandResult&&)>&& completionHandler)
{
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("handle"_s, toplevelBrowsingContext);
m_host->sendCommandToBackend("closeBrowsingContext"_s, WTFMove(parameters), [this, protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) mutable {
if (!m_host->isConnected()) {
// Closing the browsing context made the browser quit.
completionHandler(CommandResult::success(JSON::Array::create()));
return;
}
if (response.isError) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
getWindowHandles([this, completionHandler = WTFMove(completionHandler)](CommandResult&& result) {
if (!m_host->isConnected()) {
// Closing the browsing context made the browser quit.
completionHandler(CommandResult::success(JSON::Array::create()));
return;
}
completionHandler(WTFMove(result));
});
});
}
void Session::closeWindow(Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
auto toplevelBrowsingContext = std::exchange(m_toplevelBrowsingContext, WTF::nullopt);
closeTopLevelBrowsingContext(toplevelBrowsingContext.value(), WTFMove(completionHandler));
});
}
void Session::switchToWindow(const String& windowHandle, Function<void (CommandResult&&)>&& completionHandler)
{
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, windowHandle);
m_host->sendCommandToBackend("switchToBrowsingContext"_s, WTFMove(parameters), [this, protectedThis = makeRef(*this), windowHandle, completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
switchToTopLevelBrowsingContext(windowHandle);
completionHandler(CommandResult::success());
});
}
void Session::getWindowHandles(Function<void (CommandResult&&)>&& completionHandler)
{
m_host->sendCommandToBackend("getBrowsingContexts"_s, JSON::Object::create(), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError || !response.responseObject) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
RefPtr<JSON::Array> browsingContextArray;
if (!response.responseObject->getArray("contexts"_s, browsingContextArray)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
RefPtr<JSON::Array> windowHandles = JSON::Array::create();
for (unsigned i = 0; i < browsingContextArray->length(); ++i) {
RefPtr<JSON::Value> browsingContextValue = browsingContextArray->get(i);
RefPtr<JSON::Object> browsingContext;
if (!browsingContextValue->asObject(browsingContext)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
String handle;
if (!browsingContext->getString("handle"_s, handle)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
windowHandles->pushString(handle);
}
completionHandler(CommandResult::success(WTFMove(windowHandles)));
});
}
void Session::switchToFrame(RefPtr<JSON::Value>&& frameID, Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
if (frameID->isNull()) {
switchToBrowsingContext(WTF::nullopt);
completionHandler(CommandResult::success());
return;
}
handleUserPrompts([this, frameID = WTFMove(frameID), completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
if (m_currentBrowsingContext)
parameters->setString("frameHandle"_s, m_currentBrowsingContext.value());
int frameIndex;
if (frameID->asInteger(frameIndex)) {
if (frameIndex < 0 || frameIndex > USHRT_MAX) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchFrame));
return;
}
parameters->setInteger("ordinal"_s, frameIndex);
} else {
String frameElementID = extractElementID(*frameID);
if (!frameElementID.isEmpty())
parameters->setString("nodeHandle"_s, frameElementID);
else {
String frameName;
if (!frameID->asString(frameName)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchFrame));
return;
}
parameters->setString("name"_s, frameName);
}
}
m_host->sendCommandToBackend("resolveChildFrameHandle"_s, WTFMove(parameters), [this, protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError || !response.responseObject) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
String frameHandle;
if (!response.responseObject->getString("result"_s, frameHandle)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
switchToBrowsingContext(frameHandle);
completionHandler(CommandResult::success());
});
});
}
void Session::switchToParentFrame(Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
if (!m_currentBrowsingContext) {
completionHandler(CommandResult::success());
return;
}
handleUserPrompts([this, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
parameters->setString("frameHandle"_s, m_currentBrowsingContext.value());
m_host->sendCommandToBackend("resolveParentFrameHandle"_s, WTFMove(parameters), [this, protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError || !response.responseObject) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
String frameHandle;
if (!response.responseObject->getString("result"_s, frameHandle)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
switchToBrowsingContext(frameHandle);
completionHandler(CommandResult::success());
});
});
}
void Session::getToplevelBrowsingContextRect(Function<void (CommandResult&&)>&& completionHandler)
{
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("handle"_s, m_toplevelBrowsingContext.value());
m_host->sendCommandToBackend("getBrowsingContext"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError || !response.responseObject) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
RefPtr<JSON::Object> browsingContext;
if (!response.responseObject->getObject("context"_s, browsingContext)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
RefPtr<JSON::Object> windowOrigin;
double x, y;
if (!browsingContext->getObject("windowOrigin"_s, windowOrigin)
|| !windowOrigin->getDouble("x"_s, x)
|| !windowOrigin->getDouble("y"_s, y)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
RefPtr<JSON::Object> windowSize;
double width, height;
if (!browsingContext->getObject("windowSize"_s, windowSize)
|| !windowSize->getDouble("width"_s, width)
|| !windowSize->getDouble("height"_s, height)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
auto windowRect = JSON::Object::create();
windowRect->setDouble("x"_s, x);
windowRect->setDouble("y"_s, y);
windowRect->setDouble("width"_s, width);
windowRect->setDouble("height"_s, height);
completionHandler(CommandResult::success(WTFMove(windowRect)));
});
}
void Session::getWindowRect(Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
getToplevelBrowsingContextRect(WTFMove(completionHandler));
});
}
void Session::setWindowRect(Optional<double> x, Optional<double> y, Optional<double> width, Optional<double> height, Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, x, y, width, height, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("handle"_s, m_toplevelBrowsingContext.value());
if (x && y) {
RefPtr<JSON::Object> windowOrigin = JSON::Object::create();
windowOrigin->setDouble("x", x.value());
windowOrigin->setDouble("y", y.value());
parameters->setObject("origin"_s, WTFMove(windowOrigin));
}
if (width && height) {
RefPtr<JSON::Object> windowSize = JSON::Object::create();
windowSize->setDouble("width", width.value());
windowSize->setDouble("height", height.value());
parameters->setObject("size"_s, WTFMove(windowSize));
}
m_host->sendCommandToBackend("setWindowFrameOfBrowsingContext"_s, WTFMove(parameters), [this, protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)] (SessionHost::CommandResponse&& response) mutable {
if (response.isError) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
getToplevelBrowsingContextRect(WTFMove(completionHandler));
});
});
}
void Session::maximizeWindow(Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("handle"_s, m_toplevelBrowsingContext.value());
m_host->sendCommandToBackend("maximizeWindowOfBrowsingContext"_s, WTFMove(parameters), [this, protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)] (SessionHost::CommandResponse&& response) mutable {
if (response.isError) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
getToplevelBrowsingContextRect(WTFMove(completionHandler));
});
});
}
void Session::minimizeWindow(Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("handle"_s, m_toplevelBrowsingContext.value());
m_host->sendCommandToBackend("hideWindowOfBrowsingContext"_s, WTFMove(parameters), [this, protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)] (SessionHost::CommandResponse&& response) mutable {
if (response.isError) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
getToplevelBrowsingContextRect(WTFMove(completionHandler));
});
});
}
void Session::fullscreenWindow(Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
parameters->setString("function"_s, String(EnterFullscreenJavaScript, sizeof(EnterFullscreenJavaScript)));
parameters->setArray("arguments"_s, JSON::Array::create());
parameters->setBoolean("expectsImplicitCallbackArgument"_s, true);
m_host->sendCommandToBackend("evaluateJavaScriptFunction"_s, WTFMove(parameters), [this, protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) mutable {
if (response.isError || !response.responseObject) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
String valueString;
if (!response.responseObject->getString("result"_s, valueString)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
RefPtr<JSON::Value> resultValue;
if (!JSON::Value::parseJSON(valueString, resultValue)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
getToplevelBrowsingContextRect(WTFMove(completionHandler));
});
});
}
RefPtr<JSON::Object> Session::createElement(RefPtr<JSON::Value>&& value)
{
RefPtr<JSON::Object> valueObject;
if (!value->asObject(valueObject))
return nullptr;
String elementID;
if (!valueObject->getString("session-node-" + id(), elementID))
return nullptr;
RefPtr<JSON::Object> elementObject = JSON::Object::create();
elementObject->setString(webElementIdentifier(), elementID);
return elementObject;
}
RefPtr<JSON::Object> Session::createElement(const String& elementID)
{
RefPtr<JSON::Object> elementObject = JSON::Object::create();
elementObject->setString("session-node-" + id(), elementID);
return elementObject;
}
RefPtr<JSON::Object> Session::extractElement(JSON::Value& value)
{
String elementID = extractElementID(value);
return !elementID.isEmpty() ? createElement(elementID) : nullptr;
}
String Session::extractElementID(JSON::Value& value)
{
RefPtr<JSON::Object> valueObject;
if (!value.asObject(valueObject))
return emptyString();
String elementID;
if (!valueObject->getString(webElementIdentifier(), elementID))
return emptyString();
return elementID;
}
void Session::computeElementLayout(const String& elementID, OptionSet<ElementLayoutOption> options, Function<void (Optional<Rect>&&, Optional<Point>&&, bool, RefPtr<JSON::Object>&&)>&& completionHandler)
{
ASSERT(m_toplevelBrowsingContext.value());
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
parameters->setString("frameHandle"_s, m_currentBrowsingContext.valueOr(emptyString()));
parameters->setString("nodeHandle"_s, elementID);
parameters->setBoolean("scrollIntoViewIfNeeded"_s, options.contains(ElementLayoutOption::ScrollIntoViewIfNeeded));
parameters->setString("coordinateSystem"_s, options.contains(ElementLayoutOption::UseViewportCoordinates) ? "LayoutViewport"_s : "Page"_s);
m_host->sendCommandToBackend("computeElementLayout"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) mutable {
if (response.isError || !response.responseObject) {
completionHandler(WTF::nullopt, WTF::nullopt, false, WTFMove(response.responseObject));
return;
}
RefPtr<JSON::Object> rectObject;
if (!response.responseObject->getObject("rect"_s, rectObject)) {
completionHandler(WTF::nullopt, WTF::nullopt, false, nullptr);
return;
}
Optional<int> elementX;
Optional<int> elementY;
RefPtr<JSON::Object> elementPosition;
if (rectObject->getObject("origin"_s, elementPosition)) {
int x, y;
if (elementPosition->getInteger("x"_s, x) && elementPosition->getInteger("y"_s, y)) {
elementX = x;
elementY = y;
}
}
if (!elementX || !elementY) {
completionHandler(WTF::nullopt, WTF::nullopt, false, nullptr);
return;
}
Optional<int> elementWidth;
Optional<int> elementHeight;
RefPtr<JSON::Object> elementSize;
if (rectObject->getObject("size"_s, elementSize)) {
int width, height;
if (elementSize->getInteger("width"_s, width) && elementSize->getInteger("height"_s, height)) {
elementWidth = width;
elementHeight = height;
}
}
if (!elementWidth || !elementHeight) {
completionHandler(WTF::nullopt, WTF::nullopt, false, nullptr);
return;
}
Rect rect = { { elementX.value(), elementY.value() }, { elementWidth.value(), elementHeight.value() } };
bool isObscured;
if (!response.responseObject->getBoolean("isObscured"_s, isObscured)) {
completionHandler(WTF::nullopt, WTF::nullopt, false, nullptr);
return;
}
RefPtr<JSON::Object> inViewCenterPointObject;
if (!response.responseObject->getObject("inViewCenterPoint"_s, inViewCenterPointObject)) {
completionHandler(rect, WTF::nullopt, isObscured, nullptr);
return;
}
int inViewCenterPointX, inViewCenterPointY;
if (!inViewCenterPointObject->getInteger("x"_s, inViewCenterPointX)
|| !inViewCenterPointObject->getInteger("y"_s, inViewCenterPointY)) {
completionHandler(WTF::nullopt, WTF::nullopt, isObscured, nullptr);
return;
}
Point inViewCenterPoint = { inViewCenterPointX, inViewCenterPointY };
completionHandler(rect, inViewCenterPoint, isObscured, nullptr);
});
}
void Session::findElements(const String& strategy, const String& selector, FindElementsMode mode, const String& rootElementID, Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, strategy, selector, mode, rootElementID, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Array> arguments = JSON::Array::create();
arguments->pushString(JSON::Value::create(strategy)->toJSONString());
if (rootElementID.isEmpty())
arguments->pushString(JSON::Value::null()->toJSONString());
else
arguments->pushString(createElement(rootElementID)->toJSONString());
arguments->pushString(JSON::Value::create(selector)->toJSONString());
arguments->pushString(JSON::Value::create(mode == FindElementsMode::Single)->toJSONString());
arguments->pushString(JSON::Value::create(m_implicitWaitTimeout.milliseconds())->toJSONString());
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
if (m_currentBrowsingContext)
parameters->setString("frameHandle"_s, m_currentBrowsingContext.value());
parameters->setString("function"_s, String(FindNodesJavaScript, sizeof(FindNodesJavaScript)));
parameters->setArray("arguments"_s, WTFMove(arguments));
parameters->setBoolean("expectsImplicitCallbackArgument"_s, true);
// If there's an implicit wait, use one second more as callback timeout.
if (m_implicitWaitTimeout)
parameters->setInteger("callbackTimeout"_s, Seconds(m_implicitWaitTimeout + 1_s).millisecondsAs<int>());
m_host->sendCommandToBackend("evaluateJavaScriptFunction"_s, WTFMove(parameters), [this, protectedThis = makeRef(*this), mode, completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError || !response.responseObject) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
String valueString;
if (!response.responseObject->getString("result"_s, valueString)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
RefPtr<JSON::Value> resultValue;
if (!JSON::Value::parseJSON(valueString, resultValue)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
switch (mode) {
case FindElementsMode::Single: {
RefPtr<JSON::Object> elementObject = createElement(WTFMove(resultValue));
if (!elementObject) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchElement));
return;
}
completionHandler(CommandResult::success(WTFMove(elementObject)));
break;
}
case FindElementsMode::Multiple: {
RefPtr<JSON::Array> elementsArray;
if (!resultValue->asArray(elementsArray)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchElement));
return;
}
RefPtr<JSON::Array> elementObjectsArray = JSON::Array::create();
unsigned elementsArrayLength = elementsArray->length();
for (unsigned i = 0; i < elementsArrayLength; ++i) {
if (auto elementObject = createElement(elementsArray->get(i)))
elementObjectsArray->pushObject(WTFMove(elementObject));
}
completionHandler(CommandResult::success(WTFMove(elementObjectsArray)));
break;
}
}
});
});
}
void Session::getActiveElement(Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
parameters->setString("function"_s, "function() { return document.activeElement; }"_s);
parameters->setArray("arguments"_s, JSON::Array::create());
m_host->sendCommandToBackend("evaluateJavaScriptFunction"_s, WTFMove(parameters), [this, protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError || !response.responseObject) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
String valueString;
if (!response.responseObject->getString("result"_s, valueString)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
RefPtr<JSON::Value> resultValue;
if (!JSON::Value::parseJSON(valueString, resultValue)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
RefPtr<JSON::Object> elementObject = createElement(WTFMove(resultValue));
if (!elementObject) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchElement));
return;
}
completionHandler(CommandResult::success(WTFMove(elementObject)));
});
});
}
void Session::isElementSelected(const String& elementID, Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, elementID, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Array> arguments = JSON::Array::create();
arguments->pushString(createElement(elementID)->toJSONString());
arguments->pushString(JSON::Value::create("selected")->toJSONString());
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
if (m_currentBrowsingContext)
parameters->setString("frameHandle"_s, m_currentBrowsingContext.value());
parameters->setString("function"_s, String(ElementAttributeJavaScript, sizeof(ElementAttributeJavaScript)));
parameters->setArray("arguments"_s, WTFMove(arguments));
m_host->sendCommandToBackend("evaluateJavaScriptFunction"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError || !response.responseObject) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
String valueString;
if (!response.responseObject->getString("result"_s, valueString)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
RefPtr<JSON::Value> resultValue;
if (!JSON::Value::parseJSON(valueString, resultValue)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
if (resultValue->isNull()) {
completionHandler(CommandResult::success(JSON::Value::create(false)));
return;
}
String booleanResult;
if (!resultValue->asString(booleanResult) || booleanResult != "true") {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
completionHandler(CommandResult::success(JSON::Value::create(true)));
});
});
}
void Session::getElementText(const String& elementID, Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, elementID, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Array> arguments = JSON::Array::create();
arguments->pushString(createElement(elementID)->toJSONString());
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
if (m_currentBrowsingContext)
parameters->setString("frameHandle"_s, m_currentBrowsingContext.value());
// FIXME: Add an atom to properly implement this instead of just using innerText.
parameters->setString("function"_s, "function(element) { return element.innerText.replace(/^[^\\S\\xa0]+|[^\\S\\xa0]+$/g, '') }"_s);
parameters->setArray("arguments"_s, WTFMove(arguments));
m_host->sendCommandToBackend("evaluateJavaScriptFunction"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError || !response.responseObject) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
String valueString;
if (!response.responseObject->getString("result"_s, valueString)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
RefPtr<JSON::Value> resultValue;
if (!JSON::Value::parseJSON(valueString, resultValue)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
completionHandler(CommandResult::success(WTFMove(resultValue)));
});
});
}
void Session::getElementTagName(const String& elementID, Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, elementID, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Array> arguments = JSON::Array::create();
arguments->pushString(createElement(elementID)->toJSONString());
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
if (m_currentBrowsingContext)
parameters->setString("frameHandle"_s, m_currentBrowsingContext.value());
parameters->setString("function"_s, "function(element) { return element.tagName.toLowerCase() }"_s);
parameters->setArray("arguments"_s, WTFMove(arguments));
m_host->sendCommandToBackend("evaluateJavaScriptFunction"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError || !response.responseObject) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
String valueString;
if (!response.responseObject->getString("result"_s, valueString)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
RefPtr<JSON::Value> resultValue;
if (!JSON::Value::parseJSON(valueString, resultValue)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
completionHandler(CommandResult::success(WTFMove(resultValue)));
});
});
}
void Session::getElementRect(const String& elementID, Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, elementID, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
computeElementLayout(elementID, { }, [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](Optional<Rect>&& rect, Optional<Point>&&, bool, RefPtr<JSON::Object>&& error) {
if (!rect || error) {
completionHandler(CommandResult::fail(WTFMove(error)));
return;
}
RefPtr<JSON::Object> rectObject = JSON::Object::create();
rectObject->setInteger("x"_s, rect.value().origin.x);
rectObject->setInteger("y"_s, rect.value().origin.y);
rectObject->setInteger("width"_s, rect.value().size.width);
rectObject->setInteger("height"_s, rect.value().size.height);
completionHandler(CommandResult::success(WTFMove(rectObject)));
});
});
}
void Session::isElementEnabled(const String& elementID, Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, elementID, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Array> arguments = JSON::Array::create();
arguments->pushString(createElement(elementID)->toJSONString());
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
if (m_currentBrowsingContext)
parameters->setString("frameHandle"_s, m_currentBrowsingContext.value());
parameters->setString("function"_s, "function(element) { return element.disabled === undefined ? true : !element.disabled }"_s);
parameters->setArray("arguments"_s, WTFMove(arguments));
m_host->sendCommandToBackend("evaluateJavaScriptFunction"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError || !response.responseObject) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
String valueString;
if (!response.responseObject->getString("result"_s, valueString)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
RefPtr<JSON::Value> resultValue;
if (!JSON::Value::parseJSON(valueString, resultValue)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
completionHandler(CommandResult::success(WTFMove(resultValue)));
});
});
}
void Session::isElementDisplayed(const String& elementID, Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, elementID, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Array> arguments = JSON::Array::create();
arguments->pushString(createElement(elementID)->toJSONString());
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
if (m_currentBrowsingContext)
parameters->setString("frameHandle"_s, m_currentBrowsingContext.value());
parameters->setString("function"_s, String(ElementDisplayedJavaScript, sizeof(ElementDisplayedJavaScript)));
parameters->setArray("arguments"_s, WTFMove(arguments));
m_host->sendCommandToBackend("evaluateJavaScriptFunction"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError || !response.responseObject) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
String valueString;
if (!response.responseObject->getString("result"_s, valueString)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
RefPtr<JSON::Value> resultValue;
if (!JSON::Value::parseJSON(valueString, resultValue)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
completionHandler(CommandResult::success(WTFMove(resultValue)));
});
});
}
void Session::getElementAttribute(const String& elementID, const String& attribute, Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, elementID, attribute, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Array> arguments = JSON::Array::create();
arguments->pushString(createElement(elementID)->toJSONString());
arguments->pushString(JSON::Value::create(attribute)->toJSONString());
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
if (m_currentBrowsingContext)
parameters->setString("frameHandle"_s, m_currentBrowsingContext.value());
parameters->setString("function"_s, String(ElementAttributeJavaScript, sizeof(ElementAttributeJavaScript)));
parameters->setArray("arguments"_s, WTFMove(arguments));
m_host->sendCommandToBackend("evaluateJavaScriptFunction"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError || !response.responseObject) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
String valueString;
if (!response.responseObject->getString("result"_s, valueString)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
RefPtr<JSON::Value> resultValue;
if (!JSON::Value::parseJSON(valueString, resultValue)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
completionHandler(CommandResult::success(WTFMove(resultValue)));
});
});
}
void Session::getElementProperty(const String& elementID, const String& property, Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, elementID, property, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Array> arguments = JSON::Array::create();
arguments->pushString(createElement(elementID)->toJSONString());
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
if (m_currentBrowsingContext)
parameters->setString("frameHandle"_s, m_currentBrowsingContext.value());
parameters->setString("function"_s, makeString("function(element) { return element.", property, "; }"));
parameters->setArray("arguments"_s, WTFMove(arguments));
m_host->sendCommandToBackend("evaluateJavaScriptFunction"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError || !response.responseObject) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
String valueString;
if (!response.responseObject->getString("result"_s, valueString)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
RefPtr<JSON::Value> resultValue;
if (!JSON::Value::parseJSON(valueString, resultValue)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
completionHandler(CommandResult::success(WTFMove(resultValue)));
});
});
}
void Session::getElementCSSValue(const String& elementID, const String& cssProperty, Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, elementID, cssProperty, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Array> arguments = JSON::Array::create();
arguments->pushString(createElement(elementID)->toJSONString());
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
if (m_currentBrowsingContext)
parameters->setString("frameHandle"_s, m_currentBrowsingContext.value());
parameters->setString("function"_s, makeString("function(element) { return document.defaultView.getComputedStyle(element).getPropertyValue('", cssProperty, "'); }"));
parameters->setArray("arguments"_s, WTFMove(arguments));
m_host->sendCommandToBackend("evaluateJavaScriptFunction"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError || !response.responseObject) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
String valueString;
if (!response.responseObject->getString("result"_s, valueString)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
RefPtr<JSON::Value> resultValue;
if (!JSON::Value::parseJSON(valueString, resultValue)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
completionHandler(CommandResult::success(WTFMove(resultValue)));
});
});
}
void Session::waitForNavigationToComplete(Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
if (m_currentBrowsingContext)
parameters->setString("frameHandle"_s, m_currentBrowsingContext.value());
parameters->setInteger("pageLoadTimeout"_s, m_pageLoadTimeout.millisecondsAs<int>());
if (auto pageLoadStrategy = pageLoadStrategyString())
parameters->setString("pageLoadStrategy"_s, pageLoadStrategy.value());
m_host->sendCommandToBackend("waitForNavigationToComplete"_s, WTFMove(parameters), [this, protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError) {
auto result = CommandResult::fail(WTFMove(response.responseObject));
switch (result.errorCode()) {
case CommandResult::ErrorCode::NoSuchWindow:
// Window was closed, reset the top level browsing context and ignore the error.
m_toplevelBrowsingContext = WTF::nullopt;
break;
case CommandResult::ErrorCode::NoSuchFrame:
// Navigation destroyed the current frame, switch to top level browsing context and ignore the error.
switchToBrowsingContext(WTF::nullopt);
break;
default:
completionHandler(WTFMove(result));
return;
}
}
completionHandler(CommandResult::success());
});
}
void Session::selectOptionElement(const String& elementID, Function<void (CommandResult&&)>&& completionHandler)
{
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
parameters->setString("frameHandle"_s, m_currentBrowsingContext.valueOr(emptyString()));
parameters->setString("nodeHandle"_s, elementID);
m_host->sendCommandToBackend("selectOptionElement"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
completionHandler(CommandResult::success());
});
}
void Session::elementClick(const String& elementID, Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, elementID, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
OptionSet<ElementLayoutOption> options = { ElementLayoutOption::ScrollIntoViewIfNeeded, ElementLayoutOption::UseViewportCoordinates };
computeElementLayout(elementID, options, [this, protectedThis = makeRef(*this), elementID, completionHandler = WTFMove(completionHandler)](Optional<Rect>&& rect, Optional<Point>&& inViewCenter, bool isObscured, RefPtr<JSON::Object>&& error) mutable {
if (!rect || error) {
completionHandler(CommandResult::fail(WTFMove(error)));
return;
}
if (isObscured) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::ElementClickIntercepted));
return;
}
if (!inViewCenter) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::ElementNotInteractable));
return;
}
getElementTagName(elementID, [this, elementID, inViewCenter = WTFMove(inViewCenter), completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
bool isOptionElement = false;
if (!result.isError()) {
String tagName;
if (result.result()->asString(tagName))
isOptionElement = tagName == "option";
}
Function<void (CommandResult&&)> continueAfterClickFunction = [this, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
waitForNavigationToComplete(WTFMove(completionHandler));
};
if (isOptionElement)
selectOptionElement(elementID, WTFMove(continueAfterClickFunction));
else
performMouseInteraction(inViewCenter.value().x, inViewCenter.value().y, MouseButton::Left, MouseInteraction::SingleClick, WTFMove(continueAfterClickFunction));
});
});
});
}
void Session::elementClear(const String& elementID, Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, elementID, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Array> arguments = JSON::Array::create();
arguments->pushString(createElement(elementID)->toJSONString());
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
if (m_currentBrowsingContext)
parameters->setString("frameHandle"_s, m_currentBrowsingContext.value());
parameters->setString("function"_s, String(FormElementClearJavaScript, sizeof(FormElementClearJavaScript)));
parameters->setArray("arguments"_s, WTFMove(arguments));
m_host->sendCommandToBackend("evaluateJavaScriptFunction"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
completionHandler(CommandResult::success());
});
});
}
String Session::virtualKeyForKey(UChar key, KeyModifier& modifier)
{
// §17.4.2 Keyboard Actions.
// https://www.w3.org/TR/webdriver/#keyboard-actions
modifier = KeyModifier::None;
switch (key) {
case 0xE001U:
return "Cancel"_s;
case 0xE002U:
return "Help"_s;
case 0xE003U:
return "Backspace"_s;
case 0xE004U:
return "Tab"_s;
case 0xE005U:
return "Clear"_s;
case 0xE006U:
return "Return"_s;
case 0xE007U:
return "Enter"_s;
case 0xE008U:
case 0xE050U:
modifier = KeyModifier::Shift;
return "Shift"_s;
case 0xE009U:
case 0xE051U:
modifier = KeyModifier::Control;
return "Control"_s;
case 0xE00AU:
case 0xE052U:
modifier = KeyModifier::Alternate;
return "Alternate"_s;
case 0xE00BU:
return "Pause"_s;
case 0xE00CU:
return "Escape"_s;
case 0xE00DU:
return "Space"_s;
case 0xE00EU:
case 0xE054U:
return "PageUp"_s;
case 0xE00FU:
case 0xE055U:
return "PageDown"_s;
case 0xE010U:
case 0xE056U:
return "End"_s;
case 0xE011U:
case 0xE057U:
return "Home"_s;
case 0xE012U:
case 0xE058U:
return "LeftArrow"_s;
case 0xE013U:
case 0xE059U:
return "UpArrow"_s;
case 0xE014U:
case 0xE05AU:
return "RightArrow"_s;
case 0xE015U:
case 0xE05BU:
return "DownArrow"_s;
case 0xE016U:
case 0xE05CU:
return "Insert"_s;
case 0xE017U:
case 0xE05DU:
return "Delete"_s;
case 0xE018U:
return "Semicolon"_s;
case 0xE019U:
return "Equals"_s;
case 0xE01AU:
return "NumberPad0"_s;
case 0xE01BU:
return "NumberPad1"_s;
case 0xE01CU:
return "NumberPad2"_s;
case 0xE01DU:
return "NumberPad3"_s;
case 0xE01EU:
return "NumberPad4"_s;
case 0xE01FU:
return "NumberPad5"_s;
case 0xE020U:
return "NumberPad6"_s;
case 0xE021U:
return "NumberPad7"_s;
case 0xE022U:
return "NumberPad8"_s;
case 0xE023U:
return "NumberPad9"_s;
case 0xE024U:
return "NumberPadMultiply"_s;
case 0xE025U:
return "NumberPadAdd"_s;
case 0xE026U:
return "NumberPadSeparator"_s;
case 0xE027U:
return "NumberPadSubtract"_s;
case 0xE028U:
return "NumberPadDecimal"_s;
case 0xE029U:
return "NumberPadDivide"_s;
case 0xE031U:
return "Function1"_s;
case 0xE032U:
return "Function2"_s;
case 0xE033U:
return "Function3"_s;
case 0xE034U:
return "Function4"_s;
case 0xE035U:
return "Function5"_s;
case 0xE036U:
return "Function6"_s;
case 0xE037U:
return "Function7"_s;
case 0xE038U:
return "Function8"_s;
case 0xE039U:
return "Function9"_s;
case 0xE03AU:
return "Function10"_s;
case 0xE03BU:
return "Function11"_s;
case 0xE03CU:
return "Function12"_s;
case 0xE03DU:
case 0xE053U:
modifier = KeyModifier::Meta;
return "Meta"_s;
default:
break;
}
return String();
}
void Session::elementSendKeys(const String& elementID, const String& text, Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, elementID, text, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
// FIXME: move this to an atom.
static const char focusScript[] =
"function focus(element) {"
" let doc = element.ownerDocument || element;"
" let prevActiveElement = doc.activeElement;"
" if (element != prevActiveElement && prevActiveElement)"
" prevActiveElement.blur();"
" element.focus();"
" let tagName = element.tagName.toUpperCase();"
" let isTextElement = tagName === 'TEXTAREA' || (tagName === 'INPUT' && element.type === 'text');"
" if (isTextElement && element.selectionEnd == 0)"
" element.setSelectionRange(element.value.length, element.value.length);"
" if (element != doc.activeElement)"
" throw new Error('cannot focus element');"
"}";
RefPtr<JSON::Array> arguments = JSON::Array::create();
arguments->pushString(createElement(elementID)->toJSONString());
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
if (m_currentBrowsingContext)
parameters->setString("frameHandle"_s, m_currentBrowsingContext.value());
parameters->setString("function"_s, focusScript);
parameters->setArray("arguments"_s, WTFMove(arguments));
m_host->sendCommandToBackend("evaluateJavaScriptFunction"_s, WTFMove(parameters), [this, protectedThis = makeRef(*this), text, completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) mutable {
if (response.isError || !response.responseObject) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
unsigned stickyModifiers = 0;
auto textLength = text.length();
Vector<KeyboardInteraction> interactions;
interactions.reserveInitialCapacity(textLength);
for (unsigned i = 0; i < textLength; ++i) {
auto key = text[i];
KeyboardInteraction interaction;
KeyModifier modifier;
auto virtualKey = virtualKeyForKey(key, modifier);
if (!virtualKey.isNull()) {
interaction.key = virtualKey;
if (modifier != KeyModifier::None) {
stickyModifiers ^= modifier;
if (stickyModifiers & modifier)
interaction.type = KeyboardInteractionType::KeyPress;
else
interaction.type = KeyboardInteractionType::KeyRelease;
}
} else
interaction.text = String(&key, 1);
interactions.uncheckedAppend(WTFMove(interaction));
}
// Reset sticky modifiers if needed.
if (stickyModifiers) {
if (stickyModifiers & KeyModifier::Shift)
interactions.append({ KeyboardInteractionType::KeyRelease, WTF::nullopt, Optional<String>("Shift"_s) });
if (stickyModifiers & KeyModifier::Control)
interactions.append({ KeyboardInteractionType::KeyRelease, WTF::nullopt, Optional<String>("Control"_s) });
if (stickyModifiers & KeyModifier::Alternate)
interactions.append({ KeyboardInteractionType::KeyRelease, WTF::nullopt, Optional<String>("Alternate"_s) });
if (stickyModifiers & KeyModifier::Meta)
interactions.append({ KeyboardInteractionType::KeyRelease, WTF::nullopt, Optional<String>("Meta"_s) });
}
performKeyboardInteractions(WTFMove(interactions), WTFMove(completionHandler));
});
});
}
RefPtr<JSON::Value> Session::handleScriptResult(RefPtr<JSON::Value>&& resultValue)
{
RefPtr<JSON::Array> resultArray;
if (resultValue->asArray(resultArray)) {
RefPtr<JSON::Array> returnValueArray = JSON::Array::create();
unsigned resultArrayLength = resultArray->length();
for (unsigned i = 0; i < resultArrayLength; ++i)
returnValueArray->pushValue(handleScriptResult(resultArray->get(i)));
return returnValueArray;
}
if (auto element = createElement(RefPtr<JSON::Value>(resultValue)))
return element;
RefPtr<JSON::Object> resultObject;
if (resultValue->asObject(resultObject)) {
RefPtr<JSON::Object> returnValueObject = JSON::Object::create();
auto end = resultObject->end();
for (auto it = resultObject->begin(); it != end; ++it)
returnValueObject->setValue(it->key, handleScriptResult(WTFMove(it->value)));
return returnValueObject;
}
return WTFMove(resultValue);
}
void Session::executeScript(const String& script, RefPtr<JSON::Array>&& argumentsArray, ExecuteScriptMode mode, Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, script, argumentsArray = WTFMove(argumentsArray), mode, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Array> arguments = JSON::Array::create();
unsigned argumentsLength = argumentsArray->length();
for (unsigned i = 0; i < argumentsLength; ++i) {
if (auto argument = argumentsArray->get(i)) {
if (auto element = extractElement(*argument))
arguments->pushString(element->toJSONString());
else
arguments->pushString(argument->toJSONString());
}
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
if (m_currentBrowsingContext)
parameters->setString("frameHandle"_s, m_currentBrowsingContext.value());
parameters->setString("function"_s, "function(){" + script + '}');
parameters->setArray("arguments"_s, WTFMove(arguments));
if (mode == ExecuteScriptMode::Async) {
parameters->setBoolean("expectsImplicitCallbackArgument"_s, true);
if (m_scriptTimeout)
parameters->setInteger("callbackTimeout"_s, m_scriptTimeout.millisecondsAs<int>());
}
m_host->sendCommandToBackend("evaluateJavaScriptFunction"_s, WTFMove(parameters), [this, protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) mutable {
if (response.isError || !response.responseObject) {
auto result = CommandResult::fail(WTFMove(response.responseObject));
if (result.errorCode() == CommandResult::ErrorCode::UnexpectedAlertOpen)
completionHandler(CommandResult::success());
else
completionHandler(WTFMove(result));
return;
}
String valueString;
if (!response.responseObject->getString("result"_s, valueString)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
RefPtr<JSON::Value> resultValue;
if (!JSON::Value::parseJSON(valueString, resultValue)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
completionHandler(CommandResult::success(handleScriptResult(WTFMove(resultValue))));
});
});
}
static String mouseButtonForAutomation(MouseButton button)
{
switch (button) {
case MouseButton::None:
return "None"_s;
case MouseButton::Left:
return "Left"_s;
case MouseButton::Middle:
return "Middle"_s;
case MouseButton::Right:
return "Right"_s;
}
RELEASE_ASSERT_NOT_REACHED();
}
void Session::performMouseInteraction(int x, int y, MouseButton button, MouseInteraction interaction, Function<void (CommandResult&&)>&& completionHandler)
{
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("handle"_s, m_toplevelBrowsingContext.value());
RefPtr<JSON::Object> position = JSON::Object::create();
position->setInteger("x"_s, x);
position->setInteger("y"_s, y);
parameters->setObject("position"_s, WTFMove(position));
parameters->setString("button"_s, mouseButtonForAutomation(button));
switch (interaction) {
case MouseInteraction::Move:
parameters->setString("interaction"_s, "Move"_s);
break;
case MouseInteraction::Down:
parameters->setString("interaction"_s, "Down"_s);
break;
case MouseInteraction::Up:
parameters->setString("interaction"_s, "Up"_s);
break;
case MouseInteraction::SingleClick:
parameters->setString("interaction"_s, "SingleClick"_s);
break;
case MouseInteraction::DoubleClick:
parameters->setString("interaction"_s, "DoubleClick"_s);
break;
}
parameters->setArray("modifiers"_s, JSON::Array::create());
m_host->sendCommandToBackend("performMouseInteraction"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
completionHandler(CommandResult::success());
});
}
void Session::performKeyboardInteractions(Vector<KeyboardInteraction>&& interactions, Function<void (CommandResult&&)>&& completionHandler)
{
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("handle"_s, m_toplevelBrowsingContext.value());
RefPtr<JSON::Array> interactionsArray = JSON::Array::create();
for (const auto& interaction : interactions) {
RefPtr<JSON::Object> interactionObject = JSON::Object::create();
switch (interaction.type) {
case KeyboardInteractionType::KeyPress:
interactionObject->setString("type"_s, "KeyPress"_s);
break;
case KeyboardInteractionType::KeyRelease:
interactionObject->setString("type"_s, "KeyRelease"_s);
break;
case KeyboardInteractionType::InsertByKey:
interactionObject->setString("type"_s, "InsertByKey"_s);
break;
}
if (interaction.key)
interactionObject->setString("key"_s, interaction.key.value());
if (interaction.text)
interactionObject->setString("text"_s, interaction.text.value());
interactionsArray->pushObject(WTFMove(interactionObject));
}
parameters->setArray("interactions"_s, WTFMove(interactionsArray));
m_host->sendCommandToBackend("performKeyboardInteractions"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
completionHandler(CommandResult::success());
});
}
static Optional<Session::Cookie> parseAutomationCookie(const JSON::Object& cookieObject)
{
Session::Cookie cookie;
if (!cookieObject.getString("name"_s, cookie.name))
return WTF::nullopt;
if (!cookieObject.getString("value"_s, cookie.value))
return WTF::nullopt;
String path;
if (cookieObject.getString("path"_s, path))
cookie.path = path;
String domain;
if (cookieObject.getString("domain"_s, domain))
cookie.domain = domain;
bool secure;
if (cookieObject.getBoolean("secure"_s, secure))
cookie.secure = secure;
bool httpOnly;
if (cookieObject.getBoolean("httpOnly"_s, httpOnly))
cookie.httpOnly = httpOnly;
bool session = false;
cookieObject.getBoolean("session"_s, session);
if (!session) {
double expiry;
if (cookieObject.getDouble("expires"_s, expiry))
cookie.expiry = expiry;
}
return cookie;
}
static RefPtr<JSON::Object> builtAutomationCookie(const Session::Cookie& cookie)
{
RefPtr<JSON::Object> cookieObject = JSON::Object::create();
cookieObject->setString("name"_s, cookie.name);
cookieObject->setString("value"_s, cookie.value);
cookieObject->setString("path"_s, cookie.path.valueOr("/"));
cookieObject->setString("domain"_s, cookie.domain.valueOr(emptyString()));
cookieObject->setBoolean("secure"_s, cookie.secure.valueOr(false));
cookieObject->setBoolean("httpOnly"_s, cookie.httpOnly.valueOr(false));
cookieObject->setBoolean("session"_s, !cookie.expiry);
cookieObject->setDouble("expires"_s, cookie.expiry.valueOr(0));
return cookieObject;
}
static RefPtr<JSON::Object> serializeCookie(const Session::Cookie& cookie)
{
RefPtr<JSON::Object> cookieObject = JSON::Object::create();
cookieObject->setString("name"_s, cookie.name);
cookieObject->setString("value"_s, cookie.value);
if (cookie.path)
cookieObject->setString("path"_s, cookie.path.value());
if (cookie.domain)
cookieObject->setString("domain"_s, cookie.domain.value());
if (cookie.secure)
cookieObject->setBoolean("secure"_s, cookie.secure.value());
if (cookie.httpOnly)
cookieObject->setBoolean("httpOnly"_s, cookie.httpOnly.value());
if (cookie.expiry)
cookieObject->setInteger("expiry"_s, cookie.expiry.value());
return cookieObject;
}
void Session::getAllCookies(Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
m_host->sendCommandToBackend("getAllCookies"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) mutable {
if (response.isError || !response.responseObject) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
RefPtr<JSON::Array> cookiesArray;
if (!response.responseObject->getArray("cookies"_s, cookiesArray)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
RefPtr<JSON::Array> cookies = JSON::Array::create();
for (unsigned i = 0; i < cookiesArray->length(); ++i) {
RefPtr<JSON::Value> cookieValue = cookiesArray->get(i);
RefPtr<JSON::Object> cookieObject;
if (!cookieValue->asObject(cookieObject)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
auto cookie = parseAutomationCookie(*cookieObject);
if (!cookie) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
cookies->pushObject(serializeCookie(cookie.value()));
}
completionHandler(CommandResult::success(WTFMove(cookies)));
});
});
}
void Session::getNamedCookie(const String& name, Function<void (CommandResult&&)>&& completionHandler)
{
getAllCookies([name, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Array> cookiesArray;
result.result()->asArray(cookiesArray);
for (unsigned i = 0; i < cookiesArray->length(); ++i) {
RefPtr<JSON::Value> cookieValue = cookiesArray->get(i);
RefPtr<JSON::Object> cookieObject;
cookieValue->asObject(cookieObject);
String cookieName;
cookieObject->getString("name"_s, cookieName);
if (cookieName == name) {
completionHandler(CommandResult::success(WTFMove(cookieObject)));
return;
}
}
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchCookie));
});
}
void Session::addCookie(const Cookie& cookie, Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, cookie = builtAutomationCookie(cookie), completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
parameters->setObject("cookie"_s, WTFMove(cookie));
m_host->sendCommandToBackend("addSingleCookie"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
completionHandler(CommandResult::success());
});
});
}
void Session::deleteCookie(const String& name, Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, name, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
parameters->setString("cookieName"_s, name);
m_host->sendCommandToBackend("deleteSingleCookie"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
completionHandler(CommandResult::success());
});
});
}
void Session::deleteAllCookies(Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
m_host->sendCommandToBackend("deleteAllCookies"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
completionHandler(CommandResult::success());
});
});
}
InputSource& Session::getOrCreateInputSource(const String& id, InputSource::Type type, Optional<PointerType> pointerType)
{
auto addResult = m_activeInputSources.add(id, InputSource());
if (addResult.isNewEntry)
addResult.iterator->value = { type, pointerType };
return addResult.iterator->value;
}
Session::InputSourceState& Session::inputSourceState(const String& id)
{
return m_inputStateTable.ensure(id, [] { return InputSourceState(); }).iterator->value;
}
static const char* automationSourceType(InputSource::Type type)
{
switch (type) {
case InputSource::Type::None:
return "Null";
case InputSource::Type::Pointer:
return "Mouse";
case InputSource::Type::Key:
return "Keyboard";
}
RELEASE_ASSERT_NOT_REACHED();
}
static const char* automationOriginType(PointerOrigin::Type type)
{
switch (type) {
case PointerOrigin::Type::Viewport:
return "Viewport";
case PointerOrigin::Type::Pointer:
return "Pointer";
case PointerOrigin::Type::Element:
return "Element";
}
RELEASE_ASSERT_NOT_REACHED();
}
void Session::performActions(Vector<Vector<Action>>&& actionsByTick, Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, actionsByTick = WTFMove(actionsByTick), completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
// First check if we have actions and whether we need to resolve any pointer move element origin.
unsigned actionsCount = 0;
for (const auto& tick : actionsByTick)
actionsCount += tick.size();
if (!actionsCount) {
completionHandler(CommandResult::success());
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("handle"_s, m_toplevelBrowsingContext.value());
if (m_currentBrowsingContext)
parameters->setString("frameHandle"_s, m_currentBrowsingContext.value());
RefPtr<JSON::Array> inputSources = JSON::Array::create();
for (const auto& inputSource : m_activeInputSources) {
RefPtr<JSON::Object> inputSourceObject = JSON::Object::create();
inputSourceObject->setString("sourceId"_s, inputSource.key);
inputSourceObject->setString("sourceType"_s, automationSourceType(inputSource.value.type));
inputSources->pushObject(WTFMove(inputSourceObject));
}
parameters->setArray("inputSources"_s, WTFMove(inputSources));
RefPtr<JSON::Array> steps = JSON::Array::create();
for (const auto& tick : actionsByTick) {
RefPtr<JSON::Array> states = JSON::Array::create();
for (const auto& action : tick) {
RefPtr<JSON::Object> state = JSON::Object::create();
auto& currentState = inputSourceState(action.id);
state->setString("sourceId"_s, action.id);
switch (action.type) {
case Action::Type::None:
state->setDouble("duration"_s, action.duration.value());
break;
case Action::Type::Pointer: {
switch (action.subtype) {
case Action::Subtype::PointerUp:
currentState.pressedButton = WTF::nullopt;
break;
case Action::Subtype::PointerDown:
currentState.pressedButton = action.button.value();
break;
case Action::Subtype::PointerMove: {
state->setString("origin"_s, automationOriginType(action.origin->type));
RefPtr<JSON::Object> location = JSON::Object::create();
location->setInteger("x"_s, action.x.value());
location->setInteger("y"_s, action.y.value());
state->setObject("location"_s, WTFMove(location));
if (action.origin->type == PointerOrigin::Type::Element)
state->setString("nodeHandle"_s, action.origin->elementID.value());
FALLTHROUGH;
}
case Action::Subtype::Pause:
if (action.duration)
state->setDouble("duration"_s, action.duration.value());
break;
case Action::Subtype::PointerCancel:
currentState.pressedButton = WTF::nullopt;
break;
case Action::Subtype::KeyUp:
case Action::Subtype::KeyDown:
ASSERT_NOT_REACHED();
}
if (currentState.pressedButton)
state->setString("pressedButton"_s, mouseButtonForAutomation(currentState.pressedButton.value()));
break;
}
case Action::Type::Key:
switch (action.subtype) {
case Action::Subtype::KeyUp:
if (currentState.pressedVirtualKey)
currentState.pressedVirtualKey = WTF::nullopt;
else
currentState.pressedKey = WTF::nullopt;
break;
case Action::Subtype::KeyDown: {
KeyModifier modifier;
auto virtualKey = virtualKeyForKey(action.key.value()[0], modifier);
if (!virtualKey.isNull())
currentState.pressedVirtualKey = virtualKey;
else
currentState.pressedKey = action.key.value();
break;
}
case Action::Subtype::Pause:
if (action.duration)
state->setDouble("duration"_s, action.duration.value());
break;
case Action::Subtype::PointerUp:
case Action::Subtype::PointerDown:
case Action::Subtype::PointerMove:
case Action::Subtype::PointerCancel:
ASSERT_NOT_REACHED();
}
if (currentState.pressedKey)
state->setString("pressedCharKey"_s, currentState.pressedKey.value());
if (currentState.pressedVirtualKey) {
// FIXME: support parsing and tracking multiple virtual keys.
Ref<JSON::Array> virtualKeys = JSON::Array::create();
virtualKeys->pushString(currentState.pressedVirtualKey.value());
state->setArray("pressedVirtualKeys"_s, WTFMove(virtualKeys));
}
break;
}
states->pushObject(WTFMove(state));
}
RefPtr<JSON::Object> stepStates = JSON::Object::create();
stepStates->setArray("states"_s, WTFMove(states));
steps->pushObject(WTFMove(stepStates));
}
parameters->setArray("steps"_s, WTFMove(steps));
m_host->sendCommandToBackend("performInteractionSequence"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)] (SessionHost::CommandResponse&& response) {
if (response.isError) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
completionHandler(CommandResult::success());
});
});
}
void Session::releaseActions(Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
m_activeInputSources.clear();
m_inputStateTable.clear();
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("handle"_s, m_toplevelBrowsingContext.value());
m_host->sendCommandToBackend("cancelInteractionSequence"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
completionHandler(CommandResult::success());
});
}
void Session::dismissAlert(Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
m_host->sendCommandToBackend("dismissCurrentJavaScriptDialog"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
completionHandler(CommandResult::success());
});
}
void Session::acceptAlert(Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
m_host->sendCommandToBackend("acceptCurrentJavaScriptDialog"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
completionHandler(CommandResult::success());
});
}
void Session::getAlertText(Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
m_host->sendCommandToBackend("messageOfCurrentJavaScriptDialog"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError || !response.responseObject) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
String valueString;
if (!response.responseObject->getString("message"_s, valueString)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
completionHandler(CommandResult::success(JSON::Value::create(valueString)));
});
}
void Session::sendAlertText(const String& text, Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("browsingContextHandle"_s, m_toplevelBrowsingContext.value());
parameters->setString("userInput"_s, text);
m_host->sendCommandToBackend("setUserInputForCurrentJavaScriptPrompt"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) {
if (response.isError) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
completionHandler(CommandResult::success());
});
}
void Session::takeScreenshot(Optional<String> elementID, Optional<bool> scrollIntoView, Function<void (CommandResult&&)>&& completionHandler)
{
if (!m_toplevelBrowsingContext) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::NoSuchWindow));
return;
}
handleUserPrompts([this, elementID, scrollIntoView, completionHandler = WTFMove(completionHandler)](CommandResult&& result) mutable {
if (result.isError()) {
completionHandler(WTFMove(result));
return;
}
RefPtr<JSON::Object> parameters = JSON::Object::create();
parameters->setString("handle"_s, m_toplevelBrowsingContext.value());
if (m_currentBrowsingContext)
parameters->setString("frameHandle"_s, m_currentBrowsingContext.value());
if (elementID)
parameters->setString("nodeHandle"_s, elementID.value());
else
parameters->setBoolean("clipToViewport"_s, true);
if (scrollIntoView.valueOr(false))
parameters->setBoolean("scrollIntoViewIfNeeded"_s, true);
m_host->sendCommandToBackend("takeScreenshot"_s, WTFMove(parameters), [protectedThis = makeRef(*this), completionHandler = WTFMove(completionHandler)](SessionHost::CommandResponse&& response) mutable {
if (response.isError || !response.responseObject) {
completionHandler(CommandResult::fail(WTFMove(response.responseObject)));
return;
}
String data;
if (!response.responseObject->getString("data"_s, data)) {
completionHandler(CommandResult::fail(CommandResult::ErrorCode::UnknownError));
return;
}
completionHandler(CommandResult::success(JSON::Value::create(data)));
});
});
}
} // namespace WebDriver