feat(react utils): add ref merge util

abelflopes · Dec 19, 2024 · 42683b3 · 42683b3
1 parent 322f21d
commit 42683b3
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/packages/utils/react/src/index.ts b/packages/utils/react/src/index.ts
@@ -9,3 +9,5 @@ export * from "./click-outside";
 export * from "./next-render";
 
 export * from "./children-without-fragments";
+
+export * from "./merge-refs";
diff --git a/packages/utils/react/src/merge-refs.ts b/packages/utils/react/src/merge-refs.ts
@@ -0,0 +1,13 @@
+import { type ForwardedRef, type MutableRefObject, type RefCallback } from "react";
+
+export const megeRefs =
+  <T>(refs: Array<ForwardedRef<T> | MutableRefObject<T | null> | RefCallback<T>>): RefCallback<T> =>
+  (r) => {
+    refs.forEach((thisRef) => {
+      if (!thisRef) return;
+
+      if (typeof thisRef === "object" && "current" in thisRef) thisRef.current = r;
+      else if (typeof thisRef === "function") thisRef(r);
+      else throw new Error("Unuspported ref type to merge");
+    });
+  };
Original file line number	Diff line number	Diff line change
Expand Up		@@ -9,3 +9,5 @@ export * from "./click-outside";
		export * from "./next-render";

		export * from "./children-without-fragments";

		export * from "./merge-refs";