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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
|
// Copyright 2023 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/tabs/organization/tab_data.h"
#include "base/functional/bind.h"
#include "chrome/browser/ui/browser_window/public/browser_window_interface.h"
#include "chrome/browser/ui/tabs/organization/tab_sensitivity_cache.h"
#include "chrome/browser/ui/tabs/tab_model.h"
#include "chrome/browser/ui/tabs/tab_strip_model.h"
#include "components/tabs/public/tab_interface.h"
#include "content/public/browser/navigation_handle.h"
#include "content/public/browser/web_contents.h"
#include "url/gurl.h"
TabData::TabData(tabs::TabInterface* tab)
: WebContentsObserver(tab->GetContents()),
tab_(tab->GetHandle()),
original_url_(tab->GetContents()->GetLastCommittedURL()) {
original_tab_strip_model_ =
tab->GetBrowserWindowInterface()->GetTabStripModel();
original_tab_strip_model_->AddObserver(this);
tab_subscriptions_.push_back(tab->RegisterWillDetach(
base::BindRepeating(&TabData::OnTabWillDetach, base::Unretained(this))));
tab_subscriptions_.push_back(
tab->RegisterWillDiscardContents(base::BindRepeating(
&TabData::OnTabWillDiscardContents, base::Unretained(this))));
Observe(tab->GetContents());
}
TabData::~TabData() {
if (original_tab_strip_model_) {
original_tab_strip_model_->RemoveObserver(this);
}
for (auto& observer : observers_) {
observer.OnTabDataDestroyed(tab_id());
}
}
void TabData::AddObserver(TabData::Observer* observer) {
observers_.AddObserver(observer);
}
void TabData::RemoveObserver(TabData::Observer* observer) {
observers_.RemoveObserver(observer);
}
bool TabData::IsValidForOrganizing(
std::optional<tab_groups::TabGroupId> allowed_group_id) const {
// if the model or the tab have been destroyed, then it's not valid.
if (!original_tab_strip_model_ || !tab_.Get() || !tab_.Get()->GetContents()) {
return false;
}
// If the web_contents is no longer the same URL, then it's not valid.
if (original_url_ != tab_.Get()->GetContents()->GetLastCommittedURL()) {
return false;
}
// All non http(s) schemes are invalid.
if (!original_url_.SchemeIsHTTPOrHTTPS()) {
return false;
}
// Tab is not valid if it does not exist in the TabStripModel any more.
// NOTE: we will hear of this from the TabStripModel in OnTabStripModelChanged
// below, but IsValidForOrganizing might be called from another
// TabStripModelObserver first - most notably from a TabData's in another
// TabOrganization - so we cannot assume our web_contents_ is already nullptr.
if (tab_.Get()->GetBrowserWindowInterface()->GetTabStripModel() !=
original_tab_strip_model_) {
return false;
}
// Tab is not valid if it is grouped and does not belong to
// |allowed_group_id|.
const std::optional<tab_groups::TabGroupId> tab_group_id =
tab_.Get()->GetGroup();
if (tab_group_id.has_value() &&
(!allowed_group_id.has_value() ||
tab_group_id.value() != allowed_group_id.value())) {
return false;
}
if (tab_.Get()->IsPinned()) {
return false;
}
return true;
}
void TabData::OnTabStripModelDestroyed(TabStripModel* tab_strip_model) {
if (original_tab_strip_model_ == tab_strip_model) {
original_tab_strip_model_ = nullptr;
tab_ = tabs::TabHandle::Null();
NotifyObserversOfUpdate();
}
}
void TabData::DidFinishNavigation(
content::NavigationHandle* navigation_handle) {
NotifyObserversOfUpdate();
}
void TabData::OnTabWillDetach(tabs::TabInterface* tab,
tabs::TabInterface::DetachReason reason) {
if (tab_.Get() == nullptr) {
return;
}
CHECK_EQ(tab, tab_.Get());
// Drop this tab, whether the reason is deletion or moving to another window.
// Either way, this tab will no longer be valid for organizing.
tab_ = tabs::TabHandle::Null();
Observe(nullptr);
NotifyObserversOfUpdate();
}
void TabData::OnTabWillDiscardContents(tabs::TabInterface* tab,
content::WebContents* old_contents,
content::WebContents* new_contents) {
if (tab_.Get() == nullptr) {
return;
}
CHECK_EQ(tab, tab_.Get());
Observe(new_contents);
NotifyObserversOfUpdate();
}
void TabData::NotifyObserversOfUpdate() {
for (auto& observer : observers_) {
observer.OnTabDataUpdated(this);
}
}
|
