Remove imperative handle

sanalabs · May 7, 2024 · bffcd95 · bffcd95
1 parent f0cd6c4
commit bffcd95
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/auto-text-size-react.tsx b/src/auto-text-size-react.tsx
@@ -34,7 +34,9 @@ export function AutoTextSize({
 
   useEffect(() => updateTextSizeRef.current?.(), [children]);
 
-  useImperativeHandle(onUpdateRef, () => onUpdate)
+  useEffect(() => {
+    onUpdateRef.current = onUpdate;
+  }, [onUpdate]);
 
   const refCallback = useCallback(
     (innerEl: HTMLElement | null) => {
@@ -51,7 +53,7 @@ export function AutoTextSize({
         maxFontSizePx,
         fontSizePrecisionPx,
         // Not passing the callback ref directly so we can always get the latest value
-        onUpdate: () =>  onUpdateRef.current(),
+        onUpdate: () => onUpdateRef.current?.(),
       });
     },
     [mode, minFontSizePx, maxFontSizePx, fontSizePrecisionPx]