accept filenames with .lc extension

1 files changed, 5 insertions, 2 deletions

diff --git a/tool/Compiler.hs b/tool/Compiler.hs
index aab95deb..1cdc9fd9 100644
--- a/tool/Compiler.hs
+++ b/tool/Compiler.hs
@@ -32,8 +32,11 @@ main = do
 
 compile :: Config -> IO ()
 compile Config{..} = do
-  pplRes <- compileMain [".", sourceDir] backend srcName
+  let dropExt n | takeExtension n == ".lc"  = dropExtension n
+      dropExt n = n
+      baseName = dropExt srcName
+  pplRes <- compileMain [".", sourceDir] backend baseName
   case pplRes of
     Left err -> putStrLn err
     Right ppl -> do
-      B.writeFile (srcName <> ".json") $ encode ppl
+      B.writeFile (baseName <> ".json") $ encode ppl