Add ErrorConsole UI for Extension Install Warnings

chrome/browser/extensions/error_console/error_console.cc

 // Copyright 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/extensions/error_console/error_console.h"
 
 #include <list>
+#include <vector>
 
 #include "base/lazy_instance.h"
 #include "base/stl_util.h"
+#include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/chrome_notification_types.h"
+#include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/extensions/extension_system.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/common/extensions/extension.h"
+#include "chrome/common/extensions/extension_set.h"
+#include "chrome/common/pref_names.h"
 #include "content/public/browser/notification_details.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/notification_source.h"
 #include "extensions/common/constants.h"
 
 namespace extensions {
 
 namespace {
 
 const size_t kMaxErrorsPerExtension = 100;
 
 // Iterate through an error list and remove and delete all errors which were
 // from an incognito context.
 void DeleteIncognitoErrorsFromList(ErrorConsole::ErrorList* list) {
   ErrorConsole::ErrorList::iterator iter = list->begin();
   while (iter != list->end()) {
     if ((*iter)->from_incognito()) {
       delete *iter;
       iter = list->erase(iter);
     } else {
       ++iter;
     }
   }
 }
 
+// Iterate through an error list and remove and delete all errors of a given
+// |type|.
+void DeleteErrorsOfTypeFromList(ErrorConsole::ErrorList* list,
+                                ExtensionError::Type type) {
+  ErrorConsole::ErrorList::iterator iter = list->begin();
+  while (iter != list->end()) {
+    if ((*iter)->type() == type) {
+      delete *iter;
+      iter = list->erase(iter);
+    } else {
+      ++iter;
+    }
+  }
+}
+
 base::LazyInstance<ErrorConsole::ErrorList> g_empty_error_list =
     LAZY_INSTANCE_INITIALIZER;
 
 }  // namespace
 
 void ErrorConsole::Observer::OnErrorConsoleDestroyed() {
 }
 
-ErrorConsole::ErrorConsole(Profile* profile) : profile_(profile) {
-  registrar_.Add(this,
-                 chrome::NOTIFICATION_PROFILE_DESTROYED,
-                 content::NotificationService::AllBrowserContextsAndSources());
-  registrar_.Add(this,
-                 chrome::NOTIFICATION_EXTENSION_UNINSTALLED,
-                 content::Source<Profile>(profile_));
+ErrorConsole::ErrorConsole(Profile* profile,
+                           ExtensionService* extension_service)
+     : enabled_(false), profile_(profile) {
+  pref_registrar_.Init(profile_->GetPrefs());
+  pref_registrar_.Add(prefs::kExtensionsUIDeveloperMode,
+                      base::Bind(&ErrorConsole::OnPrefChanged,
+                                 base::Unretained(this)));
+
+  if (profile_->GetPrefs()->GetBoolean(prefs::kExtensionsUIDeveloperMode))
+    Enable(extension_service);
 }
 
 ErrorConsole::~ErrorConsole() {
   FOR_EACH_OBSERVER(Observer, observers_, OnErrorConsoleDestroyed());
   RemoveAllErrors();
 }
 
 // static
 ErrorConsole* ErrorConsole::Get(Profile* profile) {
   return ExtensionSystem::Get(profile)->error_console();
 }
 
-void ErrorConsole::ReportError(scoped_ptr<const ExtensionError> scoped_error) {
+void ErrorConsole::ReportError(scoped_ptr<const ExtensionError> error) {
   DCHECK(thread_checker_.CalledOnValidThread());
 
-  const ExtensionError* error = scoped_error.release();
-  // If there are too many errors for an extension already, limit ourselves to
-  // the most recent ones.
-  ErrorList* error_list = &errors_[error->extension_id()];
-  if (error_list->size() >= kMaxErrorsPerExtension) {
-    delete error_list->front();
-    error_list->pop_front();
+  if (!enabled_ || !Extension::IdIsValid(error->extension_id()))
+    return;
+
+  // This will be assigned when we place the error in the list, releasing the
+  // scoped_ptr.
+  const ExtensionError* weak_error = NULL;
+  ErrorList* extension_errors = &errors_[error->extension_id()];
+
+  // First, check if it's a duplicate.
+  for (size_t i = 0; i < extension_errors->size(); ++i) {
+    // If we find a duplicate error, replace the error with the new version.
+    // This can be useful for runtime errors, where we can link to the latest
+    // context, inspectable view, etc.
+    if (error->IsEqual(extension_errors->at(i))) {
+      delete extension_errors->at(i);
+      extension_errors->at(i) = error.release();
+      weak_error = extension_errors->at(i);
+    }
   }
-  error_list->push_back(error);
 
-  FOR_EACH_OBSERVER(Observer, observers_, OnErrorAdded(error));
+  // If we didn't replace a previous error, then we add the error to the end of
+  // the list.
+  if (!weak_error) {
+    // If there are too many errors for an extension already, limit ourselves to
+    // the most recent ones.
+    if (extension_errors->size() >= kMaxErrorsPerExtension) {
+      delete extension_errors->front();
+      extension_errors->pop_front();
+    }
+    extension_errors->push_back(error.release());
+    weak_error = extension_errors->back();
+  }
+
+  FOR_EACH_OBSERVER(Observer, observers_, OnErrorAdded(weak_error));
 }
 
 const ErrorConsole::ErrorList& ErrorConsole::GetErrorsForExtension(
     const std::string& extension_id) const {
   ErrorMap::const_iterator iter = errors_.find(extension_id);
   if (iter != errors_.end())
     return iter->second;
   return g_empty_error_list.Get();
 }
 
 void ErrorConsole::AddObserver(Observer* observer) {
   DCHECK(thread_checker_.CalledOnValidThread());
   observers_.AddObserver(observer);
 }
 
 void ErrorConsole::RemoveObserver(Observer* observer) {
   DCHECK(thread_checker_.CalledOnValidThread());
   observers_.RemoveObserver(observer);
 }
 
+void ErrorConsole::OnPrefChanged() {
+  bool developer_mode =
+      profile_->GetPrefs()->GetBoolean(prefs::kExtensionsUIDeveloperMode);
+
+  if (developer_mode && !enabled_)
+    Enable(ExtensionSystem::Get(profile_)->extension_service());
+  else if (!developer_mode && enabled_)
+    Disable();
+}
+
+void ErrorConsole::Enable(ExtensionService* extension_service) {
+  enabled_ = true;
+
+  notification_registrar_.Add(
+      this,
+      chrome::NOTIFICATION_PROFILE_DESTROYED,
+      content::NotificationService::AllBrowserContextsAndSources());
+  notification_registrar_.Add(
+      this,
+      chrome::NOTIFICATION_EXTENSION_UNINSTALLED,
+      content::Source<Profile>(profile_));
+  notification_registrar_.Add(
+      this,
+      chrome::NOTIFICATION_EXTENSION_INSTALLED,
+      content::Source<Profile>(profile_));
+
+  if (extension_service) {
+    // Get manifest errors for extensions already installed.
+    const ExtensionSet* extensions = extension_service->extensions();
+    for (ExtensionSet::const_iterator iter = extensions->begin();
+         iter != extensions->end(); ++iter) {
+      AddManifestErrorsForExtension(iter->get());
+    }
+  }
+}
+
+void ErrorConsole::Disable() {
+  notification_registrar_.RemoveAll();
+  RemoveAllErrors();
+  enabled_ = false;
+}
+
+void ErrorConsole::AddManifestErrorsForExtension(const Extension* extension) {
+  const std::vector<InstallWarning>& warnings =
+      extension->install_warnings();
+  for (std::vector<InstallWarning>::const_iterator iter = warnings.begin();
+       iter != warnings.end(); ++iter) {
+    ReportError(scoped_ptr<const ExtensionError>(new ManifestError(
+        extension->id(),
+        base::UTF8ToUTF16(iter->message),
+        base::UTF8ToUTF16(iter->key),
+        base::UTF8ToUTF16(iter->specific))));
+  }
+}
+
 void ErrorConsole::RemoveIncognitoErrors() {
   for (ErrorMap::iterator iter = errors_.begin();
        iter != errors_.end(); ++iter) {
     DeleteIncognitoErrorsFromList(&(iter->second));
   }
 }
 
 void ErrorConsole::RemoveErrorsForExtension(const std::string& extension_id) {
   ErrorMap::iterator iter = errors_.find(extension_id);
   if (iter != errors_.end()) {
@@ skipping 19 matching lines @@
       if (profile->IsOffTheRecord() && profile_->IsSameProfile(profile))
         RemoveIncognitoErrors();
       break;
     }
     case chrome::NOTIFICATION_EXTENSION_UNINSTALLED:
       // No need to check the profile here, since we registered to only receive
       // notifications from our own.
       RemoveErrorsForExtension(
           content::Details<Extension>(details).ptr()->id());
       break;
+    case chrome::NOTIFICATION_EXTENSION_INSTALLED: {
+      const InstalledExtensionInfo* info =
+          content::Details<InstalledExtensionInfo>(details).ptr();
+
+      // We don't want to have manifest errors from previous installs. We want
+      // to keep runtime errors, though, because extensions are reloaded on a
+      // refresh of chrome:extensions, and we don't want to wipe our history
+      // whenever that happens.
+      ErrorMap::iterator iter = errors_.find(info->extension->id());
+      if (iter != errors_.end()) {
+        DeleteErrorsOfTypeFromList(&(iter->second),
+                                   ExtensionError::MANIFEST_ERROR);
+      }
+
+      AddManifestErrorsForExtension(info->extension);
+      break;
+    }
     default:
       NOTREACHED();
   }
 }
 
 }  // namespace extensions