Generate back links for notes

1 files changed, 8 insertions, 1 deletions

diff --git a/src/main/java/org/zwobble/mammoth/DocumentConverter.java b/src/main/java/org/zwobble/mammoth/DocumentConverter.java
index 6f59cba..47e24ee 100644
--- a/src/main/java/org/zwobble/mammoth/DocumentConverter.java
+++ b/src/main/java/org/zwobble/mammoth/DocumentConverter.java
@@ -66,7 +66,14 @@ public class DocumentConverter {
 
     private HtmlNode convertToHtml(Note note) {
         String id = generateNoteHtmlId(note.getNoteType(), note.getId());
-        return Html.element("li", map("id", id), convertToHtml(note.getBody()));
+        String referenceId = generateNoteRefHtmlId(note.getNoteType(), note.getId());
+        ImmutableList.Builder<HtmlNode> children = ImmutableList.builder();
+        children.addAll(convertToHtml(note.getBody()));
+        // TODO: should be collapsible
+        children.add(Html.element("p", list(
+            Html.text(" "),
+            Html.element("a", map("href", "#" + referenceId), list(Html.text("↑"))))));
+        return Html.element("li", map("id", id), children.build());
     }
 
     private List<HtmlNode> convertToHtml(List<DocumentElement> elements) {