diff options
Diffstat (limited to 'src/Haddock/Utils.hs')
| -rw-r--r-- | src/Haddock/Utils.hs | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/src/Haddock/Utils.hs b/src/Haddock/Utils.hs index 8f5b087c..8c9cf5b1 100644 --- a/src/Haddock/Utils.hs +++ b/src/Haddock/Utils.hs @@ -40,6 +40,7 @@ module Haddock.Utils ( ) where import Haddock.Types +import Haddock.GHC.Utils import GHC import SrcLoc @@ -89,7 +90,8 @@ restrictTo names (L loc decl) = L loc $ case decl of [] -> TyClD (d { tcdND = DataType, tcdCons = [] }) [con] -> TyClD (d { tcdCons = [con] }) TyClD d | isClassDecl d -> - TyClD (d { tcdSigs = restrictDecls names (tcdSigs d) }) + TyClD (d { tcdSigs = restrictDecls names (tcdSigs d), + tcdATs = restrictATs names (tcdATs d) }) _ -> decl restrictCons :: [Name] -> [LConDecl Name] -> [LConDecl Name] @@ -117,6 +119,11 @@ restrictDecls names decls = filter keep decls where keep d = fromJust (sigName d) `elem` names -- has to have a name, since it's a class method type signature + +restrictATs :: [Name] -> [LTyClDecl Name] -> [LTyClDecl Name] +restrictATs names ats = [ at | at <- ats , tcdName (unL at) `elem` names ] + + -- ----------------------------------------------------------------------------- -- Filename mangling functions stolen from s main/DriverUtil.lhs. |