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
|
// Copyright 2019 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/extensions/extension_browser_window_helper.h"
#include "chrome/browser/extensions/app_tab_helper.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/browser/ui/browser_command_controller.h"
#include "chrome/browser/ui/browser_window.h"
#include "chrome/browser/ui/tabs/tab_enums.h"
#include "chrome/browser/ui/tabs/tab_strip_model.h"
#include "chrome/browser/ui/tabs/tab_utils.h"
#include "chrome/browser/ui/web_applications/app_browser_controller.h"
#include "chrome/common/url_constants.h"
#include "content/public/browser/render_frame_host.h"
#include "content/public/browser/web_contents.h"
#include "content/public/common/url_constants.h"
#include "extensions/browser/unloaded_extension_reason.h"
#include "extensions/common/constants.h"
#include "url/origin.h"
namespace extensions {
namespace {
// Returns true if the given |web_contents| should be closed when the extension
// is unloaded.
bool ShouldCloseTabOnExtensionUnload(const Extension* extension,
Browser* browser,
content::WebContents* web_contents) {
// Case 1: A "regular" extension page, e.g. chrome-extension://<id>/page.html.
// Note: we check the tuple or precursor tuple in order to close any
// windows with opaque origins that were opened by extensions, and may
// still be running code.
const url::SchemeHostPort& tuple_or_precursor_tuple =
web_contents->GetPrimaryMainFrame()
->GetLastCommittedOrigin()
.GetTupleOrPrecursorTupleIfOpaque();
if (tuple_or_precursor_tuple.scheme() == extensions::kExtensionScheme &&
tuple_or_precursor_tuple.host() == extension->id()) {
// Edge-case: Chrome URL overrides (such as NTP overrides) are handled
// differently (reloaded), and managed by ExtensionWebUI. Ignore them.
if (!web_contents->GetLastCommittedURL().SchemeIs(
content::kChromeUIScheme)) {
return true;
}
}
// Case 2: Check if the page is a page associated with a hosted app, which
// can have non-extension schemes. For example, the Gmail hosted app would
// have a URL of https://mail.google.com.
if (AppTabHelper::FromWebContents(web_contents)->GetExtensionAppId() ==
extension->id()) {
return true;
}
return false;
}
// Unmutes the given |contents| if it was muted by the extension with
// |extension_id|.
void UnmuteIfMutedByExtension(content::WebContents* contents,
const ExtensionId& extension_id) {
LastMuteMetadata::CreateForWebContents(contents); // Ensures metadata exists.
LastMuteMetadata* const metadata =
LastMuteMetadata::FromWebContents(contents);
if (metadata->reason == TabMutedReason::EXTENSION &&
metadata->extension_id == extension_id) {
SetTabAudioMuted(contents, false, TabMutedReason::EXTENSION, extension_id);
}
}
} // namespace
ExtensionBrowserWindowHelper::ExtensionBrowserWindowHelper(Browser* browser)
: browser_(browser) {
registry_observation_.Observe(ExtensionRegistry::Get(browser_->profile()));
}
ExtensionBrowserWindowHelper::~ExtensionBrowserWindowHelper() = default;
void ExtensionBrowserWindowHelper::OnExtensionLoaded(
content::BrowserContext* browser_context,
const Extension* extension) {
browser_->command_controller()->ExtensionStateChanged();
}
void ExtensionBrowserWindowHelper::OnExtensionUnloaded(
content::BrowserContext* browser_context,
const Extension* extension,
UnloadedExtensionReason reason) {
browser_->command_controller()->ExtensionStateChanged();
// Clean up any tabs from |extension|, unless it was terminated. In the
// terminated case (as when the extension crashed), we let the sad tabs stay.
if (reason != extensions::UnloadedExtensionReason::TERMINATE)
CleanUpTabsOnUnload(extension);
}
void ExtensionBrowserWindowHelper::CleanUpTabsOnUnload(
const Extension* extension) {
TabStripModel* tab_strip_model = browser_->tab_strip_model();
// Iterate backwards as we may remove items while iterating.
for (int i = tab_strip_model->count() - 1; i >= 0; --i) {
content::WebContents* web_contents = tab_strip_model->GetWebContentsAt(i);
if (ShouldCloseTabOnExtensionUnload(extension, browser_, web_contents)) {
// Do not close the last tab if it belongs to the extension. Instead
// replace it with the default NTP.
if (tab_strip_model->count() == 1) {
const GURL new_tab_url(chrome::kChromeUINewTabURL);
// Replace the extension page with default NTP. This behavior is similar
// to how Chrome URL overrides (such as NTP overrides) are handled by
// ExtensionWebUI.
web_contents->GetController().LoadURL(new_tab_url, content::Referrer(),
ui::PAGE_TRANSITION_RELOAD,
std::string());
} else {
tab_strip_model->CloseWebContentsAt(i, TabCloseTypes::CLOSE_NONE);
}
} else {
UnmuteIfMutedByExtension(web_contents, extension->id());
}
}
}
} // namespace extensions
|
