import PropTypes from 'prop-types'; import React, { Component } from 'react'; import { Switch, Route } from 'react-router-dom'; import { getEnvironmentType } from '../../../app/scripts/lib/util'; import { ENVIRONMENT_TYPE_NOTIFICATION } from '../../../shared/constants/app'; import { MILLISECOND } from '../../../shared/constants/time'; import { DEFAULT_ROUTE } from '../../helpers/constants/routes'; import PermissionPageContainer from '../../components/app/permission-page-container'; import ChooseAccount from './choose-account'; import PermissionsRedirect from './redirect'; const APPROVE_TIMEOUT = MILLISECOND * 1200; export default class PermissionConnect extends Component { static propTypes = { approvePermissionsRequest: PropTypes.func.isRequired, rejectPermissionsRequest: PropTypes.func.isRequired, getRequestAccountTabIds: PropTypes.func.isRequired, getCurrentWindowTab: PropTypes.func.isRequired, accounts: PropTypes.array.isRequired, currentAddress: PropTypes.string.isRequired, origin: PropTypes.string, showNewAccountModal: PropTypes.func.isRequired, newAccountNumber: PropTypes.number.isRequired, nativeCurrency: PropTypes.string, permissionsRequest: PropTypes.object, addressLastConnectedMap: PropTypes.object.isRequired, lastConnectedInfo: PropTypes.object.isRequired, permissionsRequestId: PropTypes.string, history: PropTypes.object.isRequired, connectPath: PropTypes.string.isRequired, confirmPermissionPath: PropTypes.string.isRequired, page: PropTypes.string.isRequired, targetDomainMetadata: PropTypes.shape({ extensionId: PropTypes.string, icon: PropTypes.string, host: PropTypes.string.isRequired, name: PropTypes.string.isRequired, origin: PropTypes.string.isRequired, }), }; static defaultProps = { origin: '', nativeCurrency: '', permissionsRequest: undefined, permissionsRequestId: '', }; static contextTypes = { t: PropTypes.func, }; state = { redirecting: false, selectedAccountAddresses: new Set([this.props.currentAddress]), permissionsApproved: null, origin: this.props.origin, targetDomainMetadata: this.props.targetDomainMetadata || {}, }; beforeUnload = () => { const { permissionsRequestId, rejectPermissionsRequest } = this.props; const { permissionsApproved } = this.state; if (permissionsApproved === null && permissionsRequestId) { rejectPermissionsRequest(permissionsRequestId); } }; removeBeforeUnload = () => { const environmentType = getEnvironmentType(); if (environmentType === ENVIRONMENT_TYPE_NOTIFICATION) { window.removeEventListener('beforeunload', this.beforeUnload); } }; componentDidMount() { const { getCurrentWindowTab, getRequestAccountTabIds, permissionsRequest, history, } = this.props; getCurrentWindowTab(); getRequestAccountTabIds(); if (!permissionsRequest) { history.push(DEFAULT_ROUTE); return; } const environmentType = getEnvironmentType(); if (environmentType === ENVIRONMENT_TYPE_NOTIFICATION) { window.addEventListener('beforeunload', this.beforeUnload); } } static getDerivedStateFromProps(props, state) { const { permissionsRequest, targetDomainMetadata } = props; const { targetDomainMetadata: savedMetadata } = state; if ( permissionsRequest && savedMetadata.origin !== targetDomainMetadata?.origin ) { return { targetDomainMetadata }; } return null; } componentDidUpdate(prevProps) { const { permissionsRequest, lastConnectedInfo } = this.props; const { redirecting, origin } = this.state; if (!permissionsRequest && prevProps.permissionsRequest && !redirecting) { const accountsLastApprovedTime = lastConnectedInfo[origin]?.lastApproved || 0; const initialAccountsLastApprovedTime = prevProps.lastConnectedInfo[origin]?.lastApproved || 0; const approved = accountsLastApprovedTime > initialAccountsLastApprovedTime; this.redirect(approved); } } selectAccounts = (addresses) => { this.setState( { selectedAccountAddresses: addresses, }, () => this.props.history.push(this.props.confirmPermissionPath), ); }; redirect(approved) { const { history } = this.props; this.setState({ redirecting: true, permissionsApproved: approved, }); this.removeBeforeUnload(); if (approved) { setTimeout(() => history.push(DEFAULT_ROUTE), APPROVE_TIMEOUT); } else { history.push(DEFAULT_ROUTE); } } cancelPermissionsRequest = async (requestId) => { const { rejectPermissionsRequest } = this.props; if (requestId) { await rejectPermissionsRequest(requestId); this.redirect(false); } }; goBack() { const { history, connectPath } = this.props; history.push(connectPath); } renderTopBar() { const { redirecting } = this.state; const { page } = this.props; const { t } = this.context; return redirecting ? null : (
{page === '2' ? (
this.goBack()} > {t('back')}
) : null}
{t('xOfY', [page, '2'])}
); } render() { const { approvePermissionsRequest, accounts, showNewAccountModal, newAccountNumber, nativeCurrency, permissionsRequest, addressLastConnectedMap, permissionsRequestId, connectPath, confirmPermissionPath, } = this.props; const { selectedAccountAddresses, permissionsApproved, redirecting, targetDomainMetadata, } = this.state; return (
{this.renderTopBar()} {redirecting && permissionsApproved ? ( ) : ( ( this.selectAccounts(addresses)} selectNewAccountViaModal={(handleAccountClick) => { showNewAccountModal({ onCreateNewAccount: (address) => handleAccountClick(address), newAccountNumber, }); }} addressLastConnectedMap={addressLastConnectedMap} cancelPermissionsRequest={(requestId) => this.cancelPermissionsRequest(requestId) } permissionsRequestId={permissionsRequestId} selectedAccountAddresses={selectedAccountAddresses} targetDomainMetadata={targetDomainMetadata} /> )} /> ( { approvePermissionsRequest(...args); this.redirect(true); }} rejectPermissionsRequest={(requestId) => this.cancelPermissionsRequest(requestId) } selectedIdentities={accounts.filter((account) => selectedAccountAddresses.has(account.address), )} targetDomainMetadata={targetDomainMetadata} /> )} /> )}
); } }