Update Console domains for RTDB SDK (#4154)

Author: schmidt-sebastian

packages/database/src/realtime/BrowserPollConnection.ts

@@ -28,7 +28,7 @@ import { StatsManager } from '../core/stats/StatsManager';
 import { PacketReceiver } from './polling/PacketReceiver';
 import {
   APPLICATION_ID_PARAM,
-  FORGE_DOMAIN,
+  FORGE_DOMAIN_RE,
   FORGE_REF,
   LAST_SESSION_PARAM,
   LONG_POLLING,
@@ -222,8 +222,8 @@ export class BrowserPollConnection implements Transport {
       }
       if (
         typeof location !== 'undefined' &&
-        location.href &&
-        location.href.indexOf(FORGE_DOMAIN) !== -1
+        location.hostname &&
+        FORGE_DOMAIN_RE.test(location.hostname)
       ) {
         urlParams[REFERER_PARAM] = FORGE_REF;
       }

packages/database/src/realtime/Constants.ts

@@ -25,7 +25,9 @@ export const REFERER_PARAM = 'r';
 
 export const FORGE_REF = 'f';
 
-export const FORGE_DOMAIN = 'firebaseio.com';
+// Matches console.firebase.google.com, firebase-console-*.corp.google.com and
+// firebase.corp.google.com
+export const FORGE_DOMAIN_RE = /(console\.firebase|firebase-console-\w+\.corp|firebase\.corp)\.google\.com/;
 
 export const LAST_SESSION_PARAM = 'ls';
 

packages/database/src/realtime/WebSocketConnection.ts

@@ -16,16 +16,11 @@
  */
 
 import { RepoInfo } from '../core/RepoInfo';
-import {
-  assert,
-  jsonEval,
-  stringify,
-  isNodeSdk
-} from '@firebase/util';
+import { assert, jsonEval, stringify, isNodeSdk } from '@firebase/util';
 import { logWrapper, splitStringBySize } from '../core/util/util';
 import { StatsManager } from '../core/stats/StatsManager';
 import {
-  FORGE_DOMAIN,
+  FORGE_DOMAIN_RE,
   FORGE_REF,
   LAST_SESSION_PARAM,
   PROTOCOL_VERSION,
@@ -121,8 +116,8 @@ export class WebSocketConnection implements Transport {
     if (
       !isNodeSdk() &&
       typeof location !== 'undefined' &&
-      location.href &&
-      location.href.indexOf(FORGE_DOMAIN) !== -1
+      location.hostname &&
+      FORGE_DOMAIN_RE.test(location.hostname)
     ) {
       urlParams[REFERER_PARAM] = FORGE_REF;
     }