src/chrome/renderer/extensions/render_view_observer_natives.cc

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