| 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/ui/unload_controller.h" | 5 #include "chrome/browser/ui/unload_controller.h" |
| 6 | 6 |
| 7 #include "base/logging.h" | 7 #include "base/logging.h" |
| 8 #include "base/message_loop.h" | 8 #include "base/message_loop.h" |
| 9 #include "chrome/browser/ui/browser.h" | 9 #include "chrome/browser/ui/browser.h" |
| 10 #include "chrome/browser/ui/browser_tabstrip.h" | 10 #include "chrome/browser/ui/browser_tabstrip.h" |
| 11 #include "chrome/browser/ui/tab_contents/core_tab_helper.h" |
| 11 #include "chrome/browser/ui/tabs/tab_strip_model.h" | 12 #include "chrome/browser/ui/tabs/tab_strip_model.h" |
| 12 #include "chrome/browser/ui/tabs/tab_strip_model_delegate.h" | 13 #include "chrome/browser/ui/tabs/tab_strip_model_delegate.h" |
| 13 #include "chrome/common/chrome_notification_types.h" | 14 #include "chrome/common/chrome_notification_types.h" |
| 14 #include "content/public/browser/notification_service.h" | 15 #include "content/public/browser/notification_service.h" |
| 15 #include "content/public/browser/notification_source.h" | 16 #include "content/public/browser/notification_source.h" |
| 16 #include "content/public/browser/notification_types.h" | 17 #include "content/public/browser/notification_types.h" |
| 17 #include "content/public/browser/render_view_host.h" | 18 #include "content/public/browser/render_view_host.h" |
| 18 #include "content/public/browser/web_contents.h" | 19 #include "content/public/browser/web_contents.h" |
| 19 #include "content/public/browser/web_contents_delegate.h" | 20 #include "content/public/browser/web_contents_delegate.h" |
| 20 | 21 |
| (...skipping 194 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 215 ClearUnloadState(contents); | 216 ClearUnloadState(contents); |
| 216 } | 217 } |
| 217 | 218 |
| 218 bool UnloadController::DetachWebContents(content::WebContents* contents) { | 219 bool UnloadController::DetachWebContents(content::WebContents* contents) { |
| 219 int index = browser_->tab_strip_model()->GetIndexOfWebContents(contents); | 220 int index = browser_->tab_strip_model()->GetIndexOfWebContents(contents); |
| 220 if (index != TabStripModel::kNoTab && | 221 if (index != TabStripModel::kNoTab && |
| 221 contents->NeedToFireBeforeUnload()) { | 222 contents->NeedToFireBeforeUnload()) { |
| 222 tabs_needing_unload_ack_.insert(contents); | 223 tabs_needing_unload_ack_.insert(contents); |
| 223 browser_->tab_strip_model()->DetachWebContentsAt(index); | 224 browser_->tab_strip_model()->DetachWebContentsAt(index); |
| 224 contents->SetDelegate(detached_delegate_.get()); | 225 contents->SetDelegate(detached_delegate_.get()); |
| 225 contents->OnUnloadDetachedStarted(); | 226 CoreTabHelper* core_tab_helper = CoreTabHelper::FromWebContents(contents); |
| 227 core_tab_helper->OnUnloadDetachedStarted(); |
| 226 return true; | 228 return true; |
| 227 } | 229 } |
| 228 return false; | 230 return false; |
| 229 } | 231 } |
| 230 | 232 |
| 231 void UnloadController::ProcessPendingTabs() { | 233 void UnloadController::ProcessPendingTabs() { |
| 232 if (!is_attempting_to_close_browser_) { | 234 if (!is_attempting_to_close_browser_) { |
| 233 // Because we might invoke this after a delay it's possible for the value of | 235 // Because we might invoke this after a delay it's possible for the value of |
| 234 // is_attempting_to_close_browser_ to have changed since we scheduled the | 236 // is_attempting_to_close_browser_ to have changed since we scheduled the |
| 235 // task. | 237 // task. |
| 236 return; | 238 return; |
| 237 } | 239 } |
| 238 | 240 |
| 239 if (tab_needing_before_unload_ack_ != NULL) { | 241 if (tab_needing_before_unload_ack_ != NULL) { |
| 240 // Wait for |BeforeUnloadFired| before proceeding. | 242 // Wait for |BeforeUnloadFired| before proceeding. |
| 241 return; | 243 return; |
| 242 } | 244 } |
| 243 | 245 |
| 244 // Process a beforeunload handler. | 246 // Process a beforeunload handler. |
| 245 if (!tabs_needing_before_unload_.empty()) { | 247 if (!tabs_needing_before_unload_.empty()) { |
| 246 WebContentsSet::iterator it = tabs_needing_before_unload_.begin(); | 248 WebContentsSet::iterator it = tabs_needing_before_unload_.begin(); |
| 247 content::WebContents* contents = *it; | 249 content::WebContents* contents = *it; |
| 248 tabs_needing_before_unload_.erase(it); | 250 tabs_needing_before_unload_.erase(it); |
| 249 // Null check render_view_host here as this gets called on a PostTask and | 251 // Null check render_view_host here as this gets called on a PostTask and |
| 250 // the tab's render_view_host may have been nulled out. | 252 // the tab's render_view_host may have been nulled out. |
| 251 if (contents->GetRenderViewHost()) { | 253 if (contents->GetRenderViewHost()) { |
| 252 tab_needing_before_unload_ack_ = contents; | 254 tab_needing_before_unload_ack_ = contents; |
| 253 contents->OnCloseStarted(); | 255 |
| 256 CoreTabHelper* core_tab_helper = CoreTabHelper::FromWebContents(contents); |
| 257 core_tab_helper->OnCloseStarted(); |
| 258 |
| 254 contents->GetRenderViewHost()->FirePageBeforeUnload(false); | 259 contents->GetRenderViewHost()->FirePageBeforeUnload(false); |
| 255 } else { | 260 } else { |
| 256 ProcessPendingTabs(); | 261 ProcessPendingTabs(); |
| 257 } | 262 } |
| 258 return; | 263 return; |
| 259 } | 264 } |
| 260 | 265 |
| 261 // Process all the unload handlers. (The beforeunload handlers have finished.) | 266 // Process all the unload handlers. (The beforeunload handlers have finished.) |
| 262 if (!tabs_needing_unload_.empty()) { | 267 if (!tabs_needing_unload_.empty()) { |
| 263 browser_->OnWindowClosing(); | 268 browser_->OnWindowClosing(); |
| 264 | 269 |
| 265 // Run unload handlers detached since no more interaction is possible. | 270 // Run unload handlers detached since no more interaction is possible. |
| 266 WebContentsSet::iterator it = tabs_needing_unload_.begin(); | 271 WebContentsSet::iterator it = tabs_needing_unload_.begin(); |
| 267 while (it != tabs_needing_unload_.end()) { | 272 while (it != tabs_needing_unload_.end()) { |
| 268 WebContentsSet::iterator current = it++; | 273 WebContentsSet::iterator current = it++; |
| 269 content::WebContents* contents = *current; | 274 content::WebContents* contents = *current; |
| 270 tabs_needing_unload_.erase(current); | 275 tabs_needing_unload_.erase(current); |
| 271 // Null check render_view_host here as this gets called on a PostTask | 276 // Null check render_view_host here as this gets called on a PostTask |
| 272 // and the tab's render_view_host may have been nulled out. | 277 // and the tab's render_view_host may have been nulled out. |
| 273 if (contents->GetRenderViewHost()) { | 278 if (contents->GetRenderViewHost()) { |
| 274 contents->OnUnloadStarted(); | 279 CoreTabHelper* core_tab_helper = |
| 280 CoreTabHelper::FromWebContents(contents); |
| 281 core_tab_helper->OnUnloadStarted(); |
| 275 DetachWebContents(contents); | 282 DetachWebContents(contents); |
| 276 contents->GetRenderViewHost()->ClosePage(); | 283 contents->GetRenderViewHost()->ClosePage(); |
| 277 } | 284 } |
| 278 } | 285 } |
| 279 | 286 |
| 280 // Get the browser hidden. | 287 // Get the browser hidden. |
| 281 if (browser_->tab_strip_model()->empty()) { | 288 if (browser_->tab_strip_model()->empty()) { |
| 282 browser_->TabStripEmpty(); | 289 browser_->TabStripEmpty(); |
| 283 } else { | 290 } else { |
| 284 browser_->tab_strip_model()->CloseAllTabs(); // tabs not needing unload | 291 browser_->tab_strip_model()->CloseAllTabs(); // tabs not needing unload |
| (...skipping 21 matching lines...) Expand all Loading... |
| 306 tab_needing_before_unload_ack_ == NULL && | 313 tab_needing_before_unload_ack_ == NULL && |
| 307 tabs_needing_unload_.empty() && | 314 tabs_needing_unload_.empty() && |
| 308 tabs_needing_unload_ack_.empty(); | 315 tabs_needing_unload_ack_.empty(); |
| 309 } | 316 } |
| 310 | 317 |
| 311 void UnloadController::CancelWindowClose() { | 318 void UnloadController::CancelWindowClose() { |
| 312 // Closing of window can be canceled from a beforeunload handler. | 319 // Closing of window can be canceled from a beforeunload handler. |
| 313 DCHECK(is_attempting_to_close_browser_); | 320 DCHECK(is_attempting_to_close_browser_); |
| 314 tabs_needing_before_unload_.clear(); | 321 tabs_needing_before_unload_.clear(); |
| 315 if (tab_needing_before_unload_ack_ != NULL) { | 322 if (tab_needing_before_unload_ack_ != NULL) { |
| 316 tab_needing_before_unload_ack_->OnCloseCanceled(); | 323 |
| 324 CoreTabHelper* core_tab_helper = |
| 325 CoreTabHelper::FromWebContents(tab_needing_before_unload_ack_); |
| 326 core_tab_helper->OnCloseCanceled(); |
| 317 tab_needing_before_unload_ack_ = NULL; | 327 tab_needing_before_unload_ack_ = NULL; |
| 318 } | 328 } |
| 319 for (WebContentsSet::iterator it = tabs_needing_unload_.begin(); | 329 for (WebContentsSet::iterator it = tabs_needing_unload_.begin(); |
| 320 it != tabs_needing_unload_.end(); it++) { | 330 it != tabs_needing_unload_.end(); it++) { |
| 321 content::WebContents* contents = *it; | 331 content::WebContents* contents = *it; |
| 322 contents->OnCloseCanceled(); | 332 |
| 333 CoreTabHelper* core_tab_helper = CoreTabHelper::FromWebContents(contents); |
| 334 core_tab_helper->OnCloseCanceled(); |
| 323 } | 335 } |
| 324 tabs_needing_unload_.clear(); | 336 tabs_needing_unload_.clear(); |
| 325 | 337 |
| 326 // No need to clear tabs_needing_unload_ack_. Those tabs are already detached. | 338 // No need to clear tabs_needing_unload_ack_. Those tabs are already detached. |
| 327 | 339 |
| 328 is_attempting_to_close_browser_ = false; | 340 is_attempting_to_close_browser_ = false; |
| 329 | 341 |
| 330 content::NotificationService::current()->Notify( | 342 content::NotificationService::current()->Notify( |
| 331 chrome::NOTIFICATION_BROWSER_CLOSE_CANCELLED, | 343 chrome::NOTIFICATION_BROWSER_CLOSE_CANCELLED, |
| 332 content::Source<Browser>(browser_), | 344 content::Source<Browser>(browser_), |
| (...skipping 24 matching lines...) Expand all Loading... |
| 357 } | 369 } |
| 358 | 370 |
| 359 void UnloadController::PostTaskForProcessPendingTabs() { | 371 void UnloadController::PostTaskForProcessPendingTabs() { |
| 360 base::MessageLoop::current()->PostTask( | 372 base::MessageLoop::current()->PostTask( |
| 361 FROM_HERE, | 373 FROM_HERE, |
| 362 base::Bind(&UnloadController::ProcessPendingTabs, | 374 base::Bind(&UnloadController::ProcessPendingTabs, |
| 363 weak_factory_.GetWeakPtr())); | 375 weak_factory_.GetWeakPtr())); |
| 364 } | 376 } |
| 365 | 377 |
| 366 } // namespace chrome | 378 } // namespace chrome |
| OLD | NEW |