1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/ui/webui/tab_strip/tab_before_unload_tracker.h"
#include <memory>
#include "base/memory/raw_ptr.h"
#include "content/public/browser/web_contents.h"
#include "content/public/browser/web_contents_observer.h"
namespace tab_strip_ui {
TabBeforeUnloadTracker::TabBeforeUnloadTracker(
TabCloseCancelledCallback cancelled_callback)
: cancelled_callback_(std::move(cancelled_callback)) {}
TabBeforeUnloadTracker::~TabBeforeUnloadTracker() = default;
void TabBeforeUnloadTracker::Observe(content::WebContents* contents) {
observers_[contents] = std::make_unique<TabObserver>(contents, this);
}
void TabBeforeUnloadTracker::Unobserve(content::WebContents* contents) {
observers_.erase(contents);
}
void TabBeforeUnloadTracker::OnBeforeUnloadDialogCancelled(
content::WebContents* contents) {
cancelled_callback_.Run(contents);
}
class TabBeforeUnloadTracker::TabObserver
: public content::WebContentsObserver {
public:
TabObserver(content::WebContents* contents, TabBeforeUnloadTracker* tracker)
: content::WebContentsObserver(contents), tracker_(tracker) {}
~TabObserver() override = default;
// content::WebContentsObserver
void WebContentsDestroyed() override { tracker_->Unobserve(web_contents()); }
void BeforeUnloadDialogCancelled() override {
tracker_->OnBeforeUnloadDialogCancelled(web_contents());
}
private:
raw_ptr<TabBeforeUnloadTracker> tracker_;
};
} // namespace tab_strip_ui
|
