metamask-extension/ui/app/pages/connected-sites/connected-sites.component.js

import PropTypes from 'prop-types'
import React, { Component } from 'react'
import ConnectedSitesList from '../../components/app/connected-sites-list'
import Popover from '../../components/ui/popover/popover.component'
import Button from '../../components/ui/button'

export default class ConnectedSites extends Component {
  static contextTypes = {
    t: PropTypes.func,
  }

  static defaultProps = {
    tabToConnect: null,
  }

  static propTypes = {
    accountLabel: PropTypes.string.isRequired,
    closePopover: PropTypes.func.isRequired,
    connectedDomains: PropTypes.arrayOf(PropTypes.object).isRequired,
    domainHostCount: PropTypes.objectOf(PropTypes.number).isRequired,
    disconnectAllAccounts: PropTypes.func.isRequired,
    disconnectAccount: PropTypes.func.isRequired,
    getOpenMetamaskTabsIds: PropTypes.func.isRequired,
    permittedAccountsByOrigin: PropTypes.objectOf(
      PropTypes.arrayOf(PropTypes.string),
    ).isRequired,
    tabToConnect: PropTypes.object,
    requestAccountsPermission: PropTypes.func.isRequired,
  }

  state = {
    sitePendingDisconnect: null,
  }

  componentDidMount() {
    const { getOpenMetamaskTabsIds } = this.props
    getOpenMetamaskTabsIds()
  }

  setPendingDisconnect = (domainKey) => {
    this.setState({
      sitePendingDisconnect: {
        domainKey,
      },
    })
  }

  clearPendingDisconnect = () => {
    this.setState({
      sitePendingDisconnect: null,
    })
  }

  disconnectAccount = () => {
    const { disconnectAccount } = this.props
    const { sitePendingDisconnect } = this.state

    disconnectAccount(sitePendingDisconnect.domainKey)
    this.clearPendingDisconnect()
  }

  disconnectAllAccounts = () => {
    const { disconnectAllAccounts } = this.props
    const { sitePendingDisconnect } = this.state

    disconnectAllAccounts(sitePendingDisconnect.domainKey)
    this.clearPendingDisconnect()
  }

  renderConnectedSitesList() {
    return (
      <ConnectedSitesList
        domainHostCount={this.props.domainHostCount}
        connectedDomains={this.props.connectedDomains}
        onDisconnect={this.setPendingDisconnect}
      />
    )
  }

  renderConnectedSitesPopover() {
    const {
      accountLabel,
      closePopover,
      connectedDomains,
      tabToConnect,
      requestAccountsPermission,
    } = this.props
    const { t } = this.context

    return (
      <Popover
        className="connected-sites"
        title={t('connectedSites')}
        subtitle={
          connectedDomains.length
            ? t('connectedSitesDescription', [accountLabel])
            : t('connectedSitesEmptyDescription', [accountLabel])
        }
        onClose={closePopover}
        footer={
          tabToConnect ? (
            <a
              className="connected-sites__text-button"
              onClick={requestAccountsPermission}
            >
              {t('connectManually')}
            </a>
          ) : null
        }
        footerClassName="connected-sites__add-site-manually"
      >
        {this.renderConnectedSitesList()}
      </Popover>
    )
  }

  renderDisconnectPopover() {
    const { closePopover, permittedAccountsByOrigin } = this.props
    const { t } = this.context
    const {
      sitePendingDisconnect: { domainKey },
    } = this.state

    const numPermittedAccounts = permittedAccountsByOrigin[domainKey].length

    return (
      <Popover
        className="connected-sites"
        title={t('disconnectPrompt', [domainKey])}
        subtitle={t('disconnectAllAccountsConfirmationDescription')}
        onClose={closePopover}
        footer={
          <>
            <div className="connected-sites__footer-row">
              <Button type="secondary" onClick={this.clearPendingDisconnect}>
                {t('cancel')}
              </Button>
              <Button type="primary" onClick={this.disconnectAccount}>
                {t('disconnect')}
              </Button>
            </div>
            {numPermittedAccounts > 1 ? (
              <div className="connected-sites__footer-row">
                <a
                  className="connected-sites__text-button"
                  onClick={this.disconnectAllAccounts}
                >
                  {t('disconnectAllAccounts')}
                </a>
              </div>
            ) : null}
          </>
        }
        footerClassName="connected-sites__confirmation"
      />
    )
  }

  render() {
    const { sitePendingDisconnect } = this.state
    return sitePendingDisconnect
      ? this.renderDisconnectPopover()
      : this.renderConnectedSitesPopover()
  }
}