如何查找文件夹的所有子文件夹？

Question

我想枚举Haskell中某个文件夹的所有子文件夹。获取所有文件夹内容很容易，有一个getDirectoryContents函数。但是我该如何过滤它们呢？因为getDirectoryContents返回一个IO [FilePath]，而filter需要[a]，所以我不能直接把这两个放在一起。(显然，我是一条有单数和do-notation的新鲜鱼。)

getAllFolders :: FilePath -> IO [FilePath]
getAllFolder path = do
    allItems <- getDirectoryContents path
    -- now what? the predicate is doesDirectoryExist

Answer 1

getAllFolders path = do
    contents <- getDirectoryContents path
    -- do something with contents now, it's a plain [FilePath]

问题是谓词doesDirectoryExist的类型是FilePath -> IO Bool。对于这样的事情，有

ghci> :t Control.Monad.filterM
Control.Monad.filterM :: Monad m => (a -> m Bool) -> [a] -> m [a]

Control.Monad中定义的filterM，因此

getAllFolders path = do
    contents <- getDirectoryContents path
    filterM doesDirectoryExist contents

或者，不将目录的内容绑定到名称，

getAllFolders path = getDirectoryContents path >>= filterM doesDirectoryExist

和无指针：

getAllFolders = getDirectoryContents >=> filterM doesDirectoryExist

Answer 2

看起来Control.Monad提供的filterM就是答案：

getAllFolders :: FilePath -> IO [FilePath]
getAllFolders path = do
    allItems <- getDirectoryContents path
    justFolders <- filterM doesDirectoryExist allItems
    return justFolders