From 4746aa52f85f4dc3ce8e195f0a5fd8afe2d54378 Mon Sep 17 00:00:00 2001
From: tomsmeding <tom.smeding@gmail.com>
Date: Mon, 18 Nov 2019 18:35:40 +0100
Subject: Fix includes relative to lisp file

---
 Main.hs | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

(limited to 'Main.hs')

diff --git a/Main.hs b/Main.hs
index b56edfe..ef98b62 100644
--- a/Main.hs
+++ b/Main.hs
@@ -1,3 +1,4 @@
+{-# LANGUAGE LambdaCase, TupleSections #-}
 module Main where
 
 import System.Environment
@@ -16,13 +17,12 @@ usage = do
 
 main :: IO ()
 main = do
-    clargs <- getArgs
-    source <- case clargs of
-        [] -> getContents
-        [fname] -> readFile fname
+    (mfname, source) <- getArgs >>= \case
+        [] -> (Nothing,) <$> getContents
+        [arg] -> (Just arg,) <$> readFile arg
         _ -> usage >> exitFailure
 
-    prog <- parseProgram source >>= either (die . show) return
+    prog <- parseProgram mfname source >>= either (die . show) return
     irprog <- either die return (compileProgram prog)
     let opt = optimise irprog
     -- print opt
-- 
cgit v1.2.3-54-g00ecf