Set encoding for handles at compile time

Author: Abigail Magalhães

bin/Amc/Explain/TH.hs

@@ -1,21 +1,33 @@
 {-# LANGUAGE TemplateHaskell #-}
 module Amc.Explain.TH where
 
+import Control.Exception
+
 import Language.Haskell.TH.Syntax
 import Language.Haskell.TH
 
 import System.FilePath
+import System.IO
 
 qReadFile :: FilePath -> Q String
 qReadFile path = do
   qAddDependentFile path
-  str <- qRunIO (readFile path)
-  length str `seq` pure str
+  qRunIO $
+    withFile path ReadMode $ \handle -> do
+      hSetEncoding handle utf8
+      s <- hGetContents handle
+      _ <- evaluate (length s)
+      pure s
 
 qReadFilesList :: FilePath -> Q [FilePath]
 qReadFilesList path = do
   qAddDependentFile path
-  contents <- qRunIO (readFile path)
+  contents <- qRunIO $
+    withFile path ReadMode $ \handle -> do
+      hSetEncoding handle utf8
+      s <- hGetContents handle
+      _ <- evaluate (length s)
+      pure s
   let files = lines contents
       fileP = map (head . words) files
   pure fileP

tests/driver/Main.hs

@@ -20,6 +20,8 @@ import qualified Test.Lua.Parser as LParser
 import qualified Test.Frontend.Amc as Amc
 import qualified Test.Lsp as Lsp
 
+import GHC.IO.Encoding
+
 tests :: IO TestTree
 tests = testGroup "Tests" <$> sequence
   -- These two will timeout if you run 10000 of them.
@@ -40,7 +42,9 @@ tests = testGroup "Tests" <$> sequence
   ]
 
 main :: IO ()
-main = tests >>= defaultMainWithIngredients ingredients where
+main = locale *> test where
+  locale = setLocaleEncoding utf8
+  test = tests >>= defaultMainWithIngredients ingredients
   ingredients =
     [ rerunning
       [ listingTests