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#include "chrome/browser/repost_form_warning_controller.h"
6
7#include "content/browser/tab_contents/tab_contents.h"
8#include "content/common/notification_source.h"
9
10RepostFormWarningController::RepostFormWarningController(
11    TabContents* tab_contents)
12    : tab_contents_(tab_contents),
13      window_(NULL) {
14  NavigationController* controller = &tab_contents->controller();
15  registrar_.Add(this, NotificationType::LOAD_START,
16                 Source<NavigationController>(controller));
17  registrar_.Add(this, NotificationType::TAB_CLOSING,
18                 Source<NavigationController>(controller));
19  registrar_.Add(this, NotificationType::REPOST_WARNING_SHOWN,
20                 Source<NavigationController>(controller));
21}
22
23RepostFormWarningController::~RepostFormWarningController() {
24  // If we end up here, the constrained window has been closed, so make sure we
25  // don't close it again.
26  window_ = NULL;
27  // Make sure everything is cleaned up.
28  Cancel();
29}
30
31void RepostFormWarningController::Show(
32    ConstrainedWindowDelegate* window_delegate) {
33  window_ = tab_contents_->CreateConstrainedDialog(window_delegate);
34}
35
36void RepostFormWarningController::Cancel() {
37  if (tab_contents_) {
38    tab_contents_->controller().CancelPendingReload();
39    CloseDialog();
40  }
41}
42
43void RepostFormWarningController::Continue() {
44  if (tab_contents_) {
45    tab_contents_->controller().ContinuePendingReload();
46    // If we reload the page, the dialog will be closed anyway.
47  }
48}
49
50void RepostFormWarningController::Observe(NotificationType type,
51                                const NotificationSource& source,
52                                const NotificationDetails& details) {
53  // Close the dialog if we load a page (because reloading might not apply to
54  // the same page anymore) or if the tab is closed, because then we won't have
55  // a navigation controller anymore.
56  if (tab_contents_ &&
57      (type == NotificationType::LOAD_START ||
58       type == NotificationType::TAB_CLOSING ||
59       type == NotificationType::REPOST_WARNING_SHOWN)) {
60    DCHECK_EQ(Source<NavigationController>(source).ptr(),
61              &tab_contents_->controller());
62    Cancel();
63  }
64}
65
66void RepostFormWarningController::CloseDialog() {
67  // Make sure we won't do anything when |Cancel()| is called again.
68  tab_contents_ = NULL;
69  if (window_) {
70    window_->CloseConstrainedWindow();
71  }
72}
73