From dfc2f0faec4b89c3f13425c6e370dce4ad476445 Mon Sep 17 00:00:00 2001
From: Crozzers <captaincrozzers@gmail.com>
Date: Sat, 17 Feb 2024 14:29:02 +0000
Subject: [PATCH] Add `prepend` arg to toc extra

---
 lib/markdown2.py                   |  2 +-
 test/tm-cases/toc_prepend.html     | 40 ++++++++++++++++++++++++++++++
 test/tm-cases/toc_prepend.opts     |  1 +
 test/tm-cases/toc_prepend.tags     |  1 +
 test/tm-cases/toc_prepend.text     | 20 +++++++++++++++
 test/tm-cases/toc_prepend.toc_html | 20 +++++++++++++++
 6 files changed, 83 insertions(+), 1 deletion(-)
 create mode 100644 test/tm-cases/toc_prepend.html
 create mode 100644 test/tm-cases/toc_prepend.opts
 create mode 100644 test/tm-cases/toc_prepend.tags
 create mode 100644 test/tm-cases/toc_prepend.text
 create mode 100644 test/tm-cases/toc_prepend.toc_html

diff --git a/lib/markdown2.py b/lib/markdown2.py
index 3a1db281..01252ff8 100755
--- a/lib/markdown2.py
+++ b/lib/markdown2.py
@@ -513,7 +513,7 @@ def toc_sort(entry):
             self._toc_html = calculate_toc_html(self._toc)
 
             # Prepend toc html to output
-            if self.cli:
+            if self.cli or (self.extras['toc'] is not None and self.extras['toc'].get('prepend', False)):
                 text = '{}\n{}'.format(self._toc_html, text)
 
         text += "\n"
diff --git a/test/tm-cases/toc_prepend.html b/test/tm-cases/toc_prepend.html
new file mode 100644
index 00000000..7aa0a0e5
--- /dev/null
+++ b/test/tm-cases/toc_prepend.html
@@ -0,0 +1,40 @@
+<ul>
+  <li><a href="#readme-for-blah">README for Blah</a>
+  <ul>
+    <li><a href="#introduction">Introduction</a></li>
+    <li><a href="#the-meat">The Meat</a>
+    <ul>
+      <li><a href="#beef">Beef</a>
+      <ul>
+        <li><a href="#steak">Steak</a></li>
+        <li><a href="#burgers">Burgers</a></li>
+      </ul></li>
+      <li><a href="#chicken">Chicken</a></li>
+      <li><a href="#pork">Pork</a>
+      <ul>
+        <li><a href="#mmmmmmmm-bacon">Mmmmmmmm, bacon</a></li>
+      </ul></li>
+    </ul></li>
+  </ul></li>
+  <li><a href="#at-the-top-level-again">At the <em>top</em> level again!?</a></li>
+</ul>
+
+<h1 id="readme-for-blah">README for Blah</h1>
+
+<h2 id="introduction">Introduction</h2>
+
+<h2 id="the-meat">The Meat</h2>
+
+<h3 id="beef">Beef</h3>
+
+<h5 id="steak">Steak</h5>
+
+<h5 id="burgers">Burgers</h5>
+
+<h3 id="chicken">Chicken</h3>
+
+<h3 id="pork">Pork</h3>
+
+<h4 id="mmmmmmmm-bacon">Mmmmmmmm, bacon</h4>
+
+<h1 id="at-the-top-level-again">At the <em>top</em> level again!?</h1>
diff --git a/test/tm-cases/toc_prepend.opts b/test/tm-cases/toc_prepend.opts
new file mode 100644
index 00000000..261151cd
--- /dev/null
+++ b/test/tm-cases/toc_prepend.opts
@@ -0,0 +1 @@
+{"extras": {"toc": {"prepend": True}}}
diff --git a/test/tm-cases/toc_prepend.tags b/test/tm-cases/toc_prepend.tags
new file mode 100644
index 00000000..2b2472e0
--- /dev/null
+++ b/test/tm-cases/toc_prepend.tags
@@ -0,0 +1 @@
+toc extra
diff --git a/test/tm-cases/toc_prepend.text b/test/tm-cases/toc_prepend.text
new file mode 100644
index 00000000..34f629a6
--- /dev/null
+++ b/test/tm-cases/toc_prepend.text
@@ -0,0 +1,20 @@
+# README for Blah
+
+## Introduction
+
+## The Meat
+
+### Beef
+
+##### Steak
+
+##### Burgers
+
+### Chicken
+
+### Pork
+
+#### Mmmmmmmm, bacon
+
+# At the *top* level again!?
+
diff --git a/test/tm-cases/toc_prepend.toc_html b/test/tm-cases/toc_prepend.toc_html
new file mode 100644
index 00000000..08288703
--- /dev/null
+++ b/test/tm-cases/toc_prepend.toc_html
@@ -0,0 +1,20 @@
+<ul>
+  <li><a href="#readme-for-blah">README for Blah</a>
+  <ul>
+    <li><a href="#introduction">Introduction</a></li>
+    <li><a href="#the-meat">The Meat</a>
+    <ul>
+      <li><a href="#beef">Beef</a>
+      <ul>
+        <li><a href="#steak">Steak</a></li>
+        <li><a href="#burgers">Burgers</a></li>
+      </ul></li>
+      <li><a href="#chicken">Chicken</a></li>
+      <li><a href="#pork">Pork</a>
+      <ul>
+        <li><a href="#mmmmmmmm-bacon">Mmmmmmmm, bacon</a></li>
+      </ul></li>
+    </ul></li>
+  </ul></li>
+  <li><a href="#at-the-top-level-again">At the <em>top</em> level again!?</a></li>
+</ul>