revert flag elements stuff

laander · laander · commit 1b0dcabbd54f · 2025-09-29T14:50:55.000+02:00
diff --git a/packages/react-sdk/src/index.tsx b/packages/react-sdk/src/index.tsx
@@ -124,7 +124,7 @@ type ProviderContextType = {
   };
   provider: boolean;
   activeFlags: Set<string>;
-  registerActiveFlag: (flagKey: string, element?: HTMLElement) => void;
+  registerActiveFlag: (flagKey: string) => void;
   unregisterActiveFlag: (flagKey: string) => void;
 };
 
@@ -188,7 +188,6 @@ export function ReflagProvider({
   const [featuresLoading, setFlagsLoading] = useState(true);
   const [rawFlags, setRawFlags] = useState<RawFlags>({});
   const [activeFlags, setActiveFlags] = useState<Set<string>>(new Set());
-  const [flagElements, setFlagElements] = useState<Map<string, HTMLElement>>(new Map());
 
   const clientRef = useRef<ReflagClient>();
   const contextKeyRef = useRef<string>();
@@ -236,23 +235,13 @@ export function ReflagProvider({
     // eslint-disable-next-line react-hooks/exhaustive-deps -- should only run once
   }, [contextKey]);
 
-  const registerActiveFlag = useCallback((flagKey: string, element?: HTMLElement) => {
+  const registerActiveFlag = useCallback((flagKey: string) => {
     setActiveFlags((prev) => {
       const newSet = new Set(prev).add(flagKey);
       // Sync with browser SDK client
       clientRef.current?.setActiveFlags(newSet);
       return newSet;
     });
-    
-    if (element) {
-      setFlagElements((prev) => {
-        const newMap = new Map(prev);
-        newMap.set(flagKey, element);
-        // Sync with browser SDK client
-        clientRef.current?.setFlagElements(newMap);
-        return newMap;
-      });
-    }
   }, []);
 
   const unregisterActiveFlag = useCallback((flagKey: string) => {
@@ -263,14 +252,6 @@ export function ReflagProvider({
       clientRef.current?.setActiveFlags(newSet);
       return newSet;
     });
-    
-    setFlagElements((prev) => {
-      const newMap = new Map(prev);
-      newMap.delete(flagKey);
-      // Sync with browser SDK client
-      clientRef.current?.setFlagElements(newMap);
-      return newMap;
-    });
   }, []);
 
   const context: ProviderContextType = {
