Formatting feature initial commit for ChromeOS Tree

chrome/browser/chromeos/extensions/file_browser_event_router.cc

 // Copyright (c) 2011 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/chromeos/extensions/file_browser_event_router.h"
 
+#include "base/bind.h"
 #include "base/json/json_writer.h"
 #include "base/memory/singleton.h"
+#include "base/message_loop.h"
 #include "base/stl_util.h"
 #include "base/values.h"
 #include "chrome/browser/chromeos/cros/cros_library.h"
 #include "chrome/browser/chromeos/login/user_manager.h"
 #include "chrome/browser/chromeos/notifications/system_notification.h"
 #include "chrome/browser/extensions/extension_event_names.h"
 #include "chrome/browser/extensions/extension_event_router.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/extensions/file_manager_util.h"
 #include "chrome/browser/profiles/profile.h"
 #include "content/browser/browser_thread.h"
 #include "grit/generated_resources.h"
 #include "grit/theme_resources.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "webkit/fileapi/file_system_types.h"
 #include "webkit/fileapi/file_system_util.h"
 
-const char kDiskAddedEventType[] = "added";
-const char kDiskRemovedEventType[] = "removed";
+namespace {
+  const char kDiskAddedEventType[] = "added";
+  const char kDiskRemovedEventType[] = "removed";
 
-const char kPathChanged[] = "changed";
-const char kPathWatchError[] = "error";
+  const char kPathChanged[] = "changed";
+  const char kPathWatchError[] = "error";
 
-const char* DeviceTypeToString(chromeos::DeviceType type) {
-  switch (type) {
-    case chromeos::FLASH:
-      return "flash";
-    case chromeos::HDD:
-      return "hdd";
-    case chromeos::OPTICAL:
-      return "optical";
-    default:
-      break;
+  const char* DeviceTypeToString(chromeos::DeviceType type) {
+    switch (type) {
+      case chromeos::FLASH:
+        return "flash";
+      case chromeos::HDD:
+        return "hdd";
+      case chromeos::OPTICAL:
+        return "optical";
+      default:
+        break;
+    }
+    return "undefined";
   }
-  return "undefined";
+
+  DictionaryValue* DiskToDictionaryValue(
+      const chromeos::MountLibrary::Disk* disk) {
+    DictionaryValue* result = new DictionaryValue();
+    result->SetString("mountPath", disk->mount_path());
+    result->SetString("devicePath", disk->device_path());
+    result->SetString("label", disk->device_label());
+    result->SetString("deviceType", DeviceTypeToString(disk->device_type()));
+    result->SetInteger("totalSizeKB", disk->total_size() / 1024);
+    result->SetBoolean("readOnly", disk->is_read_only());
+    return result;
+  }
 }
 
-DictionaryValue* DiskToDictionaryValue(
-    const chromeos::MountLibrary::Disk* disk) {
-  DictionaryValue* result = new DictionaryValue();
-  result->SetString("mountPath", disk->mount_path());
-  result->SetString("devicePath", disk->device_path());
-  result->SetString("label", disk->device_label());
-  result->SetString("deviceType", DeviceTypeToString(disk->device_type()));
-  result->SetInteger("totalSizeKB", disk->total_size() / 1024);
-  result->SetBoolean("readOnly", disk->is_read_only());
-  return result;
+void HideFileBrowserNotificationExternally(const std::string& category,
+    const std::string& system_path, ExtensionFileBrowserEventRouter* that) {
+  that->HideFileBrowserNotification(category, system_path);
 }
 
 ExtensionFileBrowserEventRouter::ExtensionFileBrowserEventRouter()
     : delegate_(new ExtensionFileBrowserEventRouter::FileWatcherDelegate()),
       profile_(NULL) {
 }
 
 ExtensionFileBrowserEventRouter::~ExtensionFileBrowserEventRouter() {
   DCHECK(file_watchers_.empty());
   STLDeleteValues(&file_watchers_);
@@ skipping 80 matching lines @@
 
 void ExtensionFileBrowserEventRouter::DiskChanged(
     chromeos::MountLibraryEventType event,
     const chromeos::MountLibrary::Disk* disk) {
   if (event == chromeos::MOUNT_DISK_ADDED) {
     OnDiskAdded(disk);
   } else if (event == chromeos::MOUNT_DISK_REMOVED) {
     OnDiskRemoved(disk);
   } else if (event == chromeos::MOUNT_DISK_CHANGED) {
     OnDiskChanged(disk);
+  } else if (event == chromeos::MOUNT_DISK_MOUNTED) {
+    OnDiskMounted(disk);

[tbarzic, 2011/07/26 22:49:50]

I added separate mount event handler, so you can remove these ;)

[sidor, 2011/07/28 00:20:51]

Done.

+  } else if (event == chromeos::MOUNT_DISK_UNMOUNTED) {
+    OnDiskUnmounted(disk);
   }
 }
 
 void ExtensionFileBrowserEventRouter::DeviceChanged(
     chromeos::MountLibraryEventType event,
     const std::string& device_path) {
   if (event == chromeos::MOUNT_DEVICE_ADDED) {
     OnDeviceAdded(device_path);
   } else if (event == chromeos::MOUNT_DEVICE_REMOVED) {
     OnDeviceRemoved(device_path);
   } else if (event == chromeos::MOUNT_DEVICE_SCANNED) {
     OnDeviceScanned(device_path);
+  } else if (event == chromeos::MOUNT_FORMATTING_STARTED) {
+    OnFormattingStarted(device_path);
+  } else if (event == chromeos::MOUNT_FORMATTING_FINISHED) {
+    OnFormattingFinished(device_path);
   }
 }
 
 void ExtensionFileBrowserEventRouter::HandleFileWatchNotification(
     const FilePath& local_path, bool got_error) {
   base::AutoLock lock(lock_);
   WatcherMap::const_iterator iter = file_watchers_.find(local_path);
   if (iter == file_watchers_.end()) {
     NOTREACHED();
     return;
@@ skipping 48 matching lines @@
   DictionaryValue* disk_info = DiskToDictionaryValue(disk);
   mount_info->Set("volumeInfo", disk_info);
 
   std::string args_json;
   base::JSONWriter::Write(&args, false /* pretty_print */, &args_json);
   profile_->GetExtensionEventRouter()->DispatchEventToRenderers(
       extension_event_names::kOnFileBrowserDiskChanged, args_json, NULL,
       GURL());
 }
 
+
 void ExtensionFileBrowserEventRouter::OnDiskAdded(
     const chromeos::MountLibrary::Disk* disk) {
   VLOG(1) << "Disk added: " << disk->device_path();
   if (disk->device_path().empty()) {
     VLOG(1) << "Empty system path for " << disk->device_path();
     return;
   }
   if (disk->is_parent()) {
     if (!disk->has_media()) {
-      HideDeviceNotification(disk->system_path());
+      HideFileBrowserNotification("MOUNT", disk->system_path());
       return;
     }
   }
 
   // If disk is not mounted yet, give it a try.
   if (disk->mount_path().empty()) {
     // Initiate disk mount operation.
     chromeos::MountLibrary* lib =
         chromeos::CrosLibrary::Get()->GetMountLibrary();
     lib->MountPath(disk->device_path().c_str());
   }
+  DispatchMountEvent(disk, true);
 }
 
 void ExtensionFileBrowserEventRouter::OnDiskRemoved(
     const chromeos::MountLibrary::Disk* disk) {
   VLOG(1) << "Disk removed: " << disk->device_path();
-  HideDeviceNotification(disk->system_path());
+  HideFileBrowserNotification("MOUNT", disk->system_path());
   MountPointMap::iterator iter = mounted_devices_.find(disk->device_path());
   if (iter == mounted_devices_.end())
     return;
 
   chromeos::MountLibrary* lib =
       chromeos::CrosLibrary::Get()->GetMountLibrary();
   // TODO(zelidrag): This for some reason does not work as advertized.
   // we might need to clean up mount directory on FILE thread here as well.
   lib->UnmountPath(disk->device_path().c_str());
-
   DispatchMountEvent(disk, false);
   mounted_devices_.erase(iter);
 }
 
 void ExtensionFileBrowserEventRouter::OnDiskChanged(
     const chromeos::MountLibrary::Disk* disk) {
   VLOG(1) << "Disk changed : " << disk->device_path();
   if (!disk->mount_path().empty()) {
-    HideDeviceNotification(disk->system_path());
+    HideFileBrowserNotification("MOUNT", disk->system_path());
     // Remember this mount point.
     if (mounted_devices_.find(disk->device_path()) == mounted_devices_.end()) {
       mounted_devices_.insert(
           std::pair<std::string, std::string>(disk->device_path(),
                                               disk->mount_path()));
       DispatchMountEvent(disk, true);
-      HideDeviceNotification(disk->system_path());
+      HideFileBrowserNotification("MOUNT", disk->system_path());
       FileManagerUtil::ShowFullTabUrl(profile_, FilePath(disk->mount_path()));
     }
   }
 }
 
+void ExtensionFileBrowserEventRouter::OnDiskMounted(
+    const chromeos::MountLibrary::Disk* disk) {
+  DispatchMountEvent(disk, true);
+}
+
+void ExtensionFileBrowserEventRouter::OnDiskUnmounted(
+    const chromeos::MountLibrary::Disk* disk) {
+  DispatchMountEvent(disk, false);
+}
+
 void ExtensionFileBrowserEventRouter::OnDeviceAdded(
     const std::string& device_path) {
   VLOG(1) << "Device added : " << device_path;
   // TODO(zelidrag): Find better icon here.
-  ShowDeviceNotification(device_path, IDR_PAGEINFO_INFO,
+  ShowFileBrowserNotification("MOUNT", device_path, IDR_PAGEINFO_INFO,
+      l10n_util::GetStringUTF16(IDS_REMOVABLE_DEVICE_DETECTION_TITLE),
       l10n_util::GetStringUTF16(IDS_REMOVABLE_DEVICE_SCANNING_MESSAGE));
-
 }
 
 void ExtensionFileBrowserEventRouter::OnDeviceRemoved(
     const std::string& system_path) {
-  HideDeviceNotification(system_path);
+  HideFileBrowserNotification("MOUNT", system_path);
 }
 
 void ExtensionFileBrowserEventRouter::OnDeviceScanned(
     const std::string& device_path) {
   VLOG(1) << "Device scanned : " << device_path;
 }
 
-void ExtensionFileBrowserEventRouter::ShowDeviceNotification(
-    const std::string& system_path, int icon_resource_id,
-    const string16& message) {
-  NotificationMap::iterator iter = FindNotificationForPath(system_path);
-  std::string mount_path;
-  if (iter != notifications_.end()) {
-    iter->second->Show(message, false, false);
+void ExtensionFileBrowserEventRouter::OnFormattingStarted(
+    const std::string& device_path) {
+  if (device_path[0] == '!') {
+    ShowFileBrowserNotification("FORMAT_FINISHED", device_path.substr(1, -1),
+        IDR_PAGEINFO_WARNING_MAJOR,
+        l10n_util::GetStringUTF16(IDS_FORMATTING_OF_DEVICE_FINISHED_TITLE),
+        l10n_util::GetStringUTF16(IDS_FORMATTING_STARTED_FAILURE_MESSAGE));
   } else {
-    if (!profile_) {
-      NOTREACHED();
-      return;
-    }
-    chromeos::SystemNotification* notification =
-        new chromeos::SystemNotification(
-            profile_,
-            system_path,
-            icon_resource_id,
-            l10n_util::GetStringUTF16(IDS_REMOVABLE_DEVICE_DETECTION_TITLE));
-    notifications_.insert(NotificationMap::value_type(system_path,
-        linked_ptr<chromeos::SystemNotification>(notification)));
-    notification->Show(message, false, false);
+    ShowFileBrowserNotification("FORMAT", device_path, IDR_PAGEINFO_INFO,
+        l10n_util::GetStringUTF16(IDS_FORMATTING_OF_DEVICE_PENDING_TITLE),
+        l10n_util::GetStringUTF16(IDS_FORMATTING_OF_DEVICE_PENDING_MESSAGE));
   }
 }
 
-void ExtensionFileBrowserEventRouter::HideDeviceNotification(
-    const std::string& system_path) {
-  NotificationMap::iterator iter = FindNotificationForPath(system_path);
+void ExtensionFileBrowserEventRouter::OnFormattingFinished(
+    const std::string& device_path) {
+  if (device_path[0] == '!') {
+    HideFileBrowserNotification("FORMAT", device_path.substr(1, -1));
+    ShowFileBrowserNotification("FORMAT_FINISHED", device_path.substr(1, -1),

[tbarzic, 2011/07/26 22:49:50]

you don't need -1 here.

[sidor, 2011/07/28 00:20:51]

Done.

+        IDR_PAGEINFO_WARNING_MAJOR,
+        l10n_util::GetStringUTF16(IDS_FORMATTING_OF_DEVICE_FINISHED_TITLE),
+        l10n_util::GetStringUTF16(IDS_FORMATTING_FINISHED_FAILURE_MESSAGE));
+  } else {
+    HideFileBrowserNotification("FORMAT", device_path);
+    ShowFileBrowserNotification("FORMAT_FINISHED", device_path,
+        IDR_PAGEINFO_INFO,
+        l10n_util::GetStringUTF16(IDS_FORMATTING_OF_DEVICE_FINISHED_TITLE),
+        l10n_util::GetStringUTF16(IDS_FORMATTING_FINISHED_SUCCESS_MESSAGE));
+    // Hide it after a couple of seconds
+    MessageLoop::current()->PostDelayedTask(FROM_HERE,
+        base::Bind(&HideFileBrowserNotificationExternally, "FORMAT_FINISHED",
+        device_path, this),
+        4000);
+    chromeos::MountLibrary* lib =
+      chromeos::CrosLibrary::Get()->GetMountLibrary();
+    lib->MountPath(device_path.c_str());

[tbarzic, 2011/07/26 22:49:50]

MountLibrary interface has changed, this now has to be.
lib->MountPath(device_path.c_str(), 
               chromeos::MOUNT_TYPE_DEVICE,
               chromeos::MountPathOptions());

[sidor, 2011/07/28 00:20:51]

Done.

+  }
+}
+
+void ExtensionFileBrowserEventRouter::ShowFileBrowserNotification(
+        const std::string& category, const std::string& system_path,

[tbarzic, 2011/07/26 22:49:50]

Since you are not using category and system_path separately in this method,
maybe passing notification_id as parameter would be better idea than calculating
it inside.

[sidor, 2011/07/28 00:20:51]

Nope, Look at the use of this function in code.

+        int icon_resource_id, const string16& title, const string16& message) {
+  std::string notification_id = category + system_path;
+  // New notification always created because, it might have been closed by now.
+  NotificationMap::iterator iter = FindNotificationForPath(notification_id);
+  if (iter != notifications_.end())
+    notifications_.erase(iter);
+  if (!profile_) {
+    NOTREACHED();
+    return;
+  }
+  chromeos::SystemNotification* notification =
+      new chromeos::SystemNotification(
+          profile_,
+          notification_id,
+          icon_resource_id,
+          title);
+  notifications_.insert(NotificationMap::value_type(notification_id,
+      linked_ptr<chromeos::SystemNotification>(notification)));
+  notification->Show(message, false, false);
+}
+
+void ExtensionFileBrowserEventRouter::HideFileBrowserNotification(
+    const std::string& category, const std::string& system_path) {
+  NotificationMap::iterator iter = FindNotificationForPath(
+      category + system_path);
   if (iter != notifications_.end()) {
     iter->second->Hide();
     notifications_.erase(iter);
   }
 }
 
 ExtensionFileBrowserEventRouter::NotificationMap::iterator
     ExtensionFileBrowserEventRouter::FindNotificationForPath(

[tbarzic, 2011/07/26 22:49:50]

I would change this name to FindNotificationForId

[sidor, 2011/07/28 00:20:51]

Done.

-        const std::string& system_path) {
+        const std::string& notification_id) {
   for (NotificationMap::iterator iter = notifications_.begin();
        iter != notifications_.end();
        ++iter) {
     const std::string& notification_device_path = iter->first;
     // Doing a sub string match so that we find if this new one is a subdevice
     // of another already inserted device.
-    if (StartsWithASCII(system_path, notification_device_path, true)) {
+    if (StartsWithASCII(notification_id, notification_device_path, true)) {
       return iter;
     }
   }
   return notifications_.end();
 }
 
 
 // ExtensionFileBrowserEventRouter::WatcherDelegate methods.
 ExtensionFileBrowserEventRouter::FileWatcherDelegate::FileWatcherDelegate() {
 }
@@ skipping 17 matching lines @@
           local_path,
           true));     // got_error
 }
 
 void
 ExtensionFileBrowserEventRouter::FileWatcherDelegate::HandleFileWatchOnUIThread(
      const FilePath& local_path, bool got_error) {
   ExtensionFileBrowserEventRouter::GetInstance()->HandleFileWatchNotification(
       local_path, got_error);
 }