import PropTypes from 'prop-types' import React, { Component } from 'react' import { Switch, Route } from 'react-router-dom' import ChooseAccount from './choose-account' import { getEnvironmentType } from '../../../../app/scripts/lib/util' import { ENVIRONMENT_TYPE_FULLSCREEN, ENVIRONMENT_TYPE_NOTIFICATION, ENVIRONMENT_TYPE_POPUP, } from '../../../../app/scripts/lib/enums' import { DEFAULT_ROUTE, } from '../../helpers/constants/routes' import PermissionPageContainer from '../../components/app/permission-page-container' 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, originName: PropTypes.string, showNewAccountModal: PropTypes.func.isRequired, newAccountNumber: PropTypes.number.isRequired, nativeCurrency: PropTypes.string, permissionsRequest: PropTypes.object, addressLastConnectedMap: PropTypes.object, permissionsRequestId: PropTypes.string, domains: PropTypes.object, history: PropTypes.object.isRequired, connectPath: PropTypes.string.isRequired, confirmPermissionPath: PropTypes.string.isRequired, page: PropTypes.string.isRequired, redirecting: PropTypes.bool, targetDomainMetadata: PropTypes.object, } static defaultProps = { originName: '', nativeCurrency: '', permissionsRequest: undefined, addressLastConnectedMap: {}, permissionsRequestId: '', domains: {}, redirecting: false, } static contextTypes = { t: PropTypes.func, } state = { redirecting: false, selectedAccountAddresses: this.props.accounts.length === 1 ? new Set([this.props.accounts[0].address]) : new Set(), permissionAccepted: null, originName: this.props.originName, } beforeUnload = () => { const { permissionsRequestId, rejectPermissionsRequest } = this.props const { permissionAccepted } = this.state if (permissionAccepted === null && permissionsRequestId) { rejectPermissionsRequest(permissionsRequestId) } } removeBeforeUnload = () => { if (getEnvironmentType() === ENVIRONMENT_TYPE_FULLSCREEN || getEnvironmentType() === ENVIRONMENT_TYPE_NOTIFICATION) { window.removeEventListener('beforeunload', this.beforeUnload) } } componentDidUpdate (prevProps) { const { domains, permissionsRequest, redirecting } = this.props const { originName } = this.state if (!permissionsRequest && prevProps.permissionsRequest && !redirecting) { const permissionDataForDomain = (domains && domains[originName]) || {} const permissionsForDomain = permissionDataForDomain.permissions || [] const prevPermissionDataForDomain = (prevProps.domains && prevProps.domains[originName]) || {} const prevPermissionsForDomain = prevPermissionDataForDomain.permissions || [] const addedAPermission = permissionsForDomain.length > prevPermissionsForDomain.length if (addedAPermission) { this.redirectFlow(true) } else { this.redirectFlow(false) } } } selectAccounts = (addresses) => { this.setState({ selectedAccountAddresses: addresses, }, () => this.props.history.push(this.props.confirmPermissionPath)) } redirectFlow (accepted) { const { history } = this.props this.setState({ redirecting: true, permissionAccepted: accepted, }) this.removeBeforeUnload() if (getEnvironmentType() === ENVIRONMENT_TYPE_FULLSCREEN) { setTimeout(async () => { const currentTab = await global.platform.currentTab() global.platform.closeTab(currentTab.id) }, 2000) } else if (getEnvironmentType() === ENVIRONMENT_TYPE_NOTIFICATION) { setTimeout(async () => { global.platform.closeCurrentWindow() }, 2000) } else if (getEnvironmentType() === ENVIRONMENT_TYPE_POPUP) { history.push(DEFAULT_ROUTE) } } componentDidMount () { const { getCurrentWindowTab, getRequestAccountTabIds, permissionsRequest, history, } = this.props getCurrentWindowTab() getRequestAccountTabIds() if (!permissionsRequest) { return history.push(DEFAULT_ROUTE) } if (getEnvironmentType() === ENVIRONMENT_TYPE_FULLSCREEN || getEnvironmentType() === ENVIRONMENT_TYPE_NOTIFICATION) { window.addEventListener('beforeunload', this.beforeUnload) } } cancelPermissionsRequest = async (requestId) => { const { history, rejectPermissionsRequest } = this.props if (requestId) { await rejectPermissionsRequest(requestId) if (getEnvironmentType() === ENVIRONMENT_TYPE_FULLSCREEN || getEnvironmentType() === ENVIRONMENT_TYPE_NOTIFICATION) { window.close() } else if (getEnvironmentType() === ENVIRONMENT_TYPE_POPUP) { history.push(DEFAULT_ROUTE) } } } goBack () { const { history, connectPath } = this.props history.push(connectPath) } renderTopBar () { const { redirecting } = this.state const { page } = this.props const { t } = this.context return !redirecting ? (
{ page === '2' ? (
this.goBack()}> { t('back') }
) : null }
{ t('xOfY', [ page, '2' ]) }
) : null } render () { const { approvePermissionsRequest, rejectPermissionsRequest, accounts, showNewAccountModal, newAccountNumber, nativeCurrency, permissionsRequest, addressLastConnectedMap, permissionsRequestId, connectPath, confirmPermissionPath, targetDomainMetadata, } = this.props const { selectedAccountAddresses, permissionAccepted, originName, redirecting, } = this.state return (
{ this.renderTopBar() } ( 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(request, accounts) this.redirectFlow(true) }} rejectPermissionsRequest={(requestId) => { rejectPermissionsRequest(requestId) this.redirectFlow(false) }} selectedIdentities={accounts.filter((account) => selectedAccountAddresses.has(account.address))} redirect={redirecting} permissionRejected={ permissionAccepted === false } cachedOrigin={originName} /> )} />
) } }