如何在Template Haskell中检查引用的数据构造函数名称？

Question

我正在尝试学习一些模板Haskell.作为练习,我写的,可以生成诸如函数isLeft和isRight(灵感来自这个问题).这是我谦虚的尝试:

isA connam = do
    ConE nam <- connam
    nn <- newName "p"
    lamE [varP nn] $ caseE (varE nn) [
                       match (conP nam [wildP]) ( normalB [| True |] ) [],
                       match wildP ( normalB [| False |] ) [] 
                     ]

问题是它只适用于单参数构造函数.罪魁祸首是conP nam [wildP]模式.理想情况下,它看起来应该是conP nam (replicate (numArgs nam) wildP),numArgs函数返回构造函数的参数数量.但是我该如何编写这样的功能呢？我想我需要访问相关的数据声明,但我不知道如何.

这里有另外一个关于这个功能的问题.

Answer 1

虽然您可以使用reify并检查类型来确定数据构造函数的arity,但使用记录模式生成与arity无关的代码要容易得多:

isFoo :: Bar -> Bool
isFoo p = case p of
    (Foo {}) -> True     -- Valid no matter what the arity of Foo is
    _        -> False

这可以通过替代来实现conP用recP你的代码.

isA connam = do
    ConE nam <- connam
    nn <- newName "p"
    lamE [varP nn] $ caseE (varE nn) [
                       match (recP nam []) ( normalB [| True |] ) [],
                       match wildP ( normalB [| False |] ) [] 
                     ]