Catching/hijacking stdout w haskell

Question

Jak zdefiniować "catchOutput" tak, aby główne wyjścia tylko "bar"?

To znaczy, w jaki sposób można uzyskać dostęp zarówno do strumienia wyjściowego (standardowego), jak i rzeczywistego wyjścia działania io?

catchOutput :: IO a -> IO (a,String) 
catchOutput = undefined 

doSomethingWithOutput :: IO a -> IO() 
doSomethingWithOutput io = do 
    (_ioOutp, stdOutp) <- catchOutput io 
    if stdOutp == "foo" 
     then putStrLn "bar" 
     else putStrLn "fail!" 

main = doSomethingWithOutput (putStr "foo") 

Najlepszym „rozwiązanie” hipotetyczny znalazłem do tej pory obejmuje przekierowanie stdout, inspired by this, do strumienia pliku, a następnie odczyt z tego pliku (Poza tym jest super-brzydki nie byłem w stanie czytać bezpośrednio po napisaniu z pliku Czy możliwe jest utworzenie "niestandardowego strumienia bufora", który nie musi być zapisany w pliku?). Chociaż to "trochę" przypomina boczny tor.

Inny kąt wydaje się używać "hGetContents stdout", jeśli to ma robić to, co myślę, że powinien. Ale nie mam pozwolenia na czytanie ze stdouta. Mimo, że googlowałem, wydaje się, że było to used.

Answer 1

Dlaczego zamiast tego należy użyć monitu pisarskiego? Na przykład,

import Control.Monad.Writer 

doSomethingWithOutput :: WriterT String IO a -> IO() 
doSomethingWithOutput io = do 
    (_, res) <- runWriterT io 
    if res == "foo" 
     then putStrLn "bar" 
     else putStrLn "fail!" 

main = doSomethingWithOutput (tell "foo") 

Alternatywnie, można modyfikować swoje wewnętrzne działania, aby wziąć Handle zapisu zamiast stdout. Następnie możesz użyć czegoś takiego, jak knob, aby utworzyć uchwyt pliku w pamięci, który możesz przekazać do działania wewnętrznego, a następnie sprawdzić jego zawartość.

Answer 2

Jak @hammar wskazał, można użyć pokrętła, aby utworzyć plik w pamięci, ale można też użyć hDuplicate i hDuplicateTo zmienić stdout do pliku pamięci iz powrotem. Coś jak następujący kod całkowicie niesprawdzonych:

catchOutput io = do 
    knob <- newKnob (pack []) 
    let before = do 
     h <- newFileHandle knob "<stdout>" WriteMode 
     stdout' <- hDuplicate stdout 
     hDuplicateTo h stdout 
     hClose h 
     return stdout' 
     after stdout' = do 
     hDuplicateTo stdout' stdout 
     hClose stdout' 
    a <- bracket_ before after io 
    bytes <- Data.Knob.getContents knob 
    return (a, unpack bytes) 

Answer 3

użyłem następującą funkcję dla testu jednostkowego funkcja, która drukuje na standardowe wyjście.

import GHC.IO.Handle 
import System.IO 
import System.Directory 

catchOutput :: IO() -> IO String 
catchOutput f = do 
    tmpd <- getTemporaryDirectory 
    (tmpf, tmph) <- openTempFile tmpd "haskell_stdout" 
    stdout_dup <- hDuplicate stdout 
    hDuplicateTo tmph stdout 
    hClose tmph 
    f 
    hDuplicateTo stdout_dup stdout 
    str <- readFile tmpf 
    removeFile tmpf 
    return str 

Nie mam pewności co do podejścia do zapisu w pamięci, ale działa dobrze dla niewielkiej ilości danych wyjściowych z plikiem tymczasowym.

Answer 4

Istnieje kilka pakietów na Hackage, które obiecują to zrobić: io-capture i cicho. cicho wydaje się być utrzymywany i działa również na Windows (io-capture działa tylko na systemie Unix). Z dyskretnie, należy użyć przechwytywania:

import System.IO.Silently 

main = do 
    (output, _) <- capture $ putStr "hello" 
    putStrLn $ output ++ " world" 

pamiętać, że działa poprzez przekierowanie wyjścia do pliku tymczasowego, a następnie ją przeczytać ... Ale tak długo, jak to działa!