1 files changed, 29 insertions, 71 deletions

diff --git a/static/js/codecopy.js b/static/js/codecopy.js
index c1f3a75..9e00c79 100644
--- a/static/js/codecopy.js
+++ b/static/js/codecopy.js
@@ -1,76 +1,34 @@
-// https://aaronluna.dev/blog/add-copy-button-to-code-blocks-hugo-chroma/
-// Further modified for Zola by Jieiku: https://github.com/Jieiku
-(function() {
-  // create copy button
-  function ccb(highlightDiv) {
-    const button = document.createElement("button");
-    button.className = "cc ";
-    button.type = "button";
-    button.innerHTML = "Copy";
-    button.addEventListener("click", () =>
-      ccc(button, highlightDiv)
-    );
+const changeIcon = (copyDiv, className) => {
+    copyDiv.classList.add(className);
+    setTimeout(() => copyDiv.classList.remove(className), 2500);
+};
 
-    // addCopyButtonToDom
-    const wrapper = document.createElement("div");
-    wrapper.className = "ccw";
-    wrapper.appendChild(button);
-    highlightDiv.parentNode.insertBefore(wrapper, highlightDiv);
-    wrapper.appendChild(highlightDiv);
-  }
-
-  // copy code clipboard
-  async function ccc(button, highlightDiv) {
-    //look for table, if so do it different
-    const { length } = highlightDiv.querySelectorAll("table");
-    let codeToCopy = "";
-    if (length > 0) {
-      const items = highlightDiv.querySelectorAll(":last-child > tr > td:last-child");
-      items.forEach(item => {
-        codeToCopy = codeToCopy + item.innerText;
-        //console.log(codeToCopy);
-      });
-    } else {
-      codeToCopy = highlightDiv.querySelector(":last-child > code").innerText;
-    }
+const copyCodeAndChangeIcon = async (copyDiv, block) => {
+    const code = block.querySelector('table') ? getTableCode(block) : getNonTableCode(block);
     try {
-      result = await navigator.permissions.query({ name: "clipboard-write" });
-      if (result.state == "granted" || result.state == "prompt") {
-        await navigator.clipboard.writeText(codeToCopy);
-      } else {
-        cce(codeToCopy, highlightDiv);//copyCodeBlockExecCommand
-      }
-    } catch (_) {
-      cce(codeToCopy, highlightDiv);//copyCodeBlockExecCommand
-    } finally {
-      //codeWasCopied
-      //button.blur();
-      button.innerHTML = "Copied";
-      setTimeout(function () {
-        button.innerHTML = "Copy";
-      }, 2000);
+        await navigator.clipboard.writeText(code);
+        changeIcon(copyDiv, "yes");
+    } catch (error) {
+        changeIcon(copyDiv, "err");
     }
-  }
+};
+
+const getNonTableCode = (block) => {
+    return [...block.querySelectorAll('code')]
+        .map(code => code.textContent)
+        .join('');
+};
 
-  //copyCodeBlockExecCommand
-  function cce(codeToCopy, highlightDiv) {
-    const textArea = document.createElement("textArea");
-    textArea.contentEditable = "true";
-    textArea.readOnly = "false";
-    textArea.className = "copyable-text-area";
-    textArea.value = codeToCopy;
-    highlightDiv.insertBefore(textArea, highlightDiv.firstChild);
-    const range = document.createRange();
-    range.selectNodeContents(textArea);
-    const sel = window.getSelection();
-    sel.removeAllRanges();
-    sel.addRange(range);
-    textArea.setSelectionRange(0, 999999);
-    document.execCommand("copy");
-    highlightDiv.removeChild(textArea);
-  }
+const getTableCode = (block) => {
+    return [...block.querySelectorAll('tr')]
+        .map(row => row.querySelector('td:last-child')?.innerText ?? '')
+        .join('');
+};
 
-  document
-    .querySelectorAll("pre")
-    .forEach((highlightDiv) => ccb(highlightDiv));
-})();
\ No newline at end of file
+document.querySelectorAll("pre").forEach((block) => {
+    const copyDiv = document.createElement("div");
+    copyDiv.className = "cc svgs svgh copy";
+    copyDiv.innerHTML = " ";
+    block.prepend(copyDiv);
+    copyDiv.addEventListener("click", () => copyCodeAndChangeIcon(copyDiv, block));;
+});