Show device attach web intent picker dialog only if we have registered services.

chrome/browser/intents/device_attached_intent_source.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/intents/device_attached_intent_source.h"
 
 #include <string>
 
+#include "base/bind.h"
 #include "base/file_path.h"
+#include "base/string16.h"
+#include "base/memory/ref_counted.h"
 #include "base/utf_string_conversions.h"
+#include "chrome/browser/intents/web_intents_registry.h"
+#include "chrome/browser/intents/web_intents_registry_factory.h"
+#include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_window.h"
 #include "content/public/browser/web_intents_dispatcher.h"
 #include "content/public/browser/web_contents_delegate.h"
 #include "webkit/fileapi/file_system_types.h"
 #include "webkit/fileapi/isolated_context.h"
 #include "webkit/glue/web_intent_data.h"
+#include "webkit/glue/web_intent_service_data.h"
 #include "webkit/fileapi/media/media_file_system_config.h"
 
 #if defined(SUPPORT_MEDIA_FILESYSTEM)
 #include "webkit/fileapi/media/media_device_map_service.h"
 
 using fileapi::MediaDeviceMapService;
 #endif
 
 using base::SystemMonitor;
 using content::WebContentsDelegate;
+using webkit_glue::WebIntentServiceData;
+
+namespace {
+
+// Specifies device attached web intent kAction.
+const char kAction[] = "chrome-extension://attach";
+
+// Specifies device attached web intent type.
+const char kIntentType[] = "chrome-extension://filesystem";
+
+// Dispatch intent only when there is a list of registered services. This is a
+// helper class that stores the attached media device information and decides
+// whether to dispatch an intent or not.
+class DispatchIntentTaskHelper
+    : public base::RefCountedThreadSafe<DispatchIntentTaskHelper> {
+ public:
+  DispatchIntentTaskHelper(
+      const base::WeakPtr<DeviceAttachedIntentSource> source,
+      SystemMonitor::MediaDeviceInfo device_info)
+      : source_(source),
+        device_info_(device_info) {
+  }
+
+  // Query callback for WebIntentsRegistry::GetIntentServices function.
+  void MayDispatchIntentForService(
+      const std::vector<WebIntentServiceData>& services) {
+    if (!services.empty() && source_)
+      source_->DispatchIntentsForService(device_info_);
+  }
+
+ private:
+  friend class base::RefCountedThreadSafe<DispatchIntentTaskHelper>;
+
+  ~DispatchIntentTaskHelper() {}
+
+  // A weak pointer to |DeviceAttachedIntentSource| object to dispatch a
+  // web intent.
+  base::WeakPtr<DeviceAttachedIntentSource> source_;
+
+  // Store the device info. This is used while registering the device as file
+  // system.
+  const SystemMonitor::MediaDeviceInfo device_info_;
+
+  DISALLOW_COPY_AND_ASSIGN(DispatchIntentTaskHelper);
+};
+
+}  // namespace
 
 DeviceAttachedIntentSource::DeviceAttachedIntentSource(
     Browser* browser, WebContentsDelegate* delegate)
     : browser_(browser), delegate_(delegate) {
   SystemMonitor* sys_monitor = SystemMonitor::Get();
   if (sys_monitor)
     sys_monitor->AddDevicesChangedObserver(this);
 }
 
 DeviceAttachedIntentSource::~DeviceAttachedIntentSource() {
   SystemMonitor* sys_monitor = SystemMonitor::Get();
   if (sys_monitor)
     sys_monitor->RemoveDevicesChangedObserver(this);
 }
 
 void DeviceAttachedIntentSource::OnMediaDeviceAttached(
     const std::string& id,
     const string16& name,
-    base::SystemMonitor::MediaDeviceType type,
+    base::SystemMonitor::MediaDeviceType device_type,
     const FilePath::StringType& location) {
   if (!browser_->window()->IsActive())
     return;
 
+  // TODO(kmadhusu): Dispatch intents on incognito window.
+  if (browser_->profile()->IsOffTheRecord())
+    return;
+
   // Only handle FilePaths for now.
   // TODO(kmadhusu): Handle all device types. http://crbug.com/140353.
-  if (type != SystemMonitor::TYPE_PATH)
+  if (device_type != SystemMonitor::TYPE_PATH)
     return;
 
   // Sanity checks for |device_path|.
   const FilePath device_path(location);
   if (!device_path.IsAbsolute() || device_path.ReferencesParent())
     return;
 
+  SystemMonitor::MediaDeviceInfo device_info(id, name, device_type, location);
+  scoped_refptr<DispatchIntentTaskHelper> task = new DispatchIntentTaskHelper(
+      AsWeakPtr(), device_info);
+  WebIntentsRegistryFactory::GetForProfile(browser_->profile())->
+      GetIntentServices(
+          UTF8ToUTF16(kAction), UTF8ToUTF16(kIntentType),
+          base::Bind(&DispatchIntentTaskHelper::MayDispatchIntentForService,
+                     task.get()));
+}
+
+void DeviceAttachedIntentSource::DispatchIntentsForService(
+    const base::SystemMonitor::MediaDeviceInfo& device_info) {
   // Store the media device info locally.
-  SystemMonitor::MediaDeviceInfo device_info(id, name, type, location);
-  device_id_map_.insert(std::make_pair(id, device_info));
+  device_id_map_.insert(std::make_pair(device_info.unique_id, device_info));
 
-  std::string device_name(UTF16ToUTF8(name));
+  std::string device_name(UTF16ToUTF8(device_info.name));
+  const FilePath device_path(device_info.location);
 
-  // Register device path as an isolated file system.
   // TODO(kinuko, kmadhusu): Use a different file system type for MTP.
-  const std::string filesystem_id =
-      fileapi::IsolatedContext::GetInstance()->RegisterFileSystemForPath(
-          fileapi::kFileSystemTypeNativeMedia, device_path, &device_name);
+  const std::string fs_id = fileapi::IsolatedContext::GetInstance()->
+      RegisterFileSystemForPath(fileapi::kFileSystemTypeNativeMedia,
+                                device_path, &device_name);
 
-  CHECK(!filesystem_id.empty());
+  DCHECK(!fs_id.empty());
   webkit_glue::WebIntentData intent(
-      ASCIIToUTF16("chrome-extension://attach"),
-      ASCIIToUTF16("chrome-extension://filesystem"),
-      device_name,
-      filesystem_id);
+      UTF8ToUTF16(kAction), UTF8ToUTF16(kIntentType), device_name, fs_id);
 
   delegate_->WebIntentDispatch(NULL  /* no WebContents */,
                                content::WebIntentsDispatcher::Create(intent));
 }
 
 void DeviceAttachedIntentSource::OnMediaDeviceDetached(const std::string& id) {
   DeviceIdToInfoMap::iterator it = device_id_map_.find(id);
   if (it == device_id_map_.end())
     return;
 
   // TODO(kmadhusu, vandebo): Clean up this code. http://crbug.com/140340.
 
   FilePath path(it->second.location);
   fileapi::IsolatedContext::GetInstance()->RevokeFileSystemByPath(path);
   switch (it->second.type) {
     case SystemMonitor::TYPE_MTP:
 #if defined(SUPPORT_MEDIA_FILESYSTEM)
       MediaDeviceMapService::GetInstance()->RemoveMediaDevice(
           it->second.location);
 #endif
       break;
     case SystemMonitor::TYPE_PATH:
       break;
   }
   device_id_map_.erase(it);
 }