1 files changed, 14 insertions, 16 deletions
diff --git a/app/Main.hs b/app/Main.hs
index 9c62589..60a44f0 100644
--- a/app/Main.hs
+++ b/app/Main.hs
@@ -7,30 +7,28 @@ import           System.Environment             ( getArgs )
 import           Term
 import           Transpile
 
-transpile :: IO ()
-transpile = do
-  args <- getArgs
-  file <- readFile (head args)
+transpile :: String -> IO ()
+transpile path = do
+  file <- readFile path
   let term = fromBLC file
-  putStrLn $ "input: " ++ show term
-  let ski = transpileSKI term
-  putStrLn $ "SKI transpiled: " ++ show ski
+  let ski  = transpileSKI term
   let birb = transpileBirb ski
-  putStrLn $ "Birb transpiled: " ++ show birb
-  return ()
+  putStrLn $ concatMap show birb
 
-reduce :: IO ()
-reduce = do
-  args <- getArgs
-  file <- readFile (head args)
+reduce :: String -> IO ()
+reduce path = do
+  file <- readFile path
   let termified   = fromBirbs file
   let rebirbified = fromTerm termified
   putStrLn $ "input: " ++ rebirbified
   normalBirbs <- nf termified
   let retermified = fromTerm normalBirbs
   putStrLn $ "reduced: " ++ retermified
-  return ()
 
 main :: IO ()
-main = transpile
--- main = mapM_ (bruteForce "...") [1 .. 10]
+main = do
+  args <- getArgs
+  case args of
+    ["transpile", path] -> transpile path
+    ["reduce"   , path] -> reduce path
+    _                   -> putStrLn "Usage: birb [transpile|reduce] <file>"