Add listeners registration to messaging-compat

Author: zwu52

common/api-review/messaging-exp.api.md

@@ -67,6 +67,9 @@ export function onBackgroundMessage(messaging: FirebaseMessaging, nextOrObserver
 // @public
 export function onMessage(messaging: FirebaseMessaging, nextOrObserver: NextFn<MessagePayload> | Observer<MessagePayload>): Unsubscribe;
 
+// @internal
+export function _registerListeners(messaging: FirebaseMessaging): void;
+
 export { Unsubscribe }
 
 

packages-exp/messaging-compat/src/messaging-compat.ts

@@ -22,6 +22,7 @@ import {
 import {
   FirebaseMessaging,
   MessagePayload,
+  _registerListeners,
   deleteToken,
   getToken,
   onBackgroundMessage,
@@ -88,22 +89,11 @@ function isSwSupported(): boolean {
 }
 
 export class MessagingCompatImpl implements MessagingCompat, _FirebaseService {
-  swRegistration?: ServiceWorkerRegistration;
-  vapidKey?: string;
-
-  onBackgroundMessageHandler:
-    | NextFn<MessagePayload>
-    | Observer<MessagePayload>
-    | null = null;
-
-  onMessageHandler:
-    | NextFn<MessagePayload>
-    | Observer<MessagePayload>
-    | null = null;
-
   constructor(readonly app: AppCompat, readonly _delegate: FirebaseMessaging) {
     this.app = app;
     this._delegate = _delegate;
+
+    _registerListeners(_delegate);
   }
 
   async getToken(options?: {

packages-exp/messaging-exp/src/helpers/register.ts

@@ -23,9 +23,17 @@ import {
 } from '@firebase/component';
 import { ERROR_FACTORY, ErrorCode } from '../util/errors';
 import { isSwSupported, isWindowSupported } from '../api/isSupported';
+import {
+  onNotificationClick,
+  onPush,
+  onSubChange
+} from '../listeners/sw-listeners';
 
+import { FirebaseMessaging } from '../interfaces/public-types';
 import { MessagingService } from '../messaging-service';
+import { ServiceWorkerGlobalScope } from '../util/sw-types';
 import { _registerComponent } from '@firebase/app-exp';
+import { messageEventListener } from '../listeners/window-listener';
 
 const WindowMessagingFactory: InstanceFactory<'messaging-exp'> = (
   container: ComponentContainer
@@ -86,3 +94,32 @@ export function registerMessagingInSw(): void {
     new Component('messaging-exp', SwMessagingFactory, ComponentType.PUBLIC)
   );
 }
+
+declare const self: ServiceWorkerGlobalScope;
+/**
+ * Conditionally registers the listeners. Theses registrations are done in `getMessagingInSw` and
+ * `getMessagingInWindow` for the v9 SDK. This method exists for `messaging-compat` because the
+ * injected messaging instance to `messaging-compat` isn't created through the `getMessaging`
+ * method. Thus the main package of v9 needs to export this method to support `messaging-compat`.
+ *
+ * @internal
+ */
+export function _registerListeners(messaging: FirebaseMessaging): void {
+  if (!!navigator) {
+    // in window
+    navigator.serviceWorker.addEventListener('message', e =>
+      messageEventListener(messaging as MessagingService, e)
+    );
+  } else {
+    // in sw
+    self.addEventListener('push', e => {
+      e.waitUntil(onPush(e, messaging as MessagingService));
+    });
+    self.addEventListener('pushsubscriptionchange', e => {
+      e.waitUntil(onSubChange(e, messaging as MessagingService));
+    });
+    self.addEventListener('notificationclick', e => {
+      e.waitUntil(onNotificationClick(e));
+    });
+  }
+}

packages-exp/messaging-exp/src/index.ts

@@ -33,6 +33,7 @@ export {
   getMessagingInWindow as getMessaging,
   onBackgroundMessage
 } from './api';
+export { _registerListeners } from './helpers/register';
 export { isWindowSupported as isSupported } from './api/isSupported';
 export * from './interfaces/public-types';