Create registerMessagingCompat.ts

Author: zwu52

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

@@ -0,0 +1,60 @@
+/**
+ * @license
+ * Copyright 2020 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import {
+  Component,
+  ComponentContainer,
+  ComponentType,
+  InstanceFactory
+} from '@firebase/component';
+import firebase, { _FirebaseNamespace } from '@firebase/app-compat';
+
+import { MessagingCompatImpl } from './messaging-compat';
+
+declare module '@firebase/component' {
+  interface NameServiceMapping {
+    'messaging-compat': MessagingCompatImpl;
+  }
+}
+
+const messagingCompatFactory: InstanceFactory<'messaging-compat'> = (
+  container: ComponentContainer
+) => {
+  if (self && 'ServiceWorkerGlobalScope' in self) {
+    // in sw
+    return new MessagingCompatImpl(
+      container.getProvider('app-compat').getImmediate(),
+      container.getProvider('messaging-sw-exp').getImmediate()
+    );
+  } else {
+    // in window
+    return new MessagingCompatImpl(
+      container.getProvider('app-compat').getImmediate(),
+      container.getProvider('messaging-exp').getImmediate()
+    );
+  }
+};
+
+export function registerMessagingCompat(): void {
+  (firebase as _FirebaseNamespace).INTERNAL.registerComponent(
+    new Component(
+      'messaging-compat',
+      messagingCompatFactory,
+      ComponentType.PUBLIC
+    )
+  );
+}