src/chrome/browser/extensions/extension_warning_service.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/extension_warning_service.h"
   6
   7 #include "chrome/browser/browser_process.h"
   8 #include "chrome/browser/extensions/extension_service.h"
   9 #include "chrome/browser/profiles/profile.h"
  10 #include "chrome/browser/profiles/profile_manager.h"
  11 #include "content/public/browser/browser_thread.h"
  12 #include "extensions/browser/extension_registry.h"
  13 #include "extensions/browser/extension_system.h"
  14
  15 using content::BrowserThread;
  16
  17 namespace extensions {
  18
  19 ExtensionWarningService::ExtensionWarningService(Profile* profile)
  20     : profile_(profile), extension_registry_observer_(this) {
  21   DCHECK(CalledOnValidThread());
  22   if (profile_) {
  23     extension_registry_observer_.Add(
  24         ExtensionRegistry::Get(profile_->GetOriginalProfile()));
  25   }
  26 }
  27
  28 ExtensionWarningService::~ExtensionWarningService() {}
  29
  30 void ExtensionWarningService::ClearWarnings(
  31     const std::set<ExtensionWarning::WarningType>& types) {
  32   DCHECK(CalledOnValidThread());
  33   bool deleted_anything = false;
  34   for (ExtensionWarningSet::iterator i = warnings_.begin();
  35        i != warnings_.end();) {
  36     if (types.find(i->warning_type()) != types.end()) {
  37       deleted_anything = true;
  38       warnings_.erase(i++);
  39     } else {
  40       ++i;
  41     }
  42   }
  43
  44   if (deleted_anything)
  45     NotifyWarningsChanged();
  46 }
  47
  48 std::set<ExtensionWarning::WarningType>
  49 ExtensionWarningService::GetWarningTypesAffectingExtension(
  50     const std::string& extension_id) const {
  51   DCHECK(CalledOnValidThread());
  52   std::set<ExtensionWarning::WarningType> result;
  53   for (ExtensionWarningSet::const_iterator i = warnings_.begin();
  54        i != warnings_.end(); ++i) {
  55     if (i->extension_id() == extension_id)
  56       result.insert(i->warning_type());
  57   }
  58   return result;
  59 }
  60
  61 std::vector<std::string>
  62 ExtensionWarningService::GetWarningMessagesForExtension(
  63     const std::string& extension_id) const {
  64   DCHECK(CalledOnValidThread());
  65   std::vector<std::string> result;
  66
  67   const ExtensionService* extension_service =
  68       ExtensionSystem::Get(profile_)->extension_service();
  69
  70   for (ExtensionWarningSet::const_iterator i = warnings_.begin();
  71        i != warnings_.end(); ++i) {
  72     if (i->extension_id() == extension_id)
  73       result.push_back(i->GetLocalizedMessage(extension_service->extensions()));
  74   }
  75   return result;
  76 }
  77
  78 void ExtensionWarningService::AddWarnings(
  79     const ExtensionWarningSet& warnings) {
  80   DCHECK(CalledOnValidThread());
  81   size_t old_size = warnings_.size();
  82
  83   warnings_.insert(warnings.begin(), warnings.end());
  84
  85   if (old_size != warnings_.size())
  86     NotifyWarningsChanged();
  87 }
  88
  89 // static
  90 void ExtensionWarningService::NotifyWarningsOnUI(
  91     void* profile_id,
  92     const ExtensionWarningSet& warnings) {
  93   DCHECK_CURRENTLY_ON(BrowserThread::UI);
  94   Profile* profile = reinterpret_cast<Profile*>(profile_id);
  95   if (!profile ||
  96       !g_browser_process->profile_manager() ||
  97       !g_browser_process->profile_manager()->IsValidProfile(profile)) {
  98     return;
  99   }
 100
 101   extensions::ExtensionWarningService* warning_service =
 102       extensions::ExtensionSystem::Get(profile)->warning_service();
 103
 104   warning_service->AddWarnings(warnings);
 105 }
 106
 107 void ExtensionWarningService::AddObserver(Observer* observer) {
 108   observer_list_.AddObserver(observer);
 109 }
 110
 111 void ExtensionWarningService::RemoveObserver(Observer* observer) {
 112   observer_list_.RemoveObserver(observer);
 113 }
 114
 115 void ExtensionWarningService::NotifyWarningsChanged() {
 116   FOR_EACH_OBSERVER(Observer, observer_list_, ExtensionWarningsChanged());
 117 }
 118
 119 void ExtensionWarningService::OnExtensionUnloaded(
 120     content::BrowserContext* browser_context,
 121     const Extension* extension,
 122     UnloadedExtensionInfo::Reason reason) {
 123   // Unloading one extension might have solved the problems of others.
 124   // Therefore, we clear warnings of this type for all extensions.
 125   std::set<ExtensionWarning::WarningType> warning_types =
 126       GetWarningTypesAffectingExtension(extension->id());
 127   ClearWarnings(warning_types);
 128 }
 129
 130 }  // namespace extensions