exchangemarket/setup/reset-password.js

const fs = require('fs');
const getInput = require('./getInput');

const floGlobals = require('../public/floGlobals');
require('../src/set_globals');
require('../src/lib');
require('../src/floCrypto');

console.log(__dirname);

function validateKey(privKey) {
    return new Promise((resolve, reject) => {
        if (floCrypto.verifyPrivKey(privKey, floGlobals.adminID))
            return resolve(privKey);
        else {
            getInput.Text('Incorrect Private Key! Re-Enter: (Cancel)', 'Cancel').then(value => {
                if (value === 'Cancel')
                    return reject(true);
                validateKey(value)
                    .then(result => resolve(result))
                    .catch(error => reject(error))
            });
        }
    })
}

function getPassword() {
    return new Promise((resolve, reject) => {
        getInput.Text(`Enter a password [Minimum 8 characters]`, 'Cancel').then(value1 => {
            if (value1 === 'Cancel')
                return reject(true);
            else if (value1.length < 8) {
                console.log('Password length must be minimum of 8 characters');
                getPassword()
                    .then(result => resolve(result))
                    .catch(error => reject(error))
            } else {
                getInput.Text(`Re-enter password`).then(value2 => {
                    if (value1 !== value2) {
                        console.log('Passwords doesnot match! Try again.');
                        getPassword()
                            .then(result => resolve(result))
                            .catch(error => reject(error))
                    } else 
                        resolve(value1);
                })
            }
        });
    })
}

function resetPassword() {
    return new Promise((resolve, reject) => {
        getInput.Text(`Enter private key for adminID (${floGlobals.adminID})`).then(value => {
            validateKey(value).then(privKey => {
                getPassword().then(password => {
                    let encrypted = Crypto.AES.encrypt(privKey, password);
                    let randNum = floCrypto.randInt(10, 15);
                    let splitShares = floCrypto.createShamirsSecretShares(encrypted, randNum, randNum);
                    fs.writeFile(__dirname + '/../args/keys.json', JSON.stringify(splitShares), 'utf8', (err) => {
                        if (err) {
                            console.error(err);
                            return reject(false);
                        }
                        console.log('Password reset successful!');
                        resolve(true);
                    })
                }).catch(error => {
                    console.log('Password reset cancelled!');
                    reject(true);
                })
            }).catch(error => {
                console.log('Password reset cancelled!');
                reject(true);
            })
        })
    })
}

if (!module.parent)
    resetPassword().then(_ => null).catch(error => console.error(error));
else
    module.exports = resetPassword;