fix(comparison-table): Fix comparison table not growing

getPopsure · Jan 18, 2024 · dc394ce · dc394ce
1 parent 208aac5
commit dc394ce
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/src/lib/components/comparisonTable/hooks/useComparisonTable.ts b/src/lib/components/comparisonTable/hooks/useComparisonTable.ts
@@ -1,5 +1,5 @@
 import debounce from 'lodash.debounce';
-import { useCallback, useEffect, useRef, useState } from 'react';
+import { useCallback, useEffect, useLayoutEffect, useRef, useState } from 'react';
 
 import { ArrowValues } from '../components/TableArrows';
 import generateId from '../../../util/generateId';
@@ -51,6 +51,18 @@ export const useComparisonTable = () => {
     };
   }, []);
 
+  useLayoutEffect(() => {
+    const resetHeaderWidth = () => {
+      if (headerRef.current) {
+        setHeaderWidth(headerRef.current.clientWidth);
+      }
+    }
+
+    window.addEventListener('resize', resetHeaderWidth);
+
+    return () => window.removeEventListener('resize', resetHeaderWidth);
+  }, []);
+
   const handleTableScroll = () => {
     if (!headerRef.current) {
       return;