From 2920f6ca1f09f42aca9bd5b1e8ee5817c4b382b3 Mon Sep 17 00:00:00 2001
From: Greg Sexton <gregsexton@gmail.com>
Date: Sat, 19 Apr 2014 12:36:49 +0100
Subject: [PATCH] Search for node containing point + misc

---
 origami.el | 45 +++++++++++++++++++++++++++++++++------------
 1 file changed, 33 insertions(+), 12 deletions(-)

diff --git a/origami.el b/origami.el
index 7a5c548..e81b6c4 100644
--- a/origami.el
+++ b/origami.el
@@ -64,7 +64,8 @@
         (vector beg end open sorted-children data)))))
 
 (defun origami-fold-top-level-node (&optional children)
-  (origami-fold-node 0 0 t children))
+  ;; TODO: fix min and max
+  (origami-fold-node 0 10000 t children))
 
 (defun origami-fold-beg (node) (when node (aref node 0)))
 
@@ -72,6 +73,15 @@
 
 (defun origami-fold-open-p (node) (when node (aref node 2)))
 
+(defun origami-fold-open-set (path value)
+  (let* ((old-node (-last-item path))
+         (new-node (origami-fold-node (origami-fold-beg old-node)
+                                      (origami-fold-end old-node)
+                                      value
+                                      (origami-fold-children old-node)
+                                      (origami-fold-data old-node))))
+    (origami-fold-assoc path new-node)))
+
 (defun origami-fold-children (node &optional children)
   (when node
     (if children
@@ -90,15 +100,6 @@ used to nil out data. This mutates the node."
         (aset node 4 data)
       (aref node 4))))
 
-(defun origami-fold-open-set (path value)
-  (let* ((old-node (-last-item path))
-         (new-node (origami-fold-node (origami-fold-beg old-node)
-                                      (origami-fold-end old-node)
-                                      value
-                                      (origami-fold-children old-node)
-                                      (origami-fold-data old-node))))
-    (origami-fold-assoc path new-node)))
-
 (defun origami-fold-range-equal (a b)
   (and (equal (origami-fold-beg a) (origami-fold-beg b))
        (equal (origami-fold-end a) (origami-fold-end b))))
@@ -145,6 +146,21 @@ used to nil out data. This mutates the node."
   (-each (origami-fold-children node) f)
   (funcall f node))
 
+(defun origami-fold-find-node-containing (tree point)
+  "Return the path to the most specific (deepest) node that
+contains point, or null."
+  (cl-labels
+      ((node-contains? (point node)
+                       (and (<= (origami-fold-beg node) point)
+                            (>= (origami-fold-end node) point)))
+       (get-child-containing (children point)
+                             (-first (lambda (n) (node-contains? point n)) children)))
+    (when tree
+      (when (node-contains? point tree)
+        (-if-let (child (get-child-containing (origami-fold-children tree) point))
+            (cons tree (origami-fold-find-node-containing child point))
+          (list tree))))))
+
 ;;; overlay manipulation
 
 (defun origami-create-overlay (beg end buffer text)
@@ -297,13 +313,16 @@ used to nil out data. This mutates the node."
 
 (defun origami-store-cached-tree (buffer tree)
   ;; TODO:
-  )
+  tree)
 
 (defun origami-build-tree (buffer parser)
   (when parser
     (with-current-buffer buffer
       (let ((contents (buffer-string)))
-        (origami-run-parser parser (origami-content 0 contents))))))
+        (-> parser
+          (origami-run-parser (origami-content 0 contents))
+          car
+          origami-fold-top-level-node)))))
 
 (defun origami-get-parser (buffer)
   ;; TODO: remove hardcoding!
@@ -339,8 +358,10 @@ otherwise fetch cached tree."
 (defun origami-open-node (buffer point)
   (interactive (list (current-buffer) (point)))
   (let ((tree (origami-get-fold-tree buffer)))
+    (debug-msg "tree: %s" tree)
     (-when-let (path (origami-fold-find-node-containing tree point))
       ;; TODO: don't actually compute + apply the diff
+      (debug-msg "path: %s" path)
       (origami-store-cached-tree buffer
                                  (origami-fold-open-set path t)))))