在 haskell 中捕获/劫持标准输出

Question

如何定义“catchOutput”以便运行主输出仅“bar”？

也就是说，如何分别访问 io 操作的输出流（stdout）和实际输出？

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")

到目前为止，我发现的最好的假设“解决方案”包括转移标准输出， 受此启发，到文件流，然后从该文件中读取（除了超级难看之外，我还无法在从文件写入后直接读取。是否有可能创建一个不必存储在文件中的“自定义缓冲流”？）。尽管这感觉“有点”像是一条岔路。

另一个角度似乎使用“hGetContents stdout”，如果这应该做我认为应该做的事情。但我没有获得从标准输出读取的权限。尽管谷歌搜索似乎表明它已经已使用。

Answer 1

我使用以下函数对打印到标准输出的函数进行单元测试。

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

我不确定内存文件方法，但对于使用临时文件的少量输出来说它可以正常工作。

Answer 2

Hackage 上有一些软件包承诺可以做到这一点：io-capture 和默默地。似乎默默地被维护并且也可以在 Windows 上工作（io-capture 仅在 Unix 上工作）。静默地，您可以使用 capture ：

import System.IO.Silently

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

请注意，它的工作原理是将输出重定向到临时文件，然后读取它......但只要它有效！

Answer 3

为什么不直接使用 writer monad 呢？例如，

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")

或者，您可以修改内部操作以采用 Handle 进行写入，而不是 stdout。然后，您可以使用类似 knob创建一个内存中的文件句柄，您可以将其传递给内部操作，然后检查其内容。

Answer 4

正如 @hammar 指出的，您可以使用旋钮创建内存中文件，但您也可以使用 hDuplicate 和 hDuplicateTo 更改 stdout > 到内存文件，然后再返回。类似于以下完全未经测试的代码：

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)