feat: migrate hooks to useBFF hook

src/components/app/data/hooks/useBFF.js

@@ -1,9 +1,7 @@
 import { useLocation, useParams } from 'react-router-dom';
 import { useQuery } from '@tanstack/react-query';
 import { logError } from '@edx/frontend-platform/logging';
-import useEnterpriseCustomer from './useEnterpriseCustomer';
 import { resolveBFFQuery } from '../queries';
-import useEnterpriseFeatures from './useEnterpriseFeatures';
 
 /**
  * Uses the route to determine which API call to make for the BFF
@@ -19,18 +17,12 @@ export default function useBFF({
   bffQueryOptions = {},
   fallbackQueryConfig = null,
 }) {
-  const { data: enterpriseCustomer } = useEnterpriseCustomer();
-  const { data: enterpriseFeatures } = useEnterpriseFeatures();
   const { enterpriseSlug } = useParams();
   const location = useLocation();
 
   // Determine the BFF query to use based on the current location
   const matchedBFFQuery = resolveBFFQuery(
     location.pathname,
-    {
-      enterpriseCustomerUuid: enterpriseCustomer.uuid,
-      enterpriseFeatures,
-    },
   );
 
   // Determine which query to call, the original hook or the new BFF

src/components/app/data/hooks/useEnterpriseCustomer.js

@@ -9,14 +9,15 @@ export default function useEnterpriseCustomer(queryOptions = {}) {
   const { select, ...queryOptionsRest } = queryOptions;
   return useEnterpriseLearner({
     ...queryOptionsRest,
-    select: (enterpriseLearner) => {
+    select: (data) => {
+      const transformedData = data.enterpriseCustomer || data.transformed.enterpriseCustomer;
       if (select) {
         return select({
-          original: enterpriseLearner,
-          transformed: enterpriseLearner.enterpriseCustomer,
+          original: data,
+          transformed: transformedData,
         });
       }
-      return enterpriseLearner.enterpriseCustomer;
+      return transformedData;
     },
   });
 }

src/components/app/data/hooks/useEnterpriseFeatures.js

@@ -5,13 +5,14 @@ export default function useEnterpriseFeatures(queryOptions = {}) {
   return useEnterpriseLearner({
     ...queryOptionsRest,
     select: (data) => {
+      const transformedData = data.enterpriseFeatures || data.transformed.enterpriseFeatures;
       if (select) {
         return select({
           original: data,
-          transformed: data?.enterpriseFeatures,
+          transformed: transformedData,
         });
       }
-      return data?.enterpriseFeatures;
+      return transformedData;
     },
   });
 }

src/components/app/data/hooks/useEnterpriseLearner.js

@@ -1,8 +1,8 @@
 import { AppContext } from '@edx/frontend-platform/react';
-import { useQuery } from '@tanstack/react-query';
 import { useContext } from 'react';
 import { useParams } from 'react-router-dom';
 import { queryEnterpriseLearner } from '../queries';
+import useBFF from './useBFF';
 
 /**
  * Retrieves the enterprise learner data for the authenticated user.
@@ -12,8 +12,32 @@ import { queryEnterpriseLearner } from '../queries';
 export default function useEnterpriseLearner(queryOptions = {}) {
   const { authenticatedUser } = useContext(AppContext);
   const { enterpriseSlug } = useParams();
-  return useQuery({
-    ...queryEnterpriseLearner(authenticatedUser.username, enterpriseSlug),
-    ...queryOptions,
+  const { select, ...queryOptionsRest } = queryOptions;
+
+  return useBFF({
+    bffQueryOptions: {
+      ...queryOptionsRest,
+      select: (data) => {
+        const transformedData = {
+          enterpriseCustomer: data.enterpriseCustomer,
+          allLinkedEnterpriseCustomerUsers: data.allLinkedEnterpriseCustomerUsers,
+          enterpriseFeatures: data.enterpriseFeatures,
+        };
+        // When custom `select` function is provided in `queryOptions`, call it with original and transformed data.
+        if (select) {
+          return select({
+            original: data,
+            transformed: transformedData,
+          });
+        }
+
+        // Otherwise, return the transformed data.
+        return transformedData;
+      },
+    },
+    fallbackQueryConfig: {
+      ...queryEnterpriseLearner(authenticatedUser.username, enterpriseSlug),
+      ...queryOptions,
+    },
   });
 }