src/chrome/browser/chromeos/extensions/file_manager/device_event_router.h

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef CHROME_BROWSER_CHROMEOS_EXTENSIONS_FILE_MANAGER_DEVICE_EVENT_ROUTER_H_
   6 #define CHROME_BROWSER_CHROMEOS_EXTENSIONS_FILE_MANAGER_DEVICE_EVENT_ROUTER_H_
   7
   8 #include <map>
   9 #include <string>
  10
  11 #include "base/memory/weak_ptr.h"
  12 #include "base/threading/thread_checker.h"
  13 #include "chrome/browser/chromeos/file_manager/volume_manager_observer.h"
  14 #include "chrome/common/extensions/api/file_manager_private.h"
  15 #include "chromeos/dbus/power_manager_client.h"
  16
  17 namespace file_manager {
  18
  19 enum DeviceState {
  20   // Device is not being hard unplugged.
  21   DEVICE_STATE_USUAL,
  22   // Device is hard unplugged.
  23   DEVICE_HARD_UNPLUGGED,
  24   // Device is hard unplugged and reported to the JavaScript side.
  25   DEVICE_HARD_UNPLUGGED_AND_REPORTED
  26 };
  27
  28 // Event router for device events.
  29 class DeviceEventRouter : public VolumeManagerObserver,
  30                           public chromeos::PowerManagerClient::Observer {
  31  public:
  32   DeviceEventRouter();
  33
  34   // |overriding_time_delta| overrides time delta of delayed tasks for testing
  35   // |so that the tasks are executed by RunLoop::RunUntilIdle.
  36   explicit DeviceEventRouter(base::TimeDelta overriding_time_delta);
  37
  38   virtual ~DeviceEventRouter();
  39
  40   // Turns the startup flag on, and then turns it off after few seconds.
  41   void Startup();
  42
  43   // VolumeManagerObserver overrides.
  44   virtual void OnDiskAdded(const chromeos::disks::DiskMountManager::Disk& disk,
  45                            bool mounting) OVERRIDE;
  46   virtual void OnDiskRemoved(
  47       const chromeos::disks::DiskMountManager::Disk& disk) OVERRIDE;
  48   virtual void OnDeviceAdded(const std::string& device_path) OVERRIDE;
  49   virtual void OnDeviceRemoved(const std::string& device_path) OVERRIDE;
  50   virtual void OnVolumeMounted(chromeos::MountError error_code,
  51                                const VolumeInfo& volume_info) OVERRIDE;
  52   virtual void OnVolumeUnmounted(chromeos::MountError error_code,
  53                                  const VolumeInfo& volume_info) OVERRIDE;
  54   virtual void OnFormatStarted(const std::string& device_path,
  55                                bool success) OVERRIDE;
  56   virtual void OnFormatCompleted(const std::string& device_path,
  57                                  bool success) OVERRIDE;
  58
  59   // PowerManagerClient::Observer overrides.
  60   virtual void SuspendImminent() OVERRIDE;
  61   virtual void SuspendDone(const base::TimeDelta& sleep_duration) OVERRIDE;
  62
  63   bool is_resuming() const { return is_resuming_; }
  64   bool is_starting_up() const { return is_starting_up_; }
  65
  66  protected:
  67   // Handles a device event containing |type| and |device_path|.
  68   virtual void OnDeviceEvent(
  69       extensions::api::file_manager_private::DeviceEventType type,
  70       const std::string& device_path) = 0;
  71   // Returns external storage is disabled or not.
  72   virtual bool IsExternalStorageDisabled() = 0;
  73
  74  private:
  75   void StartupDelayed();
  76   void OnDeviceAddedDelayed(const std::string& device_path);
  77   void SuspendDoneDelayed();
  78
  79   // Obtains device state of the device having |device_path|.
  80   DeviceState GetDeviceState(const std::string& device_path) const;
  81
  82   // Sets device state to the device having |device_path|.
  83   void SetDeviceState(const std::string& device_path, DeviceState state);
  84
  85   // Whther to use zero time delta for testing or not.
  86   const base::TimeDelta resume_time_delta_;
  87   const base::TimeDelta startup_time_delta_;
  88
  89   // Whether the profile is starting up or not.
  90   bool is_starting_up_;
  91
  92   // Whether the system is resuming or not.
  93   bool is_resuming_;
  94
  95   // Map of device path and device state.
  96   std::map<std::string, DeviceState> device_states_;
  97
  98   // Thread checker.
  99   base::ThreadChecker thread_checker_;
 100
 101   // Note: This should remain the last member so it'll be destroyed and
 102   // invalidate the weak pointers before any other members are destroyed.
 103   base::WeakPtrFactory<DeviceEventRouter> weak_factory_;
 104   DISALLOW_COPY_AND_ASSIGN(DeviceEventRouter);
 105 };
 106 }  // namespace file_manager
 107
 108 #endif  // CHROME_BROWSER_CHROMEOS_EXTENSIONS_FILE_MANAGER_DEVICE_EVENT_ROUTER_H_