import React from 'react'; import { connect } from 'react-redux'; import { withRouter, RouteComponentProps } from 'react-router-dom'; import Modal, { IButton } from 'components/ui/Modal'; import { AppState } from 'reducers'; import { resetWallet, TResetWallet } from 'actions/wallet'; interface Props extends RouteComponentProps<{}> { // State wallet: AppState['wallet']['inst']; // Actions resetWallet: TResetWallet; } interface State { nextLocation: RouteComponentProps<{}>['location'] | null; openModal: boolean; } class LogOutPromptClass extends React.Component { constructor(props) { super(props); this.state = { nextLocation: null, openModal: false }; this.props.history.block(nextLocation => { if (this.props.wallet && nextLocation.pathname !== this.props.location.pathname) { const isSubTab = nextLocation.pathname.split('/')[1] === this.props.location.pathname.split('/')[1]; if (!isSubTab) { this.setState({ openModal: true, nextLocation }); return false; } } }); } public render() { const buttons: IButton[] = [ { text: 'Log Out', type: 'primary', onClick: this.onConfirm }, { text: 'Cancel', type: 'default', onClick: this.onCancel } ]; return (

Leaving this page will log you out. Are you sure you want to continue?

); } private onCancel = () => { this.setState({ nextLocation: null, openModal: false }); }; private onConfirm = () => { const { nextLocation: next } = this.state; this.props.resetWallet(); this.setState( { openModal: false, nextLocation: null }, () => { if (next) { this.props.history.push(`${next.pathname}${next.search}${next.hash}`); } } ); }; } function mapStateToProps(state: AppState) { return { wallet: state.wallet.inst }; } export default connect(mapStateToProps, { resetWallet })(withRouter(LogOutPromptClass));