InjectedBundleScriptWorld.cpp [plain text]
#include "config.h"
#include "InjectedBundleScriptWorld.h"
#include <WebCore/DOMWrapperWorld.h>
#include <WebCore/ScriptController.h>
#include <wtf/HashMap.h>
#include <wtf/NeverDestroyed.h>
#include <wtf/text/StringConcatenate.h>
#include <wtf/text/WTFString.h>
namespace WebKit {
using namespace WebCore;
typedef HashMap<DOMWrapperWorld*, InjectedBundleScriptWorld*> WorldMap;
static WorldMap& allWorlds()
{
static NeverDestroyed<WorldMap> map;
return map;
}
static String uniqueWorldName()
{
static uint64_t uniqueWorldNameNumber = 0;
return makeString("UniqueWorld_"_s, String::number(uniqueWorldNameNumber++));
}
Ref<InjectedBundleScriptWorld> InjectedBundleScriptWorld::create()
{
return adoptRef(*new InjectedBundleScriptWorld(ScriptController::createWorld(), uniqueWorldName()));
}
Ref<InjectedBundleScriptWorld> InjectedBundleScriptWorld::create(const String& name)
{
return adoptRef(*new InjectedBundleScriptWorld(ScriptController::createWorld(), name));
}
Ref<InjectedBundleScriptWorld> InjectedBundleScriptWorld::getOrCreate(DOMWrapperWorld& world)
{
if (&world == &mainThreadNormalWorld())
return normalWorld();
if (InjectedBundleScriptWorld* existingWorld = allWorlds().get(&world))
return *existingWorld;
return adoptRef(*new InjectedBundleScriptWorld(world, uniqueWorldName()));
}
InjectedBundleScriptWorld* InjectedBundleScriptWorld::find(const String& name)
{
for (auto* world : allWorlds().values()) {
if (world->name() == name)
return world;
}
return nullptr;
}
InjectedBundleScriptWorld& InjectedBundleScriptWorld::normalWorld()
{
static InjectedBundleScriptWorld& world = adoptRef(*new InjectedBundleScriptWorld(mainThreadNormalWorld(), String())).leakRef();
return world;
}
InjectedBundleScriptWorld::InjectedBundleScriptWorld(DOMWrapperWorld& world, const String& name)
: m_world(world)
, m_name(name)
{
ASSERT(!allWorlds().contains(m_world.ptr()));
allWorlds().add(m_world.ptr(), this);
}
InjectedBundleScriptWorld::~InjectedBundleScriptWorld()
{
ASSERT(allWorlds().contains(m_world.ptr()));
allWorlds().remove(m_world.ptr());
}
const DOMWrapperWorld& InjectedBundleScriptWorld::coreWorld() const
{
return m_world;
}
DOMWrapperWorld& InjectedBundleScriptWorld::coreWorld()
{
return m_world;
}
void InjectedBundleScriptWorld::clearWrappers()
{
m_world->clearWrappers();
}
void InjectedBundleScriptWorld::makeAllShadowRootsOpen()
{
m_world->setShadowRootIsAlwaysOpen();
}
void InjectedBundleScriptWorld::disableOverrideBuiltinsBehavior()
{
m_world->disableOverrideBuiltinsBehavior();
}
}