From e567e041339a4c9f81dfa003b2beb2174acf182f Mon Sep 17 00:00:00 2001
From: Mystikfluu <Mystikfluu@gmail.com>
Date: Fri, 19 Aug 2022 19:51:08 +0200
Subject: [PATCH] enhance simple e2ee

---
 js/dms.js | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/js/dms.js b/js/dms.js
index 8a0c6d4..d081528 100644
--- a/js/dms.js
+++ b/js/dms.js
@@ -52,6 +52,8 @@ function set_keys(s_key) {
   })
   
   encryption_keys = packed
+
+  main()
 }
 
 async function postMessage() {
@@ -66,7 +68,7 @@ async function postMessage() {
     let text = document.getElementById("post-text").value
 
     if(typeof encrypt == "function" && encryption_keys != "") {
-      text = encrypt(test,{
+      text = encrypt(text,{
         packed: encryption_keys
       })
     }
@@ -245,6 +247,9 @@ async function createPost(username,text,time,specialtext,postid,isbot,reply_id,a
     }
   }
 
+  if(typeof decrypt == "function" && encryption_keys != "") {
+    text = decrypt(text,{packed:encryption_keys})
+  }
   newDiv.appendChild(newP)
   newDiv.innerHTML += filterPost(text)
   newDiv.id = postid