From 6d51409fd3f0321d99e66fbed4187af36abe1aac Mon Sep 17 00:00:00 2001
From: Peter Bengtsson <mail@peterbe.com>
Date: Fri, 6 Dec 2019 04:08:11 -0500
Subject: [PATCH] scoop up the right pre elements' text content (#6194)

Fixes #6161
---
 kuma/javascript/src/live-examples.js | 18 ++++++------------
 1 file changed, 6 insertions(+), 12 deletions(-)

diff --git a/kuma/javascript/src/live-examples.js b/kuma/javascript/src/live-examples.js
index b779ced309c..740e6ed0e62 100644
--- a/kuma/javascript/src/live-examples.js
+++ b/kuma/javascript/src/live-examples.js
@@ -48,20 +48,14 @@ export function addLiveExampleButtons(rootElement) {
         // direct siblings and also descendants of direct siblings
         // (because sometimes some of the source code is tucked inside
         // a hidden div).
-        let html = rootElement.querySelector(
-            `#${sectid} ~ pre[class*=html], #${sectid} ~ * pre[class*=html]`
-        );
-        let css = rootElement.querySelector(
-            `#${sectid} ~ pre[class*=css], #${sectid} ~ * pre[class*=css]`
-        );
-        let js = rootElement.querySelector(
-            `#${sectid} ~ pre[class*=js], #${sectid} ~ * pre[class*=js]`
-        );
+        let html = section.querySelectorAll('pre[class*=html]');
+        let css = section.querySelectorAll('pre[class*=css]');
+        let js = section.querySelectorAll('pre[class*=js]');
 
         // Now get the source code out of those pre elements
-        let htmlCode = html ? html.textContent : '';
-        let jsCode = js ? js.textContent : '';
-        let cssCode = css ? css.textContent : '';
+        let htmlCode = [...html].map(p => p.textContent).join('');
+        let jsCode = [...js].map(p => p.textContent).join('');
+        let cssCode = [...css].map(p => p.textContent).join('');
 
         // If we found any source code, then add buttons
         if (htmlCode || cssCode || jsCode) {