clean stale notification sizes

Author: zombieJ

src/hooks/useListPosition/useSizes.ts

@@ -12,12 +12,29 @@ export type NodeSizeMap = Record<string, NodeSize>;
  */
 export default function useSizes() {
   const [sizeMap, setSizeMap] = React.useState<NodeSizeMap>({});
+  const cleanUpMapRef = React.useRef<Record<string, ReturnType<typeof setTimeout>>>({});
 
   const setNodeSize = React.useCallback((key: string, node: HTMLDivElement | null) => {
     if (!node) {
+      cleanUpMapRef.current[key] = setTimeout(() => {
+        delete cleanUpMapRef.current[key];
+
+        setSizeMap((prevSizeMap) => {
+          if (!(key in prevSizeMap)) {
+            return prevSizeMap;
+          }
+
+          const nextSizeMap = { ...prevSizeMap };
+          delete nextSizeMap[key];
+          return nextSizeMap;
+        });
+      });
       return;
     }
 
+    clearTimeout(cleanUpMapRef.current[key]);
+    delete cleanUpMapRef.current[key];
+
     const nextSize = {
       width: node.offsetWidth,
       height: node.offsetHeight,
@@ -36,5 +53,14 @@ export default function useSizes() {
     });
   }, []);
 
+  React.useEffect(
+    () => () => {
+      Object.values(cleanUpMapRef.current).forEach((timer) => {
+        clearTimeout(timer);
+      });
+    },
+    [],
+  );
+
   return [sizeMap, setNodeSize] as const;
 }