DevTools: add backend for discovering connected ADB devices.

chrome/browser/ui/webui/inspect_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/inspect_ui.h"
 
 #include <set>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
@@ skipping 40 matching lines @@
 using content::DevToolsManager;
 using content::RenderProcessHost;
 using content::RenderViewHost;
 using content::RenderViewHostDelegate;
 using content::RenderWidgetHost;
 using content::WebContents;
 using content::WebUIMessageHandler;
 using content::WorkerService;
 using content::WorkerServiceObserver;
 
+namespace {
+
 static const char kDataFile[] = "targets-data.json";
+static const char kAdbQuery[] = "adb-query/";
+static const char kLocalXhr[] = "local-xhr/";
 
 static const char kExtensionTargetType[]  = "extension";
 static const char kPageTargetType[]  = "page";
 static const char kWorkerTargetType[]  = "worker";
 
 static const char kInspectCommand[]  = "inspect";
 static const char kTerminateCommand[]  = "terminate";
 
 static const char kTargetTypeField[]  = "type";
 static const char kAttachedField[]  = "attached";
 static const char kProcessIdField[]  = "processId";
 static const char kRouteIdField[]  = "routeId";
 static const char kUrlField[]  = "url";
 static const char kNameField[]  = "name";
 static const char kFaviconUrlField[] = "favicon_url";
 static const char kPidField[]  = "pid";
 
-namespace {
-
 DictionaryValue* BuildTargetDescriptor(
     const std::string& target_type,
     bool attached,
     const GURL& url,
     const std::string& name,
     const GURL& favicon_url,
     int process_id,
     int route_id,
     base::ProcessHandle handle = base::kNullProcessHandle) {
   DictionaryValue* target_data = new DictionaryValue();
@@ skipping 71 matching lines @@
         worker_info[i].route_id,
         worker_info[i].handle));
   }
 
   std::string json_string;
   base::JSONWriter::Write(rvh_list, &json_string);
 
   callback.Run(base::RefCountedString::TakeString(&json_string));
 }
 
-bool HandleRequestCallback(
+bool HandleDataRequestCallback(
     const std::string& path,
     const content::WebUIDataSource::GotDataCallback& callback) {
-  if (path != kDataFile)
-    return false;
-
   std::set<RenderViewHost*> tab_rvhs;
   for (TabContentsIterator it; !it.done(); it.Next())
     tab_rvhs.insert(it->GetRenderViewHost());
 
   scoped_ptr<ListValue> rvh_list(new ListValue());
 
   for (RenderProcessHost::iterator it(RenderProcessHost::AllHostsIterator());
        !it.IsAtEnd(); it.Advance()) {
     RenderProcessHost* render_process_host = it.GetCurrentValue();
     DCHECK(render_process_host);
@@ skipping 18 matching lines @@
     }
   }
 
   BrowserThread::PostTask(
       BrowserThread::IO,
       FROM_HERE,
       base::Bind(&SendDescriptors, base::Owned(rvh_list.release()), callback));
   return true;
 }
 
