fix(firebase): add FirebaseUniversalAuthService

Author: patrickmichalina

src/modules/firebase/auth/app.auth.module.ts

@@ -1,17 +1,19 @@
 import { NgModule } from '@angular/core'
 import { AngularFireAuthModule } from 'angularfire2/auth'
+import { CookieService } from '../../cookies/browser'
+import { FirebaseUniversalAuthService } from './browser.auth.service'
 import {
   FIREBASE_AUTH_COOKIE_STO_KEY,
   FIREBASE_AUTH_COOKIE_FACTORY
 } from './tokens'
-import { CookieService } from '../../cookies/browser'
 
 // tslint:disable:no-this
 // tslint:disable-next-line:no-class
 @NgModule({
   imports: [AngularFireAuthModule],
   exports: [AngularFireAuthModule],
   providers: [
+    FirebaseUniversalAuthService,
     { provide: FIREBASE_AUTH_COOKIE_FACTORY, useClass: CookieService },
     { provide: FIREBASE_AUTH_COOKIE_STO_KEY, useValue: 'firebaseJWT' }
   ]

src/modules/firebase/auth/browser.auth.service.ts

@@ -0,0 +1,21 @@
+import { AngularFireAuth } from 'angularfire2/auth'
+import { Injectable, Inject } from '@angular/core'
+import { startWith } from 'rxjs/operators'
+import { TransferState, StateKey } from '@angular/platform-browser'
+import { FIREBASE_AUTH_OBJ_TS } from './tokens'
+
+// tslint:disable:no-this
+// tslint:disable-next-line:no-class
+@Injectable()
+export class FirebaseUniversalAuthService {
+  constructor(
+    private auth: AngularFireAuth,
+    private ts: TransferState,
+    @Inject(FIREBASE_AUTH_OBJ_TS) private tsKey: StateKey<any>
+  ) {}
+
+  readonly fbAuth = this.auth
+  readonly user = this.auth.user.pipe(
+    startWith(this.ts.get(this.tsKey, undefined))
+  )
+}

src/modules/firebase/auth/server.auth.module.ts

@@ -1,4 +1,4 @@
-import { NgModule } from '@angular/core'
+import { NgModule, APP_INITIALIZER } from '@angular/core'
 import { AngularFireAuth } from 'angularfire2/auth'
 import { FirebaseServerAuth } from './server.auth.service'
 import {
@@ -7,8 +7,11 @@ import {
 } from './server.common'
 import { initializeApp, credential, auth, apps } from 'firebase-admin'
 import { EnvironmentService } from '../../environment'
-import { FIREBASE_AUTH_COOKIE_STO_KEY } from './tokens'
+import { FIREBASE_AUTH_COOKIE_STO_KEY, FIREBASE_AUTH_OBJ_TS } from './tokens'
 import { CookieService } from '../../cookies/browser'
+import { TransferState, StateKey } from '@angular/platform-browser'
+import { take, tap } from 'rxjs/operators'
+import { FirebaseUniversalAuthService } from './browser.auth.service'
 
 function firebaseAdminAppAlreadyExists() {
   return apps.length ? true : false
@@ -35,10 +38,30 @@ export function getUserJwt(cs: CookieService, key: string) {
   return cs.get(key)
 }
 
+export function onBootstrap(
+  transferState: TransferState,
+  auth: FirebaseServerAuth,
+  tsKey: StateKey<string>
+) {
+  return () => {
+    return auth.user
+      .pipe(
+        take(1),
+        tap(user => transferState.set(tsKey, user))
+      )
+      .toPromise()
+  }
+}
+
 // tslint:disable-next-line:no-class
 @NgModule({
   providers: [
-    { provide: AngularFireAuth, useClass: FirebaseServerAuth },
+    FirebaseServerAuth,
+    { provide: AngularFireAuth, useExisting: FirebaseServerAuth },
+    {
+      provide: FirebaseUniversalAuthService,
+      useExisting: FirebaseServerAuth
+    },
     {
       provide: FIREBASE_AUTH_SERVER_ADMIN_APP,
       useFactory: fbAdminFactory,
@@ -48,6 +71,12 @@ export function getUserJwt(cs: CookieService, key: string) {
       provide: FIREBASE_AUTH_SERVER_USER_JWT,
       useFactory: getUserJwt,
       deps: [CookieService, FIREBASE_AUTH_COOKIE_STO_KEY]
+    },
+    {
+      provide: APP_INITIALIZER,
+      useFactory: onBootstrap,
+      deps: [TransferState, FirebaseServerAuth, FIREBASE_AUTH_OBJ_TS],
+      multi: true
     }
   ]
 })

src/modules/firebase/auth/server.auth.service.ts

@@ -1,13 +1,13 @@
 import { Injectable, Inject } from '@angular/core'
 import {
   FIREBASE_AUTH_SERVER_ADMIN_APP,
-  FIREBASE_AUTH_SERVER_USER_JWT,
-  FIREBASE_AUTH_OBJ_TS
+  FIREBASE_AUTH_SERVER_USER_JWT
 } from './server.common'
 import { auth } from 'firebase-admin'
 import { of } from 'rxjs'
 import { flatMap, catchError, map, tap } from 'rxjs/operators'
 import { TransferState } from '@angular/platform-browser'
+import { FIREBASE_AUTH_OBJ_TS } from './tokens'
 
 function validateToken(auth: auth.Auth, jwt: string) {
   return of(auth.verifyIdToken(jwt))

src/modules/firebase/auth/server.common.ts

@@ -1,12 +1,9 @@
 import { InjectionToken } from '@angular/core'
 import { auth } from 'firebase-admin'
-import { makeStateKey } from '@angular/platform-browser'
 
 export const FIREBASE_AUTH_SERVER_ADMIN_APP = new InjectionToken<auth.Auth>(
   'fng.fb.svr.auth.admin'
 )
 export const FIREBASE_AUTH_SERVER_USER_JWT = new InjectionToken<string>(
   'fng.fb.svr.auth.jwt'
 )
-
-export const FIREBASE_AUTH_OBJ_TS = makeStateKey('fng.fb.auth.ts')

src/modules/firebase/auth/tokens.ts

@@ -1,9 +1,11 @@
 import { ICookieGetSet } from './app.common'
 import { InjectionToken } from '@angular/core'
+import { makeStateKey } from '@angular/platform-browser'
 
 export const FIREBASE_AUTH_COOKIE_FACTORY = new InjectionToken<ICookieGetSet>(
   'fng.auth.ck.get.set'
 )
 export const FIREBASE_AUTH_COOKIE_STO_KEY = new InjectionToken<string>(
   'fng.auth.ck.sto'
 )
+export const FIREBASE_AUTH_OBJ_TS = makeStateKey('fng.fb.auth.ts')