{-# OPTIONS_GHC -Wunused-imports #-}
module Language.Haskell.LSP.Test.Compat where
-import Control.Concurrent.Chan
-import Control.Monad.IO.Class
-import Data.Conduit
+import Data.Maybe
+import System.IO
-#ifdef mingw32_HOST_OS
-
-import qualified System.Win32.Process as P (getCurrentProcessId)
-getProcessID :: IO Int
-getProcessID = fromIntegral <$> P.getCurrentProcessId
+#if MIN_VERSION_process(1,6,3)
+-- We have to hide cleanupProcess for process-1.6.3.0
+-- cause it is in the public api for 1.6.3.0 versions
+-- shipped with ghc >= 8.6 and < 8.6.4
+import System.Process hiding (getPid, cleanupProcess)
+# if MIN_VERSION_process(1,6,4)
+import qualified System.Process (getPid, cleanupProcess)
+# else
+import qualified System.Process (getPid)
+# endif
+#else
+import System.Process
+import System.Process.Internals
+import Control.Concurrent.MVar
+#endif
+#ifdef mingw32_HOST_OS
+import qualified System.Win32.Process
#else
+import qualified System.Posix.Process
+#endif
-import qualified System.Posix.Process as P (getProcessID)
-getProcessID :: IO Int
-getProcessID = fromIntegral <$> P.getProcessID
+getCurrentProcessID :: IO Int
+#ifdef mingw32_HOST_OS
+getCurrentProcessID = fromIntegral <$> System.Win32.Process.getCurrentProcessId
+#else
+getCurrentProcessID = fromIntegral <$> System.Posix.Process.getProcessID
#endif
-#if MIN_VERSION_conduit(1,3,0)
-chanSource :: MonadIO m => Chan o -> ConduitT i o m b
+getProcessID :: ProcessHandle -> IO Int
+getProcessID p = fromIntegral . fromJust <$> getProcessID' p
+ where
+#if MIN_VERSION_process(1,6,3)
+ getProcessID' = System.Process.getPid
#else
-chanSource :: MonadIO m => Chan o -> ConduitM i o m b
+#if MIN_VERSION_process(1,6,0)
+ getProcessID' (ProcessHandle mh _ _) = do
+#else
+ getProcessID' (ProcessHandle mh _) = do
+#endif
+ p_ <- readMVar mh
+ case p_ of
+#ifdef mingw32_HOST_OS
+ OpenHandle h -> do
+ pid <- System.Win32.Process.getProcessId h
+ return $ Just pid
+#else
+ OpenHandle pid -> return $ Just pid
+#endif
+ _ -> return Nothing
#endif
-chanSource c = do
- x <- liftIO $ readChan c
- yield x
- chanSource c
+cleanupRunningProcess :: (Maybe Handle, Maybe Handle, Maybe Handle, ProcessHandle) -> IO ()
+cleanupRunningProcess p@(_, _, _, ph) =
+ getProcessExitCode ph >>= maybe (cleanupProcess p) (const $ return ())
+
+cleanupProcess :: (Maybe Handle, Maybe Handle, Maybe Handle, ProcessHandle) -> IO ()
+#if MIN_VERSION_process(1,6,4)
+cleanupProcess = System.Process.cleanupProcess
+#else
+cleanupProcess (mb_stdin, mb_stdout, mb_stderr, ph) = do
+
+ terminateProcess ph
+ -- Note, it's important that other threads that might be reading/writing
+ -- these handles also get killed off, since otherwise they might be holding
+ -- the handle lock and prevent us from closing, leading to deadlock.
+ maybe (return ()) hClose mb_stdin
+ maybe (return ()) hClose mb_stdout
+ maybe (return ()) hClose mb_stderr
+
+ return ()
+#endif