From b3fa50bf64494b2e096af966d41b964543b287f6 Mon Sep 17 00:00:00 2001
From: Rob Rix <rob.rix@github.com>
Date: Fri, 23 Oct 2015 19:08:10 -0400
Subject: [PATCH] Map the original diff.

---
 prototype/UI/index.html | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/prototype/UI/index.html b/prototype/UI/index.html
index e930ffbbc..a0495b9b5 100644
--- a/prototype/UI/index.html
+++ b/prototype/UI/index.html
@@ -185,9 +185,15 @@
     <div id="after"></div>
     <script type="text/javascript">
     loadJSON('diff.json', function (json) {
-        var model = new Diff(json.diff);
-        document.getElementById("before").appendChild(toDOM(model, "before", json["before"]));
-        document.getElementById("after").appendChild(toDOM(model, "after", json["after"]));
+        var diff = diffFromJSON(json.diff);
+        var mapped = diff.map(function(value) {
+            return {
+                before: value.before != null ? toDOM(value.before, "before", source) : document.createTextNode(""),
+                before: value.after != null ? toDOM(value.after, "after", source) : document.createTextNode(""),
+            };
+        });
+        document.getElementById("before").appendChild(toDOM(mapped, "before", json["before"]));
+        document.getElementById("after").appendChild(toDOM(mapped, "after", json["after"]));
     });
     </script>
   </body>