Refactor renderer-side script injection

extensions/renderer/script_injection_manager.h

Index: extensions/renderer/script_injection_manager.h
diff --git a/extensions/renderer/script_injection_manager.h b/extensions/renderer/script_injection_manager.h
new file mode 100644
index 0000000000000000000000000000000000000000..a7b58108e067dfb829729f2629561187b999b5a8
--- /dev/null
+++ b/extensions/renderer/script_injection_manager.h
@@ -0,0 +1,103 @@
+// 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 EXTENSIONS_RENDERER_SCRIPT_INJECTION_MANAGER_H_
+#define EXTENSIONS_RENDERER_SCRIPT_INJECTION_MANAGER_H_
+
+#include <map>
+#include <set>
+#include <string>
+
+#include "base/callback.h"
+#include "base/memory/scoped_vector.h"
+#include "base/memory/shared_memory.h"
+#include "base/scoped_observer.h"
+#include "extensions/common/user_script.h"
+#include "extensions/renderer/script_injection.h"
+#include "extensions/renderer/user_script_set.h"
+
+struct ExtensionMsg_ExecuteCode_Params;
+
+namespace blink {
+class WebFrame;
+class WebLocalFrame;
+}
+
+namespace content {
+class RenderView;
+}
+
+namespace extensions {
+class Extension;
+class ExtensionSet;
+
+// The ScriptInjectionManager manages extensions injecting scripts into frames
+// via both content/user scripts and tabs.executeScript(). It is responsible for
+// maintaining any pending injections awaiting permission or the appropriate
+// load point, and injecting them when ready.
+class ScriptInjectionManager : public UserScriptSet::Observer {
+ public:
+  ScriptInjectionManager(const ExtensionSet* extensions,
+                         UserScriptSet* user_script_set);
+  virtual ~ScriptInjectionManager();
+
+  // Notifies that a new render view has been created.
+  void OnRenderViewCreated(content::RenderView* render_view);
+
+ private:
+  // A RenderViewObserver implementation which watches the various render views
+  // in order to notify the ScriptInjectionManager of different document load
+  // states.
+  class RVOHelper;
+
+  typedef std::map<blink::WebFrame*, UserScript::RunLocation> FrameStatusMap;
+
+  // UserScriptSet::Observer implementation.
+  virtual void OnUserScriptsUpdated(
+      const std::set<std::string>& changed_extensions,
+      const std::vector<UserScript*>& scripts) OVERRIDE;
+
+  // Notifies that an RVOHelper should be removed.
+  void RemoveObserver(RVOHelper* helper);
+
+  // Invalidate any pending tasks associated with |frame|.
+  void InvalidateForFrame(blink::WebFrame* frame);
+
+  // Inject appropriate scripts into |frame|.
+  void InjectScripts(blink::WebFrame* frame,
+                     UserScript::RunLocation run_location);
+
+  // Handle the ExecuteCode extension message.
+  void HandleExecuteCode(const ExtensionMsg_ExecuteCode_Params& params,
+                         content::RenderView* render_view);
+
+  // Handle the GrantInjectionPermission extension message.
+  void HandlePermitScriptInjection(int request_id);
+
+  // Extensions metadata, owned by Dispatcher (which owns this object).
+  const ExtensionSet* extensions_;
+
+  // The map of active web frames to their corresponding statuses. The
+  // RunLocation of the frame corresponds to the last location that has ran.
+  FrameStatusMap frame_statuses_;
+
+  // The collection of RVOHelpers.
+  ScopedVector<RVOHelper> rvo_helpers_;
+
+  // The set of UserScripts associated with extensions. Owned by the Dispatcher.
+  UserScriptSet* user_script_set_;
+
+  // Pending injections which are waiting for either the proper run location or
+  // user consent.
+  ScopedVector<ScriptInjection> pending_injections_;
+
+  ScopedObserver<UserScriptSet, UserScriptSet::Observer>
+      user_script_set_observer_;
+
+  DISALLOW_COPY_AND_ASSIGN(ScriptInjectionManager);
+};
+
+}  // namespace extensions
+
+#endif  // EXTENSIONS_RENDERER_SCRIPT_INJECTION_MANAGER_H_