OLD | NEW |
| (Empty) |
1 // Copyright (c) 2012 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/extensions/extension_navigation_observer.h" | |
6 | |
7 #include "chrome/browser/extensions/extension_install_ui.h" | |
8 #include "chrome/browser/extensions/extension_service.h" | |
9 #include "chrome/browser/profiles/profile.h" | |
10 #include "content/public/browser/navigation_controller.h" | |
11 #include "content/public/browser/navigation_entry.h" | |
12 #include "content/public/browser/notification_service.h" | |
13 #include "content/public/browser/notification_types.h" | |
14 | |
15 using content::NavigationController; | |
16 using content::NavigationEntry; | |
17 | |
18 ExtensionNavigationObserver::ExtensionNavigationObserver(Profile* profile) | |
19 : profile_(profile) { | |
20 RegisterForNotifications(); | |
21 } | |
22 | |
23 ExtensionNavigationObserver::~ExtensionNavigationObserver() {} | |
24 | |
25 void ExtensionNavigationObserver::Observe( | |
26 int type, | |
27 const content::NotificationSource& source, | |
28 const content::NotificationDetails& details) { | |
29 if (type != content::NOTIFICATION_NAV_ENTRY_COMMITTED) { | |
30 NOTREACHED(); | |
31 return; | |
32 } | |
33 | |
34 NavigationController* controller = | |
35 content::Source<NavigationController>(source).ptr(); | |
36 if (!profile_->IsSameProfile( | |
37 Profile::FromBrowserContext(controller->GetBrowserContext()))) | |
38 return; | |
39 | |
40 PromptToEnableExtensionIfNecessary(controller); | |
41 } | |
42 | |
43 void ExtensionNavigationObserver::RegisterForNotifications() { | |
44 registrar_.Add(this, content::NOTIFICATION_NAV_ENTRY_COMMITTED, | |
45 content::NotificationService::AllSources()); | |
46 } | |
47 | |
48 void ExtensionNavigationObserver::PromptToEnableExtensionIfNecessary( | |
49 NavigationController* nav_controller) { | |
50 // Bail out if we're already running a prompt. | |
51 if (!in_progress_prompt_extension_id_.empty()) | |
52 return; | |
53 | |
54 NavigationEntry* nav_entry = nav_controller->GetActiveEntry(); | |
55 if (!nav_entry) | |
56 return; | |
57 | |
58 ExtensionService* extension_service = profile_->GetExtensionService(); | |
59 const extensions::Extension* extension = | |
60 extension_service->disabled_extensions()-> | |
61 GetExtensionOrAppByURL(ExtensionURLInfo(nav_entry->GetURL())); | |
62 if (!extension) | |
63 return; | |
64 | |
65 // Try not to repeatedly prompt the user about the same extension. | |
66 if (prompted_extensions_.find(extension->id()) != prompted_extensions_.end()) | |
67 return; | |
68 prompted_extensions_.insert(extension->id()); | |
69 | |
70 extensions::ExtensionPrefs* extension_prefs = | |
71 extension_service->extension_prefs(); | |
72 if (extension_prefs->DidExtensionEscalatePermissions(extension->id())) { | |
73 // Keep track of the extension id and nav controller we're prompting for. | |
74 // These must be reset in InstallUIProceed and InstallUIAbort. | |
75 in_progress_prompt_extension_id_ = extension->id(); | |
76 in_progress_prompt_navigation_controller_ = nav_controller; | |
77 | |
78 extension_install_prompt_.reset( | |
79 ExtensionInstallUI::CreateInstallPromptWithWebContents( | |
80 nav_controller->GetWebContents())); | |
81 extension_install_prompt_->ConfirmReEnable(this, extension); | |
82 } | |
83 } | |
84 | |
85 void ExtensionNavigationObserver::InstallUIProceed() { | |
86 ExtensionService* extension_service = profile_->GetExtensionService(); | |
87 const extensions::Extension* extension = extension_service->GetExtensionById( | |
88 in_progress_prompt_extension_id_, true); | |
89 NavigationController* nav_controller = | |
90 in_progress_prompt_navigation_controller_; | |
91 CHECK(extension); | |
92 CHECK(nav_controller); | |
93 | |
94 in_progress_prompt_extension_id_ = ""; | |
95 in_progress_prompt_navigation_controller_ = NULL; | |
96 extension_install_prompt_.reset(); | |
97 | |
98 // Grant permissions, re-enable the extension, and then reload the tab. | |
99 extension_service->GrantPermissionsAndEnableExtension( | |
100 extension, extension_install_prompt_->record_oauth2_grant()); | |
101 nav_controller->Reload(true); | |
102 } | |
103 | |
104 void ExtensionNavigationObserver::InstallUIAbort(bool user_initiated) { | |
105 ExtensionService* extension_service = profile_->GetExtensionService(); | |
106 const extensions::Extension* extension = extension_service->GetExtensionById( | |
107 in_progress_prompt_extension_id_, true); | |
108 | |
109 in_progress_prompt_extension_id_ = ""; | |
110 in_progress_prompt_navigation_controller_ = NULL; | |
111 extension_install_prompt_.reset(); | |
112 | |
113 std::string histogram_name = user_initiated ? | |
114 "Extensions.Permissions_ReEnableCancel" : | |
115 "Extensions.Permissions_ReEnableAbort"; | |
116 ExtensionService::RecordPermissionMessagesHistogram( | |
117 extension, histogram_name.c_str()); | |
118 } | |
OLD | NEW |