feat(auth): expose internal utils for SSR

Author: posva

src/auth/index.ts

@@ -77,6 +77,7 @@ export function VueFireAuth(initialUser?: _Nullable<User>): VueFireModule {
 /**
  * Key to be used to inject the auth instance into components. It allows avoiding to call `getAuth()`, which isn't tree
  * shakable.
+ * @internal
  */
 export const _VueFireAuthKey = Symbol('VueFireAuth')
 
@@ -91,17 +92,37 @@ export function VueFireAuthWithDependencies({
   initialUser,
 }: VueFireAuthOptions): VueFireModule {
   return (firebaseApp: FirebaseApp, app: App) => {
-    const user = getGlobalScope(firebaseApp, app).run(() =>
-      ref<_Nullable<User>>(initialUser)
-    )!
-    // this should only be on client
-    authUserMap.set(firebaseApp, user)
-    const auth = initializeAuth(firebaseApp, dependencies)
-    app.provide(_VueFireAuthKey, auth)
+    const [user, auth] = _VueFireAuthInit(
+      firebaseApp,
+      app,
+      initialUser,
+      dependencies
+    )
     setupOnAuthStateChanged(user, auth)
   }
 }
 
+/**
+ * initializes auth for both the server and client.
+ * @internal
+ */
+export function _VueFireAuthInit(
+  firebaseApp: FirebaseApp,
+  app: App,
+  initialUser: _Nullable<User>,
+  dependencies: AuthDependencies
+) {
+  const user = getGlobalScope(firebaseApp, app).run(() =>
+    ref<_Nullable<User>>(initialUser)
+  )!
+  // TODO: Is it okay to have it both server and client?
+  authUserMap.set(firebaseApp, user)
+  const auth = initializeAuth(firebaseApp, dependencies)
+  app.provide(_VueFireAuthKey, auth)
+
+  return [user, auth] as const
+}
+
 /**
  * Retrieves the Firebase Auth instance. **Returns `null` on the server**. When using this function on the client in
  * TypeScript, you can force the type with `useFirebaseAuth()!`.

src/index.ts

@@ -90,7 +90,9 @@ export {
   type VueFireAuthOptions,
   VueFireAuth,
   VueFireAuthWithDependencies,
+  _VueFireAuthInit,
   useFirebaseAuth,
+  _VueFireAuthKey,
   getCurrentUser,
   updateCurrentUserProfile,
 } from './auth'

src/server/auth.ts

@@ -1,5 +1,5 @@
 import type { FirebaseApp } from 'firebase/app'
-import type { User } from 'firebase/auth'
+import { type User } from 'firebase/auth'
 import type { DecodedIdToken, UserRecord } from 'firebase-admin/auth'
 import { App, ref } from 'vue'
 import { authUserMap, _setInitialUser } from '../auth/user'
@@ -9,6 +9,7 @@ import type { App as AdminApp } from 'firebase-admin/app'
 import { getAuth as getAdminAuth } from 'firebase-admin/auth'
 import { logger } from './logging'
 import { isFirebaseError } from './utils'
+import { _VueFireAuthKey } from '../auth'
 
 // MUST be named `__session` to be kept in Firebase context, therefore this name is hardcoded
 // https://firebase.google.com/docs/hosting/manage-cache#using_cookies
@@ -21,10 +22,10 @@ export const AUTH_COOKIE_NAME = '__session'
 export function VueFireAuthServer(
   firebaseApp: FirebaseApp,
   app: App<unknown>,
-  userRecord: _Nullable<User>
+  initialUser: _Nullable<User>
 ) {
   const user = getGlobalScope(firebaseApp, app).run(() =>
-    ref<_Nullable<User>>(userRecord)
+    ref<_Nullable<User>>(initialUser)
   )!
   authUserMap.set(firebaseApp, user)
   _setInitialUser(firebaseApp, user)