75 lines
1.5 KiB
Rust
75 lines
1.5 KiB
Rust
import rustc::syntax::ast;
|
|
import rustc::syntax::print::pprust;
|
|
import rustc::middle::ast_map;
|
|
import astsrv::seq_srv;
|
|
|
|
export run;
|
|
|
|
fn run(
|
|
srv: astsrv::seq_srv,
|
|
doc: doc::cratedoc
|
|
) -> doc::cratedoc {
|
|
let fold = fold::fold({
|
|
fold_fn: fn~(
|
|
f: fold::fold<astsrv::seq_srv>,
|
|
d: doc::fndoc
|
|
) -> doc::fndoc {
|
|
fold_fn(f, d)
|
|
}
|
|
with *fold::default_seq_fold(srv)
|
|
});
|
|
fold.fold_crate(fold, doc)
|
|
}
|
|
|
|
fn fold_fn(
|
|
fold: fold::fold<astsrv::seq_srv>,
|
|
doc: doc::fndoc
|
|
) -> doc::fndoc {
|
|
|
|
let srv = fold.ctxt;
|
|
|
|
fn add_ret_ty(
|
|
doc: option<doc::retdoc>,
|
|
tystr: str
|
|
) -> option<doc::retdoc> {
|
|
alt doc {
|
|
some(doc) {
|
|
fail "unimplemented";
|
|
}
|
|
none. {
|
|
some({
|
|
desc: none,
|
|
ty: some(tystr)
|
|
})
|
|
}
|
|
}
|
|
}
|
|
|
|
let retty = srv.exec {|ctxt|
|
|
alt ctxt.map.get(doc.id) {
|
|
ast_map::node_item(@{
|
|
node: ast::item_fn(decl, _, _), _
|
|
}) {
|
|
pprust::ty_to_str(decl.output)
|
|
}
|
|
}
|
|
};
|
|
|
|
~{
|
|
return: add_ret_ty(doc.return, retty)
|
|
with *doc
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
|
|
#[test]
|
|
fn should_add_fn_ret_types() {
|
|
let source = "fn a() -> int { }";
|
|
let srv = astsrv::mk_seq_srv_from_str(source);
|
|
let doc = extract::from_srv(srv, "");
|
|
let doc = run(srv, doc);
|
|
assert option::get(doc.topmod.fns[0].return).ty == some("int");
|
|
}
|
|
}
|