feat: support parentPopupState in render props, fix unmounted issues

Author: jedwards1211

src/hooks.js

@@ -1,6 +1,6 @@
 // @flow
 
-import { useState } from 'react'
+import { useState, useRef, useEffect } from 'react'
 
 if (!useState) {
   throw new Error(
@@ -19,6 +19,7 @@ import {
   bindPopper,
   type Variant,
   type PopupState,
+  type CoreState,
 } from './core'
 
 export { bindTrigger, bindToggle, bindHover, bindMenu, bindPopover, bindPopper }
@@ -34,7 +35,16 @@ export function usePopupState({
   variant: Variant,
 }): PopupState {
   const [state, _setState] = useState(initCoreState)
-  const setState = nextState => _setState({ ...state, ...nextState })
+  const isMounted = useRef(true)
+  useEffect(
+    () => () => {
+      isMounted.current = false
+    },
+    []
+  )
+  const setState = (nextState: $Shape<CoreState>) => {
+    if (isMounted.current) _setState({ ...state, ...nextState })
+  }
 
   return createPopupState({
     state,

src/index.js

@@ -24,11 +24,14 @@ export type Props = {
   popupId?: string,
   children: (props: InjectedProps) => ?React.Node,
   variant: Variant,
+  parentPopupState?: ?InjectedProps,
 }
 
 export default class PopupState extends React.Component<Props, CoreState> {
   state: CoreState = initCoreState
 
+  _mounted: boolean = true
+
   static propTypes = {
     /**
      * The render function.
@@ -64,16 +67,29 @@ export default class PopupState extends React.Component<Props, CoreState> {
      * component.
      */
     variant: PropTypes.oneOf(['popover', 'popper']).isRequired,
+    /**
+     *
+     */
+    parentPopupState: PropTypes.object,
+  }
+
+  componentWillUnmount() {
+    this._mounted = false
+  }
+
+  _setStateIfMounted = (state: $Shape<CoreState>) => {
+    if (this._mounted) this.setState(state)
   }
 
   render(): React.Node | null {
-    const { children, popupId, variant } = this.props
+    const { children, popupId, variant, parentPopupState } = this.props
 
     const popupState = createPopupState({
       state: this.state,
-      setState: state => this.setState(state),
+      setState: this._setStateIfMounted,
       popupId,
       variant,
+      parentPopupState,
     })
 
     const result = children(popupState)