Introduce getHideableCopyButtonElements()

AA-Turner · AA-Turner · commit 5cf2065afd5e · 2023-04-11T15:15:18.000+01:00
diff --git a/python_docs_theme/static/copybutton.js b/python_docs_theme/static/copybutton.js
@@ -1,6 +1,36 @@
+// ``function*`` denotes a generator in JavaScript, see
+// https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Statements/function*
+function* getHideableCopyButtonElements(rootElement) {
+    // yield all elements with the "go" (Generic.Output),
+    // "gp" (Generic.Prompt), or "gt" (Generic.Traceback) CSS class
+    for (const el of rootElement.querySelectorAll('.go, .gp, .gt')) {
+        yield el
+    }
+    // tracebacks (.gt) contain bare text elements that need to be
+    // wrapped in a span to hide or show the element
+    for (let el of rootElement.querySelectorAll('.gt')) {
+        while ((el = el.nextSibling) && el.nodeType !== Node.DOCUMENT_NODE) {
+            // stop wrapping text nodes when we hit the next output or
+            // prompt element
+            if (el.nodeType === Node.ELEMENT_NODE && el.matches(".gp, .go")) {
+                break
+            }
+            // if the node is a text node with content, wrap it in a
+            // span element so that we can control visibility
+            if (el.nodeType === Node.TEXT_NODE && el.textContent.trim()) {
+                const wrapper = document.createElement('span')
+                el.after(wrapper)
+                wrapper.appendChild(el)
+                el = wrapper
+            }
+            yield el
+        }
+    }
+}
+
 
 const loadCopyButton = () => {
-    /* Add a [>>>] button on the top-right corner of code samples to hide
+    /* Add a [>>>] button in the top-right corner of code samples to hide
      * the >>> and ... prompts and the output and thus make the code
      * copyable. */
     const hide_text = 'Hide the prompts and output'
@@ -18,24 +48,16 @@ const loadCopyButton = () => {
         const codeEl = buttonEl.nextElementSibling
         if (buttonEl.dataset.hidden === 'false') {
             // hide the code output
-            codeEl.querySelectorAll('.go, .gp, .gt').forEach(el => el.hidden = true)
-            codeEl.querySelectorAll('.gt').forEach(el => {
-                while ((el = el.nextSibling) && el.nodeType !== Node.DOCUMENT_NODE) {
-                    if (el.nodeType === Node.ELEMENT_NODE && el.matches(".gp, .go")) break;
-                    if (el.nodeType === Node.TEXT_NODE && el.textContent.trim()) {
-                        const wrapper = document.createElement('span');
-                        el.after(wrapper);
-                        wrapper.appendChild(el);
-                        el = wrapper
-                    }
-                    el.hidden = true
-                }
-            })
+            for (const el of getHideableCopyButtonElements(codeEl)) {
+                el.hidden = true
+            }
             buttonEl.title = show_text
             buttonEl.dataset.hidden = "true"
         } else {
             // show the code output
-            codeEl.childNodes.forEach(el => el.hidden = false)
+            for (const el of getHideableCopyButtonElements(codeEl)) {
+                el.hidden = false
+            }
             buttonEl.title = hide_text
             buttonEl.dataset.hidden = "false"
         }
