fix(useSize): prevents accessing iframe's property when it's not defined

ref #442
streamich · Jul 4, 2019 · c9b5cdc · c9b5cdc
1 parent cf68e36
commit c9b5cdc
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/useSize.tsx b/src/useSize.tsx
@@ -45,7 +45,13 @@ const useSize = (
   };
 
   useEffect(() => {
-    const iframe: HTMLIFrameElement = ref.current!;
+    const iframe: HTMLIFrameElement = ref.current;
+
+    if(!iframe) {
+      // iframe will be undefined if component is already unmounted
+      return;
+    }
+
     if (iframe.contentWindow) {
       window = iframe.contentWindow!;
       onWindow(window);