src/extensions/renderer/user_script_set_manager.h

   1 // Copyright 2014 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 #ifndef EXTENSIONS_RENDERER_USER_SCRIPT_SET_MANAGER_H_
   6 #define EXTENSIONS_RENDERER_USER_SCRIPT_SET_MANAGER_H_
   7
   8 #include <map>
   9 #include <set>
  10 #include <string>
  11
  12 #include "base/memory/linked_ptr.h"
  13 #include "base/memory/shared_memory.h"
  14 #include "base/observer_list.h"
  15 #include "content/public/renderer/render_process_observer.h"
  16 #include "extensions/common/extension.h"
  17 #include "extensions/common/user_script.h"
  18 #include "extensions/renderer/user_script_set.h"
  19
  20 namespace IPC {
  21 class Message;
  22 }
  23
  24 namespace blink {
  25 class WebFrame;
  26 }
  27
  28 namespace extensions {
  29
  30 class ExtensionSet;
  31 class ScriptInjection;
  32
  33 // Manager for separate UserScriptSets, one for each shared memory region.
  34 // Regions are organized as follows:
  35 // static_scripts -- contains all extensions' scripts that are statically
  36 //                   declared in the extension manifest.
  37 // programmatic_scripts -- one region per extension containing only
  38 //                         programmatically-declared scripts, instantiated
  39 //                         when an extension first creates a declarative rule
  40 //                         that would, if triggered, request a script injection.
  41 class UserScriptSetManager : public content::RenderProcessObserver {
  42  public:
  43   // Like a UserScriptSet::Observer, but automatically subscribes to all sets
  44   // associated with the manager.
  45   class Observer {
  46    public:
  47     virtual void OnUserScriptsUpdated(
  48         const std::set<std::string>& changed_extensions,
  49         const std::vector<UserScript*>& scripts) = 0;
  50   };
  51
  52   UserScriptSetManager(const ExtensionSet* extensions);
  53
  54   virtual ~UserScriptSetManager();
  55
  56   void AddObserver(Observer* observer);
  57   void RemoveObserver(Observer* observer);
  58
  59   const UserScriptSet* GetProgrammaticScriptsByExtension(
  60       const ExtensionId& extensionId);
  61
  62   // Put all injections from |static_scripts| and each of
  63   // |programmatic_scripts_| into |injections|.
  64   void GetAllInjections(ScopedVector<ScriptInjection>* injections,
  65                         blink::WebFrame* web_frame,
  66                         int tab_id,
  67                         UserScript::RunLocation run_location);
  68
  69   // Get active extension IDs from |static_scripts| and each of
  70   // |programmatic_scripts_|.
  71   void GetAllActiveExtensionIds(std::set<std::string>* ids) const;
  72
  73   const UserScriptSet* static_scripts() const { return &static_scripts_; }
  74
  75  private:
  76   // Map for per-extension sets that may be defined programmatically.
  77   typedef std::map<ExtensionId, linked_ptr<UserScriptSet> > UserScriptSetMap;
  78
  79   // content::RenderProcessObserver implementation.
  80   virtual bool OnControlMessageReceived(const IPC::Message& message) OVERRIDE;
  81
  82   // Handle the UpdateUserScripts extension message.
  83   void OnUpdateUserScripts(base::SharedMemoryHandle shared_memory,
  84                            const ExtensionId& extension_id,
  85                            const std::set<std::string>& changed_extensions);
  86
  87   // Scripts statically defined in extension manifests.
  88   UserScriptSet static_scripts_;
  89
  90   // Scripts programmatically-defined through API calls (initialized and stored
  91   // per-extension).
  92   UserScriptSetMap programmatic_scripts_;
  93
  94   // The set of all known extensions. Owned by the Dispatcher.
  95   const ExtensionSet* extensions_;
  96
  97   // The associated observers.
  98   ObserverList<Observer> observers_;
  99
 100   DISALLOW_COPY_AND_ASSIGN(UserScriptSetManager);
 101 };
 102
 103 }  // namespace extensions
 104
 105 #endif  // EXTENSIONS_RENDERER_USER_SCRIPT_SET_MANAGER_H_