Introduce --remote-debugging-frontend switch for custom remote debugging front-end

chrome/browser/ui/webui/devtools_ui.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 "chrome/browser/ui/webui/devtools_ui.h"
 
+#include "base/command_line.h"
 #include "base/macros.h"
 #include "base/memory/ref_counted_memory.h"
 #include "base/strings/string_util.h"
 #include "base/strings/stringprintf.h"
 #include "chrome/browser/profiles/profile.h"
+#include "chrome/common/chrome_switches.h"
 #include "chrome/common/url_constants.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/devtools_frontend_host.h"
 #include "content/public/browser/url_data_source.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/web_ui.h"
 #include "content/public/common/user_agent.h"
+#include "net/base/filename_util.h"
+#include "net/base/load_flags.h"
 #include "net/url_request/url_fetcher.h"
 #include "net/url_request/url_fetcher_delegate.h"
 #include "net/url_request/url_request_context_getter.h"
 
 using content::BrowserThread;
 using content::WebContents;
 
 namespace {
 
 std::string PathWithoutParams(const std::string& path) {
@@ skipping 79 matching lines @@
                                int render_process_id,
                                int render_frame_id,
                                const GotDataCallback& callback);
 
   // Serves remote DevTools frontend from hard-coded App Engine domain.
   void StartRemoteDataRequest(const std::string& path,
                               int render_process_id,
                               int render_frame_id,
                               const GotDataCallback& callback);
 
+  // Serves remote DevTools frontend from any endpoint, passed through
+  // command-line flag.
+  void StartCustomDataRequest(const GURL& url,
+                              int render_process_id,
+                              int render_frame_id,
+                              const GotDataCallback& callback);
+
   ~DevToolsDataSource() override;
 
   scoped_refptr<net::URLRequestContextGetter> request_context_;
 
   using PendingRequestsMap = std::map<const net::URLFetcher*, GotDataCallback>;
   PendingRequestsMap pending_;
 
   DISALLOW_COPY_AND_ASSIGN(DevToolsDataSource);
 };
 
@@ skipping 32 matching lines @@
   // Serve request from remote location.
   std::string remote_path_prefix(chrome::kChromeUIDevToolsRemotePath);
   remote_path_prefix += "/";
   if (base::StartsWith(path, remote_path_prefix,
                        base::CompareCase::INSENSITIVE_ASCII)) {
     StartRemoteDataRequest(path.substr(remote_path_prefix.length()),
                            render_process_id, render_frame_id, callback);
     return;
   }
 
+  base::FilePath remote_frontend_path =
+      base::CommandLine::ForCurrentProcess()->GetSwitchValuePath(
+          switches::kRemoteDebuggingFrontend);
+
+  if (remote_frontend_path.empty()) {
+    callback.Run(NULL);
+    return;
+  }
+
+  // Serve request from custom location.
+  std::string custom_path_prefix(chrome::kChromeUIDevToolsCustomPath);
+  custom_path_prefix += "/";
+
+  // Generated files should be fetched from bundle.
+  if (base::StartsWith(path, custom_path_prefix + "gen/",
+                       base::CompareCase::INSENSITIVE_ASCII)) {
+    StartBundledDataRequest(path.substr(custom_path_prefix.length()),
+                            render_process_id, render_frame_id, callback);
+    return;
+  }
+  if (base::StartsWith(path, custom_path_prefix,
+                       base::CompareCase::INSENSITIVE_ASCII)) {
+    // Create a dummy url to extract path.
+    GURL dummy_url =
+        GURL("http://example.com/" + path.substr(custom_path_prefix.length()));
+    std::string dummy_url_path = dummy_url.path();
+    base::FilePath file_path(base::FilePath::StringType(dummy_url_path.begin(),
+                                                        dummy_url_path.end()));
+    file_path = remote_frontend_path.Append(file_path);
+    GURL url = net::FilePathToFileURL(file_path);
+    StartCustomDataRequest(url, render_process_id, render_frame_id, callback);
+    return;
+  }
+
   callback.Run(NULL);
 }
 
 std::string DevToolsDataSource::GetMimeType(const std::string& path) const {
   return GetMimeTypeForPath(path);
 }
 
 bool DevToolsDataSource::ShouldAddContentSecurityPolicy() const {
   return false;
 }
@@ skipping 36 matching lines @@
         new base::RefCountedStaticMemory(kHttpNotFound, strlen(kHttpNotFound)));
     return;
   }
   net::URLFetcher* fetcher =
       net::URLFetcher::Create(url, net::URLFetcher::GET, this).release();
   pending_[fetcher] = callback;
   fetcher->SetRequestContext(request_context_.get());
   fetcher->Start();
 }
 
+void DevToolsDataSource::StartCustomDataRequest(
+    const GURL& url,
+    int render_process_id,
+    int render_frame_id,
+    const content::URLDataSource::GotDataCallback& callback) {
+  if (!url.is_valid()) {
+    callback.Run(
+        new base::RefCountedStaticMemory(kHttpNotFound, strlen(kHttpNotFound)));
+    return;
+  }
+  net::URLFetcher* fetcher =
+      net::URLFetcher::Create(url, net::URLFetcher::GET, this).release();
+  pending_[fetcher] = callback;
+  fetcher->SetRequestContext(request_context_.get());
+  fetcher->SetLoadFlags(net::LOAD_DISABLE_CACHE);
+  fetcher->Start();
+}
+
 void DevToolsDataSource::OnURLFetchComplete(const net::URLFetcher* source) {
   DCHECK(source);
   PendingRequestsMap::iterator it = pending_.find(source);
   DCHECK(it != pending_.end());
   std::string response;
   source->GetResponseAsString(&response);
   delete source;
   it->second.Run(base::RefCountedString::TakeString(&response));
   pending_.erase(it);
 }
@@ skipping 28 matching lines @@
       bindings_(web_ui->GetWebContents()) {
   web_ui->SetBindings(0);
   Profile* profile = Profile::FromWebUI(web_ui);
   content::URLDataSource::Add(
       profile,
       new DevToolsDataSource(profile->GetRequestContext()));
 }
 
 DevToolsUI::~DevToolsUI() {
 }