Cool trees

1 files changed, 15 insertions, 4 deletions

diff --git a/src/Fun/Compiler.hs b/src/Fun/Compiler.hs
index a8134ed..4d9a67a 100644
--- a/src/Fun/Compiler.hs
+++ b/src/Fun/Compiler.hs
@@ -6,10 +6,21 @@ import           Fun.Tree
 import           System.Exit
 import           System.IO
 
-genTrace :: [String] -> String
-genTrace xs =
-  "Trace of expectance: "
-    ++ foldr (\a b -> a ++ if b == "" then b else " -> " ++ b) "" xs
+traceBranch :: Int -> Trace -> String
+traceBranch c (StringTrace t) = "\n" ++ (replicate c ' ') ++ t
+traceBranch c (OrTrace t1 t2) =
+  "\n"
+    ++ (replicate c ' ')
+    ++ "<either>"
+    ++ (traceTree (c + 2) t1)
+    ++ (traceTree (c + 2) t2)
+
+traceTree :: Int -> [Trace] -> String -- TODO: Indent/arrow first map
+traceTree c ts = foldr join "" (map (traceBranch c) ts)
+  where join = (\a b -> a ++ if b == "" then b else " " ++ b)
+
+genTrace :: [Trace] -> String
+genTrace ts = "Trace of expectance:\n" ++ traceTree 0 ts
 
 parse :: String -> Either String Program -- TODO: Should be tree
 parse file = case program file of