Remove useAssignment hook

Author: jcobis

packages/compass-telemetry/src/experimentation-provider.tsx

@@ -1,14 +1,7 @@
-import React, { createContext, useContext, useRef } from 'react';
+import React, { createContext, useRef } from 'react';
 import type { types } from '@mongodb-js/mdb-experiment-js';
-import type { typesReact } from '@mongodb-js/mdb-experiment-js/react';
 import type { ExperimentTestName } from './growth-experiments';
 
-type UseAssignmentHook = (
-  experimentName: ExperimentTestName,
-  trackIsInSample: boolean,
-  options?: typesReact.UseAssignmentOptions<types.TypeData>
-) => typesReact.UseAssignmentResponse<types.TypeData>;
-
 type AssignExperimentFn = (
   experimentName: ExperimentTestName,
   options?: types.AssignOptions<string>
@@ -21,22 +14,11 @@ type GetAssignmentFn = (
 ) => Promise<types.SDKAssignment<ExperimentTestName, string> | null>;
 
 interface CompassExperimentationProviderContextValue {
-  useAssignment: UseAssignmentHook;
   assignExperiment: AssignExperimentFn;
   getAssignment: GetAssignmentFn;
 }
 
 const initialContext: CompassExperimentationProviderContextValue = {
-  useAssignment() {
-    return {
-      assignment: null,
-      asyncStatus: null,
-      error: null,
-      isLoading: false,
-      isError: false,
-      isSuccess: true,
-    };
-  },
   assignExperiment() {
     return Promise.resolve(null);
   },
@@ -51,18 +33,15 @@ export const ExperimentationContext =
 // Provider component that accepts MMS experiment utils as props
 export const CompassExperimentationProvider: React.FC<{
   children: React.ReactNode;
-  useAssignment: UseAssignmentHook;
   assignExperiment: AssignExperimentFn;
   getAssignment: GetAssignmentFn;
-}> = ({ children, useAssignment, assignExperiment, getAssignment }) => {
+}> = ({ children, assignExperiment, getAssignment }) => {
   // Use useRef to keep the functions up-to-date; Use mutation pattern to maintain the
   // same object reference to prevent unnecessary re-renders of consuming components
   const { current: contextValue } = useRef({
-    useAssignment,
     assignExperiment,
     getAssignment,
   });
-  contextValue.useAssignment = useAssignment;
   contextValue.assignExperiment = assignExperiment;
   contextValue.getAssignment = getAssignment;
 
@@ -72,8 +51,3 @@ export const CompassExperimentationProvider: React.FC<{
     </ExperimentationContext.Provider>
   );
 };
-
-// Hook for components to access experiment assignment
-export const useAssignment = (...args: Parameters<UseAssignmentHook>) => {
-  return useContext(ExperimentationContext).useAssignment(...args);
-};

packages/compass-telemetry/src/provider.tsx

@@ -162,4 +162,3 @@ export const useFireExperimentViewed = ({
 
 export type { TrackFunction };
 export { ExperimentTestName, ExperimentTestGroup } from './growth-experiments';
-export { useAssignment } from './experimentation-provider';