fix extension support, by selecting all enabled extensions

bitc · Jul 25, 2017 · 94a4543 · 94a4543
1 parent 07ea161
commit 94a4543
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/Main.hs b/src/Main.hs
@@ -17,7 +17,7 @@ module Main (main) where
 import Data.List (isPrefixOf, partition)
 import Data.Vector(Vector, fromList)
 import Language.Haskell.Exts.Annotated (parseFileContentsWithMode, ParseMode(..), knownExtensions, ParseResult(ParseOk, ParseFailed))
-import Language.Haskell.Exts.Extension (Language(..))
+import Language.Haskell.Exts.Extension (Language(..), Extension(EnableExtension, DisableExtension))
 import System.Environment (getArgs, getProgName)
 import System.IO (hPutStrLn, stderr)
 import qualified Data.Text as T (unpack, lines, pack, unlines)
@@ -55,12 +55,14 @@ processFile file ignore_parse_error = do
         parseMode = ParseMode
             { parseFilename = file
             , baseLanguage = Haskell2010
-            , extensions = knownExtensions
+            , extensions = filter isEnabled knownExtensions
             , ignoreLanguagePragmas = False
             , ignoreLinePragmas = True
             , fixities = Nothing
             , ignoreFunctionArity = False
             }
+        isEnabled (EnableExtension _) = True
+        isEnabled _ = False
 
 loadFile :: FilePath -> IO (String, Vector String)
 loadFile file = do