Extend the capability of SystemMonitor to support watching storage free space change

chrome/browser/system_monitor/removable_device_notifications_window_win.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/system_monitor/removable_device_notifications_window_win.h"
 
 #include <windows.h>
 #include <dbt.h>
 
 #include <string>
@@ skipping 61 matching lines @@
           RemovableDeviceNotificationsWindowWin::WndProcThunk>,
       0, 0, 0, NULL, NULL, NULL, NULL, NULL,
       &window_class);
   instance_ = window_class.hInstance;
   atom_ = RegisterClassEx(&window_class);
   DCHECK(atom_);
 
   window_ = CreateWindow(MAKEINTATOM(atom_), 0, 0, 0, 0, 0, 0, 0, 0, instance_,
                          0);
   SetWindowLongPtr(window_, GWLP_USERDATA, reinterpret_cast<LONG_PTR>(this));
+
+  if (SystemMonitor::Get())
+    SystemMonitor::Get()->set_storage_free_space_delegate(this);
 }
 
 RemovableDeviceNotificationsWindowWin::~RemovableDeviceNotificationsWindowWin(
     ) {
+  if (SystemMonitor::Get() &&
+      SystemMonitor::Get()->storage_free_space_delegate())
+    SystemMonitor::Get()->set_storage_free_space_delegate(NULL);
+
   if (window_)
     DestroyWindow(window_);
 
   if (atom_)
     UnregisterClass(MAKEINTATOM(atom_), instance_);
 }
 
 LRESULT RemovableDeviceNotificationsWindowWin::OnDeviceChange(UINT event_type,
                                                               DWORD data) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
@@ skipping 58 matching lines @@
     const std::string& id,
     const FilePath::StringType& device_name,
     const FilePath& path) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
 
   SystemMonitor::Get()->ProcessRemovableStorageAttached(id,
                                                         device_name,
                                                         path.value());
 }
 
+void RemovableDeviceNotificationsWindowWin::StartWatchingStorage(
+    const FilePath& path) {
+  NOTIMPLEMENTED();
+}
+
+void RemovableDeviceNotificationsWindowWin::StopWatchingStorage(
+    const FilePath& path) {
+  NOTIMPLEMENTED();
+}
+
 LRESULT CALLBACK RemovableDeviceNotificationsWindowWin::WndProc(
     HWND hwnd, UINT message, WPARAM wparam, LPARAM lparam) {
   switch (message) {
     case WM_DEVICECHANGE:
       return OnDeviceChange(static_cast<UINT>(wparam),
                             static_cast<DWORD>(lparam));
     default:
       break;
   }
 
   return ::DefWindowProc(hwnd, message, wparam, lparam);
 }
 
 // static
 LRESULT CALLBACK RemovableDeviceNotificationsWindowWin::WndProcThunk(
     HWND hwnd,
     UINT message,
     WPARAM wparam,
     LPARAM lparam) {
   RemovableDeviceNotificationsWindowWin* msg_wnd =
       reinterpret_cast<RemovableDeviceNotificationsWindowWin*>(
           GetWindowLongPtr(hwnd, GWLP_USERDATA));
   if (msg_wnd)
     return msg_wnd->WndProc(hwnd, message, wparam, lparam);
   return ::DefWindowProc(hwnd, message, wparam, lparam);
 }
 
 }  // namespace chrome