rustdoc: Build mod and fn nodes from the AST

src/rustdoc/extract.rs

@@ -1,11 +1,69 @@
-#[doc = "Converting the Rust AST to the rustdoc document model"];
-
 import rustc::syntax::ast;
+import rustc::syntax::visit;
+
+export extract;
 
 #[doc = "Converts the Rust AST to the rustdoc document model"]
-fn extract(crate: @ast::crate) -> doc::cratedoc {
-    {
-        mods: []
+fn extract(
+    crate: @ast::crate
+) -> doc::cratedoc {
+    ~{
+        topmod: top_moddoc_from_crate(crate),
+    }
+}
+
+fn top_moddoc_from_crate(
+    crate: @ast::crate
+) -> doc::moddoc {
+    moddoc_from_mod(crate.node.module, "crate", crate.node.attrs)
+}
+
+fn moddoc_from_mod(
+    module: ast::_mod,
+    name: ast::ident,
+    _attrs: [ast::attribute]
+
+) -> doc::moddoc {
+    ~{
+        name: name,
+        mods: doc::modlist(
+            vec::filter_map(module.items) {|item|
+                alt item.node {
+                  ast::item_mod(m) {
+                    some(moddoc_from_mod(m, item.ident, item.attrs))
+                  }
+                  _ {
+                    none
+                  }
+                }
+            }),
+        fns: doc::fnlist(
+            vec::filter_map(module.items) {|item|
+                alt item.node {
+                  ast::item_fn(decl, typarams, _) {
+                    some(fndoc_from_fn(
+                        decl, typarams, item.ident, item.attrs))
+                  }
+                  _ {
+                    none
+                  }
+                }
+            })
+    }
+}
+
+fn fndoc_from_fn(
+    _decl: ast::fn_decl,
+    _typarams: [ast::ty_param],
+    name: ast::ident,
+    _attrs: [ast::attribute]
+) -> doc::fndoc {
+    ~{
+        name: name,
+        brief: "todo",
+        desc: none,
+        return: none,
+        args: map::new_str_hash::<str>()
     }
 }
 
@@ -17,6 +75,37 @@ mod tests {
         let source = ""; // empty crate
         let ast = parse::from_str(source);
         let doc = extract(ast);
-        assert doc.mods == [];
+        // FIXME #1535: These are boxed to prevent a crash
+        assert ~doc.topmod.mods == ~doc::modlist([]);
+        assert ~doc.topmod.fns == ~doc::fnlist([]);
+    }
+
+    #[test]
+    fn extract_mods() {
+        let source = "mod a { mod b { } mod c { } }";
+        let ast = parse::from_str(source);
+        let doc = extract(ast);
+        assert doc.topmod.mods[0].name == "a";
+        assert doc.topmod.mods[0].mods[0].name == "b";
+        assert doc.topmod.mods[0].mods[1].name == "c";
+    }
+
+    #[test]
+    fn extract_mods_deep() {
+        let source = "mod a { mod b { mod c { } } }";
+        let ast = parse::from_str(source);
+        let doc = extract(ast);
+        assert doc.topmod.mods[0].mods[0].mods[0].name == "c";
+    }
+
+    #[test]
+    fn extract_fns() {
+        let source =
+            "fn a() { } \
+             mod b { fn c() { } }";
+        let ast = parse::from_str(source);
+        let doc = extract(ast);
+        assert doc.topmod.fns[0].name == "a";
+        assert doc.topmod.mods[0].fns[0].name == "c";
     }
 }