| OLD | NEW |
| (Empty) |
| 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 #include "chrome/browser/extensions/api/system_info/system_info_api.h" | |
| 6 | |
| 7 #include <set> | |
| 8 | |
| 9 #include "base/bind.h" | |
| 10 #include "base/lazy_instance.h" | |
| 11 #include "base/memory/scoped_ptr.h" | |
| 12 #include "base/memory/singleton.h" | |
| 13 #include "base/strings/string_util.h" | |
| 14 #include "base/values.h" | |
| 15 #include "chrome/browser/browser_process.h" | |
| 16 #include "chrome/browser/extensions/api/system_display/display_info_provider.h" | |
| 17 #include "chrome/browser/extensions/api/system_storage/storage_info_provider.h" | |
| 18 #include "chrome/browser/extensions/event_router_forwarder.h" | |
| 19 #include "chrome/common/extensions/api/system_display.h" | |
| 20 #include "chrome/common/extensions/api/system_storage.h" | |
| 21 #include "components/storage_monitor/removable_storage_observer.h" | |
| 22 #include "components/storage_monitor/storage_info.h" | |
| 23 #include "components/storage_monitor/storage_monitor.h" | |
| 24 #include "content/public/browser/browser_thread.h" | |
| 25 #include "ui/gfx/display_observer.h" | |
| 26 #include "ui/gfx/screen.h" | |
| 27 | |
| 28 namespace extensions { | |
| 29 | |
| 30 using api::system_storage::StorageUnitInfo; | |
| 31 using content::BrowserThread; | |
| 32 using storage_monitor::StorageMonitor; | |
| 33 | |
| 34 namespace system_display = api::system_display; | |
| 35 namespace system_storage = api::system_storage; | |
| 36 | |
| 37 namespace { | |
| 38 | |
| 39 bool IsDisplayChangedEvent(const std::string& event_name) { | |
| 40 return event_name == system_display::OnDisplayChanged::kEventName; | |
| 41 } | |
| 42 | |
| 43 bool IsSystemStorageEvent(const std::string& event_name) { | |
| 44 return (event_name == system_storage::OnAttached::kEventName || | |
| 45 event_name == system_storage::OnDetached::kEventName); | |
| 46 } | |
| 47 | |
| 48 // Event router for systemInfo API. It is a singleton instance shared by | |
| 49 // multiple profiles. | |
| 50 class SystemInfoEventRouter : public gfx::DisplayObserver, | |
| 51 public storage_monitor::RemovableStorageObserver { | |
| 52 public: | |
| 53 static SystemInfoEventRouter* GetInstance(); | |
| 54 | |
| 55 SystemInfoEventRouter(); | |
| 56 virtual ~SystemInfoEventRouter(); | |
| 57 | |
| 58 // Add/remove event listener for the |event_name| event. | |
| 59 void AddEventListener(const std::string& event_name); | |
| 60 void RemoveEventListener(const std::string& event_name); | |
| 61 | |
| 62 private: | |
| 63 // gfx::DisplayObserver: | |
| 64 virtual void OnDisplayAdded(const gfx::Display& new_display) OVERRIDE; | |
| 65 virtual void OnDisplayRemoved(const gfx::Display& old_display) OVERRIDE; | |
| 66 virtual void OnDisplayMetricsChanged(const gfx::Display& display, | |
| 67 uint32_t metrics) OVERRIDE; | |
| 68 | |
| 69 // RemovableStorageObserver implementation. | |
| 70 virtual void OnRemovableStorageAttached( | |
| 71 const storage_monitor::StorageInfo& info) OVERRIDE; | |
| 72 virtual void OnRemovableStorageDetached( | |
| 73 const storage_monitor::StorageInfo& info) OVERRIDE; | |
| 74 | |
| 75 // Called from any thread to dispatch the systemInfo event to all extension | |
| 76 // processes cross multiple profiles. | |
| 77 void DispatchEvent(const std::string& event_name, | |
| 78 scoped_ptr<base::ListValue> args); | |
| 79 | |
| 80 // Called to dispatch the systemInfo.display.onDisplayChanged event. | |
| 81 void OnDisplayChanged(); | |
| 82 | |
| 83 // Used to record the event names being watched. | |
| 84 std::multiset<std::string> watching_event_set_; | |
| 85 | |
| 86 bool has_storage_monitor_observer_; | |
| 87 | |
| 88 DISALLOW_COPY_AND_ASSIGN(SystemInfoEventRouter); | |
| 89 }; | |
| 90 | |
| 91 static base::LazyInstance<SystemInfoEventRouter>::Leaky | |
| 92 g_system_info_event_router = LAZY_INSTANCE_INITIALIZER; | |
| 93 | |
| 94 // static | |
| 95 SystemInfoEventRouter* SystemInfoEventRouter::GetInstance() { | |
| 96 return g_system_info_event_router.Pointer(); | |
| 97 } | |
| 98 | |
| 99 SystemInfoEventRouter::SystemInfoEventRouter() | |
| 100 : has_storage_monitor_observer_(false) { | |
| 101 } | |
| 102 | |
| 103 SystemInfoEventRouter::~SystemInfoEventRouter() { | |
| 104 if (has_storage_monitor_observer_) { | |
| 105 StorageMonitor* storage_monitor = StorageMonitor::GetInstance(); | |
| 106 if (storage_monitor) | |
| 107 storage_monitor->RemoveObserver(this); | |
| 108 } | |
| 109 } | |
| 110 | |
| 111 void SystemInfoEventRouter::AddEventListener(const std::string& event_name) { | |
| 112 DCHECK_CURRENTLY_ON(BrowserThread::UI); | |
| 113 | |
| 114 watching_event_set_.insert(event_name); | |
| 115 if (watching_event_set_.count(event_name) > 1) | |
| 116 return; | |
| 117 | |
| 118 if (IsDisplayChangedEvent(event_name)) { | |
| 119 gfx::Screen* screen = DisplayInfoProvider::Get()->GetActiveScreen(); | |
| 120 if (screen) | |
| 121 screen->AddObserver(this); | |
| 122 } | |
| 123 | |
| 124 if (IsSystemStorageEvent(event_name)) { | |
| 125 if (!has_storage_monitor_observer_) { | |
| 126 has_storage_monitor_observer_ = true; | |
| 127 DCHECK(StorageMonitor::GetInstance()->IsInitialized()); | |
| 128 StorageMonitor::GetInstance()->AddObserver(this); | |
| 129 } | |
| 130 } | |
| 131 } | |
| 132 | |
| 133 void SystemInfoEventRouter::RemoveEventListener(const std::string& event_name) { | |
| 134 DCHECK_CURRENTLY_ON(BrowserThread::UI); | |
| 135 | |
| 136 std::multiset<std::string>::iterator it = | |
| 137 watching_event_set_.find(event_name); | |
| 138 if (it != watching_event_set_.end()) { | |
| 139 watching_event_set_.erase(it); | |
| 140 if (watching_event_set_.count(event_name) > 0) | |
| 141 return; | |
| 142 } | |
| 143 | |
| 144 if (IsDisplayChangedEvent(event_name)) { | |
| 145 gfx::Screen* screen = DisplayInfoProvider::Get()->GetActiveScreen(); | |
| 146 if (screen) | |
| 147 screen->RemoveObserver(this); | |
| 148 } | |
| 149 | |
| 150 if (IsSystemStorageEvent(event_name)) { | |
| 151 const std::string& other_event_name = | |
| 152 (event_name == system_storage::OnDetached::kEventName) ? | |
| 153 system_storage::OnAttached::kEventName : | |
| 154 system_storage::OnDetached::kEventName; | |
| 155 if (watching_event_set_.count(other_event_name) == 0) { | |
| 156 StorageMonitor::GetInstance()->RemoveObserver(this); | |
| 157 has_storage_monitor_observer_ = false; | |
| 158 } | |
| 159 } | |
| 160 } | |
| 161 | |
| 162 void SystemInfoEventRouter::OnRemovableStorageAttached( | |
| 163 const storage_monitor::StorageInfo& info) { | |
| 164 StorageUnitInfo unit; | |
| 165 systeminfo::BuildStorageUnitInfo(info, &unit); | |
| 166 scoped_ptr<base::ListValue> args(new base::ListValue); | |
| 167 args->Append(unit.ToValue().release()); | |
| 168 DispatchEvent(system_storage::OnAttached::kEventName, args.Pass()); | |
| 169 } | |
| 170 | |
| 171 void SystemInfoEventRouter::OnRemovableStorageDetached( | |
| 172 const storage_monitor::StorageInfo& info) { | |
| 173 scoped_ptr<base::ListValue> args(new base::ListValue); | |
| 174 std::string transient_id = | |
| 175 StorageMonitor::GetInstance()->GetTransientIdForDeviceId( | |
| 176 info.device_id()); | |
| 177 args->AppendString(transient_id); | |
| 178 | |
| 179 DispatchEvent(system_storage::OnDetached::kEventName, args.Pass()); | |
| 180 } | |
| 181 | |
| 182 void SystemInfoEventRouter::OnDisplayAdded(const gfx::Display& new_display) { | |
| 183 OnDisplayChanged(); | |
| 184 } | |
| 185 | |
| 186 void SystemInfoEventRouter::OnDisplayRemoved(const gfx::Display& old_display) { | |
| 187 OnDisplayChanged(); | |
| 188 } | |
| 189 | |
| 190 void SystemInfoEventRouter::OnDisplayMetricsChanged(const gfx::Display& display, | |
| 191 uint32_t metrics) { | |
| 192 OnDisplayChanged(); | |
| 193 } | |
| 194 | |
| 195 void SystemInfoEventRouter::OnDisplayChanged() { | |
| 196 scoped_ptr<base::ListValue> args(new base::ListValue()); | |
| 197 DispatchEvent(system_display::OnDisplayChanged::kEventName, args.Pass()); | |
| 198 } | |
| 199 | |
| 200 void SystemInfoEventRouter::DispatchEvent(const std::string& event_name, | |
| 201 scoped_ptr<base::ListValue> args) { | |
| 202 g_browser_process->extension_event_router_forwarder()-> | |
| 203 BroadcastEventToRenderers(event_name, args.Pass(), GURL()); | |
| 204 } | |
| 205 | |
| 206 void AddEventListener(const std::string& event_name) { | |
| 207 SystemInfoEventRouter::GetInstance()->AddEventListener(event_name); | |
| 208 } | |
| 209 | |
| 210 void RemoveEventListener(const std::string& event_name) { | |
| 211 SystemInfoEventRouter::GetInstance()->RemoveEventListener(event_name); | |
| 212 } | |
| 213 | |
| 214 } // namespace | |
| 215 | |
| 216 static base::LazyInstance<BrowserContextKeyedAPIFactory<SystemInfoAPI> > | |
| 217 g_factory = LAZY_INSTANCE_INITIALIZER; | |
| 218 | |
| 219 // static | |
| 220 BrowserContextKeyedAPIFactory<SystemInfoAPI>* | |
| 221 SystemInfoAPI::GetFactoryInstance() { | |
| 222 return g_factory.Pointer(); | |
| 223 } | |
| 224 | |
| 225 SystemInfoAPI::SystemInfoAPI(content::BrowserContext* context) | |
| 226 : browser_context_(context) { | |
| 227 EventRouter* router = EventRouter::Get(browser_context_); | |
| 228 router->RegisterObserver(this, system_storage::OnAttached::kEventName); | |
| 229 router->RegisterObserver(this, system_storage::OnDetached::kEventName); | |
| 230 router->RegisterObserver(this, system_display::OnDisplayChanged::kEventName); | |
| 231 } | |
| 232 | |
| 233 SystemInfoAPI::~SystemInfoAPI() { | |
| 234 } | |
| 235 | |
| 236 void SystemInfoAPI::Shutdown() { | |
| 237 EventRouter::Get(browser_context_)->UnregisterObserver(this); | |
| 238 } | |
| 239 | |
| 240 void SystemInfoAPI::OnListenerAdded(const EventListenerInfo& details) { | |
| 241 if (IsSystemStorageEvent(details.event_name)) { | |
| 242 StorageMonitor::GetInstance()->EnsureInitialized( | |
| 243 base::Bind(&AddEventListener, details.event_name)); | |
| 244 } else { | |
| 245 AddEventListener(details.event_name); | |
| 246 } | |
| 247 } | |
| 248 | |
| 249 void SystemInfoAPI::OnListenerRemoved(const EventListenerInfo& details) { | |
| 250 if (IsSystemStorageEvent(details.event_name)) { | |
| 251 StorageMonitor::GetInstance()->EnsureInitialized( | |
| 252 base::Bind(&RemoveEventListener, details.event_name)); | |
| 253 } else { | |
| 254 RemoveEventListener(details.event_name); | |
| 255 } | |
| 256 } | |
| 257 | |
| 258 } // namespace extensions | |
| OLD | NEW |