chrome/browser/extensions/navigation_observer.cc

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