1 // Copyright 2014 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #include "content/renderer/web_ui_mojo.h" 6 7 #include "content/common/view_messages.h" 8 #include "content/public/renderer/render_frame.h" 9 #include "content/public/renderer/render_view.h" 10 #include "content/renderer/web_ui_mojo_context_state.h" 11 #include "gin/per_context_data.h" 12 #include "third_party/WebKit/public/web/WebKit.h" 13 #include "third_party/WebKit/public/web/WebLocalFrame.h" 14 #include "third_party/WebKit/public/web/WebView.h" 15 #include "v8/include/v8.h" 16 17 namespace content { 18 19 namespace { 20 21 const char kWebUIMojoContextStateKey[] = "WebUIMojoContextState"; 22 23 struct WebUIMojoContextStateData : public base::SupportsUserData::Data { 24 scoped_ptr<WebUIMojoContextState> state; 25 }; 26 27 } // namespace 28 29 WebUIMojo::MainFrameObserver::MainFrameObserver(WebUIMojo* web_ui_mojo) 30 : RenderFrameObserver(RenderFrame::FromWebFrame( 31 web_ui_mojo->render_view()->GetWebView()->mainFrame())), 32 web_ui_mojo_(web_ui_mojo) { 33 } 34 35 WebUIMojo::MainFrameObserver::~MainFrameObserver() { 36 } 37 38 void WebUIMojo::MainFrameObserver::WillReleaseScriptContext( 39 v8::Handle<v8::Context> context, 40 int world_id) { 41 web_ui_mojo_->DestroyContextState(context); 42 } 43 44 void WebUIMojo::MainFrameObserver::DidFinishDocumentLoad() { 45 web_ui_mojo_->OnDidFinishDocumentLoad(); 46 } 47 48 WebUIMojo::WebUIMojo(RenderView* render_view) 49 : RenderViewObserver(render_view), 50 RenderViewObserverTracker<WebUIMojo>(render_view), 51 main_frame_observer_(this) { 52 } 53 54 WebUIMojo::~WebUIMojo() { 55 } 56 57 void WebUIMojo::CreateContextState() { 58 v8::HandleScope handle_scope(blink::mainThreadIsolate()); 59 blink::WebLocalFrame* frame = 60 render_view()->GetWebView()->mainFrame()->toWebLocalFrame(); 61 v8::Handle<v8::Context> context = frame->mainWorldScriptContext(); 62 gin::PerContextData* context_data = gin::PerContextData::From(context); 63 WebUIMojoContextStateData* data = new WebUIMojoContextStateData; 64 data->state.reset(new WebUIMojoContextState( 65 render_view()->GetWebView()->mainFrame(), context)); 66 context_data->SetUserData(kWebUIMojoContextStateKey, data); 67 } 68 69 void WebUIMojo::DestroyContextState(v8::Handle<v8::Context> context) { 70 gin::PerContextData* context_data = gin::PerContextData::From(context); 71 if (!context_data) 72 return; 73 context_data->RemoveUserData(kWebUIMojoContextStateKey); 74 } 75 76 void WebUIMojo::OnDidFinishDocumentLoad() { 77 v8::HandleScope handle_scope(blink::mainThreadIsolate()); 78 WebUIMojoContextState* state = GetContextState(); 79 if (state) 80 state->Run(); 81 } 82 83 WebUIMojoContextState* WebUIMojo::GetContextState() { 84 blink::WebLocalFrame* frame = 85 render_view()->GetWebView()->mainFrame()->toWebLocalFrame(); 86 v8::HandleScope handle_scope(blink::mainThreadIsolate()); 87 v8::Handle<v8::Context> context = frame->mainWorldScriptContext(); 88 gin::PerContextData* context_data = gin::PerContextData::From(context); 89 if (!context_data) 90 return NULL; 91 WebUIMojoContextStateData* context_state = 92 static_cast<WebUIMojoContextStateData*>( 93 context_data->GetUserData(kWebUIMojoContextStateKey)); 94 return context_state ? context_state->state.get() : NULL; 95 } 96 97 void WebUIMojo::DidCreateDocumentElement(blink::WebLocalFrame* frame) { 98 CreateContextState(); 99 } 100 101 void WebUIMojo::DidClearWindowObject(blink::WebLocalFrame* frame) { 102 if (frame != render_view()->GetWebView()->mainFrame()) 103 return; 104 105 // NOTE: this function may be called early on twice. From the constructor 106 // mainWorldScriptContext() may trigger this to be called. If we are created 107 // before the page is loaded (which is very likely), then on first load this 108 // is called. In the case of the latter we may have already supplied the 109 // handle to the context state so that if we destroy now the handle is 110 // lost. If this is the result of the first load then the contextstate should 111 // be empty and we don't need to destroy it. 112 WebUIMojoContextState* state = GetContextState(); 113 if (state && !state->module_added()) 114 return; 115 116 v8::HandleScope handle_scope(blink::mainThreadIsolate()); 117 DestroyContextState(frame->mainWorldScriptContext()); 118 } 119 120 } // namespace content 121