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 '../../../../app/scripts/lib/enums' 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 = 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 ? (