用SYB访问GHC AST

时间:2015-10-24 07:17:59

标签: haskell ghc abstract-syntax-tree scrap-your-boilerplate

我写了一个用Haskell-src-exts访问AST的程序。我正在尝试将其转换为使用GHC API。前者使用Uniplate,而后者似乎不幸的是我被迫使用SYB(文档非常稀少)。

以下是原始代码:

module Argon.Visitor (funcsCC)
    where

import Data.Data (Data)
import Data.Generics.Uniplate.Data (childrenBi, universeBi)
import Language.Haskell.Exts.Syntax
import Argon.Types (ComplexityBlock(..))


-- | Compute cyclomatic complexity of every function binding in the given AST.
funcsCC :: Data from => from -> [ComplexityBlock]
funcsCC ast = map funCC [matches | FunBind matches <- universeBi ast]

funCC :: [Match] -> ComplexityBlock
funCC [] = CC (0, 0, "<unknown>", 0)
funCC ms@(Match (SrcLoc _ l c) n _ _ _ _:_) = CC (l, c, name n, complexity ms)
    where name (Ident s)   = s
          name (Symbol s) = s

sumWith :: (a -> Int) -> [a] -> Int
sumWith f = sum . map f

complexity :: Data from => from -> Int
complexity node = 1 + visitMatches node + visitExps node

visitMatches :: Data from => from -> Int
visitMatches = sumWith descend . childrenBi
    where descend :: [Match] -> Int
          descend x = length x - 1 + sumWith visitMatches x

visitExps :: Data from => from -> Int
visitExps = sumWith inspect . universeBi
    where inspect e = visitExp e + visitOp e

visitExp :: Exp -> Int
visitExp (If {})        = 1
visitExp (MultiIf alts) = length alts - 1
visitExp (Case _ alts)  = length alts - 1
visitExp (LCase alts)   = length alts - 1
visitExp _ = 0

visitOp :: Exp -> Int
visitOp (InfixApp _ (QVarOp (UnQual (Symbol op))) _) =
  case op of
    "||" -> 1
    "&&" -> 1
    _    -> 0
visitOp _ = 0

我需要访问函数绑定,匹配和表达式。这就是我设法写的(不工作):

import Data.Generics
import qualified GHC
import Outputable  -- from the GHC package

funcs :: (Data id, Typeable id, Outputable id, Data from, Typeable from) => from -> [GHC.HsBindLR id id]
funcs ast = everything (++) (mkQ [] (\fun@(GHC.FunBind {}) -> [fun])) ast

它抱怨id有太多实例,但我不知道它到底是什么。相关的GHC模块是: http://haddock.stackage.org/lts-3.10/ghc-7.10.2/HsBinds.html

我因此而疯狂。目标是计算复杂性(正如您在原始代码中看到的那样)。我想切换到GHC API,因为它使用与编译器相同的解析器,因此它可以解析每个模块而不必担心扩展。

编辑:这就是当前代码不起作用的原因:

λ> :m +Language.Haskell.GHC.ExactPrint.Parsers GHC Data.Generics Outputable
λ> r <- Language.Haskell.GHC.ExactPrint.parseModule src/Argon/Visitor.hs
λ> let ast = snd $ (\(Right t) -> t) r
.> 
λ> :t ast
ast :: Located (HsModule RdrName)
λ> let funcs = everything (++) (mkQ [] (un@(FunBind _ _ _ _ _ _) -> [fun])) ast :: (Data id, Typeable id, Outputable id) => [HsBindLR id id]
.> 
λ> length funcs

<interactive>:12:8:
    No instance for (Data id0) arising from a use of ‘funcs’
    The type variable ‘id0’ is ambiguous
    Note: there are several potential instances:
      instance Data aeson-0.8.0.2:Data.Aeson.Types.Internal.Value
        -- Defined in ‘aeson-0.8.0.2:Data.Aeson.Types.Internal’
      instance Data attoparsec-0.12.1.6:Data.Attoparsec.Number.Number
        -- Defined in ‘attoparsec-0.12.1.6:Data.Attoparsec.Number’
      instance Data a => Data (Data.Complex.Complex a)
        -- Defined in ‘Data.Complex’
      ../..plus 367 others
    In the first argument of ‘length’, namely ‘funcs’
    In the expression: length funcs
    In an equation for ‘it’: it = length funcs

1 个答案:

答案 0 :(得分:1)

GHC AST根据树中使用的名称类型进行参数化:解析器输出一个带有RdrName名称的AST,您似乎正在使用它。 GHC Commentarythe Haddocks有更多信息。

如果您告诉编译器您正在使用ERROR CifService - No signature of method: CifService.sendQueueJMSMessage() is applicable for argument types: (java.lang.String, java.util.LinkedHashMap) values: [queue.sendEmailCC, [emailTo:abc.aries@gmail.com, emailSubject:Akyong - Reset Password, ...]] ,那么您可能会有更多的运气。

像这样:

HsBindLR RdrName RdrName