Incorporated most of main, as well as started on quickcheck

src/Main.hs

@@ -2,42 +2,81 @@
 
 module Main where
 
-import           Grammar.Par        (myLexer, pProgram)
--- import           TypeChecker.TypeChecker (typecheck)
+import           Codegen.Codegen           (compile)
+import           GHC.IO.Handle.Text        (hPutStrLn)
+import           Grammar.ErrM              (Err)
+import           Grammar.Par               (myLexer, pProgram)
+import           Grammar.Print             (printTree)
 
-import           Grammar.Print      (printTree)
-import           Renamer.RenamerM   (rename)
-import           System.Environment (getArgs)
-import           System.Exit        (exitFailure, exitSuccess)
-import           TypeChecker.AlgoW  (typecheck)
+import           LambdaLifter.LambdaLifter (lambdaLift)
+import           Renamer.Renamer           (rename)
+import           System.Environment        (getArgs)
+import           System.Exit               (exitFailure, exitSuccess)
+import           System.IO                 (stderr)
+import           TypeChecker.TypeChecker   (typecheck)
 
 main :: IO ()
-main = getArgs >>= \case
+main =
+    getArgs >>= \case
         [] -> print "Required file path missing"
-        (x : _) -> do
-            file <- readFile x
-            case pProgram (myLexer file) of
-                Left err -> do
-                    putStrLn "SYNTAX ERROR"
-                    putStrLn err
-                    exitFailure
-                Right prg -> do
-                    putStrLn ""
-                    putStrLn " ----- PARSER ----- "
-                    putStrLn ""
-                    putStrLn . printTree $ prg
-                    case typecheck (rename prg) of
-                        Left err -> do
-                            putStrLn "TYPECHECK ERROR"
-                            print err
-                            exitFailure
-                        Right prg -> do
-                            putStrLn ""
-                            putStrLn " ----- RAW ----- "
-                            putStrLn ""
-                            print prg
-                            putStrLn ""
-                            putStrLn " ----- TYPECHECKER ----- "
-                            putStrLn ""
-                            putStrLn $ printTree prg
-                            exitSuccess
+        (s : _) -> main' s
+
+main' :: String -> IO ()
+main' s = do
+    file <- readFile s
+
+    printToErr "-- Parse Tree -- "
+    parsed <- fromSyntaxErr . pProgram $ myLexer file
+    printToErr $ printTree parsed
+
+    printToErr "\n-- Renamer --"
+    let renamed = rename parsed
+    printToErr $ printTree renamed
+
+    printToErr "\n-- TypeChecker --"
+    typechecked <- fromTypeCheckerErr $ typecheck renamed
+    printToErr $ printTree typechecked
+
+    printToErr "\n-- Lambda Lifter --"
+    let lifted = lambdaLift typechecked
+    printToErr $ printTree lifted
+
+    printToErr "\n -- Printing compiler output to stdout --"
+    compiled <- fromCompilerErr $ compile lifted
+    putStrLn compiled
+    writeFile "llvm.ll" compiled
+
+    exitSuccess
+
+printToErr :: String -> IO ()
+printToErr = hPutStrLn stderr
+
+fromCompilerErr :: Err a -> IO a
+fromCompilerErr =
+    either
+        ( \err -> do
+            putStrLn "\nCOMPILER ERROR"
+            putStrLn err
+            exitFailure
+        )
+        pure
+
+fromSyntaxErr :: Err a -> IO a
+fromSyntaxErr =
+    either
+        ( \err -> do
+            putStrLn "\nSYNTAX ERROR"
+            putStrLn err
+            exitFailure
+        )
+        pure
+
+fromTypeCheckerErr :: Err a -> IO a
+fromTypeCheckerErr =
+    either
+        ( \err -> do
+            putStrLn "\nTYPECHECKER ERROR"
+            putStrLn err
+            exitFailure
+        )
+        pure