From 4603dd25d9d448e3c8d9d1280cb59c9a66906cf7 Mon Sep 17 00:00:00 2001
From: Marvin Borner
Date: Wed, 23 Jan 2019 22:31:53 +0100
Subject: Added prettier exporting and encryption check

---
 public/scripts/2_encryption.js | 25 ++++++++++++++++++-------
 1 file changed, 18 insertions(+), 7 deletions(-)

(limited to 'public/scripts/2_encryption.js')

diff --git a/public/scripts/2_encryption.js b/public/scripts/2_encryption.js
index 4d05780..9140a3a 100644
--- a/public/scripts/2_encryption.js
+++ b/public/scripts/2_encryption.js
@@ -16,10 +16,11 @@ async function generateKeys(userId, passphrase) {
         passphrase: passphrase
     };
 
-    openpgp.generateKey(options).then((key) => {
+    await openpgp.generateKey(options).then((key) => {
         localStorage.setItem('private_key', key.privateKeyArmored);
         localStorage.setItem('public_key', key.publicKeyArmored);
         localStorage.setItem('revocation_certificate', key.revocationCertificate);
+        console.log('[LOG] Successfully generated and stored keys!')
     });
 }
 
@@ -71,6 +72,17 @@ async function decrypt(data, publicKey, privateKey, passphrase) {
     })
 }
 
+/**
+ * Checks whether the user has keys
+ * @returns {boolean}
+ */
+function isEncrypted() {
+    const hasPrivateKey = localStorage.getItem('private_key') !== null;
+    const hasPublicKey = localStorage.getItem('public_key') !== null;
+    const hasRevocationCertificate = localStorage.getItem('revocation_certificate') !== null;
+    return (hasPrivateKey && hasPublicKey && hasRevocationCertificate);
+}
+
 /**
  * Just a general test case
  */
@@ -85,9 +97,8 @@ function testEncryption() {
     })
 }
 
-// TODO: Export as module or sth
-// REMEMBER: UGLY EXPORT!
-window.generateKeys = generateKeys;
-window.encrypt = encrypt;
-window.decrypt = decrypt;
-window.testEncryption = testEncryption;
+exports.generate = generateKeys;
+exports.encrypt = encrypt;
+exports.decrypt = decrypt;
+exports.check = isEncrypted;
+exports.test = testEncryption;
-- 
cgit v1.2.3