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

@@ -18,7 +18,9 @@ import qualified Test.Types.Check as TypesC
 import qualified Test.Core.Backend as Backend
 import qualified Test.Lua.Parser as LParser
 import qualified Test.Frontend.Amc as Amc
-import qualified Test.Lsp as Lsp
+mport qualified Test.Lsp as Lsp
+
+import GHC.IO.Encoding
 
 tests :: IO TestTree
 tests = testGroup "Tests" <$> sequence
@@ -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