InjectedBundleDOMWindowExtension.cpp [plain text]
#include "config.h"
#include "InjectedBundleDOMWindowExtension.h"
#include "InjectedBundleScriptWorld.h"
#include "WebFrame.h"
#include "WebFrameLoaderClient.h"
#include <WebCore/DOMWindowExtension.h>
#include <WebCore/DOMWrapperWorld.h>
#include <WebCore/Frame.h>
#include <WebCore/FrameLoader.h>
#include <wtf/HashMap.h>
#include <wtf/NeverDestroyed.h>
using namespace WebCore;
namespace WebKit {
typedef HashMap<WebCore::DOMWindowExtension*, InjectedBundleDOMWindowExtension*> ExtensionMap;
static ExtensionMap& allExtensions()
{
static NeverDestroyed<ExtensionMap> map;
return map;
}
PassRefPtr<InjectedBundleDOMWindowExtension> InjectedBundleDOMWindowExtension::create(WebFrame* frame, InjectedBundleScriptWorld* world)
{
return adoptRef(new InjectedBundleDOMWindowExtension(frame, world));
}
InjectedBundleDOMWindowExtension* InjectedBundleDOMWindowExtension::get(DOMWindowExtension* extension)
{
ASSERT(allExtensions().contains(extension));
return allExtensions().get(extension);
}
InjectedBundleDOMWindowExtension::InjectedBundleDOMWindowExtension(WebFrame* frame, InjectedBundleScriptWorld* world)
: m_coreExtension(DOMWindowExtension::create(frame->coreFrame(), world->coreWorld()))
{
allExtensions().add(m_coreExtension.get(), this);
}
InjectedBundleDOMWindowExtension::~InjectedBundleDOMWindowExtension()
{
ASSERT(allExtensions().contains(m_coreExtension.get()));
allExtensions().remove(m_coreExtension.get());
}
WebFrame* InjectedBundleDOMWindowExtension::frame() const
{
Frame* frame = m_coreExtension->frame();
if (!frame)
return nullptr;
return WebFrame::fromCoreFrame(*frame);
}
InjectedBundleScriptWorld* InjectedBundleDOMWindowExtension::world() const
{
if (!m_world)
m_world = InjectedBundleScriptWorld::getOrCreate(m_coreExtension->world());
return m_world.get();
}
}