refactor: split out fix for #5

Author: KyleKing

mdformat_footnote/plugin.py

@@ -32,47 +32,18 @@ def _footnote_renderer(node: RenderTreeNode, context: RenderContext) -> str:
     first_line = f"[^{node.meta['label']}]:"
     indent = " " * 4
     elements = []
-
-    first_child_idx = 0
-    while (
-        first_child_idx < len(node.children)
-        and node.children[first_child_idx].type == "footnote_anchor"
-    ):
-        first_child_idx += 1
-
-    if (
-        first_child_idx < len(node.children)
-        and node.children[first_child_idx].type == "paragraph"
-    ):
-        with context.indented(len(first_line) + 1):
-            first_element = node.children[first_child_idx].render(context)
-
-        first_element_lines = first_element.split("\n")
-        first_para_first_line = first_element_lines[0]
-        first_para_rest_lines = first_element_lines[1:]
-
-        with context.indented(len(indent)):
-            for child in node.children[first_child_idx + 1 :]:
-                if child.type == "footnote_anchor":
-                    continue
-                elements.append(child.render(context))
-
-        result = first_line + " " + first_para_first_line
-        if first_para_rest_lines:
-            indented_rest = textwrap.indent("\n".join(first_para_rest_lines), indent)
-            result += "\n" + indented_rest
-        if elements:
-            result += "\n\n" + textwrap.indent("\n\n".join(elements), indent)
-        return result
-
     with context.indented(len(indent)):
         for child in node.children:
             if child.type == "footnote_anchor":
                 continue
             elements.append(child.render(context))
     body = textwrap.indent("\n\n".join(elements), indent)
-    if body:
+    # if the first body element is a paragraph, we can start on the first line,
+    # otherwise we start on the second line
+    if body and node.children and node.children[0].type != "paragraph":
         body = "\n" + body
+    else:
+        body = " " + body.lstrip()
     return first_line + body
 
 

tests/test_word_wrap.py

@@ -13,10 +13,9 @@ def test_word_wrap():
     expected_output = """\
 [^a]
 
-[^a]: Ooh no, the first line of this
-    first paragraph is still wrapped
-    too wide unfortunately. Should fix
-    this.
+[^a]: Ooh no, the first line of this first
+    paragraph is still wrapped too wide
+    unfortunately. Should fix this.
 
     But this second paragraph is wrapped
     exactly as expected. Woohooo,