type: add type guard (#527)

* type: add type guard * type: fix type * fix: fix * fix: fix * fix: fix
react-component · Apr 30, 2024 · b201640 · b201640
1 parent 15a195c
commit b201640
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/src/ref.ts b/src/ref.ts
@@ -62,7 +62,9 @@ export function supportRef(nodeOrComponent: any): boolean {
   return true;
 }
 
-export function supportNodeRef(node: React.ReactNode): boolean {
+export function supportNodeRef<T = any>(
+  node: React.ReactNode,
+): node is React.ReactElement & React.RefAttributes<T> {
   if (!isValidElement(node)) {
     return false;
   }

diff --git a/tests/ref.test.tsx b/tests/ref.test.tsx
@@ -186,5 +186,17 @@ describe('ref', () => {
       expect(supportNodeRef(<FC />)).toBeFalsy();
       expect(supportNodeRef(<RefFC />)).toBeTruthy();
     });
+
+    it('ref', () => {
+      const FC: React.FC = () => <div />;
+      const RefFC = React.forwardRef<
+        HTMLDivElement,
+        React.HTMLAttributes<HTMLDivElement>
+      >((props, ref) => <div {...props} ref={ref} />);
+      const com = <FC />;
+      const refCom = <RefFC ref={React.createRef()} />;
+      expect(supportNodeRef(com) && com.ref).toBeFalsy();
+      expect(supportNodeRef(refCom) && refCom.ref).toBeTruthy();
+    });
   });
 });