• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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/ui/extensions/extension_enable_flow.h"
6 
7 #include "chrome/browser/chrome_notification_types.h"
8 #include "chrome/browser/extensions/extension_service.h"
9 #include "chrome/browser/extensions/extension_system.h"
10 #include "chrome/browser/ui/browser.h"
11 #include "chrome/browser/ui/extensions/extension_enable_flow_delegate.h"
12 #include "chrome/browser/ui/scoped_tabbed_browser_displayer.h"
13 #include "content/public/browser/notification_details.h"
14 #include "content/public/browser/notification_source.h"
15 
16 using extensions::Extension;
17 
ExtensionEnableFlow(Profile * profile,const std::string & extension_id,ExtensionEnableFlowDelegate * delegate)18 ExtensionEnableFlow::ExtensionEnableFlow(Profile* profile,
19                                          const std::string& extension_id,
20                                          ExtensionEnableFlowDelegate* delegate)
21     : profile_(profile),
22       extension_id_(extension_id),
23       delegate_(delegate),
24       parent_contents_(NULL),
25       parent_window_(NULL) {
26 }
27 
~ExtensionEnableFlow()28 ExtensionEnableFlow::~ExtensionEnableFlow() {
29 }
30 
StartForWebContents(content::WebContents * parent_contents)31 void ExtensionEnableFlow::StartForWebContents(
32     content::WebContents* parent_contents) {
33   parent_contents_ = parent_contents;
34   parent_window_ = NULL;
35   Run();
36 }
37 
StartForNativeWindow(gfx::NativeWindow parent_window)38 void ExtensionEnableFlow::StartForNativeWindow(
39     gfx::NativeWindow parent_window) {
40   parent_contents_ = NULL;
41   parent_window_ = parent_window;
42   Run();
43 }
44 
StartForCurrentlyNonexistentWindow(base::Callback<gfx::NativeWindow (void)> window_getter)45 void ExtensionEnableFlow::StartForCurrentlyNonexistentWindow(
46     base::Callback<gfx::NativeWindow(void)> window_getter) {
47   window_getter_ = window_getter;
48   Run();
49 }
50 
Run()51 void ExtensionEnableFlow::Run() {
52   ExtensionService* service =
53       extensions::ExtensionSystem::Get(profile_)->extension_service();
54   const Extension* extension = service->GetExtensionById(extension_id_, true);
55   if (!extension) {
56     extension = service->GetTerminatedExtension(extension_id_);
57     // It's possible (though unlikely) the app could have been uninstalled since
58     // the user clicked on it.
59     if (!extension)
60       return;
61     // If the app was terminated, reload it first.
62     service->ReloadExtension(extension_id_);
63 
64     // ReloadExtension reallocates the Extension object.
65     extension = service->GetExtensionById(extension_id_, true);
66 
67     // |extension| could be NULL for asynchronous load, such as the case of
68     // an unpacked extension. Wait for the load to continue the flow.
69     if (!extension) {
70       StartObserving();
71       return;
72     }
73   }
74 
75   CheckPermissionAndMaybePromptUser();
76 }
77 
CheckPermissionAndMaybePromptUser()78 void ExtensionEnableFlow::CheckPermissionAndMaybePromptUser() {
79   ExtensionService* service =
80       extensions::ExtensionSystem::Get(profile_)->extension_service();
81   const Extension* extension = service->GetExtensionById(extension_id_, true);
82   if (!extension) {
83     delegate_->ExtensionEnableFlowAborted(false);  // |delegate_| may delete us.
84     return;
85   }
86 
87   extensions::ExtensionPrefs* extension_prefs = service->extension_prefs();
88   if (!extension_prefs->DidExtensionEscalatePermissions(extension_id_)) {
89     // Enable the extension immediately if its privileges weren't escalated.
90     // This is a no-op if the extension was previously terminated.
91     service->EnableExtension(extension_id_);
92 
93     delegate_->ExtensionEnableFlowFinished();  // |delegate_| may delete us.
94     return;
95   }
96 
97   CreatePrompt();
98   prompt_->ConfirmReEnable(this, extension);
99 }
100 
CreatePrompt()101 void ExtensionEnableFlow::CreatePrompt() {
102   if (!window_getter_.is_null())
103     parent_window_ = window_getter_.Run();
104   prompt_.reset(parent_contents_ ?
105       new ExtensionInstallPrompt(parent_contents_) :
106       new ExtensionInstallPrompt(profile_, parent_window_, this));
107 }
108 
StartObserving()109 void ExtensionEnableFlow::StartObserving() {
110   registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_LOADED,
111                  content::Source<Profile>(profile_));
112   registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_LOAD_ERROR,
113                  content::Source<Profile>(profile_));
114   registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_UNINSTALLED,
115                  content::Source<Profile>(profile_));
116 }
117 
StopObserving()118 void ExtensionEnableFlow::StopObserving() {
119   registrar_.RemoveAll();
120 }
121 
Observe(int type,const content::NotificationSource & source,const content::NotificationDetails & details)122 void ExtensionEnableFlow::Observe(int type,
123                                   const content::NotificationSource& source,
124                                   const content::NotificationDetails& details) {
125   switch (type) {
126     case chrome::NOTIFICATION_EXTENSION_LOADED: {
127       const Extension* extension =
128           content::Details<const Extension>(details).ptr();
129       if (extension->id() == extension_id_) {
130         StopObserving();
131         CheckPermissionAndMaybePromptUser();
132       }
133 
134       break;
135     }
136     case chrome::NOTIFICATION_EXTENSION_LOAD_ERROR: {
137       StopObserving();
138       delegate_->ExtensionEnableFlowAborted(false);
139       break;
140     }
141     case chrome::NOTIFICATION_EXTENSION_UNINSTALLED: {
142       const Extension* extension =
143           content::Details<const Extension>(details).ptr();
144       if (extension->id() == extension_id_) {
145         StopObserving();
146         delegate_->ExtensionEnableFlowAborted(false);
147       }
148 
149       break;
150     }
151     default:
152       NOTREACHED();
153   }
154 }
155 
InstallUIProceed()156 void ExtensionEnableFlow::InstallUIProceed() {
157   ExtensionService* service =
158       extensions::ExtensionSystem::Get(profile_)->extension_service();
159 
160   // The extension can be uninstalled in another window while the UI was
161   // showing. Treat it as a cancellation and notify |delegate_|.
162   const Extension* extension = service->GetExtensionById(extension_id_, true);
163   if (!extension) {
164     delegate_->ExtensionEnableFlowAborted(true);
165     return;
166   }
167 
168   service->GrantPermissionsAndEnableExtension(extension);
169   delegate_->ExtensionEnableFlowFinished();  // |delegate_| may delete us.
170 }
171 
InstallUIAbort(bool user_initiated)172 void ExtensionEnableFlow::InstallUIAbort(bool user_initiated) {
173   delegate_->ExtensionEnableFlowAborted(user_initiated);
174   // |delegate_| may delete us.
175 }
176 
OpenURL(const content::OpenURLParams & params)177 content::WebContents* ExtensionEnableFlow::OpenURL(
178     const content::OpenURLParams& params) {
179   chrome::ScopedTabbedBrowserDisplayer displayer(
180       profile_, chrome::GetActiveDesktop());
181   return displayer.browser()->OpenURL(params);
182 }
183