import React, { Component } from 'react' import PropTypes from 'prop-types' import classnames from 'classnames' import PubNub from 'pubnub' import qrCode from 'qrcode-generator' import { DEFAULT_ROUTE } from '../../helpers/constants/routes' import Button from '../../components/ui/button' import LoadingScreen from '../../components/ui/loading-screen' const PASSWORD_PROMPT_SCREEN = 'PASSWORD_PROMPT_SCREEN' const REVEAL_SEED_SCREEN = 'REVEAL_SEED_SCREEN' const KEYS_GENERATION_TIME = 30000 export default class MobileSyncPage extends Component { static contextTypes = { t: PropTypes.func, } static propTypes = { history: PropTypes.object.isRequired, selectedAddress: PropTypes.string.isRequired, displayWarning: PropTypes.func.isRequired, fetchInfoToSync: PropTypes.func.isRequired, requestRevealSeedWords: PropTypes.func.isRequired, } state = { screen: PASSWORD_PROMPT_SCREEN, password: '', seedWords: null, error: null, syncing: false, completed: false, channelName: undefined, cipherKey: undefined, } syncing = false componentDidMount () { const passwordBox = document.getElementById('password-box') if (passwordBox) { passwordBox.focus() } } handleSubmit (event) { event.preventDefault() this.setState({ seedWords: null, error: null }) this.props.requestRevealSeedWords(this.state.password) .then((seedWords) => { this.startKeysGeneration() this.setState({ seedWords, screen: REVEAL_SEED_SCREEN }) }) .catch((error) => this.setState({ error: error.message })) } startKeysGeneration () { this.handle && clearTimeout(this.handle) this.disconnectWebsockets() this.generateCipherKeyAndChannelName() this.initWebsockets() this.handle = setTimeout(() => { this.startKeysGeneration() }, KEYS_GENERATION_TIME) } generateCipherKeyAndChannelName () { this.cipherKey = `${this.props.selectedAddress.substr(-4)}-${PubNub.generateUUID()}` this.channelName = `mm-${PubNub.generateUUID()}` this.setState({ cipherKey: this.cipherKey, channelName: this.channelName }) } initWithCipherKeyAndChannelName (cipherKey, channelName) { this.cipherKey = cipherKey this.channelName = channelName } initWebsockets () { // Make sure there are no existing listeners this.disconnectWebsockets() this.pubnub = new PubNub({ subscribeKey: process.env.PUBNUB_SUB_KEY, publishKey: process.env.PUBNUB_PUB_KEY, cipherKey: this.cipherKey, ssl: true, }) this.pubnubListener = { message: (data) => { const { channel, message } = data // handle message if (channel !== this.channelName || !message) { return false } if (message.event === 'start-sync') { this.startSyncing() } else if (message.event === 'connection-info') { this.handle && clearTimeout(this.handle) this.disconnectWebsockets() this.initWithCipherKeyAndChannelName(message.cipher, message.channel) this.initWebsockets() } else if (message.event === 'end-sync') { this.disconnectWebsockets() this.setState({ syncing: false, completed: true }) } }, } this.pubnub.addListener(this.pubnubListener) this.pubnub.subscribe({ channels: [this.channelName], withPresence: false, }) } disconnectWebsockets () { if (this.pubnub && this.pubnubListener) { this.pubnub.removeListener(this.pubnubListener) } } // Calculating a PubNub Message Payload Size. calculatePayloadSize (channel, message) { return encodeURIComponent( channel + JSON.stringify(message) ).length + 100 } chunkString (str, size) { const numChunks = Math.ceil(str.length / size) const chunks = new Array(numChunks) for (let i = 0, o = 0; i < numChunks; ++i, o += size) { chunks[i] = str.substr(o, size) } return chunks } notifyError (errorMsg) { return new Promise((resolve, reject) => { this.pubnub.publish( { message: { event: 'error-sync', data: errorMsg, }, channel: this.channelName, sendByPost: false, // true to send via post storeInHistory: false, }, (status, response) => { if (!status.error) { resolve() } else { reject(response) } }) }) } async startSyncing () { if (this.syncing) { return false } this.syncing = true this.setState({ syncing: true }) const { accounts, network, preferences, transactions } = await this.props.fetchInfoToSync() const allDataStr = JSON.stringify({ accounts, network, preferences, transactions, udata: { pwd: this.state.password, seed: this.state.seedWords, }, }) const chunks = this.chunkString(allDataStr, 17000) const totalChunks = chunks.length try { for (let i = 0; i < totalChunks; i++) { await this.sendMessage(chunks[i], i + 1, totalChunks) } } catch (e) { this.props.displayWarning('Sync failed :(') this.setState({ syncing: false }) this.syncing = false this.notifyError(e.toString()) } } sendMessage (data, pkg, count) { return new Promise((resolve, reject) => { this.pubnub.publish( { message: { event: 'syncing-data', data, totalPkg: count, currentPkg: pkg, }, channel: this.channelName, sendByPost: false, // true to send via post storeInHistory: false, }, (status, response) => { if (!status.error) { resolve() } else { reject(response) } } ) }) } componentWillUnmount () { this.disconnectWebsockets() } renderWarning (text) { return (