fix: fixes #194 - adds support for encryption events

Author: jarrodek

package-lock.json

@@ -80,6 +80,7 @@
     "codemirror": "^5.49.0",
     "crypto-js": "^3.1.9-1",
     "electron-log": "^3.0.8",
+    "electron-prompt": "^1.3.1",
     "electron-updater": "4.1.2",
     "form-data": "^2.5.1",
     "fs-extra": "^8.1.0",

package.json

@@ -1,5 +1,6 @@
 const AES = require('crypto-js/aes.js');
 const CryptoJS = require('crypto-js/crypto-js.js');
+const prompt = require('electron-prompt');
 /**
  * A class that handles `encryption-*` web events in the renderer process
  * and performs data encryption/decryption.
@@ -26,11 +27,13 @@ class EncryptionService {
 
   _decodeHandler(e) {
     const { method } = e.detail;
+    e.preventDefault();
     e.detail.result = this.decode(method, e.detail);
   }
 
   _encodeHandler(e) {
     const { method } = e.detail;
+    e.preventDefault();
     e.detail.result = this.encode(method, e.detail);
   }
 
@@ -56,11 +59,22 @@ class EncryptionService {
   }
 
   async decodeAes(data, passphrase) {
-    if (!passphrase === undefined) {
-      passphrase = prompt('Enter password to open the file.');
+    if (passphrase === undefined) {
+      const win = require('electron').remote.getCurrentWindow();
+      passphrase = await prompt({
+        title: 'File password',
+        label: 'Enter password to open the file',
+      }, win);
+      if (passphrase === null) {
+        throw new Error('Password is required to open the file.');
+      }
+    }
+    try {
+      const bytes = AES.decrypt(data, passphrase);
+      return bytes.toString(CryptoJS.enc.Utf8);
+    } catch (_) {
+      throw new Error('Invalid password.');
     }
-    const bytes = AES.decrypt(data, passphrase);
-    return bytes.toString(CryptoJS.enc.Utf8);
   }
 }
 exports.EncryptionService = EncryptionService;

scripts/packages/encryption/renderer/encryption.js

@@ -0,0 +1 @@
+exports.EncryptionService = require('./encryption').EncryptionService;

scripts/packages/encryption/renderer/index.js

@@ -0,0 +1,62 @@
+const { assert } = require('chai');
+const { EncryptionService } = require('../renderer');
+
+describe('EncryptionService', function() {
+  describe('Encrypting content', () => {
+    let instance;
+    before(() => {
+      instance = new EncryptionService();
+      instance.listen();
+    });
+
+    after(() => {
+      instance.unlisten();
+    });
+
+    it('encodes a content using AES.encrypt', async () => {
+      const e = new CustomEvent('encryption-encode', {
+        bubbles: true,
+        cancelable: true,
+        detail: {
+          method: 'aes',
+          data: 'test-data',
+          passphrase: 'test'
+        }
+      });
+      document.body.dispatchEvent(e);
+      const result = await e.detail.result;
+      assert.typeOf(result, 'string');
+      assert.notEqual(result, 'test-data');
+    });
+  });
+
+
+  describe('Decrypting content', () => {
+    const encoded = 'U2FsdGVkX19MRBu8liXK/sbNBOQWG1mewbjEBb8cTAw=';
+    let instance;
+    before(() => {
+      instance = new EncryptionService();
+      instance.listen();
+    });
+
+    after(() => {
+      instance.unlisten();
+    });
+
+    it('decodes a content using AES.decrypt and provided password', async () => {
+      const e = new CustomEvent('encryption-decode', {
+        bubbles: true,
+        cancelable: true,
+        detail: {
+          method: 'aes',
+          data: encoded,
+          passphrase: 'test'
+        }
+      });
+      document.body.dispatchEvent(e);
+      const result = await e.detail.result;
+      assert.typeOf(result, 'string');
+      assert.equal(result, 'test-data');
+    });
+  });
+});

scripts/packages/encryption/spec/encryption.renderer.spec.js

@@ -57,6 +57,7 @@ class ArcElectron extends ArcAppMixin(LitElement) {
     /* global log */
     this.log = log;
     this.sysVars = process.env;
+    this.withEncrypt = true;
   }
 
   connectedCallback() {