haskell - 计算模块中的 `Pat` s

Question

我需要计算Pat的数量在 haskell 中 Module 。我知道最简单的方法是在 AST 的每个级别上进行模式匹配，这将产生一个看起来像整个 AST 的巨大函数。我相信有某种方法可以利用像 Functor 或 State Monad 这样的类型类来依赖一些遍历树的现有函数(例如 PrettyPrint) )并跟踪计数器，但我不确定它到底是如何工作的。

Answer 1

使用 uniplate 非常简单:

import Data.Data
import Data.Generics.Uniplate.Data
import Control.Monad
import Language.Haskell.Exts

findPats :: Data a => a -> [Pat]
findPats = universeBi

test = do
  content <- readFile "Simple.hs"
  case parseModule content of
    ParseFailed _ e -> error e
    ParseOk a       -> do
      forM_ (findPats a) $ \p -> do
        putStrLn $ "got a pat: " ++ show p

本质上它只是 universeBi 函数。