[Extensions] Show a "refresh" bubble when needed with click-to-script

chrome/browser/extensions/extension_action_runner.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_ACTION_RUNNER_H_
 #define CHROME_BROWSER_EXTENSIONS_EXTENSION_ACTION_RUNNER_H_
 
 #include <stdint.h>
 
 #include <map>
 #include <set>
 #include <string>
 #include <vector>
 
 #include "base/callback.h"
 #include "base/compiler_specific.h"
 #include "base/macros.h"
+#include "base/memory/weak_ptr.h"
 #include "base/scoped_observer.h"
 #include "chrome/browser/extensions/extension_action.h"
+#include "chrome/browser/ui/toolbar/toolbar_actions_bar_bubble_delegate.h"
 #include "content/public/browser/web_contents_observer.h"
 #include "extensions/browser/blocked_action_type.h"
 #include "extensions/browser/extension_registry_observer.h"
 #include "extensions/common/permissions/permissions_data.h"
 #include "extensions/common/user_script.h"
 
 namespace content {
 class BrowserContext;
 class WebContents;
 }
@@ skipping 39 matching lines @@
   // Called when a webRequest event for the given |extension| was blocked.
   void OnWebRequestBlocked(const Extension* extension);
 
   // Returns a bitmask of BlockedActionType for the actions that have been
   // blocked for the given extension.
   int GetBlockedActions(const Extension* extension);
 
   // Returns true if the given |extension| has any blocked actions.
   bool WantsToRun(const Extension* extension);
 
+  // Runs any blocked actions the extension has, but does not handle any page
+  // refreshes for document_start/webRequest.
+  void RunForTesting(const Extension* extension);
+
   int num_page_requests() const { return num_page_requests_; }
 
+  void set_default_bubble_close_action_for_testing(
+      scoped_ptr<ToolbarActionsBarBubbleDelegate::CloseAction> action) {
+    default_bubble_close_action_for_testing_ = std::move(action);
+  }
+
 #if defined(UNIT_TEST)
   // Only used in tests.
   PermissionsData::AccessType RequiresUserConsentForScriptInjectionForTesting(
       const Extension* extension,
       UserScript::InjectionType type) {
     return RequiresUserConsentForScriptInjection(extension, type);
   }
   void RequestScriptInjectionForTesting(const Extension* extension,
                                         UserScript::RunLocation run_location,
                                         const base::Closure& callback) {
@@ skipping 43 matching lines @@
   // Grants permission for the given request to run.
   void PermitScriptInjection(int64_t request_id);
 
   // Notifies the ExtensionActionAPI of a change (either that an extension now
   // wants permission to run, or that it has been run).
   void NotifyChange(const Extension* extension);
 
   // Log metrics.
   void LogUMA() const;
 
+  // Shows the bubble to prompt the user to refresh the page to run the blocked
+  // actions for the given |extension|.
+  void ShowBlockedActionBubble(const Extension* extension);
+
+  // Called when the blocked actions bubble is closed.
+  void OnBlockedActionBubbleClosed(
+      const std::string& extension_id,
+      ToolbarActionsBarBubbleDelegate::CloseAction action);
+
   // content::WebContentsObserver implementation.
   bool OnMessageReceived(const IPC::Message& message,
                          content::RenderFrameHost* render_frame_host) override;
   void DidNavigateMainFrame(
       const content::LoadCommittedDetails& details,
       const content::FrameNavigateParams& params) override;
 
   // ExtensionRegistryObserver:
   void OnExtensionUnloaded(content::BrowserContext* browser_context,
                            const Extension* extension,
@@ skipping 16 matching lines @@
 
   // A set of ids for which the webRequest API was blocked on the page.
   std::set<std::string> web_request_blocked_;
 
   // The extensions which have been granted permission to run on the given page.
   // TODO(rdevlin.cronin): Right now, this just keeps track of extensions that
   // have been permitted to run on the page via this interface. Instead, it
   // should incorporate more fully with ActiveTab.
   std::set<std::string> permitted_extensions_;
 
+  // If true, ignore active tab being granted rather than running pending
+  // actions.
+  bool ignore_active_tab_granted_;
+
+  // If non-null, the bubble action to simulate for testing.
+  scoped_ptr<ToolbarActionsBarBubbleDelegate::CloseAction>
+      default_bubble_close_action_for_testing_;
+
   ScopedObserver<ExtensionRegistry, ExtensionRegistryObserver>
       extension_registry_observer_;
 
+  base::WeakPtrFactory<ExtensionActionRunner> weak_factory_;
+
   DISALLOW_COPY_AND_ASSIGN(ExtensionActionRunner);
 };
 
 }  // namespace extensions
 
 #endif  // CHROME_BROWSER_EXTENSIONS_EXTENSION_ACTION_RUNNER_H_