Make sure original document.title is restored only on unmount

Author: typeofweb

index.d.ts

@@ -1,3 +1,3 @@
 interface DocumentTitle {}
 
-export default function useDocumentTitle(title: string): DocumentTitle;
+export default function useDocumentTitle(title: string, retainOnUnmount?: boolean): DocumentTitle;

index.js

@@ -6,11 +6,15 @@ function useDocumentTitle(title, retainOnUnmount = false) {
 
   useEffect(() => {
     document.title = title;
+  }, [title]);
 
+  useEffect(() => {
     return () => {
-      if (!retainOnUnmount) document.title = defaultTitle.current;
+      if (!retainOnUnmount) {
+        document.title = defaultTitle.current;
+      }
     };
-  }, [title]);
+  }, []);
 }
 
 module.exports = useDocumentTitle;

index.js.flow

@@ -1,4 +1,4 @@
 interface DocumentTitle {}
-declare function useDocumentTitle(title: string): DocumentTitle;
+declare function useDocumentTitle(title: string, retainOnUnmount?: boolean): DocumentTitle;
 
 export default useDocumentTitle;