SystemMonitor: Pull device type into the device id

chrome/browser/media_gallery/media_storage_util.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.
+
+// chrome::MediaStorageUtil implementation.
+
+#include "chrome/browser/media_gallery/media_storage_util.h"
+
+#include <vector>
+
+#include "base/callback.h"
+#include "base/file_util.h"
+#include "base/logging.h"
+#include "base/system_monitor/system_monitor.h"
+#include "content/public/browser/browser_thread.h"
+
+namespace chrome {
+
+namespace {
+
+typedef std::vector<base::SystemMonitor::MediaDeviceInfo> MediaDevicesInfo;
+
+// Prefix constants for different device id spaces.
+const char kUsbMassStorageWithDCIMPrefix[] = "dcim:";
+const char kUsbMassStorageNoDCIMPrefix[] = "usb:";
+const char kOtherMassStoragePrefix[] = "path:";
+const char kUsbMtpPrefix[] = "mtp:";
+
+void EmptyPathIsFalseCallback(const MediaStorageUtil::BoolCallback& callback,
+                              FilePath path) {
+  callback.Run(!path.empty());
+}
+
+void ValidatePathOnFileThread(
+    const FilePath& path, const MediaStorageUtil::FilePathCallback& callback) {
+  DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::FILE));
+  FilePath result;
+  if (file_util::PathExists(path))
+    result = path;
+  content::BrowserThread::PostTask(content::BrowserThread::UI, FROM_HERE,
+                                   base::Bind(callback, path));
+}
+
+// TODO(vandebo) use FilePath::AppendRelativePath instead

[kmadhusu, 2012/08/20 17:13:50]

nit: TODO(vandebo):

[vandebo (ex-Chrome), 2012/08/20 18:35:14]

colon is optional

