85 lines
2.6 KiB
C++
85 lines
2.6 KiB
C++
|
// Copyright (c) 2017 GitHub, Inc.
|
||
|
// Use of this source code is governed by the MIT license that can be
|
||
|
// found in the LICENSE file.
|
||
|
|
||
|
#include "atom/renderer/atom_render_frame_observer.h"
|
||
|
|
||
|
#include "content/public/renderer/render_frame.h"
|
||
|
#include "third_party/WebKit/public/web/WebDocument.h"
|
||
|
#include "third_party/WebKit/public/web/WebLocalFrame.h"
|
||
|
#include "third_party/WebKit/public/web/WebScriptSource.h"
|
||
|
|
||
|
namespace atom {
|
||
|
|
||
|
AtomRenderFrameObserver::AtomRenderFrameObserver(
|
||
|
content::RenderFrame* frame,
|
||
|
AtomRendererClient* renderer_client)
|
||
|
: content::RenderFrameObserver(frame),
|
||
|
render_frame_(frame),
|
||
|
renderer_client_(renderer_client) {}
|
||
|
|
||
|
void AtomRenderFrameObserver::DidClearWindowObject() {
|
||
|
renderer_client_->DidClearWindowObject(render_frame_);
|
||
|
}
|
||
|
|
||
|
void AtomRenderFrameObserver::DidCreateScriptContext(
|
||
|
v8::Handle<v8::Context> context,
|
||
|
int extension_group,
|
||
|
int world_id) {
|
||
|
if (ShouldNotifyClient(world_id))
|
||
|
renderer_client_->DidCreateScriptContext(context, render_frame_);
|
||
|
|
||
|
if (renderer_client_->isolated_world() && IsMainWorld(world_id)
|
||
|
&& render_frame_->IsMainFrame()) {
|
||
|
CreateIsolatedWorldContext();
|
||
|
renderer_client_->SetupMainWorldOverrides(context);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void AtomRenderFrameObserver::WillReleaseScriptContext(
|
||
|
v8::Local<v8::Context> context,
|
||
|
int world_id) {
|
||
|
if (ShouldNotifyClient(world_id))
|
||
|
renderer_client_->WillReleaseScriptContext(context, render_frame_);
|
||
|
}
|
||
|
|
||
|
void AtomRenderFrameObserver::OnDestruct() {
|
||
|
delete this;
|
||
|
}
|
||
|
|
||
|
void AtomRenderFrameObserver::CreateIsolatedWorldContext() {
|
||
|
auto frame = render_frame_->GetWebFrame();
|
||
|
|
||
|
// This maps to the name shown in the context combo box in the Console tab
|
||
|
// of the dev tools.
|
||
|
frame->setIsolatedWorldHumanReadableName(
|
||
|
World::ISOLATED_WORLD,
|
||
|
blink::WebString::fromUTF8("Electron Isolated Context"));
|
||
|
|
||
|
// Setup document's origin policy in isolated world
|
||
|
frame->setIsolatedWorldSecurityOrigin(
|
||
|
World::ISOLATED_WORLD, frame->document().getSecurityOrigin());
|
||
|
|
||
|
// Create initial script context in isolated world
|
||
|
blink::WebScriptSource source("void 0");
|
||
|
frame->executeScriptInIsolatedWorld(
|
||
|
World::ISOLATED_WORLD, &source, 1, ExtensionGroup::MAIN_GROUP);
|
||
|
}
|
||
|
|
||
|
bool AtomRenderFrameObserver::IsMainWorld(int world_id) {
|
||
|
return world_id == World::MAIN_WORLD;
|
||
|
}
|
||
|
|
||
|
bool AtomRenderFrameObserver::IsIsolatedWorld(int world_id) {
|
||
|
return world_id == World::ISOLATED_WORLD;
|
||
|
}
|
||
|
|
||
|
bool AtomRenderFrameObserver::ShouldNotifyClient(int world_id) {
|
||
|
if (renderer_client_->isolated_world() && render_frame_->IsMainFrame())
|
||
|
return IsIsolatedWorld(world_id);
|
||
|
else
|
||
|
return IsMainWorld(world_id);
|
||
|
}
|
||
|
|
||
|
} // namespace atom
|