Added cookie support in firebase studio

Author: maneesht

common/api-review/util.api.md

@@ -474,6 +474,11 @@ export interface Subscribe<T> {
     (observer: PartialObserver<T>): Unsubscribe;
 }
 
+// Warning: (ae-missing-release-tag) "testConnectionAlive" is exported by the package, but it is missing a release tag (@alpha, @beta, @public, or @internal)
+//
+// @public (undocumented)
+export function testConnectionAlive(endpoint: string): Promise<boolean>;
+
 // Warning: (ae-missing-release-tag) "Unsubscribe" is exported by the package, but it is missing a release tag (@alpha, @beta, @public, or @internal)
 //
 // @public (undocumented)

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

@@ -18,7 +18,11 @@ import { Auth } from '../../model/public_types';
 import { AuthErrorCode } from '../errors';
 import { _assert } from '../util/assert';
 import { _castAuth } from './auth_impl';
-import { deepEqual } from '@firebase/util';
+import {
+  deepEqual,
+  isCloudWorkstation,
+  testConnectionAlive
+} from '@firebase/util';
 
 /**
  * Changes the {@link Auth} instance to communicate with the Firebase Auth Emulator, instead of production
@@ -100,6 +104,11 @@ export function connectAuthEmulator(
   if (!disableWarnings) {
     emitEmulatorWarning();
   }
+
+  if (isCloudWorkstation(host)) {
+    // Workaround to get cookies in Firebase Studio
+    testConnectionAlive(`${protocol}//${host}:${port}`).then().catch();
+  }
 }
 
 function extractProtocol(url: string): string {

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

@@ -42,6 +42,7 @@ import {
 import { RESTTransport } from '../network/transport/rest';
 
 import { MutationManager } from './Mutation';
+import { isCloudWorkstation, testConnectionAlive } from '@firebase/util';
 
 /**
  * Connector Config for calling Data Connect backend.
@@ -237,6 +238,9 @@ export function connectDataConnectEmulator(
   port?: number,
   sslEnabled = false
 ): void {
+  if (isCloudWorkstation(host)) {
+    testConnectionAlive(`https://${host}${port ? `:${port}` : ''}`);
+  }
   dc.enableEmulator({ host, port, sslEnabled });
 }
 

packages/firestore/src/lite-api/database.ts

@@ -27,7 +27,8 @@ import {
   deepEqual,
   EmulatorMockTokenOptions,
   getDefaultEmulatorHostnameAndPort,
-  isCloudWorkstation
+  isCloudWorkstation,
+  testConnectionAlive
 } from '@firebase/util';
 
 import {
@@ -333,6 +334,9 @@ export function connectFirestoreEmulator(
     emulatorOptions: firestore._getEmulatorOptions()
   };
   const newHostSetting = `${host}:${port}`;
+  if (useSsl) {
+    testConnectionAlive(`https://${newHostSetting}`);
+  }
   if (settings.host !== DEFAULT_HOST && settings.host !== newHostSetting) {
     logWarn(
       'Host has been set in both settings() and connectFirestoreEmulator(), emulator host ' +

packages/functions/src/service.ts

@@ -30,7 +30,7 @@ import { Provider } from '@firebase/component';
 import { FirebaseAuthInternalName } from '@firebase/auth-interop-types';
 import { MessagingInternalComponentName } from '@firebase/messaging-interop-types';
 import { AppCheckInternalComponentName } from '@firebase/app-check-interop-types';
-import { isCloudWorkstation } from '@firebase/util';
+import { isCloudWorkstation, testConnectionAlive } from '@firebase/util';
 
 export const DEFAULT_REGION = 'us-central1';
 
@@ -179,6 +179,9 @@ export function connectFunctionsEmulator(
   functionsInstance.emulatorOrigin = `http${
     useSsl ? 's' : ''
   }://${host}:${port}`;
+  if (useSsl) {
+    testConnectionAlive(functionsInstance.emulatorOrigin);
+  }
 }
 
 /**

packages/storage/src/service.ts

@@ -45,7 +45,8 @@ import { FirebaseStorage } from './public-types';
 import {
   createMockUserToken,
   EmulatorMockTokenOptions,
-  isCloudWorkstation
+  isCloudWorkstation,
+  testConnectionAlive
 } from '@firebase/util';
 import { Connection, ConnectionType } from './implementation/connection';
 
@@ -146,6 +147,9 @@ export function connectStorageEmulator(
 ): void {
   storage.host = `${host}:${port}`;
   const useSsl = isCloudWorkstation(host);
+  if (useSsl) {
+    testConnectionAlive(`https://${storage.host}`);
+  }
   storage._isUsingEmulator = true;
   storage._protocol = useSsl ? 'https' : 'http';
   const { mockUserToken } = options;

packages/util/src/url.ts

@@ -22,3 +22,8 @@
 export function isCloudWorkstation(host: string): boolean {
   return host.endsWith('.cloudworkstations.dev');
 }
+
+export async function testConnectionAlive(endpoint: string): Promise<boolean> {
+  const result = await fetch(endpoint);
+  return result.ok;
+}