console build for rtdb (#3709)

* console build for rtdb

* Make initStandalone for console

* move admin specific logic to the wrapper function for admin sdk

* fix lint error

Author: Feiyang1

packages/database/index.node.ts

@@ -30,13 +30,8 @@ import { setSDKVersion } from './src/core/version';
 import { CONSTANTS, isNodeSdk } from '@firebase/util';
 import { setWebSocketImpl } from './src/realtime/WebSocketConnection';
 import { Client } from 'faye-websocket';
-import {
-  Component,
-  ComponentType,
-  Provider,
-  ComponentContainer
-} from '@firebase/component';
-import { FirebaseAuthInternalName } from '@firebase/auth-interop-types';
+import { Component, ComponentType } from '@firebase/component';
+import { FirebaseAuthInternal } from '@firebase/auth-interop-types';
 
 import { name, version } from './package.json';
 
@@ -46,7 +41,7 @@ const ServerValue = Database.ServerValue;
 
 /**
  * A one off register function which returns a database based on the app and
- * passed database URL.
+ * passed database URL. (Used by the Admin SDK)
  *
  * @param app A valid FirebaseApp-like object
  * @param url A valid Firebase databaseURL
@@ -57,42 +52,16 @@ export function initStandalone(
   app: FirebaseApp,
   url: string,
   version: string,
-  nodeAdmin: boolean = true
+  nodeAdmin = true
 ) {
-  /**
-   * This should allow the firebase-admin package to provide a custom version
-   * to the backend
-   */
   CONSTANTS.NODE_ADMIN = nodeAdmin;
-  setSDKVersion(version);
-
-  /**
-   * Create a 'auth-internal' component using firebase-admin-node's implementation
-   * that implements FirebaseAuthInternal.
-   * ComponentContainer('database-admin') is just a placeholder that doesn't perform
-   * any actual function.
-   */
-  const authProvider = new Provider<FirebaseAuthInternalName>(
-    'auth-internal',
-    new ComponentContainer('database-admin')
-  );
-  authProvider.setComponent(
-    new Component(
-      'auth-internal',
-      // firebase-admin-node's app.INTERNAL implements FirebaseAuthInternal interface
-      // eslint-disable-next-line @typescript-eslint/no-explicit-any
-      () => (app as any).INTERNAL,
-      ComponentType.PRIVATE
-    )
-  );
-
-  return {
-    instance: RepoManager.getInstance().databaseFromApp(
-      app,
-      authProvider,
-      url,
-      nodeAdmin
-    ) as types.Database,
+  return INTERNAL.initStandalone({
+    app,
+    url,
+    version,
+    // firebase-admin-node's app.INTERNAL implements FirebaseAuthInternal interface
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    customAuthImpl: (app as any).INTERNAL as FirebaseAuthInternal,
     namespace: {
       Reference,
       Query,
@@ -102,8 +71,9 @@ export function initStandalone(
       INTERNAL,
       ServerValue,
       TEST_ACCESS
-    }
-  };
+    },
+    nodeAdmin
+  });
 }
 
 export function registerDatabase(instance: FirebaseNamespace) {

packages/database/src/api/internal.ts

@@ -18,6 +18,20 @@
 import { WebSocketConnection } from '../realtime/WebSocketConnection';
 import { BrowserPollConnection } from '../realtime/BrowserPollConnection';
 import { Reference } from './Reference';
+import { RepoManager } from '../core/RepoManager';
+import { setSDKVersion } from '../core/version';
+import { FirebaseApp } from '@firebase/app-types';
+import {
+  FirebaseAuthInternal,
+  FirebaseAuthInternalName
+} from '@firebase/auth-interop-types';
+import * as types from '@firebase/database-types';
+import {
+  Provider,
+  ComponentContainer,
+  Component,
+  ComponentType
+} from '@firebase/component';
 
 /**
  * INTERNAL methods for internal-use only (tests, etc.).
@@ -67,3 +81,55 @@ export const interceptServerData = function (
 ) {
   return ref.repo.interceptServerData_(callback);
 };
+
+/**
+ * Used by console to create a database based on the app,
+ * passed database URL and a custom auth implementation.
+ *
+ * @param app A valid FirebaseApp-like object
+ * @param url A valid Firebase databaseURL
+ * @param version custom version e.g. firebase-admin version
+ * @param customAuthImpl custom auth implementation
+ */
+export function initStandalone<T>({
+  app,
+  url,
+  version,
+  customAuthImpl,
+  namespace,
+  nodeAdmin = false
+}: {
+  app: FirebaseApp;
+  url: string;
+  version: string;
+  customAuthImpl: FirebaseAuthInternal;
+  namespace: T;
+  nodeAdmin?: boolean;
+}): {
+  instance: types.Database;
+  namespace: T;
+} {
+  setSDKVersion(version);
+
+  /**
+   * ComponentContainer('database-standalone') is just a placeholder that doesn't perform
+   * any actual function.
+   */
+  const authProvider = new Provider<FirebaseAuthInternalName>(
+    'auth-internal',
+    new ComponentContainer('database-standalone')
+  );
+  authProvider.setComponent(
+    new Component('auth-internal', () => customAuthImpl, ComponentType.PRIVATE)
+  );
+
+  return {
+    instance: RepoManager.getInstance().databaseFromApp(
+      app,
+      authProvider,
+      url,
+      nodeAdmin
+    ) as types.Database,
+    namespace
+  };
+}