1 // Copyright 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_EXTENSIONS_API_SYSTEM_STORAGE_STORAGE_INFO_PROVIDER_H_ 6 #define CHROME_BROWSER_EXTENSIONS_API_SYSTEM_STORAGE_STORAGE_INFO_PROVIDER_H_ 7 8 #include <set> 9 10 #include "base/lazy_instance.h" 11 #include "base/memory/ref_counted.h" 12 #include "base/observer_list_threadsafe.h" 13 #include "base/timer/timer.h" 14 #include "chrome/browser/extensions/api/system_info/system_info_provider.h" 15 #include "chrome/common/extensions/api/system_storage.h" 16 17 namespace storage_monitor { 18 class StorageInfo; 19 } 20 21 namespace extensions { 22 23 namespace systeminfo { 24 25 // Build StorageUnitInfo struct from StorageInfo instance. The |unit| 26 // parameter is the output value. 27 void BuildStorageUnitInfo(const storage_monitor::StorageInfo& info, 28 api::system_storage::StorageUnitInfo* unit); 29 30 } // namespace systeminfo 31 32 typedef std::vector<linked_ptr< 33 api::system_storage::StorageUnitInfo> > 34 StorageUnitInfoList; 35 36 class StorageInfoProvider : public SystemInfoProvider { 37 public: 38 typedef base::Callback<void(const std::string&, double)> 39 GetStorageFreeSpaceCallback; 40 41 // Get the single shared instance of StorageInfoProvider. 42 static StorageInfoProvider* Get(); 43 44 // SystemInfoProvider implementations 45 virtual void PrepareQueryOnUIThread() OVERRIDE; 46 virtual void InitializeProvider(const base::Closure& do_query_info_callback) 47 OVERRIDE; 48 49 virtual double GetStorageFreeSpaceFromTransientIdOnFileThread( 50 const std::string& transient_id); 51 52 const StorageUnitInfoList& storage_unit_info_list() const; 53 54 static void InitializeForTesting(scoped_refptr<StorageInfoProvider> provider); 55 56 protected: 57 StorageInfoProvider(); 58 59 virtual ~StorageInfoProvider(); 60 61 // Put all available storages' information into |info_|. 62 void GetAllStoragesIntoInfoList(); 63 64 // The last information filled up by QueryInfo and is accessed on multiple 65 // threads, but the whole class is being guarded by SystemInfoProvider base 66 // class. 67 // 68 // |info_| is accessed on the UI thread while |is_waiting_for_completion_| is 69 // false and on the sequenced worker pool while |is_waiting_for_completion_| 70 // is true. 71 StorageUnitInfoList info_; 72 73 private: 74 // SystemInfoProvider implementations. 75 // Override to query the available capacity of all known storage devices on 76 // the blocking pool, including fixed and removable devices. 77 virtual bool QueryInfo() OVERRIDE; 78 79 static base::LazyInstance<scoped_refptr<StorageInfoProvider> > provider_; 80 81 DISALLOW_COPY_AND_ASSIGN(StorageInfoProvider); 82 }; 83 84 } // namespace extensions 85 86 #endif // CHROME_BROWSER_EXTENSIONS_API_SYSTEM_STORAGE_STORAGE_INFO_PROVIDER_H_ 87