From adf3e1b7695e83b0bfb7e91257e11aea89bb5c84 Mon Sep 17 00:00:00 2001
From: Roman Vishnevskii <roman.vishnevsky@xored.com>
Date: Wed, 5 Apr 2017 18:00:22 +0700
Subject: [PATCH] DOM diff problem: max prefix and suffix mutable

---
 karax.nim | 39 +++++++++++++++++++++++++++++----------
 1 file changed, 29 insertions(+), 10 deletions(-)

diff --git a/karax.nim b/karax.nim
index 888a2e9..ac3b432 100644
--- a/karax.nim
+++ b/karax.nim
@@ -119,22 +119,41 @@ proc updateElement(parent, current: Node, newNode, oldNode: VNode) =
   elif newNode.kind != VNodeKind.text:
     let newLength = newNode.len
     let oldLength = oldNode.len
+    let minLength = min(newLength, oldLength)
     assert oldNode.kind == newNode.kind
     when false:
       if current.nodeName != toTag[oldNode.kind]:
         kout current.nodeName
         kout toTag[oldNode.kind]
         assert false
-    for i in 0..min(newLength, oldLength)-1:
-      updateElement(current, current.childNodes[i],
-        newNode[i],
-        oldNode[i])
-    if newLength > oldLength:
-      for i in oldLength..newLength-1:
-        current.appendChild(vnodeToDom(newNode[i]))
-    elif oldLength > newLength:
-      for i in countdown(oldLength-1, newLength):
-        current.removeChild(current.lastChild)
+    
+    var commonPrefix = 0
+    while commonPrefix < minLength and newNode[commonPrefix] == oldNode[commonPrefix]:
+      inc commonPrefix
+
+    var oldPos = oldLength - 1
+    var newPos = newLength - 1
+    while oldPos >= 0 and newPos >= 0 and newNode[newPos] == oldNode[oldPos]:
+      dec oldPos
+      dec newPos
+
+    var pos = commonPrefix
+    while pos <= newPos and pos <= oldPos:
+      updateElement(current, current.childNodes[pos],
+        newNode[pos],
+        oldNode[pos])
+      inc pos
+    
+    while pos <= newPos:
+      if newPos == newLength - 1:
+        current.appendChild(vnodeToDom(newNode[pos]))
+      else:
+        current.insertBefore(vnodeToDom(newNode[pos]), current.childNodes[pos + 1])
+      inc pos
+
+    while pos <= oldPos:
+      current.removeChild(current.childNodes[pos])
+      inc pos
 
 proc dodraw() =
   let newtree = dorender()