Passthrough node now respects indentation

Closes brandonbloom#66
lread · Oct 3, 2019 · 08b5071 · 08b5071
1 parent aa4b946
commit 08b5071
Show file tree

Hide file tree

Showing 2 changed files with 67 additions and 18 deletions.
diff --git a/src/fipp/engine.cljc b/src/fipp/engine.cljc
@@ -1,8 +1,7 @@
 (ns fipp.engine
   "See: Oleg Kiselyov, Simon Peyton-Jones, and Amr Sabry
   Lazy v. Yield: Incremental, Linear Pretty-printing"
-  (:require [clojure.string :as s]
-            [fipp.deque :as deque])
+  (:require [fipp.deque :as deque])
   #?(:clj (:import (java.io Writer))))
 
 
@@ -175,26 +174,25 @@
         ([] (rf))
         ([res] (rf res))
         ([res {:keys [op right] :as node}]
-         (let [indent (peek @tab-stops)]
+         (let [indent (peek @tab-stops)
+               format-text (fn [text width]
+                             (let [res* (if (zero? @column)
+                                          (do (vswap! column + indent)
+                                              (rf res (apply str (repeat indent \space))))
+                                          res)]
+                               (vswap! column + width)
+                               (rf res* text)))]
            (case op
              :text
-               (let [text (:text node)
-                     res* (if (zero? @column)
-                            (do (vswap! column + indent)
-                                (rf res (apply str (repeat indent \space))))
-                            res)]
-                 (vswap! column + (count text))
-                 (rf res* text))
+               (let [text (:text node)]
+                 (format-text text (count text)))
+
              :escaped
-               (let [text (:text node)
-                     res* (if (zero? @column)
-                            (do (vswap! column + indent)
-                                (rf res (apply str (repeat indent \space))))
-                            res)]
-                 (vswap! column inc)
-                 (rf res* text))
+               (format-text (:text node) 1)
+
              :pass
-               (rf res (:text node))
+               (format-text (:text node) 0)
+
              :line
                (if (zero? @fits)
                  (do

diff --git a/test/fipp/engine_test.cljc b/test/fipp/engine_test.cljc
@@ -157,6 +157,57 @@
     (is (= (ppstr [:group (repeat 5 [:span [:escaped "&#97;"] :line])] 10)
            (str (apply str (repeat 5 "&#97; ")) "\n")))))
 
+(deftest passthrough-node-test
+  ;; reminder to reader: :pass nodes are to be used with non-visible characters such as ANSI escape codes
+  (testing "indentation without :pass (as baseline for comparison with subsequent tests)"
+    (is (= (ppstr [:group
+                   "AA"
+                   [:align
+                    :line
+                    "BB"
+                    :line
+                    "CC"
+                    [:align
+                     :line
+                     "DD"]]] 6)
+           (str "AA\n"
+                "  BB\n"
+                "  CC\n"
+                "    DD\n"))))
+  (testing ":pass nodes have a width of zero and respect indentation"
+    (is (= (ppstr [:group
+                   [:span [:pass "<"] "AA" [:pass ">"]]
+                   [:align
+                    :line
+                    [:span [:pass "<"] "BB" [:pass ">>"]]
+                    :line
+                    [:span [:pass "<<<<<<<<<<"] "CC" [:pass ">>>"]]
+                    [:align
+                     :line
+                     [:span [:pass "<"] "DD" [:pass ">"]]]]] 6)
+           (str "<AA>\n"
+                "  <BB>>\n"
+                "  <<<<<<<<<<CC>>>\n"
+                "    <DD>\n"))))
+  (testing ":pass nodes can be used at to affect indent whitespace by placing them before newline"
+    (is (= (ppstr [:group
+                   "AA"
+                   [:align
+                    [:pass "<"]
+                    :line
+                    "BB"
+                    [:pass "<"]
+                    :line
+                    "CC"
+                    [:align
+                     [:pass "<"]
+                     :line
+                     [:span "DD"]]]] 6)
+           (str "AA<\n"
+                "  BB<\n"
+                "  CC<\n"
+                "    DD\n")))) )
+
 (deftest terminate-test
   (is (= (ppstr [:group "a" [:line "-" ";"] "b"] 1000)
          "a-b\n"))