Expose distiller functions to JavaScript

content/renderer/dom_automation_controller.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "content/renderer/dom_automation_controller.h"
 
 #include "base/json/json_string_value_serializer.h"
 #include "base/strings/string_util.h"
 #include "content/child/v8_value_converter_impl.h"
 #include "content/common/child_process_messages.h"
@@ skipping 40 matching lines @@
   return gin::Wrappable<DomAutomationController>::GetObjectTemplateBuilder(
              isolate)
       .SetMethod("send", &DomAutomationController::SendMsg)
       .SetMethod("setAutomationId", &DomAutomationController::SetAutomationId)
       .SetMethod("sendJSON", &DomAutomationController::SendJSON)
       .SetMethod("sendWithId", &DomAutomationController::SendWithId);
 }
 
 void DomAutomationController::OnDestruct() {}
 
+void DomAutomationController::DidCreateScriptContext(
+    v8::Local<v8::Context> context,
+    int extension_group,
+    int world_id) {
+  // Add the domAutomationController to isolated worlds as well.
+  v8::Isolate* isolate = blink::mainThreadIsolate();
+  v8::HandleScope handle_scope(isolate);
+  if (context.IsEmpty())
+    return;
+
+  v8::Context::Scope context_scope(context);
+
+  // Resuse this object instead of creating others.
+  gin::Handle<DomAutomationController> controller =
+      gin::CreateHandle(isolate, this);
+  if (controller.IsEmpty())
+    return;
+
+  v8::Local<v8::Object> global = context->Global();
+  global->Set(gin::StringToV8(isolate, "domAutomationController"),
+              controller.ToV8());
+}
+
 bool DomAutomationController::SendMsg(const gin::Arguments& args) {
   if (!render_frame())
     return false;
 
   if (automation_id_ == MSG_ROUTING_NONE)
     return false;
 
   std::string json;
   JSONStringValueSerializer serializer(&json);
   scoped_ptr<base::Value> value;
@@ skipping 44 matching lines @@
   return Send(
       new FrameHostMsg_DomOperationResponse(routing_id(), str, automation_id));
 }
 
 bool DomAutomationController::SetAutomationId(int automation_id) {
   automation_id_ = automation_id;
   return true;
 }
 
 }  // namespace content