PageRuntimeAgent.cpp [plain text]
#include "config.h"
#if ENABLE(INSPECTOR)
#include "PageRuntimeAgent.h"
#include "Document.h"
#include "InjectedScript.h"
#include "InjectedScriptManager.h"
#include "InspectorPageAgent.h"
#include "InspectorState.h"
#include "InstrumentingAgents.h"
#include "Page.h"
#include "PageConsole.h"
#include "ScriptController.h"
#include "SecurityOrigin.h"
using WebCore::TypeBuilder::Runtime::ExecutionContextDescription;
namespace WebCore {
namespace PageRuntimeAgentState {
static const char runtimeEnabled[] = "runtimeEnabled";
};
PageRuntimeAgent::PageRuntimeAgent(InstrumentingAgents* instrumentingAgents, InspectorCompositeState* state, InjectedScriptManager* injectedScriptManager, Page* page, InspectorPageAgent* pageAgent)
: InspectorRuntimeAgent(instrumentingAgents, state, injectedScriptManager)
, m_inspectedPage(page)
, m_pageAgent(pageAgent)
, m_frontend(0)
, m_mainWorldContextCreated(false)
{
m_instrumentingAgents->setPageRuntimeAgent(this);
}
PageRuntimeAgent::~PageRuntimeAgent()
{
m_instrumentingAgents->setPageRuntimeAgent(0);
}
void PageRuntimeAgent::setFrontend(InspectorFrontend* frontend)
{
m_frontend = frontend->runtime();
}
void PageRuntimeAgent::clearFrontend()
{
m_frontend = 0;
String errorString;
disable(&errorString);
}
void PageRuntimeAgent::restore()
{
if (m_state->getBoolean(PageRuntimeAgentState::runtimeEnabled)) {
String error;
enable(&error);
}
}
void PageRuntimeAgent::enable(ErrorString* errorString)
{
if (m_enabled)
return;
InspectorRuntimeAgent::enable(errorString);
m_state->setBoolean(PageRuntimeAgentState::runtimeEnabled, true);
if (m_mainWorldContextCreated)
reportExecutionContextCreation();
}
void PageRuntimeAgent::disable(ErrorString* errorString)
{
if (!m_enabled)
return;
InspectorRuntimeAgent::disable(errorString);
m_state->setBoolean(PageRuntimeAgentState::runtimeEnabled, false);
}
void PageRuntimeAgent::didCreateMainWorldContext(Frame* frame)
{
m_mainWorldContextCreated = true;
if (!m_enabled)
return;
ASSERT(m_frontend);
String frameId = m_pageAgent->frameId(frame);
ScriptState* scriptState = mainWorldScriptState(frame);
notifyContextCreated(frameId, scriptState, 0, true);
}
void PageRuntimeAgent::didCreateIsolatedContext(Frame* frame, ScriptState* scriptState, SecurityOrigin* origin)
{
if (!m_enabled)
return;
ASSERT(m_frontend);
String frameId = m_pageAgent->frameId(frame);
notifyContextCreated(frameId, scriptState, origin, false);
}
InjectedScript PageRuntimeAgent::injectedScriptForEval(ErrorString* errorString, const int* executionContextId)
{
if (!executionContextId) {
ScriptState* scriptState = mainWorldScriptState(m_inspectedPage->mainFrame());
InjectedScript result = injectedScriptManager()->injectedScriptFor(scriptState);
if (result.hasNoValue())
*errorString = "Internal error: main world execution context not found.";
return result;
}
InjectedScript injectedScript = injectedScriptManager()->injectedScriptForId(*executionContextId);
if (injectedScript.hasNoValue())
*errorString = "Execution context with given id not found.";
return injectedScript;
}
void PageRuntimeAgent::muteConsole()
{
PageConsole::mute();
}
void PageRuntimeAgent::unmuteConsole()
{
PageConsole::unmute();
}
void PageRuntimeAgent::reportExecutionContextCreation()
{
Vector<std::pair<ScriptState*, SecurityOrigin*> > isolatedContexts;
for (Frame* frame = m_inspectedPage->mainFrame(); frame; frame = frame->tree()->traverseNext()) {
if (!frame->script()->canExecuteScripts(NotAboutToExecuteScript))
continue;
String frameId = m_pageAgent->frameId(frame);
ScriptState* scriptState = mainWorldScriptState(frame);
notifyContextCreated(frameId, scriptState, 0, true);
frame->script()->collectIsolatedContexts(isolatedContexts);
if (isolatedContexts.isEmpty())
continue;
for (size_t i = 0; i< isolatedContexts.size(); i++)
notifyContextCreated(frameId, isolatedContexts[i].first, isolatedContexts[i].second, false);
isolatedContexts.clear();
}
}
void PageRuntimeAgent::notifyContextCreated(const String& frameId, ScriptState* scriptState, SecurityOrigin* securityOrigin, bool isPageContext)
{
ASSERT(securityOrigin || isPageContext);
int executionContextId = injectedScriptManager()->injectedScriptIdFor(scriptState);
String name = securityOrigin ? securityOrigin->toRawString() : "";
m_frontend->executionContextCreated(ExecutionContextDescription::create()
.setId(executionContextId)
.setIsPageContext(isPageContext)
.setName(name)
.setFrameId(frameId)
.release());
}
}
#endif // ENABLE(INSPECTOR)