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 "extensions/renderer/render_view_observer_natives.h"
6
7 #include "content/public/renderer/render_view.h"
8 #include "content/public/renderer/render_view_observer.h"
9 #include "extensions/common/extension_api.h"
10 #include "extensions/renderer/script_context.h"
11 #include "third_party/WebKit/public/web/WebFrame.h"
12 #include "third_party/WebKit/public/web/WebScopedMicrotaskSuppression.h"
13
14 namespace extensions {
15
16 namespace {
17
18 // Deletes itself when done.
19 class LoadWatcher : public content::RenderViewObserver {
20 public:
LoadWatcher(ScriptContext * context,content::RenderView * view,v8::Handle<v8::Function> cb)21 LoadWatcher(ScriptContext* context,
22 content::RenderView* view,
23 v8::Handle<v8::Function> cb)
24 : content::RenderViewObserver(view), context_(context), callback_(cb) {}
25
DidCreateDocumentElement(blink::WebLocalFrame * frame)26 virtual void DidCreateDocumentElement(blink::WebLocalFrame* frame) OVERRIDE {
27 CallbackAndDie(true);
28 }
29
DidFailProvisionalLoad(blink::WebLocalFrame * frame,const blink::WebURLError & error)30 virtual void DidFailProvisionalLoad(blink::WebLocalFrame* frame,
31 const blink::WebURLError& error)
32 OVERRIDE {
33 CallbackAndDie(false);
34 }
35
36 private:
CallbackAndDie(bool succeeded)37 void CallbackAndDie(bool succeeded) {
38 v8::Isolate* isolate = context_->isolate();
39 v8::HandleScope handle_scope(isolate);
40 v8::Handle<v8::Value> args[] = {v8::Boolean::New(isolate, succeeded)};
41 context_->CallFunction(callback_.NewHandle(isolate), 1, args);
42 delete this;
43 }
44
45 ScriptContext* context_;
46 ScopedPersistent<v8::Function> callback_;
47 DISALLOW_COPY_AND_ASSIGN(LoadWatcher);
48 };
49 } // namespace
50
RenderViewObserverNatives(ScriptContext * context)51 RenderViewObserverNatives::RenderViewObserverNatives(ScriptContext* context)
52 : ObjectBackedNativeHandler(context) {
53 RouteFunction("OnDocumentElementCreated",
54 base::Bind(&RenderViewObserverNatives::OnDocumentElementCreated,
55 base::Unretained(this)));
56 }
57
OnDocumentElementCreated(const v8::FunctionCallbackInfo<v8::Value> & args)58 void RenderViewObserverNatives::OnDocumentElementCreated(
59 const v8::FunctionCallbackInfo<v8::Value>& args) {
60 CHECK(args.Length() == 2);
61 CHECK(args[0]->IsInt32());
62 CHECK(args[1]->IsFunction());
63
64 int view_id = args[0]->Int32Value();
65
66 content::RenderView* view = content::RenderView::FromRoutingID(view_id);
67 if (!view) {
68 LOG(WARNING) << "No render view found to register LoadWatcher.";
69 return;
70 }
71
72 new LoadWatcher(context(), view, args[1].As<v8::Function>());
73
74 args.GetReturnValue().Set(true);
75 }
76
77 } // namespace extensions
78