Begin to enable extension APIs in Extension Service Worker.

extensions/browser/extension_function_dispatcher.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 EXTENSIONS_BROWSER_EXTENSION_FUNCTION_DISPATCHER_H_
 #define EXTENSIONS_BROWSER_EXTENSION_FUNCTION_DISPATCHER_H_
 
 #include <map>
 #include <string>
 #include <vector>
@@ skipping 78 matching lines @@
   // - This object outlives any RenderFrameHost's passed to created
   //   ExtensionFunctions.
   explicit ExtensionFunctionDispatcher(
       content::BrowserContext* browser_context);
   ~ExtensionFunctionDispatcher();
 
   // Message handlers.
   // The response is sent to the corresponding render view in an
   // ExtensionMsg_Response message.
   void Dispatch(const ExtensionHostMsg_Request_Params& params,
-                content::RenderFrameHost* render_frame_host);
+                content::RenderFrameHost* render_frame_host,
+                int render_process_id);
 
   // Called when an ExtensionFunction is done executing, after it has sent
   // a response (if any) to the extension.
   void OnExtensionFunctionCompleted(const Extension* extension);
 
   // See the Delegate class for documentation on these methods.
   // TODO(devlin): None of these belong here. We should kill
   // ExtensionFunctionDispatcher::Delegate.
   WindowController* GetExtensionWindowController() const;
   content::WebContents* GetAssociatedWebContents() const;
   content::WebContents* GetVisibleWebContents() const;
 
   // The BrowserContext that this dispatcher is associated with.
   content::BrowserContext* browser_context() { return browser_context_; }
 
   void set_delegate(Delegate* delegate) { delegate_ = delegate; }
 
  private:
   // For a given RenderFrameHost instance, UIThreadResponseCallbackWrapper
   // creates ExtensionFunction::ResponseCallback instances which send responses
   // to the corresponding render view in ExtensionMsg_Response messages.
   // This class tracks the lifespan of the RenderFrameHost instance, and will be
   // destroyed automatically when it goes away.
   class UIThreadResponseCallbackWrapper;
 
+  // Same as UIThreadResponseCallbackWrapper above, but applies to an extension
+  // function from an extension Service Worker.
+  class UIThreadWorkerResponseCallbackWrapper;
+
+  // Key used to store UIThreadWorkerResponseCallbackWrapper in the map
+  // |ui_thread_response_callback_wrappers_for_worker_|.
+  struct WorkerResponseCallbackMapKey;
+
   // Helper to check whether an ExtensionFunction has the required permissions.
   // This should be called after the function is fully initialized.
   // If the check fails, |callback| is run with an access-denied error and false
   // is returned. |function| must not be run in that case.
   static bool CheckPermissions(
       ExtensionFunction* function,
       const ExtensionHostMsg_Request_Params& params,
       const ExtensionFunction::ResponseCallback& callback);
 
   // Helper to create an ExtensionFunction to handle the function given by
@@ skipping 10 matching lines @@
 
   // Helper to run the response callback with an access denied error. Can be
   // called on any thread.
   static void SendAccessDenied(
       const ExtensionFunction::ResponseCallback& callback,
       functions::HistogramValue histogram_value);
 
   void DispatchWithCallbackInternal(
       const ExtensionHostMsg_Request_Params& params,
       content::RenderFrameHost* render_frame_host,
+      int render_process_id,
       const ExtensionFunction::ResponseCallback& callback);
 
+  void RemoveWorkerCallbacksForProcess(int render_process_id);
+
   content::BrowserContext* browser_context_;
 
   Delegate* delegate_;
 
   // This map doesn't own either the keys or the values. When a RenderFrameHost
   // instance goes away, the corresponding entry in this map (if exists) will be
   // removed.
   typedef std::map<content::RenderFrameHost*, UIThreadResponseCallbackWrapper*>
       UIThreadResponseCallbackWrapperMap;
   UIThreadResponseCallbackWrapperMap ui_thread_response_callback_wrappers_;
+
+  using UIThreadWorkerResponseCallbackWrapperMap =
+      std::map<WorkerResponseCallbackMapKey,
+               std::unique_ptr<UIThreadWorkerResponseCallbackWrapper>>;
+  // TODO(lazyboy): The map entries are cleared upon RenderProcessHost shutown,
+  // we should really be clearing it on service worker shutdown.
+  UIThreadWorkerResponseCallbackWrapperMap
+      ui_thread_response_callback_wrappers_for_worker_;
 };
 
 }  // namespace extensions
 
 #endif  // EXTENSIONS_BROWSER_EXTENSION_FUNCTION_DISPATCHER_H_