1 // Copyright (c) 2013 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_IMAGEBURNER_BURN_DEVICE_HANDLER_H_ 6 #define CHROME_BROWSER_CHROMEOS_IMAGEBURNER_BURN_DEVICE_HANDLER_H_ 7 8 #include <string> 9 #include <vector> 10 11 #include "base/callback.h" 12 #include "chromeos/dbus/cros_disks_client.h" 13 #include "chromeos/disks/disk_mount_manager.h" 14 15 namespace chromeos { 16 namespace imageburner { 17 18 // This is the implementation for the communication between BurnManager 19 // and DiskMountManager. 20 // The main reason this is NOT merged into BurnManager is to improve 21 // testability, since both BurnManager and DiskMountManager are singleton 22 // in real usage. 23 class BurnDeviceHandler : public disks::DiskMountManager::Observer { 24 public: 25 // Triggered when a burnable device is added or removed. 26 typedef base::Callback<void(const disks::DiskMountManager::Disk& disk)> 27 DiskCallback; 28 29 // This class takes the pointer of DiskMountManager to improve testability, 30 // although it is singleton in the real usage. 31 explicit BurnDeviceHandler(disks::DiskMountManager* disk_mount_manager); 32 virtual ~BurnDeviceHandler(); 33 34 // |add_callback| will be called when a new burnable device is added with 35 // the device's information. 36 // |remove_callback| will be called when a burnable device is removed. 37 // Note: This class is designed to connect to only one BurnManager, 38 // so it supports only single callback for each add and remove intentionally 39 // (rather than ObserverList). 40 void SetCallbacks(const DiskCallback& add_callback, 41 const DiskCallback& remove_callback); 42 43 // Returns devices on which we can burn recovery image. 44 std::vector<disks::DiskMountManager::Disk> GetBurnableDevices(); 45 46 // DiskMountManager::Observer overrides. 47 virtual void OnDiskEvent( 48 disks::DiskMountManager::DiskEvent event, 49 const disks::DiskMountManager::Disk* disk) OVERRIDE; 50 virtual void OnDeviceEvent( 51 disks::DiskMountManager::DeviceEvent event, 52 const std::string& device_path) OVERRIDE; 53 virtual void OnMountEvent( 54 disks::DiskMountManager::MountEvent event, 55 MountError error_code, 56 const disks::DiskMountManager::MountPointInfo& mount_info) OVERRIDE; 57 virtual void OnFormatEvent( 58 disks::DiskMountManager::FormatEvent event, 59 FormatError error_code, 60 const std::string& device_path) OVERRIDE; 61 62 private: 63 disks::DiskMountManager* disk_mount_manager_; // Not owned by this class. 64 DiskCallback add_callback_; 65 DiskCallback remove_callback_; 66 67 DISALLOW_COPY_AND_ASSIGN(BurnDeviceHandler); 68 }; 69 70 } // namespace imageburner 71 } // namespace chromeos 72 73 #endif // CHROME_BROWSER_CHROMEOS_IMAGEBURNER_BURN_DEVICE_HANDLER_H_ 74