import rustc::front::attr;
import core::tuple;
-export crate_attrs, fn_attrs, arg_attrs;
+export crate_attrs, mod_attrs, fn_attrs, arg_attrs;
export parse_crate, parse_mod, parse_fn;
type crate_attrs = {
) -> doc::cratedoc {
fold_crate(f, d)
},
+ fold_mod: fn~(
+ f: fold::fold<astsrv::srv>,
+ d: doc::moddoc
+ ) -> doc::moddoc {
+ fold_mod(f, d)
+ },
fold_fn: fn~(
f: fold::fold<astsrv::srv>,
d: doc::fndoc
assert doc.topmod.name == "bond";
}
+fn fold_mod(fold: fold::fold<astsrv::srv>, doc: doc::moddoc) -> doc::moddoc {
+ let srv = fold.ctxt;
+ let attrs = if doc.id == ast::crate_node_id {
+ // This is the top-level mod, use the crate attributes
+ astsrv::exec(srv) {|ctxt|
+ attr_parser::parse_mod(ctxt.ast.node.attrs)
+ }
+ } else {
+ astsrv::exec(srv) {|ctxt|
+ let attrs = alt ctxt.map.get(doc.id) {
+ ast_map::node_item(item) { item.attrs }
+ };
+ attr_parser::parse_mod(attrs)
+ }
+ };
+ let doc = fold::default_seq_fold_mod(fold, doc);
+ ret merge_mod_attrs(doc, attrs);
+
+ fn merge_mod_attrs(
+ doc: doc::moddoc,
+ attrs: attr_parser::mod_attrs
+ ) -> doc::moddoc {
+ ~{
+ brief: attrs.brief,
+ desc: attrs.desc
+ with *doc
+ }
+ }
+}
+
+#[test]
+fn fold_mod_should_extract_mod_attributes() {
+ let source = "#[doc = \"test\"] mod a { }";
+ let srv = astsrv::mk_srv_from_str(source);
+ let doc = extract::from_srv(srv, "");
+ let fold = fold::default_seq_fold(srv);
+ let doc = fold_mod(fold, doc.topmod.mods[0]);
+ assert doc.desc == some("test");
+}
+
+#[test]
+fn fold_mod_should_extract_top_mod_attributes() {
+ let source = "#[doc = \"test\"];";
+ let srv = astsrv::mk_srv_from_str(source);
+ let doc = extract::from_srv(srv, "");
+ let fold = fold::default_seq_fold(srv);
+ let doc = fold_mod(fold, doc.topmod);
+ assert doc.desc == some("test");
+}
+
fn fold_fn(
fold: fold::fold<astsrv::srv>,
doc: doc::fndoc
export fold_crate, fold_mod, fold_fn, fold_modlist, fold_fnlist;
export default_seq_fold;
export default_seq_fold_crate;
+export default_seq_fold_mod;
export default_seq_fold_fn;
export default_seq_fold_fnlist;