1 // Copyright (c) 2011 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_UPGRADE_DETECTOR_H_ 6 #define CHROME_BROWSER_UPGRADE_DETECTOR_H_ 7 8 #include "base/timer/timer.h" 9 #include "chrome/browser/idle.h" 10 #include "ui/gfx/image/image.h" 11 12 class PrefRegistrySimple; 13 14 /////////////////////////////////////////////////////////////////////////////// 15 // UpgradeDetector 16 // 17 // This class is a singleton class that monitors when an upgrade happens in the 18 // background. We basically ask Omaha what it thinks the latest version is and 19 // if our version is lower we send out a notification upon: 20 // a) Detecting an upgrade and... 21 // b) When we think the user should be notified about the upgrade. 22 // The latter happens much later, since we don't want to be too annoying. 23 // 24 class UpgradeDetector { 25 public: 26 // The Homeland Security Upgrade Advisory System. 27 enum UpgradeNotificationAnnoyanceLevel { 28 UPGRADE_ANNOYANCE_NONE = 0, // What? Me worry? 29 UPGRADE_ANNOYANCE_LOW, // Green. 30 UPGRADE_ANNOYANCE_ELEVATED, // Yellow. 31 UPGRADE_ANNOYANCE_HIGH, // Red. 32 UPGRADE_ANNOYANCE_SEVERE, // Orange. 33 UPGRADE_ANNOYANCE_CRITICAL, // Red exclamation mark. 34 }; 35 36 // The two types of icons we know about. 37 enum UpgradeNotificationIconType { 38 UPGRADE_ICON_TYPE_BADGE = 0, // For overlay badging of the wrench menu. 39 UPGRADE_ICON_TYPE_MENU_ICON, // For showing in the wrench menu. 40 }; 41 42 // Returns the singleton implementation instance. 43 static UpgradeDetector* GetInstance(); 44 45 virtual ~UpgradeDetector(); 46 47 static void RegisterPrefs(PrefRegistrySimple* registry); 48 49 // Whether the user should be notified about an upgrade. 50 bool notify_upgrade() const { return notify_upgrade_; } 51 52 // Whether the upgrade recommendation is due to Chrome being outdated. 53 bool is_outdated_install() const { 54 return upgrade_available_ == UPGRADE_NEEDED_OUTDATED_INSTALL; 55 } 56 57 // Whether the upgrade recommendation is due to Chrome being outdated AND 58 // auto-update is turned off. 59 bool is_outdated_install_no_au() const { 60 return upgrade_available_ == UPGRADE_NEEDED_OUTDATED_INSTALL_NO_AU; 61 } 62 63 // Notifify this object that the user has acknowledged the critical update 64 // so we don't need to complain about it for now. 65 void acknowledge_critical_update() { 66 critical_update_acknowledged_ = true; 67 } 68 69 // Whether the user has acknowledged the critical update. 70 bool critical_update_acknowledged() const { 71 return critical_update_acknowledged_; 72 } 73 74 // When the last upgrade was detected. 75 const base::Time& upgrade_detected_time() const { 76 return upgrade_detected_time_; 77 } 78 79 // Retrieves the right icon ID based on the degree of severity (see 80 // UpgradeNotificationAnnoyanceLevel, each level has an an accompanying icon 81 // to go with it). |type| determines which class of icons the caller wants, 82 // either an icon appropriate for badging the wrench menu or one to display 83 // within the wrench menu. 84 int GetIconResourceID(UpgradeNotificationIconType type); 85 86 UpgradeNotificationAnnoyanceLevel upgrade_notification_stage() const { 87 return upgrade_notification_stage_; 88 } 89 90 protected: 91 UpgradeDetector(); 92 93 // Sends out UPGRADE_DETECTED notification and record upgrade_detected_time_. 94 void NotifyUpgradeDetected(); 95 96 // Sends out UPGRADE_RECOMMENDED notification and set notify_upgrade_. 97 void NotifyUpgradeRecommended(); 98 99 void set_upgrade_notification_stage(UpgradeNotificationAnnoyanceLevel stage) { 100 upgrade_notification_stage_ = stage; 101 } 102 103 enum UpgradeAvailable { 104 // If no update is available and current install is recent enough. 105 UPGRADE_AVAILABLE_NONE, 106 // If a regular update is available. 107 UPGRADE_AVAILABLE_REGULAR, 108 // If a critical update to Chrome has been installed, such as a zero-day 109 // fix. 110 UPGRADE_AVAILABLE_CRITICAL, 111 // If no update to Chrome has been installed for more than the recommended 112 // time. 113 UPGRADE_NEEDED_OUTDATED_INSTALL, 114 // If no update to Chrome has been installed for more than the recommended 115 // time AND auto-update is turned off. 116 UPGRADE_NEEDED_OUTDATED_INSTALL_NO_AU, 117 } upgrade_available_; 118 119 // Whether the user has acknowledged the critical update. 120 bool critical_update_acknowledged_; 121 122 private: 123 // Initiates an Idle check. See IdleCallback below. 124 void CheckIdle(); 125 126 // The callback for the IdleCheck. Tells us whether Chrome has received any 127 // input events since the specified time. 128 void IdleCallback(IdleState state); 129 130 // When the upgrade was detected. 131 base::Time upgrade_detected_time_; 132 133 // A timer to check to see if we've been idle for long enough to show the 134 // critical warning. Should only be set if |upgrade_available_| is 135 // UPGRADE_AVAILABLE_CRITICAL. 136 base::RepeatingTimer<UpgradeDetector> idle_check_timer_; 137 138 // The stage at which the annoyance level for upgrade notifications is at. 139 UpgradeNotificationAnnoyanceLevel upgrade_notification_stage_; 140 141 // Whether we have waited long enough after detecting an upgrade (to see 142 // is we should start nagging about upgrading). 143 bool notify_upgrade_; 144 145 DISALLOW_COPY_AND_ASSIGN(UpgradeDetector); 146 }; 147 148 #endif // CHROME_BROWSER_UPGRADE_DETECTOR_H_ 149