Add PolicyProvider to ExtensionManagement

chrome/browser/extensions/extension_management.h

 // Copyright 2014 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.
 
 #ifndef CHROME_BROWSER_EXTENSIONS_EXTENSION_MANAGEMENT_H_
 #define CHROME_BROWSER_EXTENSIONS_EXTENSION_MANAGEMENT_H_
 
 #include <map>
 #include <string>
 #include <vector>
 
 #include "base/macros.h"
+#include "base/memory/scoped_ptr.h"
+#include "base/memory/singleton.h"
+#include "base/observer_list.h"
+#include "base/prefs/pref_change_registrar.h"
 #include "base/values.h"
+#include "components/keyed_service/content/browser_context_keyed_service_factory.h"
+#include "components/keyed_service/core/keyed_service.h"
+#include "extensions/browser/management_policy.h"
 #include "extensions/common/extension.h"
 #include "extensions/common/manifest.h"
 #include "extensions/common/url_pattern_set.h"
 
 class PrefService;
 
+namespace content {
+class BrowserContext;
+}  // namespace content
+
 namespace extensions {
 
 // Tracks the management policies that affect extensions and provides interfaces
 // for observing and obtaining the global settings for all extensions, as well
 // as per-extension settings.
-class ExtensionManagement {
+class ExtensionManagement : public KeyedService {
  public:
+  // Observer class for extension management settings changes.
+  class Observer {
+   public:
+    virtual ~Observer() {}
+
+    // Will be called when an extension management preference changes.
+    virtual void OnExtensionManagementPrefChanged() = 0;

[Joao da Silva, 2014/09/03 11:52:54]

Rename to "OnExtensionManagementSettingsChanged".

Prefs are an implementation detail that Observers shouldn't need to know about.

[binjin, 2014/09/03 15:31:36]

Done.

+  };
+
   // Installation mode for extensions, default is INSTALLATION_ALLOWED.
   // * INSTALLATION_ALLOWED: Extension can be installed.
   // * INSTALLATION_BLOCKED: Extension cannot be installed.
   // * INSTALLATION_FORCED: Extension will be installed automatically
   //                        and cannot be disabled.
   // * INSTALLATION_RECOMMENDED: Extension will be installed automatically but
   //                             can be disabled.
   enum InstallationMode {
     INSTALLATION_ALLOWED = 0,
     INSTALLATION_BLOCKED,
@@ skipping 24 matching lines @@
 
     std::vector<Manifest::Type> allowed_types;
     bool has_restricted_allowed_types;
   };
 
   typedef std::map<ExtensionId, IndividualSettings> SettingsIdMap;
 
   explicit ExtensionManagement(PrefService* pref_service);
   virtual ~ExtensionManagement();
 
-  // Load all extension management preferences from |pref_service|, and
-  // refresh the settings.
-  void Refresh();
+  void AddObserver(Observer* obs);
+  void RemoveObserver(Observer* obs);

[Joao da Silva, 2014/09/03 11:52:54]

Rename "obs" to "observer"

[binjin, 2014/09/03 15:31:36]

Done.

+
+  // Get the ManagementPolicy::Provider controlled by extension management
+  // policy settings.
+  ManagementPolicy::Provider* GetProvider();
+
+  // Checks if extensions are blacklisted by default, by policy. When true,
+  // this means that even extensions without an ID should be blacklisted (e.g.
+  // from the command line, or when loaded as an unpacked extension).
+  bool BlacklistedByDefault();
 
   // Helper function to read |settings_by_id_| with |id| as key. Returns a
   // constant reference to default settings if |id| does not exist.
   const IndividualSettings& ReadById(const ExtensionId& id) const;
 
   // Returns a constant reference to |global_settings_|.
   const GlobalSettings& ReadGlobalSettings() const;
 
  private:
+  // Load all extension management preferences from |pref_service|, and
+  // refresh the settings.
+  void Refresh();
+
   // Load preference with name |pref_name| and expected type |expected_type|.
   // If |force_managed| is true, only loading from the managed preference store
   // is allowed. Returns NULL if the preference is not present, not allowed to
   // be loaded from or has the wrong type.
   const base::Value* LoadPreference(const char* pref_name,
                                     bool force_managed,
                                     base::Value::Type expected_type);
 
+  void OnExtensionPrefChanged();
+  void NotifyExtensionManagementPrefChanged();
+
   // Helper function to access |settings_by_id_| with |id| as key.
   // Adds a new IndividualSettings entry to |settings_by_id_| if none exists for
   // |id| yet.
   IndividualSettings* AccessById(const ExtensionId& id);
 
   // TODO(binjin): Add |settings_by_update_url_|, and implement mechanism for
   // it.
+  IndividualSettings default_settings_;
   SettingsIdMap settings_by_id_;
-  IndividualSettings default_settings_;
   GlobalSettings global_settings_;
 
   PrefService* pref_service_;
 
+  ObserverList<Observer> observer_list_;

[binjin, 2014/09/02 20:47:49]

adding true as 2nd parameter fails the ExtensionServiceTest now, will
investigate later.

[Joao da Silva, 2014/09/03 11:52:54]

That suggests that there is some missing cleanup at shutdown.

+  PrefChangeRegistrar pref_change_registrar_;
+  scoped_ptr<ManagementPolicy::Provider> provider_;
+
   DISALLOW_COPY_AND_ASSIGN(ExtensionManagement);
 };
 
+class ExtensionManagementFactory : public BrowserContextKeyedServiceFactory {
+ public:
+  static ExtensionManagement* GetForBrowserContext(
+      content::BrowserContext* context);
+  static ExtensionManagementFactory* GetInstance();
+
+ private:
+  friend struct DefaultSingletonTraits<ExtensionManagementFactory>;
+
+  ExtensionManagementFactory();
+  virtual ~ExtensionManagementFactory();
+
+  // BrowserContextKeyedServiceExtensionManagementFactory:
+  virtual KeyedService* BuildServiceInstanceFor(
+      content::BrowserContext* context) const OVERRIDE;
+
+  DISALLOW_COPY_AND_ASSIGN(ExtensionManagementFactory);
+};
+
 }  // namespace extensions
 
 #endif  // CHROME_BROWSER_EXTENSIONS_EXTENSION_MANAGEMENT_H_