1 files changed, 23 insertions, 14 deletions
diff --git a/src/Main.hs b/src/Main.hs
index ea41afe..92e9a1d 100644
--- a/src/Main.hs
+++ b/src/Main.hs
@@ -1,16 +1,19 @@
 module Main where
 
-import qualified Language.Fiddle.Tokenizer
-import qualified Language.Fiddle.Parser
-import Language.Fiddle.Ast
-import qualified Data.Text.IO
-import qualified System.Environment as System
 import Control.Monad (forM_)
 import Control.Monad.Writer
-import qualified Language.Fiddle.Parser
-import Language.Fiddle.GenericTree (ToGenericSyntaxTree(toGenericSyntaxTree))
 import Data.Aeson (encode)
 import qualified Data.ByteString.Lazy.Char8 as BL
+import qualified Data.Text.IO
+import GHC.IO.Exception (ExitCode (ExitFailure))
+import Language.Fiddle.Ast
+import Language.Fiddle.Compiler (coloredFormat, compile_, printDiagnostic)
+import Language.Fiddle.Compiler.Stage0
+import Language.Fiddle.GenericTree (ToGenericSyntaxTree (toGenericSyntaxTree))
+import qualified Language.Fiddle.Parser
+import qualified Language.Fiddle.Tokenizer
+import qualified System.Environment as System
+import System.Exit (exitWith)
 
 main :: IO ()
 main = do
@@ -18,10 +21,16 @@ main = do
 
   case argv of
     [filePath] -> do
-       text <- Data.Text.IO.readFile filePath
-       case squeeze =<< Language.Fiddle.Parser.parseFiddleText filePath text of
-         Left pe -> putStrLn $  "Parse Error: " ++ show pe
-         Right ast -> do
-           putStrLn (BL.unpack $ encode $ toGenericSyntaxTree $ fmap (const ()) ast)
-
-    _ -> putStrLn "Wrong Args"
+      text <- Data.Text.IO.readFile filePath
+      let (diags, ma) = compile_ $ toStage1 =<< toStage0 filePath text
+      forM_ diags printDiagnostic
+      case ma of
+        Just ast -> do
+          putStrLn "\x1b[1;32mCompilation Succeeded:\x1b[0m"
+          putStrLn $ BL.unpack $ encode $ toGenericSyntaxTree ast
+        Nothing -> do
+          putStrLn "\x1b[1;31mCompilation Failed\x1b[0m"
+          exitWith (ExitFailure 1)
+    _ -> do
+      putStrLn "Wrong Args"
+      exitWith (ExitFailure 2)