Index: chrome/browser/chromeos/ui/low_disk_notification.h |
diff --git a/chrome/browser/chromeos/ui/low_disk_notification.h b/chrome/browser/chromeos/ui/low_disk_notification.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..929e2e2915f2fbb4c596b9fe797c44196991c03c |
--- /dev/null |
+++ b/chrome/browser/chromeos/ui/low_disk_notification.h |
@@ -0,0 +1,75 @@ |
+// Copyright 2016 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. |
+ |
+#ifndef CHROME_BROWSER_CHROMEOS_UI_LOW_DISK_NOTIFICATION_H_ |
+#define CHROME_BROWSER_CHROMEOS_UI_LOW_DISK_NOTIFICATION_H_ |
+ |
+#include <stdint.h> |
+ |
+#include "base/macros.h" |
+#include "base/memory/weak_ptr.h" |
+#include "base/threading/thread_checker.h" |
+#include "base/time/time.h" |
+ |
+namespace message_center { |
+class Notification; |
+class MessageCenter; |
+} |
+ |
+namespace chromeos { |
+ |
+// LowDiskNotification manages the notification informing the user of low disk |
+// space. It is responsible for both creating the notifications in message |
+// center as well as throttling notifications shown to the user. |
+// This class should be created after DBus has been initialized and destroyed |
+// before DBus has been shutdown. |
+// This class must be instantiated on the UI thread. |
+class LowDiskNotification { |
+ public: |
+ // Registers this class as the CryptohomeClient LowDiskSpaceHandler. |
+ LowDiskNotification(); |
+ |
+ // Resets CryptohomeClient LowDiskSpaceHandler. |
+ ~LowDiskNotification(); |
+ |
+ // Called when the device is running low on disk space. This is responsible |
+ // for deciding whether a notification should be shown or not and showing it |
+ // if appropriate. This must be called from the thread that instantiated this |
+ // object. |
+ void OnLowDiskSpace(uint64_t free_disk_bytes); |
+ |
+ private: |
+ friend class LowDiskNotificationTest; |
+ |
+ enum Severity { NONE = -1, MEDIUM = 0, HIGH = 1 }; |
+ |
+ // Creates a notification for the specified severity. If the severity does |
+ // not match a known value MEDIUM is used by default. |
+ std::unique_ptr<message_center::Notification> CreateNotification( |
+ Severity severity); |
+ |
+ // Gets the severity of the low disk status based on the amount of free space |
+ // left on the disk. |
+ Severity GetSeverity(uint64_t free_disk_bytes); |
+ |
+ // Sets the MessageCenter instance to use. Should only be used in tests. |
+ void SetMessageCenterForTest(message_center::MessageCenter* message_center); |
+ |
+ // Sets the minimum time to wait between notifications of the same severity. |
+ // Should only be used in tests. |
+ void SetNotificationIntervalForTest(base::TimeDelta interval); |
+ |
+ base::Time last_notification_time_; |
+ Severity last_notification_severity_; |
+ message_center::MessageCenter* message_center_; |
+ base::TimeDelta notification_interval_; |
+ base::ThreadChecker thread_checker_; |
+ base::WeakPtrFactory<LowDiskNotification> weak_ptr_factory_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(LowDiskNotification); |
+}; |
+ |
+} // namespace chromeos |
+ |
+#endif // CHROME_BROWSER_CHROMEOS_UI_LOW_DISK_NOTIFICATION_H_ |