+// Make |path| a relative path, i.e. strip the drive letter and leading /.
+FilePath MakePathRelative(const FilePath& path) {
+  if (!path.IsAbsolute())
+    return path;
+
+  FilePath relative;
+  std::vector<FilePath::StringType> components;
+  path.GetComponents(&components);
+
+  // On Windows, the first component may be the drive letter with the second
+  // being \\.
+  int start = 1;
+  if (components[1].size() == 1 && FilePath::IsSeparator(components[1][0]))
+    start = 2;
+
+  for (size_t i = start; i < components.size(); i++)
+    relative = relative.Append(components[i]);
+  return relative;
+}
+
+}  // namespace
+
+// static
+std::string MediaStorageUtil::MakeDeviceId(Type type,
+                                           const std::string& unique_id) {
+  DCHECK(!unique_id.empty());
+  switch (type) {
+    case USB_MASS_STORAGE_WITH_DCIM:
+      return std::string(kUsbMassStorageWithDCIMPrefix) + unique_id;
+    case USB_MASS_STORAGE_NO_DCIM:
+      return std::string(kUsbMassStorageNoDCIMPrefix) + unique_id;
+    case OTHER_MASS_STORAGE:
+      return std::string(kOtherMassStoragePrefix) + unique_id;
+    case USB_MTP:
+      return std::string(kUsbMtpPrefix) + unique_id;
+    default:
+      NOTREACHED();
+      return std::string();
+  }
+}
+
+// static
+void MediaStorageUtil::CrackDeviceId(const std::string& device_id,
+                                     Type* type, std::string* unique_id) {
+  size_t prefix_length = device_id.find_first_of(':');
+  std::string prefix = device_id.substr(0, prefix_length);
+
+  if (type) {
+    if (prefix == kUsbMassStorageWithDCIMPrefix) {
+      *type = USB_MASS_STORAGE_WITH_DCIM;
+    } else if (prefix == kUsbMassStorageNoDCIMPrefix) {
+      *type = USB_MASS_STORAGE_NO_DCIM;
+    } else if (prefix == kOtherMassStoragePrefix) {
+      *type = OTHER_MASS_STORAGE;
+    } else if (prefix == kUsbMtpPrefix) {
+      *type = USB_MTP;
+    } else {
+      NOTREACHED();
+    }
+  }
+
+  if (unique_id)
+    *unique_id = device_id.substr(prefix_length + 1);
+}
+
+// static
+bool MediaStorageUtil::IsMediaDevice(const std::string& device_id) {
+  Type type;
+  CrackDeviceId(device_id, &type, NULL);
+  return type == USB_MASS_STORAGE_WITH_DCIM || type == USB_MTP;
+}
+
+// static
+bool MediaStorageUtil::IsRemovableDevice(const std::string& device_id) {
+  Type type;
+  CrackDeviceId(device_id, &type, NULL);
+  return type != OTHER_MASS_STORAGE;
+}
+
+// static
+void MediaStorageUtil::IsDeviceAttached(const std::string& device_id,
+                                        const BoolCallback& callback) {
+  Type type;
+  std::string unique_id;
+  CrackDeviceId(device_id, &type, &unique_id);
+
+  switch (type) {
+    case USB_MTP:  // Fall through
+    case USB_MASS_STORAGE_WITH_DCIM: {
+      // We should be able to find media devices in SystemMonitor.
+      MediaDevicesInfo media_devices =
+          base::SystemMonitor::Get()->GetAttachedMediaDevices();
+      for (MediaDevicesInfo::const_iterator it = media_devices.begin();
+           it != media_devices.end();
+           ++it) {
+        if (it->unique_id == device_id) {
+          callback.Run(true);
+          return;
+        }
+      }
+      callback.Run(false);
+      break;
+    }
+    case USB_MASS_STORAGE_NO_DCIM:
+      FindUSBDeviceById(unique_id,
+                        base::Bind(&EmptyPathIsFalseCallback, callback));
+      break;
+    case OTHER_MASS_STORAGE:
+      // For this type, the unique_id is the path.
+      content::BrowserThread::PostTask(
+          content::BrowserThread::FILE, FROM_HERE,
+          base::Bind(&ValidatePathOnFileThread,
+                     FilePath::FromUTF8Unsafe(unique_id),
+                     base::Bind(&EmptyPathIsFalseCallback, callback)));
+      break;
+    default:
+      NOTREACHED();
+  }
+}
+
+// static
+void MediaStorageUtil::GetDeviceInfoFromPath(
+    const FilePath& path, const DeviceInfoCallback& callback) {
+  // TODO(vandebo) This needs to be implemented per platform.  Below is no

[kmadhusu, 2012/08/20 17:13:50]

TODO(vandebo):

+  // worse than what the code already does.
+  // * Find mount point parent (determines relative file path)
+  // * Search System monitor, just in case.
+  // * If it's a USB device, generate device id, else use device root path as id
+  std::string device_id =
+      MakeDeviceId(OTHER_MASS_STORAGE, path.AsUTF8Unsafe());
+  FilePath relative_path = MakePathRelative(path);
+  string16 display_name = path.BaseName().LossyDisplayName();
+
+  callback.Run(device_id, relative_path, display_name);
+  return;
+}
+
+// static
+void MediaStorageUtil::FindDevicePathById(const std::string& device_id,
+                                          const FilePathCallback& callback) {
+  Type type;
+  std::string unique_id;
+  CrackDeviceId(device_id, &type, &unique_id);
+
+  switch (type) {
+    case USB_MTP:
+      callback.Run(FilePath());
+      break;
+    case USB_MASS_STORAGE_NO_DCIM:
+      FindUSBDeviceById(unique_id, callback);
+      break;
+    case OTHER_MASS_STORAGE:
+      // For this type, the unique_id is the path.
+      content::BrowserThread::PostTask(
+          content::BrowserThread::FILE, FROM_HERE,
+          base::Bind(&ValidatePathOnFileThread,
+                     FilePath::FromUTF8Unsafe(unique_id), callback));
+      break;
+    case USB_MASS_STORAGE_WITH_DCIM: {
+      FilePath result;
+      MediaDevicesInfo media_devices =
+          base::SystemMonitor::Get()->GetAttachedMediaDevices();
+      for (MediaDevicesInfo::const_iterator it = media_devices.begin();
+           it != media_devices.end();
+           ++it) {
+        if (it->unique_id == device_id) {
+          result = FilePath(it->location);
+          break;
+        }
+      }
+      callback.Run(result);
+      break;
+    }
+    default:
+      NOTREACHED();
+  }
+}
+
+MediaStorageUtil::MediaStorageUtil() {}
+
+// static
+void MediaStorageUtil::FindUSBDeviceById(const std::string unique_id,

[kmadhusu, 2012/08/20 17:13:50]

nit: const std::string&

[vandebo (ex-Chrome), 2012/08/20 18:35:14]

Done.

+                                         const FilePathCallback& callback) {
+  // TODO(vandebo) This needs to be implemented per platform.
+  // Type is USB_MASS_STORAGE_NO_DCIM, so it's a device possibly mounted
+  // somewhere...
+  NOTREACHED();
+  callback.Run(FilePath());
+}
+
+}  // namespace chrome