Addressed comments

Author: maneesht

packages/database/src/api/Database.ts

@@ -31,7 +31,8 @@ import {
   EmulatorMockTokenOptions,
   getDefaultEmulatorHostnameAndPort,
   isCloudWorkstation,
-  pingServer
+  pingServer,
+  updateEmulatorBanner
 } from '@firebase/util';
 
 import { AppCheckTokenProvider } from '../core/AppCheckTokenProvider';
@@ -56,6 +57,7 @@ import { TransportManager } from '../realtime/TransportManager';
 import { WebSocketConnection } from '../realtime/WebSocketConnection';
 
 import { ReferenceImpl } from './Reference_impl';
+import { updateStrings } from 'yargs';
 
 export { EmulatorMockTokenOptions } from '@firebase/util';
 /**
@@ -393,6 +395,7 @@ export function connectDatabaseEmulator(
   // Workaround to get cookies in Firebase Studio
   if (isCloudWorkstation(host)) {
     void pingServer(host);
+    updateEmulatorBanner('Database', true);
   }
 
   // Modify the repo to apply emulator settings

packages/database/src/core/Repo.ts

@@ -18,10 +18,12 @@
 import {
   assert,
   contains,
+  isCloudWorkstation,
   isEmpty,
   map,
   safeGet,
-  stringify
+  stringify,
+  updateEmulatorBanner
 } from '@firebase/util';
 
 import { ValueEventRegistration } from '../api/Reference_impl';
@@ -328,6 +330,9 @@ export function repoStart(
       repo.server_.unlisten(query, tag);
     }
   });
+  if (isCloudWorkstation(repo.repoInfo_.host)) {
+    updateEmulatorBanner('Database', repo.repoInfo_.isUsingEmulator);
+  }
 }
 
 /**

packages/functions/src/service.ts

@@ -30,7 +30,11 @@ 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, pingServer } from '@firebase/util';
+import {
+  isCloudWorkstation,
+  pingServer,
+  updateEmulatorBanner
+} from '@firebase/util';
 
 export const DEFAULT_REGION = 'us-central1';
 
@@ -182,6 +186,7 @@ export function connectFunctionsEmulator(
   // Workaround to get cookies in Firebase Studio
   if (useSsl) {
     void pingServer(functionsInstance.emulatorOrigin);
+    updateEmulatorBanner('Functions', true);
   }
 }
 
@@ -195,6 +200,7 @@ export function httpsCallable<RequestData, ResponseData, StreamData = unknown>(
   name: string,
   options?: HttpsCallableOptions
 ): HttpsCallable<RequestData, ResponseData, StreamData> {
+  updateEmulatorBanner('Functions', functionsInstance.emulatorOrigin !== null);
   const callable = (
     data?: RequestData | null
   ): Promise<HttpsCallableResult> => {
@@ -225,6 +231,7 @@ export function httpsCallableFromURL<
   url: string,
   options?: HttpsCallableOptions
 ): HttpsCallable<RequestData, ResponseData, StreamData> {
+  updateEmulatorBanner('Functions', functionsInstance.emulatorOrigin !== null);
   const callable = (
     data?: RequestData | null
   ): Promise<HttpsCallableResult> => {

packages/storage/src/service.ts

@@ -46,9 +46,11 @@ import {
   createMockUserToken,
   EmulatorMockTokenOptions,
   isCloudWorkstation,
-  pingServer
+  pingServer,
+  updateEmulatorBanner
 } from '@firebase/util';
 import { Connection, ConnectionType } from './implementation/connection';
+import { updateEmail } from '@firebase/auth';
 
 export function isUrl(path?: string): boolean {
   return /^[A-Za-z]+:\/\//.test(path as string);
@@ -150,6 +152,7 @@ export function connectStorageEmulator(
   // Workaround to get cookies in Firebase Studio
   if (useSsl) {
     void pingServer(`https://${storage.host}`);
+    updateEmulatorBanner('Storage', true);
   }
   storage._isUsingEmulator = true;
   storage._protocol = useSsl ? 'https' : 'http';
@@ -324,6 +327,7 @@ export class FirebaseStorageImpl implements FirebaseStorage {
     appCheckToken: string | null,
     retry = true
   ): Request<O> {
+    updateEmulatorBanner('Service', this._isUsingEmulator);
     if (!this._deleted) {
       const request = makeRequest(
         requestInfo,

packages/util/src/emulator.ts

@@ -141,10 +141,10 @@ export function createMockUserToken(
   ].join('.');
 }
 
-interface EmulatorStatuses {
+interface EmulatorStatusMap {
   [name: string]: boolean;
 }
-const emulatorStatus: EmulatorStatuses = {};
+const emulatorStatus: EmulatorStatusMap = {};
 
 interface EmulatorSummary {
   prod: string[];
@@ -197,6 +197,10 @@ export function updateEmulatorBanner(
   }
 
   emulatorStatus[name] = isRunningEmulator;
+
+  function prefixedId(id: string) {
+    return `__firebase__banner__${id}`;
+  }
   const bannerId = '__firebase__banner';
   const summary = getEmulatorSummary();
   const showError = summary.prod.length > 0;
@@ -210,9 +214,17 @@ export function updateEmulatorBanner(
 
   function setupDom(): void {
     const banner = getOrCreateEl(bannerId);
+    const firebaseTextId = prefixedId('text');
     const firebaseText: HTMLSpanElement =
-      document.getElementById('__firebase__text') ||
-      document.createElement('span');
+      document.getElementById(firebaseTextId) || document.createElement('span');
+    const learnMoreId = prefixedId('learnmore');
+    const learnMoreLink: HTMLAnchorElement =
+      (document.getElementById(learnMoreId) as HTMLAnchorElement) ||
+      document.createElement('a');
+    const prependIconId = prefixedId('preprendIcon');
+    const prependIcon =
+      document.getElementById(prependIconId) ||
+      document.createElementNS('http://www.w3.org/2000/svg', 'svg');
     if (banner.created) {
       // update styles
       const bannerEl = banner.element;
@@ -231,19 +243,45 @@ export function updateEmulatorBanner(
       closeBtn.onclick = () => {
         tearDown();
       };
+      learnMoreLink.setAttribute('id', learnMoreId);
+      learnMoreLink.href =
+        'http://firebase.google.com/docs/studio/deploy-app#emulator ';
+      bannerEl.appendChild(prependIcon);
       bannerEl.appendChild(firebaseText);
+      bannerEl.appendChild(learnMoreLink);
       bannerEl.appendChild(closeBtn);
+      prependIcon.setAttribute('width', '24');
+      prependIcon.setAttribute('id', prependIconId);
+      prependIcon.setAttribute('height', '24');
+      prependIcon.setAttribute('viewBox', '0 0 24 24');
+      prependIcon.setAttribute('fill', 'none');
       document.body.appendChild(bannerEl);
     }
+
     if (showError) {
-      banner.element.style.background = '#cd5c5c';
-      firebaseText.innerText = `Product${
-        summary.prod.length > 0 ? 's' : ''
-      } Running in Production: ${summary.prod.join(', ')}`;
+      firebaseText.innerText = `Running in Production`;
+      prependIcon.innerHTML = `<g clip-path="url(#clip0_6013_33858)">
+<path d="M4.8 17.6L12 5.6L19.2 17.6H4.8ZM6.91667 16.4H17.0833L12 7.93333L6.91667 16.4ZM12 15.6C12.1667 15.6 12.3056 15.5444 12.4167 15.4333C12.5389 15.3111 12.6 15.1667 12.6 15C12.6 14.8333 12.5389 14.6944 12.4167 14.5833C12.3056 14.4611 12.1667 14.4 12 14.4C11.8333 14.4 11.6889 14.4611 11.5667 14.5833C11.4556 14.6944 11.4 14.8333 11.4 15C11.4 15.1667 11.4556 15.3111 11.5667 15.4333C11.6889 15.5444 11.8333 15.6 12 15.6ZM11.4 13.6H12.6V10.4H11.4V13.6Z" fill="#212121"/>
+</g>
+<defs>
+<clipPath id="clip0_6013_33858">
+<rect width="24" height="24" fill="white"/>
+</clipPath>
+</defs>`;
     } else {
-      firebaseText.innerText = 'Running in this workspace';
+      prependIcon.innerHTML = `<g clip-path="url(#clip0_6083_34804)">
+<path d="M11.4 15.2H12.6V11.2H11.4V15.2ZM12 10C12.1667 10 12.3056 9.94444 12.4167 9.83333C12.5389 9.71111 12.6 9.56667 12.6 9.4C12.6 9.23333 12.5389 9.09444 12.4167 8.98333C12.3056 8.86111 12.1667 8.8 12 8.8C11.8333 8.8 11.6889 8.86111 11.5667 8.98333C11.4556 9.09444 11.4 9.23333 11.4 9.4C11.4 9.56667 11.4556 9.71111 11.5667 9.83333C11.6889 9.94444 11.8333 10 12 10ZM12 18.4C11.1222 18.4 10.2944 18.2333 9.51667 17.9C8.73889 17.5667 8.05556 17.1111 7.46667 16.5333C6.88889 15.9444 6.43333 15.2611 6.1 14.4833C5.76667 13.7056 5.6 12.8778 5.6 12C5.6 11.1111 5.76667 10.2833 6.1 9.51667C6.43333 8.73889 6.88889 8.06111 7.46667 7.48333C8.05556 6.89444 8.73889 6.43333 9.51667 6.1C10.2944 5.76667 11.1222 5.6 12 5.6C12.8889 5.6 13.7167 5.76667 14.4833 6.1C15.2611 6.43333 15.9389 6.89444 16.5167 7.48333C17.1056 8.06111 17.5667 8.73889 17.9 9.51667C18.2333 10.2833 18.4 11.1111 18.4 12C18.4 12.8778 18.2333 13.7056 17.9 14.4833C17.5667 15.2611 17.1056 15.9444 16.5167 16.5333C15.9389 17.1111 15.2611 17.5667 14.4833 17.9C13.7167 18.2333 12.8889 18.4 12 18.4ZM12 17.2C13.4444 17.2 14.6722 16.6944 15.6833 15.6833C16.6944 14.6722 17.2 13.4444 17.2 12C17.2 10.5556 16.6944 9.32778 15.6833 8.31667C14.6722 7.30555 13.4444 6.8 12 6.8C10.5556 6.8 9.32778 7.30555 8.31667 8.31667C7.30556 9.32778 6.8 10.5556 6.8 12C6.8 13.4444 7.30556 14.6722 8.31667 15.6833C9.32778 16.6944 10.5556 17.2 12 17.2Z" fill="#212121"/>
+</g>
+<defs>
+<clipPath id="clip0_6083_34804">
+<rect width="24" height="24" fill="white"/>
+</clipPath>
+</defs>`;
+      firebaseText.innerText = 'Using emulated backend';
+      learnMoreLink.href =
+        'https://firebase.google.com/docs/studio/solution-build-with-ai';
     }
-    firebaseText.setAttribute('id', '__firebase__text');
+    firebaseText.setAttribute('id', firebaseTextId);
   }
   if (document.readyState === 'loading') {
     window.addEventListener('DOMContentLoaded', setupDom);