Fix context menu for sandbox devtools (#11933)

atom/renderer/atom_sandboxed_renderer_client.cc

@@ -22,6 +22,7 @@
 #include "ipc/ipc_message_macros.h"
 #include "native_mate/converter.h"
 #include "native_mate/dictionary.h"
+#include "third_party/WebKit/public/web/WebDocument.h"
 #include "third_party/WebKit/public/web/WebFrame.h"
 #include "third_party/WebKit/public/web/WebKit.h"
 #include "third_party/WebKit/public/web/WebLocalFrame.h"
@@ -38,6 +39,10 @@ namespace {
 const std::string kIpcKey = "ipcNative";
 const std::string kModuleCacheKey = "native-module-cache";
 
+bool IsDevTools(content::RenderFrame* render_frame) {
+  return render_frame->GetWebFrame()->GetDocument().Url()
+        .ProtocolIs("chrome-devtools");
+}
 
 v8::Local<v8::Object> GetModuleCache(v8::Isolate* isolate) {
   mate::Dictionary global(isolate, isolate->GetCurrentContext()->Global());
@@ -158,8 +163,6 @@ void AtomSandboxedRendererClient::DidCreateScriptContext(
   base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
   std::string preload_script = command_line->GetSwitchValueASCII(
       switches::kPreloadScript);
-  if (preload_script.empty())
-    return;
 
   auto isolate = context->GetIsolate();
   v8::HandleScope handle_scope(isolate);

lib/sandboxed_renderer/init.js

@@ -32,8 +32,6 @@ const preloadModules = new Map([
   ['timers', require('timers')]
 ])
 
-const preloadSrc = fs.readFileSync(preloadPath).toString()
-
 // Pass different process object to the preload script(which should not have
 // access to things like `process.atomBinding`).
 const preloadProcess = new events.EventEmitter()
@@ -54,6 +52,11 @@ function preloadRequire (module) {
   throw new Error('module not found')
 }
 
+if (window.location.protocol === 'chrome-devtools:') {
+  // Override some inspector APIs.
+  require('../renderer/inspector')
+}
+
 // Wrap the script into a function executed in global scope. It won't have
 // access to the current scope, so we'll expose a few objects as arguments:
 //
@@ -73,13 +76,16 @@ function preloadRequire (module) {
 // and any `require('electron')` calls in `preload.js` will work as expected
 // since browserify won't try to include `electron` in the bundle, falling back
 // to the `preloadRequire` function above.
-const preloadWrapperSrc = `(function(require, process, Buffer, global, setImmediate) {
-${preloadSrc}
-})`
+if (preloadPath) {
+  const preloadSrc = fs.readFileSync(preloadPath).toString()
+  const preloadWrapperSrc = `(function(require, process, Buffer, global, setImmediate) {
+  ${preloadSrc}
+  })`
 
-// eval in window scope:
-// http://www.ecma-international.org/ecma-262/5.1/#sec-10.4.2
-const geval = eval
-const preloadFn = geval(preloadWrapperSrc)
-const {setImmediate} = require('timers')
-preloadFn(preloadRequire, preloadProcess, Buffer, global, setImmediate)
+  // eval in window scope:
+  // http://www.ecma-international.org/ecma-262/5.1/#sec-10.4.2
+  const geval = eval
+  const preloadFn = geval(preloadWrapperSrc)
+  const {setImmediate} = require('timers')
+  preloadFn(preloadRequire, preloadProcess, Buffer, global, setImmediate)
+}