-content::WebUIDataSource* CreateInspectUIHTMLSource() {
-  content::WebUIDataSource* source =
-      content::WebUIDataSource::Create(chrome::kChromeUIInspectHost);
-  source->AddResourcePath("inspect.css", IDR_INSPECT_CSS);
-  source->AddResourcePath("inspect.js", IDR_INSPECT_JS);
-  source->SetDefaultResource(IDR_INSPECT_HTML);
-  source->SetRequestFilter(base::Bind(&HandleRequestCallback));
-  return source;
-}
-
 class InspectMessageHandler : public WebUIMessageHandler {
  public:
   InspectMessageHandler() {}
   virtual ~InspectMessageHandler() {}
 
  private:
   // WebUIMessageHandler implementation.
   virtual void RegisterMessages() OVERRIDE;
 
   // Callback for "openDevTools" message.
@@ skipping 119 matching lines @@
 
   InspectUI* discovery_ui_;
 };
 
 InspectUI::InspectUI(content::WebUI* web_ui)
     : WebUIController(web_ui),
       observer_(new WorkerCreationDestructionListener(this)) {
   web_ui->AddMessageHandler(new InspectMessageHandler());
 
   Profile* profile = Profile::FromWebUI(web_ui);
+  adb_bridge_ = DevToolsAdbBridge::Start(profile);
   content::WebUIDataSource::Add(profile, CreateInspectUIHTMLSource());
 
   registrar_.Add(this,
                  content::NOTIFICATION_WEB_CONTENTS_CONNECTED,
                  content::NotificationService::AllSources());
   registrar_.Add(this,
                  content::NOTIFICATION_WEB_CONTENTS_DISCONNECTED,
                  content::NotificationService::AllSources());
   registrar_.Add(this,
                  content::NOTIFICATION_WEB_CONTENTS_DESTROYED,
@@ skipping 12 matching lines @@
     const content::NotificationSource& source,
     const content::NotificationDetails& details) {
   if (source != content::Source<WebContents>(web_ui()->GetWebContents()))
     RefreshUI();
   else if (type == content::NOTIFICATION_WEB_CONTENTS_DISCONNECTED)
     StopListeningNotifications();
 }
 
 void InspectUI::StopListeningNotifications()
 {
+  adb_bridge_->Stop();

[yurys, 2013/03/11 07:21:21]

Should it be destroyed like observer_ ?

+
   if (!observer_)
     return;
   observer_->InspectUIDestroyed();
   observer_ = NULL;
   registrar_.RemoveAll();
 }
+
+content::WebUIDataSource* InspectUI::CreateInspectUIHTMLSource() {
+  content::WebUIDataSource* source =
+      content::WebUIDataSource::Create(chrome::kChromeUIInspectHost);
+  source->AddResourcePath("inspect.css", IDR_INSPECT_CSS);
+  source->AddResourcePath("inspect.js", IDR_INSPECT_JS);
+  source->SetDefaultResource(IDR_INSPECT_HTML);
+  source->SetRequestFilter(base::Bind(&InspectUI::HandleRequestCallback,
+      base::Unretained(this)));
+  return source;
+}
+
+bool InspectUI::HandleRequestCallback(
+    const std::string& path,
+    const content::WebUIDataSource::GotDataCallback& callback) {
+  if (path == kDataFile)
+    return HandleDataRequestCallback(path, callback);
+  if (path.find(kAdbQuery) == 0)
+    return HandleAdbQueryCallback(path, callback);
+  if (path.find(kLocalXhr) == 0)
+    return HandleLocalXhrCallback(path, callback);
+  return false;
+}
+
+bool InspectUI::HandleAdbQueryCallback(
+    const std::string& path,
+    const content::WebUIDataSource::GotDataCallback& callback) {
+  std::string query = path.substr(strlen(kAdbQuery));
+  adb_bridge_->Query(query, base::Bind(&InspectUI::OnAdbResponse,
+                                       base::Unretained(this), callback));
+  return true;
+}
+
+bool InspectUI::HandleLocalXhrCallback(
+    const std::string& path,
+    const content::WebUIDataSource::GotDataCallback& callback) {
+  std::string url = "http://localhost:" + path.substr(strlen(kLocalXhr));
+  adb_bridge_->Fetch(url, base::Bind(&InspectUI::OnAdbResponse,
+                                     base::Unretained(this), callback));
+  return true;
+}
+
+void InspectUI::OnAdbResponse(
+    const content::WebUIDataSource::GotDataCallback& callback,
+    const std::string& error,
+    const std::string& data) {
+  ListValue result;
+  result.AppendString(error);
+  result.AppendString(data);
+  std::string json_string;
+  base::JSONWriter::Write(&result, &json_string);
+  callback.Run(base::RefCountedString::TakeString(&json_string));
+}