DevTools: introduce --custom-devtools-frontend flag.

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_split.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/escape.h"
+#include "net/base/filename_util.h"
+#include "net/base/load_flags.h"
 #include "net/base/url_util.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 {
 
@@ skipping 199 matching lines @@
   void OnURLFetchComplete(const net::URLFetcher* source) override;
 
   // Serves bundled DevTools frontend from ResourceBundle.
   void StartBundledDataRequest(const std::string& path,
                                const GotDataCallback& callback);
 
   // Serves remote DevTools frontend from hard-coded App Engine domain.
   void StartRemoteDataRequest(const std::string& path,
                               const GotDataCallback& callback);
 
+  // Serves remote DevTools frontend from any endpoint, passed through
+  // command-line flag.
+  void StartCustomDataRequest(const GURL& url,
+                              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 31 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()),
                            callback);
     return;
   }
 
+  std::string custom_frontend_url =
+      base::CommandLine::ForCurrentProcess()->GetSwitchValueASCII(
+          switches::kCustomDevtoolsFrontend);
+
+  if (custom_frontend_url.empty()) {
+    callback.Run(NULL);
+    return;
+  }
+
+  // Serve request from custom location.
+  std::string custom_path_prefix(chrome::kChromeUIDevToolsCustomPath);
+  custom_path_prefix += "/";
+
+  if (base::StartsWith(path, custom_path_prefix,
+                       base::CompareCase::INSENSITIVE_ASCII)) {
+    GURL url = GURL(custom_frontend_url +
+                    path.substr(custom_path_prefix.length()));
+    StartCustomDataRequest(url, callback);
+    return;
+  }
+
   callback.Run(NULL);
 }
 
 std::string DevToolsDataSource::GetMimeType(const std::string& path) const {
   return GetMimeTypeForPath(path);
 }
 
 bool DevToolsDataSource::ShouldAddContentSecurityPolicy() const {
   return false;
 }
@@ skipping 32 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,
+    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 37 matching lines @@
       profile,
       new DevToolsDataSource(profile->GetRequestContext()));
 
   GURL url = web_ui->GetWebContents()->GetVisibleURL();
   if (url.spec() == SanitizeFrontendURL(url).spec())
     bindings_.reset(new DevToolsUIBindings(web_ui->GetWebContents()));
 }
 
 DevToolsUI::~DevToolsUI() {
 }