OLD | NEW |
1 // Copyright (c) 2009 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2009 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_shelf_model.h" | 5 #include "chrome/browser/extensions/extension_shelf_model.h" |
6 | 6 |
7 #include "base/stl_util-inl.h" | 7 #include "base/stl_util-inl.h" |
8 #include "chrome/browser/browser.h" | 8 #include "chrome/browser/browser.h" |
9 #include "chrome/browser/profile.h" | 9 #include "chrome/browser/profile.h" |
10 #include "chrome/browser/extensions/extension_host.h" | 10 #include "chrome/browser/extensions/extension_host.h" |
(...skipping 21 matching lines...) Expand all Loading... |
32 Source<ExtensionPrefs>(prefs_)); | 32 Source<ExtensionPrefs>(prefs_)); |
33 ready_ = service->is_ready(); | 33 ready_ = service->is_ready(); |
34 if (ready_) { | 34 if (ready_) { |
35 AddExtensions(service->extensions()); | 35 AddExtensions(service->extensions()); |
36 SortToolstrips(); | 36 SortToolstrips(); |
37 } | 37 } |
38 } | 38 } |
39 } | 39 } |
40 | 40 |
41 ExtensionShelfModel::~ExtensionShelfModel() { | 41 ExtensionShelfModel::~ExtensionShelfModel() { |
42 FOR_EACH_OBSERVER(ExtensionShelfModelObserver, observers_, | |
43 ShelfModelDeleting()); | |
44 | |
45 while (observers_.size()) | 42 while (observers_.size()) |
46 observers_.RemoveObserver(observers_.GetElementAt(0)); | 43 observers_.RemoveObserver(observers_.GetElementAt(0)); |
47 | 44 |
48 for (iterator t = toolstrips_.begin(); t != toolstrips_.end(); ++t) | 45 ExtensionToolstrips::iterator t; |
| 46 for (t = toolstrips_.begin(); t != toolstrips_.end(); ++t) |
49 delete t->host; | 47 delete t->host; |
50 toolstrips_.clear(); | 48 toolstrips_.clear(); |
51 } | 49 } |
52 | 50 |
53 void ExtensionShelfModel::AddObserver(ExtensionShelfModelObserver* observer) { | 51 void ExtensionShelfModel::AddObserver(ExtensionShelfModelObserver* observer) { |
54 observers_.AddObserver(observer); | 52 observers_.AddObserver(observer); |
55 } | 53 } |
56 | 54 |
57 void ExtensionShelfModel::RemoveObserver( | 55 void ExtensionShelfModel::RemoveObserver( |
58 ExtensionShelfModelObserver* observer) { | 56 ExtensionShelfModelObserver* observer) { |
59 observers_.RemoveObserver(observer); | 57 observers_.RemoveObserver(observer); |
60 } | 58 } |
61 | 59 |
62 void ExtensionShelfModel::AppendToolstrip(const ToolstripItem& toolstrip) { | 60 void ExtensionShelfModel::AppendToolstrip(const ToolstripItem& toolstrip) { |
63 InsertToolstripAt(count(), toolstrip); | 61 InsertToolstripAt(count(), toolstrip); |
64 } | 62 } |
65 | 63 |
66 void ExtensionShelfModel::InsertToolstripAt(int index, | 64 void ExtensionShelfModel::InsertToolstripAt(int index, |
67 const ToolstripItem& toolstrip) { | 65 const ToolstripItem& toolstrip) { |
68 toolstrips_.insert(toolstrips_.begin() + index, toolstrip); | 66 toolstrips_.insert(toolstrips_.begin() + index, toolstrip); |
69 if (ready_) { | 67 if (ready_) { |
70 FOR_EACH_OBSERVER(ExtensionShelfModelObserver, observers_, | 68 FOR_EACH_OBSERVER(ExtensionShelfModelObserver, observers_, |
71 ToolstripInsertedAt(toolstrip.host, index)); | 69 ToolstripInsertedAt(toolstrip.host, index)); |
72 } | 70 } |
73 } | 71 } |
74 | 72 |
75 void ExtensionShelfModel::RemoveToolstripAt(int index) { | 73 void ExtensionShelfModel::RemoveToolstripAt(int index) { |
76 ExtensionHost* host = ToolstripAt(index).host; | 74 ExtensionHost* host = ToolstripAt(index); |
77 FOR_EACH_OBSERVER(ExtensionShelfModelObserver, observers_, | 75 FOR_EACH_OBSERVER(ExtensionShelfModelObserver, observers_, |
78 ToolstripRemovingAt(host, index)); | 76 ToolstripRemovingAt(host, index)); |
79 toolstrips_.erase(toolstrips_.begin() + index); | 77 toolstrips_.erase(toolstrips_.begin() + index); |
80 delete host; | 78 delete host; |
81 } | 79 } |
82 | 80 |
83 void ExtensionShelfModel::MoveToolstripAt(int index, int to_index) { | 81 void ExtensionShelfModel::MoveToolstripAt(int index, int to_index) { |
84 DCHECK(index >= 0); | 82 DCHECK(index >= 0); |
85 DCHECK(to_index >= 0); | 83 DCHECK(to_index >= 0); |
86 if (index == to_index) | 84 if (index == to_index) |
87 return; | 85 return; |
88 | 86 |
89 ToolstripItem toolstrip = toolstrips_[index]; | 87 ToolstripItem toolstrip = toolstrips_[index]; |
90 toolstrips_.erase(toolstrips_.begin() + index); | 88 toolstrips_.erase(toolstrips_.begin() + index); |
91 toolstrips_.insert(toolstrips_.begin() + to_index, toolstrip); | 89 toolstrips_.insert(toolstrips_.begin() + to_index, toolstrip); |
92 | 90 |
93 FOR_EACH_OBSERVER(ExtensionShelfModelObserver, observers_, | 91 FOR_EACH_OBSERVER(ExtensionShelfModelObserver, observers_, |
94 ToolstripMoved(toolstrip.host, index, to_index)); | 92 ToolstripMoved(toolstrip.host, index, to_index)); |
95 | 93 |
96 UpdatePrefs(); | 94 UpdatePrefs(); |
97 } | 95 } |
98 | 96 |
99 int ExtensionShelfModel::IndexOfHost(ExtensionHost* host) { | 97 int ExtensionShelfModel::IndexOfToolstrip(ExtensionHost* toolstrip) { |
100 for (iterator i = toolstrips_.begin(); i != toolstrips_.end(); ++i) { | 98 ExtensionToolstrips::iterator i; |
101 if (i->host == host) | 99 for (i = toolstrips_.begin(); i != toolstrips_.end(); ++i) { |
| 100 if (i->host == toolstrip) |
102 return i - toolstrips_.begin(); | 101 return i - toolstrips_.begin(); |
103 } | 102 } |
104 return -1; | 103 return -1; |
105 } | 104 } |
106 | 105 |
107 ExtensionShelfModel::iterator ExtensionShelfModel::ToolstripForHost( | 106 ExtensionHost* ExtensionShelfModel::ToolstripAt(int index) { |
108 ExtensionHost* host) { | 107 DCHECK(index >= 0); |
109 for (iterator i = toolstrips_.begin(); i != toolstrips_.end(); ++i) { | 108 return toolstrips_[index].host; |
110 if (i->host == host) | |
111 return i; | |
112 } | |
113 return toolstrips_.end(); | |
114 } | 109 } |
115 | 110 |
116 const ExtensionShelfModel::ToolstripItem& ExtensionShelfModel::ToolstripAt( | 111 Extension::ToolstripInfo& ExtensionShelfModel::ToolstripInfoAt(int index) { |
117 int index) { | |
118 DCHECK(index >= 0); | 112 DCHECK(index >= 0); |
119 return toolstrips_[index]; | 113 return toolstrips_[index].info; |
| 114 } |
| 115 |
| 116 void* ExtensionShelfModel::ToolstripDataAt(int index) { |
| 117 DCHECK(index >= 0); |
| 118 return toolstrips_[index].data; |
120 } | 119 } |
121 | 120 |
122 void ExtensionShelfModel::SetToolstripDataAt(int index, void* data) { | 121 void ExtensionShelfModel::SetToolstripDataAt(int index, void* data) { |
123 DCHECK(index >= 0); | 122 DCHECK(index >= 0); |
124 toolstrips_[index].data = data; | 123 toolstrips_[index].data = data; |
125 } | 124 } |
126 | 125 |
127 void ExtensionShelfModel::ExpandToolstrip(iterator toolstrip, | |
128 const GURL& url, int height) { | |
129 if (toolstrip == end()) | |
130 return; | |
131 toolstrip->height = height; | |
132 toolstrip->url = url; | |
133 FOR_EACH_OBSERVER(ExtensionShelfModelObserver, observers_, | |
134 ToolstripChanged(toolstrip)); | |
135 } | |
136 | |
137 void ExtensionShelfModel::CollapseToolstrip(iterator toolstrip, | |
138 const GURL& url) { | |
139 if (toolstrip == end()) | |
140 return; | |
141 toolstrip->height = 0; | |
142 toolstrip->url = url; | |
143 FOR_EACH_OBSERVER(ExtensionShelfModelObserver, observers_, | |
144 ToolstripChanged(toolstrip)); | |
145 } | |
146 | |
147 void ExtensionShelfModel::Observe(NotificationType type, | 126 void ExtensionShelfModel::Observe(NotificationType type, |
148 const NotificationSource& source, | 127 const NotificationSource& source, |
149 const NotificationDetails& details) { | 128 const NotificationDetails& details) { |
150 switch (type.value) { | 129 switch (type.value) { |
151 case NotificationType::EXTENSIONS_LOADED: | 130 case NotificationType::EXTENSIONS_LOADED: |
152 if (ready_) | 131 if (ready_) |
153 AddExtensions(Details<ExtensionList>(details).ptr()); | 132 AddExtensions(Details<ExtensionList>(details).ptr()); |
154 break; | 133 break; |
155 case NotificationType::EXTENSION_UNLOADED: | 134 case NotificationType::EXTENSION_UNLOADED: |
156 RemoveExtension(Details<Extension>(details).ptr()); | 135 RemoveExtension(Details<Extension>(details).ptr()); |
157 break; | 136 break; |
158 case NotificationType::EXTENSIONS_READY: | 137 case NotificationType::EXTENSIONS_READY: |
159 if (browser_->profile()->GetExtensionsService()) { | 138 if (browser_->profile()->GetExtensionsService()) { |
160 AddExtensions( | 139 AddExtensions(browser_->profile()->GetExtensionsService()->extensions())
; |
161 browser_->profile()->GetExtensionsService()->extensions()); | |
162 SortToolstrips(); | 140 SortToolstrips(); |
163 } | 141 } |
164 ready_ = true; | 142 ready_ = true; |
165 break; | 143 break; |
166 case NotificationType::EXTENSION_SHELF_MODEL_CHANGED: | 144 case NotificationType::EXTENSION_SHELF_MODEL_CHANGED: |
167 // Ignore changes that this model originated. | 145 // Ignore changes that this model originated. |
168 if (Details<ExtensionShelfModel>(details).ptr() != this) | 146 if (Details<ExtensionShelfModel>(details).ptr() != this) |
169 SortToolstrips(); | 147 SortToolstrips(); |
170 break; | 148 break; |
171 default: | 149 default: |
(...skipping 10 matching lines...) Expand all Loading... |
182 return; | 160 return; |
183 | 161 |
184 for (std::vector<Extension::ToolstripInfo>::const_iterator toolstrip = | 162 for (std::vector<Extension::ToolstripInfo>::const_iterator toolstrip = |
185 extension->toolstrips().begin(); | 163 extension->toolstrips().begin(); |
186 toolstrip != extension->toolstrips().end(); ++toolstrip) { | 164 toolstrip != extension->toolstrips().end(); ++toolstrip) { |
187 GURL url = toolstrip->toolstrip; | 165 GURL url = toolstrip->toolstrip; |
188 ToolstripItem item; | 166 ToolstripItem item; |
189 item.host = manager->CreateView(extension, url, browser_); | 167 item.host = manager->CreateView(extension, url, browser_); |
190 item.info = *toolstrip; | 168 item.info = *toolstrip; |
191 item.data = NULL; | 169 item.data = NULL; |
192 item.height = 0; | |
193 AppendToolstrip(item); | 170 AppendToolstrip(item); |
194 } | 171 } |
195 } | 172 } |
196 | 173 |
197 void ExtensionShelfModel::AddExtensions(const ExtensionList* extensions) { | 174 void ExtensionShelfModel::AddExtensions(const ExtensionList* extensions) { |
198 if (extensions->size()) { | 175 if (extensions->size()) { |
199 ExtensionList::const_iterator extension = extensions->begin(); | 176 ExtensionList::const_iterator extension = extensions->begin(); |
200 for (; extension != extensions->end(); ++extension) | 177 for (; extension != extensions->end(); ++extension) |
201 AddExtension(*extension); | 178 AddExtension(*extension); |
202 } | 179 } |
203 } | 180 } |
204 | 181 |
205 void ExtensionShelfModel::RemoveExtension(Extension* extension) { | 182 void ExtensionShelfModel::RemoveExtension(Extension* extension) { |
206 bool changed = false; | 183 bool changed = false; |
207 for (int i = count() - 1; i >= 0; --i) { | 184 for (int i = count() - 1; i >= 0; --i) { |
208 ExtensionHost* t = ToolstripAt(i).host; | 185 ExtensionHost* t = ToolstripAt(i); |
209 if (t->extension()->id() == extension->id()) { | 186 if (t->extension()->id() == extension->id()) { |
210 changed = true; | 187 changed = true; |
211 RemoveToolstripAt(i); | 188 RemoveToolstripAt(i); |
212 | 189 |
213 // There can be more than one toolstrip per extension, so we have to keep | 190 // There can be more than one toolstrip per extension, so we have to keep |
214 // looping even after finding a match. | 191 // looping even after finding a match. |
215 } | 192 } |
216 } | 193 } |
217 if (changed) | 194 if (changed) |
218 UpdatePrefs(); | 195 UpdatePrefs(); |
219 } | 196 } |
220 | 197 |
221 void ExtensionShelfModel::UpdatePrefs() { | 198 void ExtensionShelfModel::UpdatePrefs() { |
222 if (!prefs_) | 199 if (!prefs_) |
223 return; | 200 return; |
224 | 201 |
225 // It's easiest to just rebuild the list each time. | 202 // It's easiest to just rebuild the list each time. |
226 ExtensionPrefs::URLList urls; | 203 ExtensionPrefs::URLList urls; |
227 for (int i = 0; i < count(); ++i) | 204 for (int i = 0; i < count(); ++i) |
228 urls.push_back(ToolstripAt(i).host->GetURL()); | 205 urls.push_back(ToolstripAt(i)->GetURL()); |
229 prefs_->SetShelfToolstripOrder(urls); | 206 prefs_->SetShelfToolstripOrder(urls); |
230 | 207 |
231 NotificationService::current()->Notify( | 208 NotificationService::current()->Notify( |
232 NotificationType::EXTENSION_SHELF_MODEL_CHANGED, | 209 NotificationType::EXTENSION_SHELF_MODEL_CHANGED, |
233 Source<ExtensionPrefs>(prefs_), | 210 Source<ExtensionPrefs>(prefs_), |
234 Details<ExtensionShelfModel>(this)); | 211 Details<ExtensionShelfModel>(this)); |
235 } | 212 } |
236 | 213 |
237 void ExtensionShelfModel::SortToolstrips() { | 214 void ExtensionShelfModel::SortToolstrips() { |
238 ExtensionPrefs::URLList urls = prefs_->GetShelfToolstripOrder(); | 215 ExtensionPrefs::URLList urls = prefs_->GetShelfToolstripOrder(); |
239 ToolstripList copy = | 216 ExtensionToolstrips copy = |
240 ToolstripList(toolstrips_.begin(), toolstrips_.end()); | 217 ExtensionToolstrips(toolstrips_.begin(), toolstrips_.end()); |
241 toolstrips_.clear(); | 218 toolstrips_.clear(); |
242 | 219 |
243 // Go through the urls and find the matching toolstrip, re-adding it to the | 220 // Go through the urls and find the matching toolstrip, re-adding it to the |
244 // new list in the proper order. | 221 // new list in the proper order. |
245 for (size_t i = 0; i < urls.size(); ++i) { | 222 for (size_t i = 0; i < urls.size(); ++i) { |
246 GURL& url = urls[i]; | 223 GURL& url = urls[i]; |
247 for (iterator toolstrip = copy.begin(); | 224 for (ExtensionToolstrips::iterator toolstrip = copy.begin(); |
248 toolstrip != copy.end(); ++toolstrip) { | 225 toolstrip != copy.end(); ++toolstrip) { |
249 if (url == toolstrip->host->GetURL()) { | 226 if (url == toolstrip->host->GetURL()) { |
250 // Note that it's technically possible for the same URL to appear in | 227 // Note that it's technically possible for the same URL to appear in |
251 // multiple toolstrips, so we don't do any testing for uniqueness. | 228 // multiple toolstrips, so we don't do any testing for uniqueness. |
252 toolstrips_.push_back(*toolstrip); | 229 toolstrips_.push_back(*toolstrip); |
253 | 230 |
254 // Remove the toolstrip from the list so we don't have to iterate over | 231 // Remove the toolstrip from the list so we don't have to iterate over |
255 // it next time. | 232 // it next time. |
256 copy.erase(toolstrip); | 233 copy.erase(toolstrip); |
257 break; | 234 break; |
258 } | 235 } |
259 } | 236 } |
260 } | 237 } |
261 | 238 |
262 // Any toolstrips remaining in |copy| were somehow missing from the prefs, | 239 // Any toolstrips remaining in |copy| were somehow missing from the prefs, |
263 // so just append them to the end. | 240 // so just append them to the end. |
264 for (iterator toolstrip = copy.begin(); | 241 for (ExtensionToolstrips::iterator toolstrip = copy.begin(); |
265 toolstrip != copy.end(); ++toolstrip) { | 242 toolstrip != copy.end(); ++toolstrip) { |
266 toolstrips_.push_back(*toolstrip); | 243 toolstrips_.push_back(*toolstrip); |
267 } | 244 } |
268 | 245 |
269 FOR_EACH_OBSERVER(ExtensionShelfModelObserver, observers_, | 246 FOR_EACH_OBSERVER(ExtensionShelfModelObserver, observers_, |
270 ShelfModelReloaded()); | 247 ShelfModelReloaded()); |
271 } | 248 } |
OLD | NEW |