aboutsummaryrefslogtreecommitdiff
path: root/haddock-api
diff options
context:
space:
mode:
Diffstat (limited to 'haddock-api')
-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