fix: fix ref issue on arrow scroll (#610)

pluralsh · Jun 14, 2024 · eea0851 · eea0851
1 parent 4c9fbdc
commit eea0851
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/components/ArrowScroll.tsx b/src/components/ArrowScroll.tsx
@@ -1,7 +1,6 @@
 import {
   Children,
   type ComponentProps,
-  type ReactElement,
   cloneElement,
   useEffect,
   useRef,
@@ -82,7 +81,7 @@ const scroll = (
   }
 }
 
-function ArrowScroll({ children, ...props }: { children?: ReactElement }) {
+function ArrowScroll({ children, ...props }: { children?: any }) {
   const containerRef = useRef<HTMLElement>(undefined)
   const [showLeftArrow, setShowLeftArrow] = useState(false)
   const [showRightArrow, setShowRightArrow] = useState(false)
@@ -117,7 +116,16 @@ function ArrowScroll({ children, ...props }: { children?: ReactElement }) {
       />
       {cloneElement(Children.only(children), {
         onScroll: checkScroll,
-        ref: containerRef,
+        ref: (node: HTMLElement) => {
+          containerRef.current = node
+          if (children.ref) {
+            if (typeof children.ref === 'function') {
+              children.ref(node)
+            } else if (children.ref) {
+              children.ref.current = node
+            }
+          }
+        },
       })}
     </ComponentWrapperSC>
   )