From e52034606e8f389865072bc4b6e997324deaa26f Mon Sep 17 00:00:00 2001
From: Sebastian Landwehr <info@sebastianlandwehr.com>
Date: Sun, 24 Nov 2024 16:49:35 +0000
Subject: [PATCH] fix: fix deep nested code in pre

---
 src/commonmark-rules.js | 13 ++++++-------
 test/turndown-test.js   | 18 ++++++++++++++++++
 2 files changed, 24 insertions(+), 7 deletions(-)

diff --git a/src/commonmark-rules.js b/src/commonmark-rules.js
index f32a8933..ab2fc554 100644
--- a/src/commonmark-rules.js
+++ b/src/commonmark-rules.js
@@ -84,15 +84,14 @@ rules.indentedCodeBlock = {
     return (
       options.codeBlockStyle === 'indented' &&
       node.nodeName === 'PRE' &&
-      node.firstChild &&
-      node.firstChild.nodeName === 'CODE'
+      node.querySelector('code')
     )
   },
 
   replacement: function (content, node, options) {
     return (
       '\n\n    ' +
-      node.firstChild.textContent.replace(/\n/g, '\n    ') +
+      node.querySelector('code').textContent.replace(/\n/g, '\n    ') +
       '\n\n'
     )
   }
@@ -103,15 +102,15 @@ rules.fencedCodeBlock = {
     return (
       options.codeBlockStyle === 'fenced' &&
       node.nodeName === 'PRE' &&
-      node.firstChild &&
-      node.firstChild.nodeName === 'CODE'
+      node.querySelector('code')
     )
   },
 
   replacement: function (content, node, options) {
-    var className = node.firstChild.getAttribute('class') || ''
+    var codeElement = node.querySelector('code')
+    var className = codeElement.getAttribute('class') || ''
     var language = (className.match(/language-(\S+)/) || [null, ''])[1]
-    var code = node.firstChild.textContent
+    var code = codeElement.textContent
 
     var fenceChar = options.fence.charAt(0)
     var fenceSize = 3
diff --git a/test/turndown-test.js b/test/turndown-test.js
index a7ab0b2a..727c3891 100644
--- a/test/turndown-test.js
+++ b/test/turndown-test.js
@@ -178,3 +178,21 @@ test('remove elements are overridden by keep', function (t) {
     'Hello <del>world</del><ins>World</ins>'
   )
 })
+
+test('deep nested code in pre is converted to code block (indented)', function (t) {
+  t.plan(1)
+  var turndownService = new TurndownService()
+  t.equal(turndownService.turndown(
+    '<pre><div><code>Hi</code></div></pre>'),
+    '    Hi'
+  )
+})
+
+test('deep nested code in pre is converted to code block (fenced)', function (t) {
+  t.plan(1)
+  var turndownService = new TurndownService({ codeBlockStyle: 'fenced' })
+  t.equal(turndownService.turndown(
+    '<pre><div><code>Hi</code></div></pre>'),
+    '```\nHi\n```'
+  )
+})