Updated comments

Author: maneesht

packages/auth/src/core/auth/emulator.ts

@@ -101,8 +101,8 @@ export function connectAuthEmulator(
     emitEmulatorWarning();
   }
 
+  // Workaround to get cookies in Firebase Studio
   if (isCloudWorkstation(host)) {
-    // Workaround to get cookies in Firebase Studio
     void pingServer(`${protocol}//${host}:${port}`);
   }
 }

packages/data-connect/src/api/DataConnect.ts

@@ -238,6 +238,7 @@ export function connectDataConnectEmulator(
   port?: number,
   sslEnabled = false
 ): void {
+  // Workaround to get cookies in Firebase Studio
   if (isCloudWorkstation(host)) {
     void pingServer(`https://${host}${port ? `:${port}` : ''}`);
   }

packages/database/src/api/Database.ts

@@ -390,6 +390,7 @@ export function connectDatabaseEmulator(
     tokenProvider = new EmulatorTokenProvider(token);
   }
 
+  // Workaround to get cookies in Firebase Studio
   if (isCloudWorkstation(host)) {
     void pingServer(host);
   }

packages/firestore/src/api/database.ts

@@ -21,7 +21,12 @@ import {
   FirebaseApp,
   getApp
 } from '@firebase/app';
-import { deepEqual, getDefaultEmulatorHostnameAndPort } from '@firebase/util';
+import {
+  deepEqual,
+  getDefaultEmulatorHostnameAndPort,
+  isCloudWorkstation,
+  pingServer
+} from '@firebase/util';
 
 import { User } from '../auth/user';
 import {
@@ -194,6 +199,11 @@ export function initializeFirestore(
     );
   }
 
+  // Workaround to get cookies in Firebase Studio
+  if (settings.host && isCloudWorkstation(settings.host)) {
+    void pingServer(settings.host);
+  }
+
   return provider.initialize({
     options: settings,
     instanceIdentifier: databaseId

packages/functions/src/service.ts

@@ -179,6 +179,7 @@ export function connectFunctionsEmulator(
   functionsInstance.emulatorOrigin = `http${
     useSsl ? 's' : ''
   }://${host}:${port}`;
+  // Workaround to get cookies in Firebase Studio
   if (useSsl) {
     void pingServer(functionsInstance.emulatorOrigin);
   }

packages/storage/src/service.ts

@@ -147,6 +147,7 @@ export function connectStorageEmulator(
 ): void {
   storage.host = `${host}:${port}`;
   const useSsl = isCloudWorkstation(host);
+  // Workaround to get cookies in Firebase Studio
   if (useSsl) {
     void pingServer(`https://${storage.host}`);
   }