encryption.js

'use strict';

const NodeRSA = require('node-rsa');
const crypto = require('crypto')

let generateKeyPair = (password) => {
    let key = new NodeRSA();
    key.generateKeyPair()
    let publicKey = Buffer.from(key.exportKey('public')).toString('base64')
    let privateKey = Buffer.from(key.exportKey('private')).toString('base64')
    if(password) {
        privateKey = encryptUsingPassword(privateKey, password)
    }
    return {publicKey, privateKey}
} 

let decryptUsingKeys = (privateKeyString, encryptedMessage) => {  
    let privateKey = Buffer.from(privateKeyString, 'base64').toString()
    let key = {
        key: privateKey,
        padding: crypto.constants.RSA_PKCS1_PADDING
    }
    let message = crypto.privateDecrypt(key, Buffer.from(encryptedMessage, 'base64')).toString()
    return message
}

let encryptUsingKeys = (publicKeyString, message) => {
    let publicKey = Buffer.from(publicKeyString, 'base64').toString()
    let key = {
      key: publicKey,
      padding: crypto.constants.RSA_PKCS1_PADDING
    }
    let encryptedMessage = crypto.publicEncrypt(key, Buffer.from(message)).toString('base64')
    return encryptedMessage
}


module.exports = {
    generate: generateKeyPair,
    decrypt: decryptUsingKeys,
    encrypt: encryptUsingKeys
}