add transport methods to update framework usage flags after transport initialization

Author: stephenarosaj

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

@@ -94,7 +94,7 @@ export class DataConnect {
   _isUsingGeneratedSdk: boolean = false;
   _isUsingTanStackSdk: boolean = false;
   _isUsingReactSdk: boolean = false;
-  _isUsingAngularFireSdk: boolean = false;
+  _isUsingAngularSdk: boolean = false;
   private _appCheckTokenProvider?: AppCheckTokenProvider;
   // @internal
   constructor(
@@ -124,17 +124,26 @@ export class DataConnect {
     if (!this._isUsingTanStackSdk) {
       this._isUsingTanStackSdk = true;
     }
+    if (this._transport) {
+      this._transport._useTanStack();
+    }
   }
   // @internal
   _useReactSdk(): void {
     if (!this._isUsingReactSdk) {
       this._isUsingReactSdk = true;
     }
+    if (this._transport) {
+      this._transport._useReact();
+    }
   }
   // @internal
-  _useAngularFireSdk(): void {
-    if (!this._isUsingAngularFireSdk) {
-      this._isUsingAngularFireSdk = true;
+  _useAngularSdk(): void {
+    if (!this._isUsingAngularSdk) {
+      this._isUsingAngularSdk = true;
+    }
+    if (this._transport) {
+      this._transport._useAngular();
     }
   }
   _delete(): Promise<void> {
@@ -188,7 +197,7 @@ export class DataConnect {
       this._isUsingGeneratedSdk,
       this._isUsingTanStackSdk,
       this._isUsingReactSdk,
-      this._isUsingAngularFireSdk
+      this._isUsingAngularSdk
     );
     if (this._transportOptions) {
       this._transport.useEmulator(

packages/data-connect/src/network/fetch.ts

@@ -33,7 +33,7 @@ function getGoogApiClientValue(_isUsingGen: boolean): string {
 function getWebFrameworkValue(
   _isUsingTanStack: boolean,
   _isUsingReact: boolean,
-  _isUsingAngularFire: boolean
+  _isUsingAngular: boolean
 ): string {
   let str = '';
   if (_isUsingTanStack) {
@@ -42,7 +42,7 @@ function getWebFrameworkValue(
   if (_isUsingReact) {
     str += ' react/';
   }
-  if (_isUsingAngularFire) {
+  if (_isUsingAngular) {
     str += ' angularfire/';
   }
   // no framework SDK used
@@ -66,7 +66,7 @@ export function dcFetch<T, U>(
   _isUsingGen: boolean,
   _isUsingTanStack: boolean,
   _isUsingReact: boolean,
-  _isUsingAngularFire: boolean
+  _isUsingAngular: boolean
 ): Promise<{ data: T; errors: Error[] }> {
   if (!connectFetch) {
     throw new DataConnectError(Code.OTHER, 'No Fetch Implementation detected!');
@@ -77,7 +77,7 @@ export function dcFetch<T, U>(
     'X-Firebase-DataConnect-Web-Frameworks': getWebFrameworkValue(
       _isUsingTanStack,
       _isUsingReact,
-      _isUsingAngularFire
+      _isUsingAngular
     )
   };
   if (accessToken) {

packages/data-connect/src/network/transport/index.ts

@@ -33,6 +33,9 @@ export interface DataConnectTransport {
   ): Promise<{ data: T; errors: Error[] }>;
   useEmulator(host: string, port?: number, sslEnabled?: boolean): void;
   onTokenChanged: (token: string | null) => void;
+  _useTanStack(): void;
+  _useReact(): void;
+  _useAngular(): void;
 }
 
 /**
@@ -48,5 +51,5 @@ export type TransportClass = new (
   _isUsingGen?: boolean,
   _isUsingTanStack?: boolean,
   _isUsingReact?: boolean,
-  _isUsingAngularFire?: boolean
+  _isUsingAngular?: boolean
 ) => DataConnectTransport;

packages/data-connect/src/network/transport/rest.ts

@@ -45,7 +45,7 @@ export class RESTTransport implements DataConnectTransport {
     transportOptions?: TransportOptions | undefined,
     private _isUsingGen = false,
     private _isUsingReact = false,
-    private _isUsingAngularFire = false,
+    private _isUsingAngular = false,
     private _isUsingTanStack = false
   ) {
     if (transportOptions) {
@@ -186,7 +186,7 @@ export class RESTTransport implements DataConnectTransport {
         this._isUsingGen,
         this._isUsingTanStack,
         this._isUsingReact,
-        this._isUsingAngularFire
+        this._isUsingAngular
       )
     );
     return withAuth;
@@ -214,9 +214,25 @@ export class RESTTransport implements DataConnectTransport {
         this._isUsingGen,
         this._isUsingTanStack,
         this._isUsingReact,
-        this._isUsingAngularFire
+        this._isUsingAngular
       );
     });
     return taskResult;
   };
+
+  _useTanStack(): void {
+    if (!this._isUsingTanStack) {
+      this._isUsingTanStack = true;
+    }
+  }
+  _useReact(): void {
+    if (!this._isUsingReact) {
+      this._isUsingReact = true;
+    }
+  }
+  _useAngular(): void {
+    if (!this._isUsingAngular) {
+      this._isUsingAngular = true;
+    }
+  }
 }