src/Language/Haskell/LSP/Test/Server.hs

   1 module Language.Haskell.LSP.Test.Server (withServer) where
   2
   3 import Control.Concurrent
   4 import Control.Monad
   5 import Data.Maybe
   6 import System.IO
   7 import System.Process
   8
   9 withServer :: String -> (Handle -> Handle -> Int -> IO a) -> IO a
  10 withServer serverExe f = do
  11   let cmd:args = words serverExe
  12       createProc = (proc cmd args) { std_in = CreatePipe, std_out = CreatePipe, std_err = CreatePipe }
  13   (Just serverIn, Just serverOut, Just serverErr, serverProc) <- createProcess createProc
  14
  15   -- Need to continuously consume to stderr else it gets blocked
  16   -- Can't pass NoStream either to std_err
  17   hSetBuffering serverErr NoBuffering
  18   errSinkThread <- forkIO $ forever $ hGetLine serverErr
  19
  20   pid <- fromIntegral . fromJust <$> getPid serverProc
  21
  22   result <- f serverIn serverOut pid
  23
  24   killThread errSinkThread
  25   terminateProcess serverProc
  26   return result