rustdoc: Prune undocumented enums
This commit is contained in:
parent
95a006ce52
commit
1d94f40c6c
2 changed files with 66 additions and 1 deletions
|
@ -8,6 +8,7 @@ export default_seq_fold_crate;
|
|||
export default_seq_fold_mod;
|
||||
export default_seq_fold_fn;
|
||||
export default_seq_fold_const;
|
||||
export default_seq_fold_enum;
|
||||
export default_seq_fold_fnlist;
|
||||
|
||||
enum fold<T> = t<T>;
|
||||
|
|
|
@ -21,9 +21,11 @@ fn run(
|
|||
fold_mod: fold_mod,
|
||||
fold_fn: fold_fn,
|
||||
fold_const: fold_const,
|
||||
fold_enum: fold_enum,
|
||||
fold_modlist: fold_modlist,
|
||||
fold_fnlist: fold_fnlist,
|
||||
fold_constlist: fold_constlist
|
||||
fold_constlist: fold_constlist,
|
||||
fold_enumlist: fold_enumlist
|
||||
with *fold::default_seq_fold(ctxt)
|
||||
});
|
||||
fold.fold_crate(fold, doc)
|
||||
|
@ -216,3 +218,65 @@ fn should_elide_undocumented_consts() {
|
|||
let doc = run(srv, doc);
|
||||
assert vec::is_empty(*doc.topmod.consts);
|
||||
}
|
||||
|
||||
fn fold_enum(fold: fold::fold<ctxt>, doc: doc::enumdoc) -> doc::enumdoc {
|
||||
let doc = ~{
|
||||
variants: vec::filter_map(doc.variants) {|variant|
|
||||
if variant.desc != none {
|
||||
some(variant)
|
||||
} else {
|
||||
none
|
||||
}
|
||||
}
|
||||
with *fold::default_seq_fold_enum(fold, doc)
|
||||
};
|
||||
fold.ctxt.have_docs =
|
||||
doc.brief != none
|
||||
|| doc.desc != none
|
||||
|| vec::is_not_empty(doc.variants);
|
||||
ret doc;
|
||||
}
|
||||
|
||||
fn fold_enumlist(
|
||||
fold: fold::fold<ctxt>,
|
||||
list: doc::enumlist
|
||||
) -> doc::enumlist {
|
||||
doc::enumlist(vec::filter_map(*list) {|doc|
|
||||
let doc = fold.fold_enum(fold, doc);
|
||||
if fold.ctxt.have_docs {
|
||||
some(doc)
|
||||
} else {
|
||||
none
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn should_elide_undocumented_enums() {
|
||||
let source = "enum a { b }";
|
||||
let srv = astsrv::mk_srv_from_str(source);
|
||||
let doc = extract::from_srv(srv, "");
|
||||
let doc = attr_pass::mk_pass()(srv, doc);
|
||||
let doc = run(srv, doc);
|
||||
assert vec::is_empty(*doc.topmod.enums);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn should_elide_undocumented_variants() {
|
||||
let source = "#[doc = \"a\"] enum a { b }";
|
||||
let srv = astsrv::mk_srv_from_str(source);
|
||||
let doc = extract::from_srv(srv, "");
|
||||
let doc = attr_pass::mk_pass()(srv, doc);
|
||||
let doc = run(srv, doc);
|
||||
assert vec::is_empty(doc.topmod.enums[0].variants);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn should_not_elide_enums_with_documented_variants() {
|
||||
let source = "enum a { #[doc = \"a\"] b }";
|
||||
let srv = astsrv::mk_srv_from_str(source);
|
||||
let doc = extract::from_srv(srv, "");
|
||||
let doc = attr_pass::mk_pass()(srv, doc);
|
||||
let doc = run(srv, doc);
|
||||
assert vec::is_not_empty(*doc.topmod.enums);
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue