escape if there is no ref.current

Author: rwieruch

package.json

@@ -1,6 +1,6 @@
 {
   "name": "use-custom-element",
-  "version": "1.0.3",
+  "version": "1.0.4",
   "description": "",
   "main": "lib/index.js",
   "scripts": {

src/index.js

@@ -4,19 +4,24 @@ const useCustomElement = (props, customMapping = {}) => {
   const ref = React.createRef();
 
   React.useLayoutEffect(() => {
-    const fns = Object.keys(props)
-      .filter(key => props[key] instanceof Function)
-      .map(key => ({
-        key: customMapping[key] || key,
-        fn: customEvent =>
-          props[key](customEvent.detail, customEvent),
-      }));
-
-    fns.forEach(({ key, fn }) =>
-      ref.current.addEventListener(key, fn),
-    );
+    let fns;
+
+    if (ref.current) {
+      fns = Object.keys(props)
+        .filter(key => props[key] instanceof Function)
+        .map(key => ({
+          key: customMapping[key] || key,
+          fn: customEvent =>
+            props[key](customEvent.detail, customEvent),
+        }));
+
+      fns.forEach(({ key, fn }) =>
+        ref.current.addEventListener(key, fn),
+      );
+    }
 
     return () =>
+      ref.current &&
       fns.forEach(({ key, fn }) =>
         ref.current.removeEventListener(key, fn),
       );