X-Git-Url: http://git.lukelau.me/?a=blobdiff_plain;f=src%2FLanguage%2FHaskell%2FLSP%2FTest%2FSession.hs;h=b8286a208c8f5e5369171b87ad62802d899b485d;hb=f91ebeacf72035e1fcb7cb6a5ed2ec270e2c0a1d;hp=8612148ce870a56b1e3cc1587567607dc60f4955;hpb=1b1df64886e90bb77c2804452945ff0d66963e0a;p=lsp-test.git diff --git a/src/Language/Haskell/LSP/Test/Session.hs b/src/Language/Haskell/LSP/Test/Session.hs index 8612148..b8286a2 100644 --- a/src/Language/Haskell/LSP/Test/Session.hs +++ b/src/Language/Haskell/LSP/Test/Session.hs @@ -34,7 +34,7 @@ import Control.Lens hiding (List) import Control.Monad import Control.Monad.IO.Class import Control.Monad.Except -#if __GLASGOW_HASKELL__ >= 806 +#if __GLASGOW_HASKELL__ == 806 import Control.Monad.Fail #endif import Control.Monad.Trans.Reader (ReaderT, runReaderT) @@ -60,12 +60,13 @@ import Language.Haskell.LSP.Types.Capabilities import Language.Haskell.LSP.Types import Language.Haskell.LSP.Types.Lens hiding (error) import Language.Haskell.LSP.VFS +import Language.Haskell.LSP.Test.Compat import Language.Haskell.LSP.Test.Decoding import Language.Haskell.LSP.Test.Exceptions import System.Console.ANSI import System.Directory import System.IO -import System.Process +import System.Process (ProcessHandle()) import System.Timeout -- | A session representing one instance of launching and connecting to a server. @@ -197,7 +198,6 @@ runSessionWithHandles :: Handle -- ^ Server in -> Session a -> IO a runSessionWithHandles serverIn serverOut serverProc serverHandler config caps rootDir exitServer session = do - absRootDir <- canonicalizePath rootDir hSetBuffering serverIn NoBuffering @@ -218,13 +218,16 @@ runSessionWithHandles serverIn serverOut serverProc serverHandler config caps ro runSession' = runSession context initState errorHandler = throwTo mainThreadId :: SessionException -> IO() - serverLauncher = forkIO $ catch (serverHandler serverOut context) errorHandler + serverListenerLauncher = + forkIO $ catch (serverHandler serverOut context) errorHandler server = (Just serverIn, Just serverOut, Nothing, serverProc) - serverFinalizer tid = finally (timeout (messageTimeout config * 1000000) + serverAndListenerFinalizer tid = + finally (timeout (messageTimeout config * 1000000) (runSession' exitServer)) (cleanupProcess server >> killThread tid) - (result, _) <- bracket serverLauncher serverFinalizer (const $ runSession' session) + (result, _) <- bracket serverListenerLauncher serverAndListenerFinalizer + (const $ runSession' session) return result updateStateC :: ConduitM FromServerMessage FromServerMessage (StateT SessionState (ReaderT SessionContext IO)) ()