openethereum/js/src/shell/Dapps/SelectAccounts/store.js

// Copyright 2015-2017 Parity Technologies (UK) Ltd.
// This file is part of Parity.

// Parity is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// Parity is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with Parity.  If not, see <http://www.gnu.org/licenses/>.

import { action, observable, transaction } from 'mobx';

export default class Store {
  @observable accounts = [];
  @observable modalOpen = false;
  @observable whitelist = [];
  @observable whitelistDefault = null;

  constructor (api) {
    this._api = api;

    this.loadWhitelist();
  }

  @action closeModal = () => {
    transaction(() => {
      const checkedAccounts = this.accounts.filter((account) => account.checked);
      const defaultAddress = (this.accounts.find((account) => account.default) || {}).address;
      const addresses = checkedAccounts.length === this.accounts.length
        ? null
        : checkedAccounts.map((account) => account.address);

      this.modalOpen = false;
      this.updateWhitelist(addresses, defaultAddress);
    });
  }

  @action openModal = (accounts) => {
    transaction(() => {
      this.accounts = Object
        .values(accounts)
        .map((account, index) => {
          return {
            address: account.address,
            checked: this.whitelist
              ? this.whitelist.includes(account.address)
              : true,
            default: this.whitelistDefault
              ? this.whitelistDefault === account.address
              : index === 0,
            description: account.meta.description,
            name: account.name
          };
        });
      this.modalOpen = true;
    });
  }

  @action selectAccount = (address) => {
    transaction(() => {
      const isSingleAccount = this.accounts.filter((account) => account.checked).length === 1;

      this.accounts = this.accounts.map((account) => {
        if (account.address === address && (!isSingleAccount || !account.checked)) {
          account.checked = !account.checked;
          account.default = false;
        }

        return account;
      });

      this.setDefaultAccount((
        this.accounts.find((account) => account.default) ||
        this.accounts.find((account) => account.checked) ||
        {}
      ).address);
    });
  }

  @action setDefaultAccount = (address) => {
    this.accounts = this.accounts.map((account) => {
      if (account.address === address) {
        account.checked = true;
        account.default = true;
      } else if (account.default) {
        account.default = false;
      }

      return account;
    });
  }

  @action setWhitelist = (whitelist, whitelistDefault) => {
    transaction(() => {
      this.whitelist = whitelist;
      this.whitelistDefault = whitelistDefault;
    });
  }

  loadWhitelist () {
    return Promise
      .all([
        this._api.parity.getNewDappsAddresses(),
        this._api.parity.getNewDappsDefaultAddress()
      ])
      .then(([whitelist, whitelistDefault]) => {
        this.setWhitelist(whitelist, whitelistDefault);
      })
      .catch((error) => {
        console.warn('loadWhitelist', error);
      });
  }

  updateWhitelist (whitelist, whitelistDefault = null) {
    return Promise
      .all([
        this._api.parity.setNewDappsAddresses(whitelist),
        this._api.parity.setNewDappsDefaultAddress(whitelistDefault)
      ])
      .then(() => {
        this.setWhitelist(whitelist, whitelistDefault);
      })
      .catch((error) => {
        console.warn('updateWhitelist', error);
      });
  }
}