chore: Update findDOMNode define

react-component · Sep 30, 2019 · 30f034f · 30f034f
1 parent e34eb0b
commit 30f034f
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/Dom/findDOMNode.ts b/src/Dom/findDOMNode.ts
@@ -3,9 +3,11 @@ import ReactDOM from 'react-dom';
 /**
  * Return if a node is a DOM node. Else will return by `findDOMNode`
  */
-export default function findDOMNode(node: React.ReactInstance | HTMLElement): Element | Text {
+export default function findDOMNode<T = Element | Text>(
+  node: React.ReactInstance | HTMLElement,
+): T {
   if (node instanceof HTMLElement) {
-    return node;
+    return (node as unknown) as T;
   }
-  return ReactDOM.findDOMNode(node);
+  return (ReactDOM.findDOMNode(node) as unknown) as T;
 }