<webview>: Implement declarativeWebRequest API

chrome/browser/extensions/api/declarative/rules_registry_service.h

 // Copyright (c) 2012 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_API_DECLARATIVE_RULES_REGISTRY_SERVICE_H__
 #define CHROME_BROWSER_EXTENSIONS_API_DECLARATIVE_RULES_REGISTRY_SERVICE_H__
 
 #include <map>
 #include <string>
 #include <vector>
@@ skipping 20 matching lines @@
 class RulesRegistryStorageDelegate;
 }
 
 namespace extensions {
 
 // This class owns all RulesRegistries implementations of an ExtensionService.
 // This class lives on the UI thread.
 class RulesRegistryService : public ProfileKeyedAPI,
                              public content::NotificationObserver {
  public:
+  typedef RulesRegistry::WebViewKey WebViewKey;
+  struct RulesRegistryKey {
+    std::string event_name;
+    WebViewKey webview_key;
+    RulesRegistryKey(const std::string event_name,
+                     const WebViewKey& webview_key)
+        : event_name(event_name),
+          webview_key(webview_key) {}
+    bool operator<(const RulesRegistryKey& other) const {
+      return (event_name < other.event_name) ||
+          ((event_name == other.event_name) &&
+          (webview_key < other.webview_key));
+    }
+  };
+
   explicit RulesRegistryService(Profile* profile);
   virtual ~RulesRegistryService();
 
   // Unregisters refptrs to concrete RulesRegistries at other objects that were
   // created by us so that the RulesRegistries can be released.
   virtual void Shutdown() OVERRIDE;
 
   // ProfileKeyedAPI implementation.
   static ProfileKeyedAPIFactory<RulesRegistryService>* GetFactoryInstance();
 
   // Convenience method to get the RulesRegistryService for a profile.
   static RulesRegistryService* Get(Profile* profile);
 
   // Registers the default RulesRegistries used in Chromium.
-  void RegisterDefaultRulesRegistries();
+  void EnsureDefaultRulesRegistriesRegistered(
+      const WebViewKey& webview_key);
 
   // Registers a RulesRegistry and wraps it in an InitializingRulesRegistry.
   void RegisterRulesRegistry(scoped_refptr<RulesRegistry> rule_registry);
 
-  // Returns the RulesRegistry for |event_name| or NULL if no such registry
-  // has been registered.
+  // Returns the RulesRegistry for |event_name| and |webview_key| or NULL if no
+  // such registry has been registered. Default rules registries (such as the
+  // WebRequest rules registry) will be created on first access.
   scoped_refptr<RulesRegistry> GetRulesRegistry(
-      const std::string& event_name) const;
+      const WebViewKey& webview_key,
+      const std::string& event_name);
 
   // Accessors for each type of rules registry.
   ContentRulesRegistry* content_rules_registry() const {
+    CHECK(content_rules_registry_);
     return content_rules_registry_;
   }
 
+  // Removes all rules registries of a given webview embedder process ID.
+  void RemoveWebViewRulesRegistries(int process_id);
+
   // For testing.
   void SimulateExtensionUnloaded(const std::string& extension_id);
  private:
   friend class ProfileKeyedAPIFactory<RulesRegistryService>;
 
-  // Maps event names to RuleRegistries that handle these events.
-  typedef std::map<std::string, scoped_refptr<RulesRegistry> >
+  // Maps <event name, webview key> to RuleRegistries that handle these
+  // events.
+  typedef std::map<RulesRegistryKey, scoped_refptr<RulesRegistry> >
       RulesRegistryMap;
 
   // Notifies all RulesRegistries that |extension_id| was unloaded.
   // It is not guaranteed that this notification is processed synchronously.
   // If extensions live on another thread, the notification is posted.
   void OnExtensionUnloaded(const std::string& extension_id);
 
   // Implementation of content::NotificationObserver.
   virtual void Observe(int type,
                        const content::NotificationSource& source,
@@ skipping 18 matching lines @@
   content::NotificationRegistrar registrar_;
 
   Profile* profile_;
 
   DISALLOW_COPY_AND_ASSIGN(RulesRegistryService);
 };
 
 }  // namespace extensions
 
 #endif  // CHROME_BROWSER_EXTENSIONS_API_DECLARATIVE_RULES_REGISTRY_SERVICE_H__