cuts

1 files changed, 25 insertions, 20 deletions

diff --git a/app/Main.hs b/app/Main.hs
index be4febe..c2f03ee 100644
--- a/app/Main.hs
+++ b/app/Main.hs
@@ -1,27 +1,32 @@
 module Main where
 
 import Interpreter
-
+import Text.Pretty.Simple
 import qualified Data.Map as M
 
+ppr :: Show a => a -> IO ()
+ppr = pPrintOpt CheckColorTty defaultOutputOptionsDarkBg {outputOptionsCompactParens = True, outputOptionsIndentAmount = 2, outputOptionsPageWidth=80}
+
 main :: IO ()
-main =
-  print $
-  prove [Goal, U (Struct (1, 2)), U (Atom 1), U (Atom 2), LastCall] $
-  M.fromList
-    [ ( (1, 2)
-      , [ [U (Atom 333), U (Atom 444), NoGoal]
-        , [ U (Atom 1)
-          , U (Atom 2)
-          , Goal
-          , U (Struct (2, 0))
-          , Call
-          , Goal
-          , U (Struct (1, 2))
-          , U (Atom 333)
-          , U (Atom 444)
-          , LastCall
+main = do
+  let (res, interp) =
+        prove [Goal, U (Struct (1, 2)), U (Atom 1), U (Atom 2), LastCall] $
+        M.fromList
+          [ ( (1, 2)
+            , [ [U (Atom 333), U (Atom 444), NoGoal]
+              , [ U (Atom 1)
+                , U (Atom 2)
+                , Goal
+                , U (Struct (2, 0))
+                , Call
+                , Goal
+                , U (Struct (5, 2))
+                , U (Atom 333)
+                , U (Atom 444)
+                , LastCall
+                ]
+              ])
+          , ((2, 0), [[NoGoal]])
           ]
-        ])
-    , ((2, 0), [[NoGoal]])
-    ]
+  ppr interp
+  ppr res