aboutsummaryrefslogtreecommitdiff
path: root/Utility/Process.hs
diff options
context:
space:
mode:
authorGravatar Joey Hess <joey@kitenet.net>2013-02-26 13:04:37 -0400
committerGravatar Joey Hess <joey@kitenet.net>2013-02-26 13:04:37 -0400
commit9816398fadebf9f1a70b51b21c2d8a6bd9e8caa4 (patch)
tree6cef77b5af56432294a2592f36be627ce18b60ed /Utility/Process.hs
parentc88f2afd3a7b811e7078c9e66d47431757d2b3df (diff)
webapp: Display any error message from git init if it fails to create a repository.
Diffstat (limited to 'Utility/Process.hs')
-rw-r--r--Utility/Process.hs47
1 files changed, 46 insertions, 1 deletions
diff --git a/Utility/Process.hs b/Utility/Process.hs
index 11a9a4f38..b2bac99a1 100644
--- a/Utility/Process.hs
+++ b/Utility/Process.hs
@@ -21,6 +21,7 @@ module Utility.Process (
createProcessSuccess,
createProcessChecked,
createBackgroundProcess,
+ processTranscript,
withHandle,
withBothHandles,
withQuietOutput,
@@ -40,6 +41,8 @@ import System.Log.Logger
import Control.Concurrent
import qualified Control.Exception as E
import Control.Monad
+import Data.Maybe
+import System.Posix.IO
import Utility.Misc
@@ -116,7 +119,10 @@ forceSuccessProcess p pid = do
ExitSuccess -> return ()
ExitFailure n -> fail $ showCmd p ++ " exited " ++ show n
-{- Waits for a ProcessHandle and returns True if it exited successfully. -}
+{- Waits for a ProcessHandle and returns True if it exited successfully.
+ - Note that using this with createProcessChecked will throw away
+ - the Bool, and is only useful to ignore the exit code of a process,
+ - while still waiting for it. -}
checkSuccessProcess :: ProcessHandle -> IO Bool
checkSuccessProcess pid = do
code <- waitForProcess pid
@@ -146,6 +152,45 @@ createProcessChecked checker p a = do
createBackgroundProcess :: CreateProcessRunner
createBackgroundProcess p a = a =<< createProcess p
+{- Runs a process, optionally feeding it some input, and
+ - returns a transcript combining its stdout and stderr, and
+ - whether it succeeded or failed. -}
+processTranscript :: String -> [String] -> (Maybe String) -> IO (String, Bool)
+processTranscript cmd opts input = do
+ (readf, writef) <- createPipe
+ readh <- fdToHandle readf
+ writeh <- fdToHandle writef
+ p@(_, _, _, pid) <- createProcess $
+ (proc cmd opts)
+ { std_in = if isJust input then CreatePipe else Inherit
+ , std_out = UseHandle writeh
+ , std_err = UseHandle writeh
+ }
+ hClose writeh
+
+ -- fork off a thread to start consuming the output
+ transcript <- hGetContents readh
+ outMVar <- newEmptyMVar
+ _ <- forkIO $ E.evaluate (length transcript) >> putMVar outMVar ()
+
+ -- now write and flush any input
+ case input of
+ Just s -> do
+ let inh = stdinHandle p
+ unless (null s) $ do
+ hPutStr inh s
+ hFlush inh
+ hClose inh
+ Nothing -> return ()
+
+ -- wait on the output
+ takeMVar outMVar
+ hClose readh
+
+ ok <- checkSuccessProcess pid
+ return (transcript, ok)
+
+
{- Runs a CreateProcessRunner, on a CreateProcess structure, that
- is adjusted to pipe only from/to a single StdHandle, and passes
- the resulting Handle to an action. -}