From 7b51c96a57c7470e8d97a90aab1932b8c2c1c6de Mon Sep 17 00:00:00 2001
From: bendtherules <abhasbhattacharya2@gmail.com>
Date: Wed, 24 Jul 2019 02:28:13 +0530
Subject: [PATCH] Add utils - Methods to check if a node corresponds to
 functional or class components

---
 src/utils.ts | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/src/utils.ts b/src/utils.ts
index cc86f0c..d03253e 100644
--- a/src/utils.ts
+++ b/src/utils.ts
@@ -1,4 +1,5 @@
-import { FiberNode, FiberNodeisHTMLLike } from './mocked-types';
+import * as React from 'react';
+import { FiberNode, FiberNodeisHTMLLike, FiberNodeForFunctionComponent, FiberNodeForComponentClass } from './mocked-types';
 
 function isNodeHtmlLike(node: FiberNode): node is FiberNodeisHTMLLike {
     return (typeof node.type === "string") || node.type === null;
@@ -8,4 +9,12 @@ function isNodeNotHtmlLike(node: FiberNode): node is Exclude<FiberNode, FiberNod
     return !isNodeHtmlLike(node);
 }
 
-export { isNodeHtmlLike, isNodeNotHtmlLike };
\ No newline at end of file
+function isNodeFunctionComponent(node: FiberNode): node is FiberNodeForFunctionComponent {
+    return isNodeNotHtmlLike(node) && node.stateNode === null;
+}
+
+function isNodeComponentClass(node: FiberNode): node is FiberNodeForComponentClass {
+    return isNodeNotHtmlLike(node) && (node.stateNode instanceof React.Component);
+}
+
+export { isNodeHtmlLike, isNodeNotHtmlLike, isNodeFunctionComponent, isNodeComponentClass };
\ No newline at end of file