@@ -27,6 +27,8 @@ data Docs = FunDoc FunDoc
27 [FunDoc] -- data constructor docs
27 [FunDoc] -- data constructor docs
28 | ClassDoc Name Docstring -- class docs
28 | ClassDoc Name Docstring -- class docs
29 [FunDoc] -- method docs
29 [FunDoc] -- method docs
30+ [(Name, Maybe Docstring)] -- parameters and their docstrings
30 deriving Show
32 deriving Show
32showDoc d | nullDocstring d = empty
34showDoc d | nullDocstring d = empty
@@ -66,16 +68,47 @@ pprintFD imp (FD n doc args ty f)
66 showArgs [] _ = []
68 showArgs [] _ = []
69−
71+pprintDocs :: Bool -> Docs -> Doc OutputAnnotation
70pprintDocs imp (FunDoc d) = pprintFD imp d
72pprintDocs imp (FunDoc d) = pprintFD imp d
71pprintDocs imp (DataDoc t args)
73pprintDocs imp (DataDoc t args)
72 = text "Data type" <+> pprintFD imp t <$>
74 = text "Data type" <+> pprintFD imp t <$>
73 nest 4 (text "Constructors:" <> line <>
75 nest 4 (text "Constructors:" <> line <>
74 vsep (map (pprintFD imp) args))
76 vsep (map (pprintFD imp) args))
75−pprintDocs imp (ClassDoc n doc meths)
77+pprintDocs imp (ClassDoc n doc meths params instances)
76− = text "Type class" <+> prettyName imp [] n <$> -- parameters?
78+ = nest 4 (text "Type class" <+> prettyName imp [] n <>
79+ if nullDocstring doc then empty else line <> renderDocstring doc)
81+ nest 4 (text "Parameters:" <$> prettyParameters)
77 nest 4 (text "Methods:" <$>
83 nest 4 (text "Methods:" <$>
78− vsep (map (pprintFD imp) meths))
84+ vsep (map (pprintFD imp) meths))
86+ nest 4 (text "Instances:" <$>
87+ vsep (if null instances' then [text "<no instances>"]
88+ else map dumpInstance instances'))
90+ if null subclasses then empty
91+ else line <$> nest 4 (text "Subclasses:" <$>
92+ vsep (map (dumpInstance . remConstraint) subclasses))
94+ params' = map (\x -> (fst x, False)) params
96+ dumpInstance :: PTerm -> Doc OutputAnnotation
97+ dumpInstance = pprintPTerm imp params' []
99+ remConstraint (PPi (Constraint _ _) _ tm _) = tm
100+ remConstraint (PPi x y z pt) = PPi x y z (remConstraint pt)
103+ hasConstraint (PPi (Constraint _ _) _ _ _) = True
104+ hasConstraint (PPi _ _ _ pt) = hasConstraint pt
105+ hasConstraint _ = False
107+ (subclasses, instances') = partition hasConstraint instances
109+ prettyParameters = if any (isJust . snd) params
110+ then vsep (map (\(n,md) -> prettyName False params' n <+> maybe empty showDoc md) params)
111+ else hsep (punctuate comma (map (prettyName False params' . fst) params))
80getDocs :: Name -> Idris Docs
113getDocs :: Name -> Idris Docs
@@ -96,11 +129,13 @@ docData n ti
96docClass :: Name -> ClassInfo -> Idris Docs
129docClass :: Name -> ClassInfo -> Idris Docs
97docClass n ci
130docClass n ci
98 = do i <- getIState
131 = do i <- getIState
99− let docstr = case lookupCtxt n (idris_docstrings i) of
133+ let docStrings = listToMaybe $ lookupCtxt n $ idris_docstrings i
101− _ -> emptyDocstring
134+ docstr = maybe emptyDocstring fst docStrings
135+ params = map (\pn -> (pn, docStrings >>= (lookup pn . snd))) (class_params ci)
136+ instances = map (delabTy i) (class_instances ci)
102− mdocs <- mapM docFun (map fst (class_methods ci))
137+ mdocs <- mapM (docFun .fst) (class_methods ci)
103− return (ClassDoc n docstr mdocs)
138+ return $ ClassDoc n docstr mdocs params instances
105docFun :: Name -> Idris FunDoc
140docFun :: Name -> Idris FunDoc