[extension SW]: Support event listener registration and event dispatching.

extensions/browser/extension_message_filter.cc

 // 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.
 
 #include "extensions/browser/extension_message_filter.h"
 
 #include "base/macros.h"
 #include "base/memory/singleton.h"
 #include "components/crx_file/id_util.h"
 #include "components/keyed_service/content/browser_context_keyed_service_shutdown_notifier_factory.h"
@@ skipping 124 matching lines @@
     IPC_MESSAGE_HANDLER(ExtensionHostMsg_WakeEventPage,
                         OnExtensionWakeEventPage)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
   return handled;
 }
 
 void ExtensionMessageFilter::OnExtensionAddListener(
     const std::string& extension_id,
     const GURL& listener_url,
-    const std::string& event_name) {
+    const std::string& event_name,
+    int worker_thread_id) {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
   if (!browser_context_)
     return;
 
   RenderProcessHost* process = RenderProcessHost::FromID(render_process_id_);
   if (!process)
     return;
 
   if (crx_file::id_util::IdIsValid(extension_id)) {
-    GetEventRouter()->AddEventListener(event_name, process, extension_id);
+    if (worker_thread_id != kNonWorkerThreadId) {
+      GetEventRouter()->AddServiceWorkerEventListener(

[Devlin, 2017/06/01 04:54:50]

nit: maybe cache EventRouter, here and below.  It's not more efficient (each
branch is only one lookup), but I think it improves readability.  I don't feel
super strongly though.

[lazyboy, 2017/06/01 23:33:29]

Done.

+          event_name, process, extension_id, worker_thread_id);
+    } else {
+      GetEventRouter()->AddEventListener(event_name, process, extension_id);
+    }
   } else if (listener_url.is_valid()) {
     GetEventRouter()->AddEventListenerForURL(event_name, process, listener_url);
   } else {
     NOTREACHED() << "Tried to add an event listener without a valid "
                  << "extension ID nor listener URL";
   }
 }
 
 void ExtensionMessageFilter::OnExtensionRemoveListener(
     const std::string& extension_id,
     const GURL& listener_url,
-    const std::string& event_name) {
+    const std::string& event_name,
+    int worker_thread_id) {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
   if (!browser_context_)
     return;
 
   RenderProcessHost* process = RenderProcessHost::FromID(render_process_id_);
   if (!process)
     return;
 
   if (crx_file::id_util::IdIsValid(extension_id)) {
-    GetEventRouter()->RemoveEventListener(event_name, process, extension_id);
+    if (worker_thread_id != kNonWorkerThreadId) {
+      GetEventRouter()->RemoveServiceWorkerEventListener(
+          event_name, process, extension_id, worker_thread_id);
+    } else {
+      GetEventRouter()->RemoveEventListener(event_name, process, extension_id);
+    }
   } else if (listener_url.is_valid()) {
     GetEventRouter()->RemoveEventListenerForURL(event_name, process,
                                                 listener_url);
   } else {
     NOTREACHED() << "Tried to remove an event listener without a valid "
                  << "extension ID nor listener URL";
   }
 }
 
 void ExtensionMessageFilter::OnExtensionAddLazyListener(
-    const std::string& extension_id, const std::string& event_name) {
+    const std::string& extension_id,
+    const std::string& event_name,
+    int worker_thread_id) {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
   if (!browser_context_)
     return;
 
-  GetEventRouter()->AddLazyEventListener(event_name, extension_id);
+  if (worker_thread_id == kNonWorkerThreadId) {
+    GetEventRouter()->AddLazyEventListener(event_name, extension_id);
+  } else {
+    GetEventRouter()->AddLazyServiceWorkerEventListener(
+        event_name, extension_id, worker_thread_id);
+  }
 }
 
 void ExtensionMessageFilter::OnExtensionRemoveLazyListener(
-    const std::string& extension_id, const std::string& event_name) {
+    const std::string& extension_id,
+    const std::string& event_name,
+    int worker_thread_id) {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
   if (!browser_context_)
     return;
 
-  GetEventRouter()->RemoveLazyEventListener(event_name, extension_id);
+  if (worker_thread_id == kNonWorkerThreadId) {
+    GetEventRouter()->RemoveLazyEventListener(event_name, extension_id);
+  } else {
+    GetEventRouter()->RemoveLazyServiceWorkerEventListener(
+        event_name, extension_id, worker_thread_id);
+  }
 }
 
 void ExtensionMessageFilter::OnExtensionAddFilteredListener(
     const std::string& extension_id,
     const std::string& event_name,
     const base::DictionaryValue& filter,
     bool lazy) {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
   if (!browser_context_)
     return;
@@ skipping 99 matching lines @@
   // The extension has no background page, so there is nothing to wake.
   SendWakeEventPageResponse(request_id, false);
 }
 
 void ExtensionMessageFilter::SendWakeEventPageResponse(int request_id,
                                                        bool success) {
   Send(new ExtensionMsg_WakeEventPageResponse(request_id, success));
 }
 
 }  // namespace extensions