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
|
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/extensions/navigation_observer.h"
#include "chrome/browser/extensions/extension_service.h"
#include "chrome/browser/profiles/profile.h"
#include "content/public/browser/navigation_controller.h"
#include "content/public/browser/navigation_entry.h"
#include "content/public/browser/notification_service.h"
#include "content/public/browser/notification_types.h"
#include "extensions/browser/extension_prefs.h"
#include "extensions/browser/extension_registry.h"
#include "extensions/browser/extension_system.h"
using content::NavigationController;
using content::NavigationEntry;
namespace extensions {
NavigationObserver::NavigationObserver(Profile* profile)
: profile_(profile), weak_factory_(this) {
RegisterForNotifications();
}
NavigationObserver::~NavigationObserver() {}
void NavigationObserver::Observe(int type,
const content::NotificationSource& source,
const content::NotificationDetails& details) {
if (type != content::NOTIFICATION_NAV_ENTRY_COMMITTED) {
NOTREACHED();
return;
}
NavigationController* controller =
content::Source<NavigationController>(source).ptr();
if (!profile_->IsSameProfile(
Profile::FromBrowserContext(controller->GetBrowserContext())))
return;
PromptToEnableExtensionIfNecessary(controller);
}
void NavigationObserver::RegisterForNotifications() {
registrar_.Add(this, content::NOTIFICATION_NAV_ENTRY_COMMITTED,
content::NotificationService::AllSources());
}
void NavigationObserver::PromptToEnableExtensionIfNecessary(
NavigationController* nav_controller) {
// Bail out if we're already running a prompt.
if (!in_progress_prompt_extension_id_.empty())
return;
NavigationEntry* nav_entry = nav_controller->GetVisibleEntry();
if (!nav_entry)
return;
ExtensionRegistry* registry = extensions::ExtensionRegistry::Get(profile_);
const Extension* extension =
registry->disabled_extensions().GetExtensionOrAppByURL(
nav_entry->GetURL());
if (!extension)
return;
// Try not to repeatedly prompt the user about the same extension.
if (prompted_extensions_.find(extension->id()) != prompted_extensions_.end())
return;
prompted_extensions_.insert(extension->id());
ExtensionPrefs* extension_prefs = ExtensionPrefs::Get(profile_);
if (extension_prefs->DidExtensionEscalatePermissions(extension->id())) {
// Keep track of the extension id and nav controller we're prompting for.
// These must be reset in OnInstallPromptDone.
in_progress_prompt_extension_id_ = extension->id();
in_progress_prompt_navigation_controller_ = nav_controller;
extension_install_prompt_.reset(
new ExtensionInstallPrompt(nav_controller->GetWebContents()));
ExtensionInstallPrompt::PromptType type =
ExtensionInstallPrompt::GetReEnablePromptTypeForExtension(profile_,
extension);
extension_install_prompt_->ShowDialog(
base::Bind(&NavigationObserver::OnInstallPromptDone,
weak_factory_.GetWeakPtr()),
extension, nullptr,
make_scoped_ptr(new ExtensionInstallPrompt::Prompt(type)),
ExtensionInstallPrompt::GetDefaultShowDialogCallback());
}
}
void NavigationObserver::OnInstallPromptDone(
ExtensionInstallPrompt::Result result) {
ExtensionService* extension_service =
extensions::ExtensionSystem::Get(profile_)->extension_service();
const Extension* extension = extension_service->GetExtensionById(
in_progress_prompt_extension_id_, true);
if (result == ExtensionInstallPrompt::Result::ACCEPTED) {
NavigationController* nav_controller =
in_progress_prompt_navigation_controller_;
CHECK(extension);
CHECK(nav_controller);
// Grant permissions, re-enable the extension, and then reload the tab.
extension_service->GrantPermissionsAndEnableExtension(extension);
nav_controller->Reload(true);
} else {
std::string histogram_name =
result == ExtensionInstallPrompt::Result::USER_CANCELED
? "ReEnableCancel"
: "ReEnableAbort";
ExtensionService::RecordPermissionMessagesHistogram(extension,
histogram_name.c_str());
}
in_progress_prompt_extension_id_ = std::string();
in_progress_prompt_navigation_controller_ = nullptr;
extension_install_prompt_.reset();
}
} // namespace extensions
|
