From af04f30d2057d69148a3ee6c752e0786203cc6d6 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sun, 14 Feb 2016 18:48:16 +0000
Subject: Generate back links for notes

---
 src/main/java/org/zwobble/mammoth/DocumentConverter.java | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

(limited to 'src/main/java/org')

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) {
-- 
cgit v1.2.3