From e024c7b69b2c9e10117e2bd9e117279d871d3a6d Mon Sep 17 00:00:00 2001
From: eddieoz <3277320+eddieoz@users.noreply.github.com>
Date: Tue, 14 May 2024 12:17:11 +0300
Subject: [PATCH] Injected button 'Encrypt' aside of 'send' button

---
 src/content.js | 93 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 93 insertions(+)

diff --git a/src/content.js b/src/content.js
index e21afe9..e49613b 100644
--- a/src/content.js
+++ b/src/content.js
@@ -355,6 +355,99 @@ async function setSessionPassphrase(passphrase) {
   sessionStorage.setItem("sessionPassphrase", passphrase);
 }
 
+// Function to replace the text in the input with the encrypted version
+function replaceTextInInput(replacementText) {
+  const messageInput = document.querySelector('[contenteditable="true"][data-testid="dmComposerTextInput"]');
+  if (messageInput) {
+    messageInput.innerText = replacementText;
+    messageInput.value = replacementText;
+    console.log('message input: ', messageInput)
+
+    // Trigger the input event to update the DOM
+    const event = new Event('input', { bubbles: true });
+    messageInput.dispatchEvent(event);
+  }
+}
+
+async function handleEncryptAndSend() {
+  const messageInput = document.querySelector('[contenteditable="true"][data-testid="dmComposerTextInput"]');
+  if (messageInput) {
+    // Select the entire text in the input box
+    const range = document.createRange();
+    range.selectNodeContents(messageInput);
+    const selection = window.getSelection();
+    selection.removeAllRanges();
+    selection.addRange(range);
+
+    // Retrieve the text from the selection
+    const messageText = selection.toString().trim();
+    console.log('original msg:', messageText);
+
+    if (messageText) {
+      const twitterHandle = findTwitterHandle();
+      const extensionUserHandle = findUsernameFromInitialState();
+
+      try {
+        const recipientPublicKey = await retrieveUserPublicKey(twitterHandle);
+        const extensionUserPublicKey = await retrieveUserPublicKeyFromPrivate(extensionUserHandle);
+
+        const base64Text = btoa(encodeURIComponent(`${messageText} 🔒`).replace(/%([0-9A-F]{2})/g, (match, p1) => String.fromCharCode('0x' + p1)));
+        const xryptDocument = {
+          "event": "xrypt.msg.new",
+          "params": {
+            "content": {
+              "type": "text",
+              "text": base64Text
+            }
+          }
+        }
+
+        const encryptedText = await encryptTextPGP(JSON.stringify(xryptDocument), [recipientPublicKey, extensionUserPublicKey]);
+        console.log('encrypted msg:', encryptedText);
+
+        // Replace the selected text with the encrypted text
+        replaceSelectedText(encryptedText);
+
+        // Optionally, click the original send button
+        const sendButton = document.querySelector('[data-testid="dmComposerSendButton"]');
+        // if (sendButton) sendButton.click();
+      } catch (err) {
+        console.error('Failed to encrypt text:', err);
+        alert('Failed to encrypt text.');
+      }
+    } else {
+      alert('Message text cannot be empty.');
+    }
+  } else {
+    alert('Message input not found.');
+  }
+}
+
+
+
+// Function to inject Encrypt button before the Send button
+function injectEncryptButton() {
+  const sendButton = document.querySelector('[data-testid="dmComposerSendButton"]');
+  if (sendButton && !document.querySelector('#encryptAndSendButton')) {
+    const encryptButton = document.createElement('button');
+    encryptButton.id = 'encryptAndSendButton';
+    encryptButton.innerText = 'Encrypt';
+    encryptButton.style.marginRight = '10px'; // Add some space between buttons
+    sendButton.parentNode.insertBefore(encryptButton, sendButton);
+
+    // Add click event listener to the new button
+    encryptButton.addEventListener('click', handleEncryptAndSend);
+  }
+}
+
+// Call the function to inject the button
+injectEncryptButton();
+
+// Observe changes in the DOM to ensure the button is always injected
+const observer = new MutationObserver(injectEncryptButton);
+observer.observe(document.body, { childList: true, subtree: true });
+
+
 // Listen for messages from the popup or background script
 chrome.runtime.onMessage.addListener((request, sender, sendResponse) => {
   if (request.action === "encryptText") {