OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/extensions/extension_warning_badge_service.h" | 5 #include "chrome/browser/extensions/extension_warning_badge_service.h" |
6 | 6 |
7 #include "base/stl_util.h" | 7 #include "base/stl_util.h" |
8 #include "chrome/app/chrome_command_ids.h" | 8 #include "chrome/app/chrome_command_ids.h" |
9 #include "chrome/browser/profiles/profile.h" | 9 #include "chrome/browser/profiles/profile.h" |
10 #include "chrome/browser/ui/browser_commands.h" | 10 #include "chrome/browser/ui/browser_commands.h" |
(...skipping 81 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
92 : profile_(profile) { | 92 : profile_(profile) { |
93 DCHECK(CalledOnValidThread()); | 93 DCHECK(CalledOnValidThread()); |
94 } | 94 } |
95 | 95 |
96 ExtensionWarningBadgeService::~ExtensionWarningBadgeService() {} | 96 ExtensionWarningBadgeService::~ExtensionWarningBadgeService() {} |
97 | 97 |
98 void ExtensionWarningBadgeService::SuppressCurrentWarnings() { | 98 void ExtensionWarningBadgeService::SuppressCurrentWarnings() { |
99 DCHECK(CalledOnValidThread()); | 99 DCHECK(CalledOnValidThread()); |
100 size_t old_size = suppressed_warnings_.size(); | 100 size_t old_size = suppressed_warnings_.size(); |
101 | 101 |
102 const ExtensionWarningSet& warnings = GetCurrentWarnings(); | 102 const WarningSet& warnings = GetCurrentWarnings(); |
103 suppressed_warnings_.insert(warnings.begin(), warnings.end()); | 103 suppressed_warnings_.insert(warnings.begin(), warnings.end()); |
104 | 104 |
105 if (old_size != suppressed_warnings_.size()) | 105 if (old_size != suppressed_warnings_.size()) |
106 UpdateBadgeStatus(); | 106 UpdateBadgeStatus(); |
107 } | 107 } |
108 | 108 |
109 const ExtensionWarningSet& | 109 const WarningSet& ExtensionWarningBadgeService::GetCurrentWarnings() const { |
110 ExtensionWarningBadgeService::GetCurrentWarnings() const { | |
111 return ExtensionSystem::Get(profile_)->warning_service()->warnings(); | 110 return ExtensionSystem::Get(profile_)->warning_service()->warnings(); |
112 } | 111 } |
113 | 112 |
114 void ExtensionWarningBadgeService::ExtensionWarningsChanged() { | 113 void ExtensionWarningBadgeService::ExtensionWarningsChanged() { |
115 DCHECK(CalledOnValidThread()); | 114 DCHECK(CalledOnValidThread()); |
116 UpdateBadgeStatus(); | 115 UpdateBadgeStatus(); |
117 } | 116 } |
118 | 117 |
119 void ExtensionWarningBadgeService::UpdateBadgeStatus() { | 118 void ExtensionWarningBadgeService::UpdateBadgeStatus() { |
120 const std::set<ExtensionWarning>& warnings = GetCurrentWarnings(); | 119 const std::set<Warning>& warnings = GetCurrentWarnings(); |
121 bool non_suppressed_warnings_exist = false; | 120 bool non_suppressed_warnings_exist = false; |
122 for (std::set<ExtensionWarning>::const_iterator i = warnings.begin(); | 121 for (std::set<Warning>::const_iterator i = warnings.begin(); |
123 i != warnings.end(); ++i) { | 122 i != warnings.end(); ++i) { |
124 if (!ContainsKey(suppressed_warnings_, *i)) { | 123 if (!ContainsKey(suppressed_warnings_, *i)) { |
125 non_suppressed_warnings_exist = true; | 124 non_suppressed_warnings_exist = true; |
126 break; | 125 break; |
127 } | 126 } |
128 } | 127 } |
129 ShowBadge(non_suppressed_warnings_exist); | 128 ShowBadge(non_suppressed_warnings_exist); |
130 } | 129 } |
131 | 130 |
132 void ExtensionWarningBadgeService::ShowBadge(bool show) { | 131 void ExtensionWarningBadgeService::ShowBadge(bool show) { |
133 GlobalErrorService* service = | 132 GlobalErrorService* service = |
134 GlobalErrorServiceFactory::GetForProfile(profile_); | 133 GlobalErrorServiceFactory::GetForProfile(profile_); |
135 GlobalError* error = service->GetGlobalErrorByMenuItemCommandID( | 134 GlobalError* error = service->GetGlobalErrorByMenuItemCommandID( |
136 ErrorBadge::GetMenuItemCommandID()); | 135 ErrorBadge::GetMenuItemCommandID()); |
137 | 136 |
138 // Activate or hide the warning badge in case the current state is incorrect. | 137 // Activate or hide the warning badge in case the current state is incorrect. |
139 if (error && !show) { | 138 if (error && !show) { |
140 service->RemoveGlobalError(error); | 139 service->RemoveGlobalError(error); |
141 delete error; | 140 delete error; |
142 } else if (!error && show) { | 141 } else if (!error && show) { |
143 service->AddGlobalError(new ErrorBadge(this)); | 142 service->AddGlobalError(new ErrorBadge(this)); |
144 } | 143 } |
145 } | 144 } |
146 | 145 |
147 } // namespace extensions | 146 } // namespace extensions |
OLD | NEW |