openethereum/js/src/mobx/hardwareStore.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, computed, observable, transaction } from 'mobx';

import Ledger from '~/3rdparty/ledger';

const HW_SCAN_INTERVAL = 5000;
let instance = null;

export default class HardwareStore {
  @observable isScanning = false;
  @observable wallets = {};
  @observable pinMatrixRequest = [];

  constructor (api) {
    this._api = api;
    this._ledger = Ledger.create(api);
    this._pollId = null;
    this.hwAccounts = {};
    this.ledgerAccounts = {};

    this._pollScan();
    this._subscribeParity();
    this._api.transport.on('close', () => {
      this._subscribeParity();
    });
  }

  isConnected (address) {
    return computed(() => !!this.wallets[address]).get();
  }

  @action setScanning = (isScanning) => {
    this.isScanning = isScanning;
  }

  @action setWallets = (wallets) => {
    this.wallets = wallets;
  }

  @action setPinMatrixRequest = (requests) => {
    this.pinMatrixRequest = requests;
  }

  _pollScan = () => {
    this._pollId = setTimeout(() => {
      this.scan().then(this._pollScan);
    }, HW_SCAN_INTERVAL);
  }

  scanTrezor () {
    return this._api.parity
      .lockedHardwareAccountsInfo()
      .then((paths) => {
        this.setPinMatrixRequest(paths.map((path) => {
          return { path: path, manufacturer: 'Trezor' };
        }));
        return {};
      })
      .catch((err) => {
        console.warn('HardwareStore::scanTrezor', err);
        return {};
      });
  }

  scanLedger () {
    if (!this._ledger.isSupported) {
      return Promise.resolve({});
    }

    return this._ledger
      .scan()
      .then((wallets) => {
        console.log('HardwareStore::scanLedger', wallets);

        return wallets.reduce((hwInfo, wallet) => {
          wallet.manufacturer = 'Ledger';
          wallet.name = 'Nano S';
          wallet.via = 'ledger';

          hwInfo[wallet.address] = wallet;

          return hwInfo;
        }, {});
      })
      .catch((error) => {
        console.warn('HardwareStore::scanLedger', error);

        return {};
      });
  }

  _subscribeParity () {
    const onError = error => {
      console.warn('HardwareStore::scanParity', error);

      return {};
    };

    return this._api.pubsub
      .subscribeAndGetResult(
        callback => this._api.pubsub.parity.hardwareAccountsInfo(callback),
        hwInfo => {
          Object
            .keys(hwInfo)
            .forEach((address) => {
              const info = hwInfo[address];

              info.address = address;
              info.via = 'parity';
            });
          this.hwAccounts = hwInfo;
          this.updateWallets();
          return hwInfo;
        },
        onError
      ).catch(onError);
  }

  scan () {
    this.setScanning(true);
    // This only scans for locked devices and does not return open devices,
    // so no need to actually wait for any results here.
    this.scanTrezor();

    // NOTE: Depending on how the hardware is configured and how the local env setup
    // is done, different results will be retrieved via Parity vs. the browser APIs
    // (latter is Chrome-only, needs the browser app enabled on a Ledger, former is
    // not intended as a network call, i.e. hw wallet is with the user)
    return this.scanLedger()
      .then((ledgerAccounts) => {
        this.ledgerAccounts = ledgerAccounts;
        transaction(() => {
          this.updateWallets();
          this.setScanning(false);
        });
      });
  }

  updateWallets () {
    this.setWallets(Object.assign({}, this.hwAccounts, this.ledgerAccounts));
  }

  createAccountInfo (entry, original = {}) {
    const { address, manufacturer, name } = entry;

    return Promise
      .all([
        original.name
          ? Promise.resolve(true)
          : this._api.parity.setAccountName(address, name),
        this._api.parity.setAccountMeta(address, Object.assign({
          description: `${manufacturer} ${name}`,
          hardware: {
            manufacturer
          },
          tags: ['hardware'],
          timestamp: Date.now()
        }, original.meta || {}))
      ])
      .catch((error) => {
        console.warn('HardwareStore::createEntry', error);
        throw error;
      });
  }

  signLedger (transaction) {
    return this._ledger.signTransaction(transaction);
  }

  pinMatrixAck (device, passcode) {
    return this._api.parity
      .hardwarePinMatrixAck(device.path, passcode)
      .then((success) => {
        this.scan();
        return success;
      });
  }

  static get (api) {
    if (!instance) {
      instance = new HardwareStore(api);
    }

    return instance;
  }
}

export {
  HW_SCAN_INTERVAL
};