aboutsummaryrefslogtreecommitdiff
path: root/haddock-api/src/Haddock
diff options
context:
space:
mode:
authorŁukasz Hanuszczak <lukasz.hanuszczak@gmail.com>2015-06-06 19:36:53 +0200
committerŁukasz Hanuszczak <lukasz.hanuszczak@gmail.com>2015-06-30 22:37:48 +0200
commit74de0021815f0642a89017fbb1fbdf18064cb5ea (patch)
tree7d0a7dc23df878901ae4ba63f93634d6d468219a /haddock-api/src/Haddock
parentd275f87c4cfa1e8da042f70659331121afa9a15c (diff)
Implement utility method for extracting variable identifiers from AST.
Diffstat (limited to 'haddock-api/src/Haddock')
-rw-r--r--haddock-api/src/Haddock/Backends/Hyperlinker/Ast.hs16
1 files changed, 16 insertions, 0 deletions
diff --git a/haddock-api/src/Haddock/Backends/Hyperlinker/Ast.hs b/haddock-api/src/Haddock/Backends/Hyperlinker/Ast.hs
index abd3ca2b..62c0d439 100644
--- a/haddock-api/src/Haddock/Backends/Hyperlinker/Ast.hs
+++ b/haddock-api/src/Haddock/Backends/Hyperlinker/Ast.hs
@@ -1,6 +1,10 @@
+{-# LANGUAGE RankNTypes #-}
+
module Haddock.Backends.Hyperlinker.Ast where
import qualified GHC
+import Data.Data
+import Control.Applicative
import Haddock.Backends.Hyperlinker.Parser
@@ -18,3 +22,15 @@ enrich src =
lookupName :: GHC.RenamedSource -> Span -> Maybe GHC.Name
lookupName = undefined
+
+everything :: (r -> r -> r) -> (forall a. Data a => a -> r)
+ -> (forall a. Data a => a -> r)
+everything k f x = foldl k (f x) (gmapQ (everything k f) x)
+
+variables :: GHC.RenamedSource -> [(GHC.SrcSpan, GHC.Name)]
+variables =
+ everything (<|>) var
+ where
+ var term = case cast term of
+ (Just (GHC.L sspan (GHC.HsVar sid))) -> pure (sspan, sid)
+ _ -> empty