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