[Win, MediaGallery] Enumerate and handle mtp device attach/detach events.

chrome/browser/system_monitor/portable_device_watcher_win.h

Index: chrome/browser/system_monitor/portable_device_watcher_win.h
diff --git a/chrome/browser/system_monitor/portable_device_watcher_win.h b/chrome/browser/system_monitor/portable_device_watcher_win.h
new file mode 100644
index 0000000000000000000000000000000000000000..a1156aa12c97b912317b5272d1b7f60a4b4bc8dc
--- /dev/null
+++ b/chrome/browser/system_monitor/portable_device_watcher_win.h
@@ -0,0 +1,116 @@
+// 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.
+
+#ifndef CHROME_BROWSER_SYSTEM_MONITOR_PORTABLE_DEVICE_WATCHER_WIN_H_
+#define CHROME_BROWSER_SYSTEM_MONITOR_PORTABLE_DEVICE_WATCHER_WIN_H_
+
+#include <portabledeviceapi.h>
+
+#include <map>
+#include <string>
+#include <vector>
+
+#include "base/memory/ref_counted.h"
+#include "base/memory/weak_ptr.h"
+#include "base/string16.h"
+#include "base/system_monitor/system_monitor.h"
+
+namespace base {
+class SequencedTaskRunner;
+}
+
+class FilePath;
+
+namespace chrome {
+
+// This class watches the portable device mount points and sends notifications
+// to base::SystemMonitor about the attached/detached Mtp devices. This is a

[Peter Kasting, 2012/10/29 21:57:19]

Nit: Mtp -> MTP (or "media transfer protocol (MTP)")

[kmadhusu, 2012/10/30 01:29:24]

Done.

+// singleton class instantiated by RemovableDeviceNotificationsWindowWin. This
+// class is created, destroyed and operates on the UI thread, except for long
+// running tasks it spins off to a SequencedTaskRunner.
+class PortableDeviceWatcherWin {
+ public:
+  typedef std::vector<string16> StorageObjectIDs;
+
+  struct DeviceStorageObject {
+    DeviceStorageObject(const string16& temporary_id,
+                        const std::string& persistent_id);
+
+    // Storage object temporary identifier, e.g. "s10001".
+    string16 object_temporary_id;
+
+    // Storage object persistent identifier,
+    // e.g. "StorageSerial:<SID-{10001,D,31080448}>:<123456789>"

[Peter Kasting, 2012/10/29 21:57:19]

Nit: Trailing period

[kmadhusu, 2012/10/30 01:29:24]

Done.

+    std::string object_persistent_id;
+  };
+  typedef std::vector<DeviceStorageObject> StorageObjects;
+
+  // Struct to store attached mtp device details.
+  struct DeviceDetails {
+    // Device name.
+    string16 name;
+
+    // Device interface path.
+    string16 location;
+
+    // Device storage details. A device can have multiple data partitions.
+    StorageObjects storage_objects;
+  };
+  typedef std::vector<DeviceDetails> Devices;
+
+  PortableDeviceWatcherWin();
+  virtual ~PortableDeviceWatcherWin();
+
+  // Must be called after the browser blocking pool is ready for use.
+  // RemovableDeviceNotificationsWindowsWin::Init() will call this function.
+  void Init();
+
+  // Processes DEV_BROADCAST_DEVICEINTERFACE messages and triggers a
+  // SystemMonitor notification if appropriate.
+  void OnWindowMessage(UINT event_type, LPARAM data);
+
+ private:
+  friend class TestPortableDeviceWatcherWin;
+
+  // Key: MTP device storage unique id.
+  // Value: Metadata for the given storage.
+  typedef std::map<std::string, base::SystemMonitor::RemovableStorageInfo>
+      MtpStorageMap;

[Peter Kasting, 2012/10/29 21:57:19]

Nit: Style guide does not rule on this, but I think capitalizing all letters of
an abbreviation (MTP) in types/names/etc. is clearer then InitialCaps style

[kmadhusu, 2012/10/30 01:29:24]

Done.

+
+  // Key: Mtp device plug and play ID string.
+  // Value: Vector of device storage objects.
+  typedef std::map<string16, StorageObjects> MtpDeviceMap;
+
+  // Helpers to enumerate existing mtp storage devices.
+  virtual void EnumerateAttachedDevices();
+  virtual void OnDidEnumerateAttachedDevices(const Devices* devices,
+                                             const bool* result);
+
+  // Helpers to handle device attach event.
+  virtual void HandleDeviceAttachEvent(const string16& pnp_device_id);
+  virtual void OnDidHandleDeviceAttachEvent(
+      const DeviceDetails* device_details, const bool* result);
+
+  //  Handles the detach event of the device specified by |pnp_device_id|.

[Peter Kasting, 2012/10/29 21:57:19]

Nit: Extra space

[kmadhusu, 2012/10/30 01:29:24]

Done.

+  void HandleDeviceDetachEvent(const string16& pnp_device_id);
+
+  // Attached media transfer protocol device map.
+  MtpDeviceMap device_map_;
+
+  // Attached media transfer protocol device storage objects map.
+  MtpStorageMap storage_map_;
+
+  // The task runner used to execute tasks that may take a long time and thus
+  // should not be performed on the UI thread.
+  scoped_refptr<base::SequencedTaskRunner> media_task_runner_;
+
+  // Used by |media_task_runner_| to create cancelable callbacks.
+  base::WeakPtrFactory<PortableDeviceWatcherWin> weak_ptr_factory_;
+
+  DISALLOW_COPY_AND_ASSIGN(PortableDeviceWatcherWin);
+};
+
+}  // namespace chrome
+
+#endif  // CHROME_BROWSER_SYSTEM_MONITOR_PORTABLE_DEVICE_WATCHER_WIN_H_