1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192 |
- // ``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 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"
- const show_text = "Show the prompts and output"
- const button = document.createElement("span")
- button.classList.add("copybutton")
- button.innerText = ">>>"
- button.title = hide_text
- button.dataset.hidden = "false"
- const buttonClick = event => {
- // define the behavior of the button when it's clicked
- event.preventDefault()
- const buttonEl = event.currentTarget
- const codeEl = buttonEl.nextElementSibling
- if (buttonEl.dataset.hidden === "false") {
- // hide the code output
- for (const el of getHideableCopyButtonElements(codeEl)) {
- el.hidden = true
- }
- buttonEl.title = show_text
- buttonEl.dataset.hidden = "true"
- } else {
- // show the code output
- for (const el of getHideableCopyButtonElements(codeEl)) {
- el.hidden = false
- }
- buttonEl.title = hide_text
- buttonEl.dataset.hidden = "false"
- }
- }
- const highlightedElements = document.querySelectorAll(
- ".highlight-python .highlight,"
- + ".highlight-python3 .highlight,"
- + ".highlight-pycon .highlight,"
- + ".highlight-pycon3 .highlight,"
- + ".highlight-default .highlight"
- )
- // create and add the button to all the code blocks that contain >>>
- highlightedElements.forEach(el => {
- el.style.position = "relative"
- // if we find a console prompt (.gp), prepend the (deeply cloned) button
- const clonedButton = button.cloneNode(true)
- // the onclick attribute is not cloned, set it on the new element
- clonedButton.onclick = buttonClick
- if (el.querySelector(".gp") !== null) {
- el.prepend(clonedButton)
- }
- })
- }
- if (document.readyState !== "loading") {
- loadCopyButton()
- } else {
- document.addEventListener("DOMContentLoaded", loadCopyButton)
- }
|