Add SDK Tracking

packages/react/src/data-connect/types.ts

@@ -13,3 +13,12 @@ export type FlattenedMutationResult<Data, Variables> = Omit<
   "data" | "toJSON"
 > &
   Data;
+
+export enum CallerSdkType {
+  Base, // Core JS SDK
+  Generated, // Generated JS SDK
+  TanstackReactCore, // Tanstack non-generated React SDK
+  GeneratedReact, // Generated React SDK
+  TanstackAngularCore, // Tanstack non-generated Angular SDK
+  GeneratedAngular // Generated Angular SDK
+}

packages/react/src/data-connect/useDataConnectMutation.ts

@@ -10,7 +10,7 @@
   type QueryRef,
   executeMutation,
 } from "firebase/data-connect";
-import type { FlattenedMutationResult } from "./types";
+import { CallerSdkType, type FlattenedMutationResult } from "./types";
 
 export type useDataConnectMutationOptions<
   TData = unknown,
@@ -48,10 +48,10 @@
     FlattenedMutationResult<Data, Variables>,
     FirebaseError,
     Variables
-  >
+  >,
+  _callerSdkType: CallerSdkType = CallerSdkType.TanstackReactCore
 ) {
   const queryClient = useQueryClient();
-
   return useMutation<
     FlattenedMutationResult<Data, Variables>,
     FirebaseError,
@@ -78,6 +78,9 @@
     },
     mutationFn: async (variables) => {
       const mutationRef = typeof ref === "function" ? ref(variables) : ref;
+
+      // @ts-expect-error function is hidden under `DataConnect`.
+      mutationRef.dataConnect._setCallerSdkType(_callerSdkType);
       const response = await executeMutation<Data, Variables>(mutationRef);
 
       return {

packages/react/src/data-connect/useDataConnectQuery.ts

@@ -6,7 +6,8 @@ import {
   executeQuery,
 } from "firebase/data-connect";
 import type { PartialBy } from "../../utils";
-import type { FlattenedQueryResult } from "./types";
+import { CallerSdkType, type FlattenedQueryResult } from "./types";
+
 
 export type useDataConnectQueryOptions<
   TData = unknown,
@@ -19,6 +20,7 @@ export function useDataConnectQuery<Data = unknown, Variables = unknown>(
     FlattenedQueryResult<Data, Variables>,
     FirebaseError
   >,
+  _callerSdkType: CallerSdkType = CallerSdkType.TanstackReactCore
 ) {
   let queryRef: QueryRef<Data, Variables>;
   let initialData: FlattenedQueryResult<Data, Variables> | undefined;
@@ -34,7 +36,8 @@ export function useDataConnectQuery<Data = unknown, Variables = unknown>(
   } else {
     queryRef = refOrResult;
   }
-
+  // @ts-expect-error function is hidden under `DataConnect`.
+  queryRef.dataConnect._setCallerSdkType(_callerSdkType);
   return useQuery<FlattenedQueryResult<Data, Variables>, FirebaseError>({
     ...options,
     initialData,