1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
// Copyright 2013 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "extensions/browser/api/system_storage/storage_info_provider.h"
#include "base/system/sys_info.h"
#include "components/storage_monitor/storage_info.h"
#include "components/storage_monitor/storage_monitor.h"
#include "content/public/browser/browser_thread.h"
using storage_monitor::StorageInfo;
using storage_monitor::StorageMonitor;
namespace extensions {
using content::BrowserThread;
using api::system_storage::StorageUnitInfo;
// Static member intialization.
base::LazyInstance<scoped_refptr<StorageInfoProvider>>::DestructorAtExit
StorageInfoProvider::provider_ = LAZY_INSTANCE_INITIALIZER;
StorageInfoProvider::StorageInfoProvider() = default;
StorageInfoProvider::~StorageInfoProvider() = default;
void StorageInfoProvider::InitializeForTesting(
scoped_refptr<StorageInfoProvider> provider) {
DCHECK(provider.get() != nullptr);
provider_.Get() = provider;
}
void StorageInfoProvider::PrepareQueryOnUIThread() {
// Get all available storage devices before invoking |QueryInfo()|.
GetAllStoragesIntoInfoList();
}
void StorageInfoProvider::InitializeProvider(
base::OnceClosure do_query_info_callback) {
DCHECK_CURRENTLY_ON(BrowserThread::UI);
// Register the |do_query_info_callback| callback to StorageMonitor.
// See the comments of StorageMonitor::EnsureInitialized about when the
// callback gets run.
StorageMonitor::GetInstance()->EnsureInitialized(
std::move(do_query_info_callback));
}
bool StorageInfoProvider::QueryInfo() {
// No info to query since we get all available storage devices' info in
// |PrepareQueryOnUIThread()|.
return true;
}
void StorageInfoProvider::GetAllStoragesIntoInfoList() {
info_.clear();
std::vector<StorageInfo> storage_list =
StorageMonitor::GetInstance()->GetAllAvailableStorages();
for (const auto& info : storage_list) {
StorageUnitInfo unit;
systeminfo::BuildStorageUnitInfo(info, &unit);
info_.push_back(std::move(unit));
}
}
double StorageInfoProvider::GetStorageFreeSpaceFromTransientIdAsync(
const std::string& transient_id) {
std::vector<StorageInfo> storage_list =
StorageMonitor::GetInstance()->GetAllAvailableStorages();
std::string device_id =
StorageMonitor::GetInstance()->GetDeviceIdForTransientId(transient_id);
// Lookup the matched storage info by |device_id|.
for (const auto& info : storage_list) {
if (device_id == info.device_id()) {
return static_cast<double>(base::SysInfo::AmountOfFreeDiskSpace(
base::FilePath(info.location())));
}
}
return -1;
}
// static
StorageInfoProvider* StorageInfoProvider::Get() {
if (provider_.Get().get() == nullptr) {
provider_.Get() = new StorageInfoProvider();
}
return provider_.Get().get();
}
} // namespace extensions
|
