UserMessageHandlersNamespace.cpp [plain text]
#include "config.h"
#include "UserMessageHandlersNamespace.h"
#if ENABLE(USER_MESSAGE_HANDLERS)
#include "DOMWrapperWorld.h"
#include "Frame.h"
#include "Page.h"
#include "UserContentController.h"
namespace WebCore {
UserMessageHandlersNamespace::UserMessageHandlersNamespace(Frame& frame)
: FrameDestructionObserver(&frame)
{
}
UserMessageHandlersNamespace::~UserMessageHandlersNamespace()
{
}
UserMessageHandler* UserMessageHandlersNamespace::handler(const AtomicString& name, DOMWrapperWorld& world)
{
if (!frame())
return nullptr;
Page* page = frame()->page();
if (!page)
return nullptr;
const auto* userContentController = page->userContentController();
if (!userContentController)
return nullptr;
const auto* userMessageHandlerDescriptors = userContentController->userMessageHandlerDescriptors();
if (!userMessageHandlerDescriptors)
return nullptr;
RefPtr<UserMessageHandlerDescriptor> descriptor = userMessageHandlerDescriptors->get(std::make_pair(name, &world));
if (!descriptor) {
m_messageHandlers.removeFirstMatching([&name, &world](Ref<UserMessageHandler>& handler) {
return handler->name() == name && &handler->world() == &world;
});
return nullptr;
}
for (auto& handler : m_messageHandlers) {
if (handler->name() == name && &handler->world() == &world)
return &handler.get();
}
m_messageHandlers.append(UserMessageHandler::create(*frame(), *descriptor));
return &m_messageHandlers.last().get();
}
}
#endif // ENABLE(USER_MESSAGE_HANDLERS)