Enforce utf8

Author: berberman

plugins/hls-eval-plugin/src/Ide/Plugin/Eval/Code.hs

@@ -12,7 +12,7 @@ import           Data.String                    (IsString)
 import qualified Data.Text                      as T
 import           Development.IDE.Types.Location (Position (..), Range (..))
 import           GHC                            (ExecOptions, ExecResult (..),
-                                                 execStmt)
+                                                 execOptions, execStmt)
 import           GHC.LanguageExtensions.Type    (Extension (..))
 import           GhcMonad                       (Ghc, liftIO, modifySession)
 import           HscTypes
@@ -24,7 +24,7 @@ import           Ide.Plugin.Eval.Types          (Language (Plain), Loc,
 import           InteractiveEval                (getContext, parseImportDecl,
                                                  runDecls, setContext)
 import           Language.LSP.Types.Lens        (line, start)
-import           System.IO.Extra                (newTempFile, readFile')
+import           System.IO.Extra                (newTempFile, readFileUTF8')
 
 -- | Return the ranges of the expression and result parts of the given test
 testRanges :: Test -> (Range, Range)
@@ -95,8 +95,13 @@ evalExtensions =
 evalSetup :: Ghc ()
 evalSetup = do
     preludeAsP <- parseImportDecl "import qualified Prelude as P"
+    encodingAsP <- parseImportDecl "import GHC.IO.Encoding as P"
     context <- getContext
-    setContext (IIDecl preludeAsP : context)
+    setContext (IIDecl encodingAsP : IIDecl preludeAsP : context)
+    execStmt "setLocaleEncoding utf8" execOptions >>= \case
+        ExecComplete (Left err) _ -> error $ "failed to set encoding in eval setup: " <> show err
+        ExecComplete (Right _) _ -> pure ()
+        ExecBreak{} -> error "breakpoints are not supported"
 
 -- | A wrapper of 'InteractiveEval.execStmt', capturing the execution result
 myExecStmt :: String -> ExecOptions -> Ghc (Either String (Maybe String))
@@ -106,7 +111,7 @@ myExecStmt stmt opts = do
     modifySession $ \hsc -> hsc {hsc_IC = setInteractivePrintName (hsc_IC hsc) evalPrint}
     result <- execStmt stmt opts >>= \case
               ExecComplete (Left err) _ -> pure $ Left $ show err
-              ExecComplete (Right _) _ -> liftIO $ Right . (\x -> if null x then Nothing else Just x ) <$> readFile' temp
+              ExecComplete (Right _) _ -> liftIO $ Right . (\x -> if null x then Nothing else Just x) <$> readFileUTF8' temp
               ExecBreak{} -> pure $ Right $ Just "breakpoints are not supported"
     liftIO purge
     pure result