Introduce ActiveScriptController; track active extension scripts

chrome/browser/extensions/active_script_controller.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_ACTIVE_SCRIPT_CONTROLLER_H_
+#define CHROME_BROWSER_EXTENSIONS_ACTIVE_SCRIPT_CONTROLLER_H_
+
+#include <map>
+#include <set>
+#include <string>
+
+#include "base/compiler_specific.h"
+#include "base/memory/linked_ptr.h"
+#include "chrome/browser/extensions/location_bar_controller_provider.h"
+
+namespace content {
+class WebContents;
+}
+
+class ExtensionAction;
+
+namespace extensions {
+class Extension;
+
+// The provider for ExtensionActions corresponding to scripts which are actively
+// running or need permission.
+class ActiveScriptController : public LocationBarControllerProvider {
+ public:
+  explicit ActiveScriptController(content::WebContents* web_contents);
+  virtual ~ActiveScriptController();
+
+  // Notify the ActiveScriptController that an extension is running a script.
+  // TODO(rdevlin.cronin): Soon, this should be ask the user for permission,
+  // rather than simply notifying them.
+  void NotifyScriptExecuting(const std::string& extension_id);
+
+  // Returns true if there is an entry for the given |extension_id|.
+  bool HasActionForExtensionId(const std::string& extension_id);
+
+  // LocationBarControllerProvider implementation.
+  virtual std::vector<ExtensionAction*> GetCurrentActions() OVERRIDE;
+  virtual LocationBarController::Action OnClicked(
+      const Extension* extension,
+      LocationBarController::MouseButton button) OVERRIDE;
+  virtual void NavigatedMainFrame(
+      const content::LoadCommittedDetails& details,
+      const content::FrameNavigateParams& params) OVERRIDE;
+
+ private:
+  // Returns an extension action for the given |extension|, creating one if
+  // necessary.
+  ExtensionAction* GetActionForExtension(const Extension* extension);
+
+  // Whether or not the ActiveScriptController is enabled (corresponding to the
+  // kActiveScriptEnforcement switch). If it is not, it acts as an empty shell,
+  // always allowing scripts to run and never displaying actions.
+  bool enabled_;
+
+  // The extensions that have called ExecuteScript on the current frame.
+  std::set<std::string> extensions_executing_scripts_;
+
+  // Script badges that have been generated for extensions. This is both those
+  // with actions already declared that are copied and normalised, and actions
+  // that get generated for extensions that haven't declared anything.
+  typedef std::map<std::string, linked_ptr<ExtensionAction> > ActiveScriptMap;
+  ActiveScriptMap script_badges_;
+
+  DISALLOW_COPY_AND_ASSIGN(ActiveScriptController);
+};
+
+}  // namespace extensions
+
+#endif  // CHROME_BROWSER_EXTENSIONS_ACTIVE_SCRIPT_CONTROLLER_H_