type: add type guard (#527)

* type: add type guard

* type: fix type

* fix: fix

* fix: fix

* fix: fix

Author: li-jia-nan

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;
   }

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();
+    });
   });
 });