ladybird/Userland/Services/WebContent/ConsoleGlobalEnvironmentExtensions.cpp
Luke Wilde 54f58e2662 LibWeb: Restore proper functionality of legacy platform objects
With the GC heap conversion, the functionality of legacy platform
objects was broken. This is because the generated implementation of one
of them was used for all of them, removing functionality such as
deletion.

This re-adds all functionality, where questions such as "does the
object support indexed properties?" is instead answered by virtual
functions instead of by the IDL generator checking the presence of
certain keywords/attributes.
2023-02-28 12:36:14 +01:00

123 lines
4.2 KiB
C++

/*
* Copyright (c) 2021-2022, Sam Atkins <atkinssj@serenityos.org>
* Copyright (c) 2022, Andreas Kling <kling@serenityos.org>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#include "ConsoleGlobalEnvironmentExtensions.h"
#include <LibJS/Runtime/Array.h>
#include <LibJS/Runtime/Completion.h>
#include <LibWeb/Bindings/ExceptionOrUtils.h>
#include <LibWeb/DOM/Document.h>
#include <LibWeb/DOM/NodeList.h>
#include <LibWeb/HTML/Window.h>
namespace WebContent {
ConsoleGlobalEnvironmentExtensions::ConsoleGlobalEnvironmentExtensions(JS::Realm& realm, Web::HTML::Window& window)
: Object(realm, nullptr)
, m_window_object(window)
{
}
JS::ThrowCompletionOr<void> ConsoleGlobalEnvironmentExtensions::initialize(JS::Realm& realm)
{
MUST_OR_THROW_OOM(Base::initialize(realm));
define_native_accessor(realm, "$0", $0_getter, nullptr, 0);
define_native_accessor(realm, "$_", $__getter, nullptr, 0);
define_native_function(realm, "$", $_function, 2, JS::default_attributes);
define_native_function(realm, "$$", $$_function, 2, JS::default_attributes);
return {};
}
void ConsoleGlobalEnvironmentExtensions::visit_edges(Visitor& visitor)
{
Base::visit_edges(visitor);
visitor.visit(m_window_object);
}
static JS::ThrowCompletionOr<ConsoleGlobalEnvironmentExtensions*> get_console(JS::VM& vm)
{
auto this_value = vm.this_value();
if (!this_value.is_object() || !is<ConsoleGlobalEnvironmentExtensions>(this_value.as_object()))
return vm.throw_completion<JS::TypeError>(JS::ErrorType::NotAnObjectOfType, "ConsoleGlobalEnvironmentExtensions");
return &static_cast<ConsoleGlobalEnvironmentExtensions&>(this_value.as_object());
}
JS_DEFINE_NATIVE_FUNCTION(ConsoleGlobalEnvironmentExtensions::$0_getter)
{
auto* console_global_object = TRY(get_console(vm));
auto& window = *console_global_object->m_window_object;
auto* inspected_node = window.associated_document().inspected_node();
if (!inspected_node)
return JS::js_undefined();
return inspected_node;
}
JS_DEFINE_NATIVE_FUNCTION(ConsoleGlobalEnvironmentExtensions::$__getter)
{
auto* console_global_object = TRY(get_console(vm));
return console_global_object->m_most_recent_result;
}
JS_DEFINE_NATIVE_FUNCTION(ConsoleGlobalEnvironmentExtensions::$_function)
{
auto* console_global_object = TRY(get_console(vm));
auto& window = *console_global_object->m_window_object;
auto selector = TRY(vm.argument(0).to_deprecated_string(vm));
if (vm.argument_count() > 1) {
auto element_value = vm.argument(1);
if (!(element_value.is_object() && is<Web::DOM::ParentNode>(element_value.as_object()))) {
return vm.throw_completion<JS::TypeError>(JS::ErrorType::NotAnObjectOfType, "Node");
}
auto& element = static_cast<Web::DOM::ParentNode&>(element_value.as_object());
return TRY(Web::Bindings::throw_dom_exception_if_needed(vm, [&]() {
return element.query_selector(selector);
}));
}
return TRY(Web::Bindings::throw_dom_exception_if_needed(vm, [&]() {
return window.associated_document().query_selector(selector);
}));
}
JS_DEFINE_NATIVE_FUNCTION(ConsoleGlobalEnvironmentExtensions::$$_function)
{
auto* console_global_object = TRY(get_console(vm));
auto& window = *console_global_object->m_window_object;
auto selector = TRY(vm.argument(0).to_deprecated_string(vm));
Web::DOM::ParentNode* element = &window.associated_document();
if (vm.argument_count() > 1) {
auto element_value = vm.argument(1);
if (!(element_value.is_object() && is<Web::DOM::ParentNode>(element_value.as_object()))) {
return vm.throw_completion<JS::TypeError>(JS::ErrorType::NotAnObjectOfType, "Node");
}
element = static_cast<Web::DOM::ParentNode*>(&element_value.as_object());
}
auto node_list = TRY(Web::Bindings::throw_dom_exception_if_needed(vm, [&]() {
return element->query_selector_all(selector);
}));
auto array = TRY(JS::Array::create(*vm.current_realm(), node_list->length()));
for (auto i = 0u; i < node_list->length(); ++i) {
// NOTE: NodeList::item_value cannot fail.
TRY(array->create_data_property_or_throw(i, MUST(node_list->item_value(i))));
}
return array;
}
}