poa-dapps-keys-generation/src/App.js

import React, { Component } from 'react';
import getWeb3 from './getWeb3'
import KeysManager from './keysManager';
import Keys from './Keys';
import swal from 'sweetalert';
import './index/index.css';
import addressGenerator from './addressGenerator'
import JSzip from 'jszip';
import FileSaver from 'file-saver';
import { constants } from './constants';
import networkAddresses from './addresses';
import Header from './Header';
import Footer from './Footer';
import Loading from './Loading';

function generateElement(msg){
  let errorNode = document.createElement("div");
  errorNode.innerHTML = `<div style="line-height: 1.6;">
    ${msg}
  </div>`;
  return errorNode;
}

class App extends Component {
  constructor(props){
    super(props);
    this.onClick = this.onClick.bind(this);
    this.saveFile = (blob) => { 
      FileSaver.saveAs(blob, `poa_network_validator_keys.zip`);
    };
    this.state = {
      web3Config: {},
      mining: null,
      isDisabledBtn: props.generateKeysIsDisabled
    }
    this.keysManager = null;
    getWeb3().then(async (web3Config) => {
      return networkAddresses(web3Config)
    }).then(async (config) => {
      const {web3Config, addresses} = config;
      this.keysManager = new KeysManager();
      await this.keysManager.init({
        web3: web3Config.web3Instance,
        netId: web3Config.netId,
        addresses,
      });
      this.setState({
        isDisabledBtn: false,
        web3Config
      })
    }).catch((error) => {
      if(error.msg){
        this.setState({isDisabledBtn: true});
        swal({
          icon: 'warning',
          title: 'Warning',
          content: error.node
        });
      }
    })
  }
  componentDidMount(){
    if(window.location.hash.indexOf('just-generate-keys') !== -1) {
      this.setState({loading:true});
      setTimeout(async () => {
        const {mining, voting, payout} = await this.generateKeys();
        this.setState({loading:false});
        await this.generateZip({mining, voting, payout, netIdName: "manualCreation"});
      }, 150)
    }
  }
  async generateKeys(cb) {
    const mining = await addressGenerator();
    const voting = await addressGenerator();
    const payout = await addressGenerator();
    this.setState({
      mining,
      voting,
      payout,
      keysGenerated: true
    })
    return {
      mining, voting, payout
    }
  }
  async generateZip({mining, voting, payout, netIdName}){
    const zip = new JSzip();
    zip.file(`${netIdName}_keys/mining_key_${mining.jsonStore.address}.json`, JSON.stringify(mining.jsonStore));
    zip.file(`${netIdName}_keys/mining_password_${mining.jsonStore.address}.txt`, mining.password);

    zip.file(`${netIdName}_keys/voting_key_${voting.jsonStore.address}.json`, JSON.stringify(voting.jsonStore));
    zip.file(`${netIdName}_keys/voting_password_${voting.jsonStore.address}.txt`, voting.password);

    zip.file(`${netIdName}_keys/payout_key_${payout.jsonStore.address}.json`, JSON.stringify(payout.jsonStore));
    zip.file(`${netIdName}_keys/payout_password_${payout.jsonStore.address}.txt`, payout.password);
    zip.generateAsync({type:"blob"}).then((blob) => {
      FileSaver.saveAs(blob, `poa_network_validator_keys.zip`);
    });
  }
  async onClick() {
    this.setState({loading:true});
    const initialKey = window.web3.eth.defaultAccount;
    let isValid
    try {
      isValid = await this.keysManager.isInitialKeyValid(initialKey);
    } catch(e) {
      isValid = false;
    }
    console.log(isValid);
    if(Number(isValid) !== 1){
      this.setState({loading:false});
      const invalidKeyMsg = `The key is invalid initial Key<br/>
      or you're connect to incorrect chain!<br/>
      Please make sure you have loaded correct initial key in Metamask.<br/><br/>
      <b>Your current selected key is</b> <i>${initialKey}</i><br/>
      <b>Current Network ID</b> is <i>${this.state.web3Config.netId}</i>`
      swal({
        icon: 'error',
        title: 'Error',
        content: generateElement(invalidKeyMsg)
      })
      return;
    }
    if(Number(isValid) === 1){
      const {mining, voting, payout} = await this.generateKeys()
      // add loading screen
      await this.keysManager.createKeys({
        mining: mining.jsonStore.address,
        voting: voting.jsonStore.address,
        payout: payout.jsonStore.address,
        sender: initialKey
      }).then(async (receipt) => {
        console.log(receipt);
        if (receipt.status === true || receipt.status === "0x1") {
          this.setState({loading: false})
          swal("Congratulations!", "Your keys are generated!", "success");
          await this.generateZip({mining, voting, payout, netIdName: this.state.web3Config.netIdName});
        } else {
          this.setState({loading: false, keysGenerated: false})
          let content = document.createElement("div");
          let msg = `Transaction failed`;
          content.innerHTML = `<div>
            Something went wrong!<br/><br/>
            Please contact Master Of Ceremony<br/><br/>
            ${msg}
          </div>`;
          swal({
            icon: 'error',
            title: 'Error',
            content: content
          });
        }
      }).catch((error) => {
        console.error(error.message);
        this.setState({loading: false, keysGenerated: false})
        let content = document.createElement("div");
        let msg;
        if (error.message.includes(constants.userDeniedTransactionPattern))
          msg = `Error: User ${constants.userDeniedTransactionPattern}`
        else
          msg = error.message
        content.innerHTML = `<div>
          Something went wrong!<br/><br/>
          Please contact Master Of Ceremony<br/><br/>
          ${msg}
        </div>`;
        swal({
          icon: 'error',
          title: 'Error',
          content: content
        });
      })
    }
  }
  render() {    
    let loader = this.state.loading ? <Loading netId={this.state.web3Config.netId}/> : '';
    let createKeyBtn = (<div className="create-keys">
              <h1>Create keys from initial key</h1>
              <h2>
                In this application, you will create mining, payout and voting keys.
                The app will make your initial key unusable after the process.
                Please proceed with care, don't lose your keys and follow instructions.
              </h2>
                <div className="create-keys-button-container">
                  <button className="create-keys-button" onClick={this.onClick} disabled={this.state.isDisabledBtn}>Generate keys</button>
                
                </div>
            </div>)
    let content;
    if(this.state.keysGenerated){
      content = <Keys mining={this.state.mining} voting={this.state.voting} payout={this.state.payout}/> 
    } else {
      content = createKeyBtn
    }
    return (
      <div className="App">
        <Header netId={this.state.web3Config.netId}/>
        {loader}
        <section className="content">
          {content}
        </section>
        <Footer netId={this.state.web3Config.netId}/>
      </div>
    );
  }
}

export default App;