95 lines
3.0 KiB
Rust
95 lines
3.0 KiB
Rust
|
#[link(name = "rustdoc_ng",
|
||
|
|
vers = "0.1.0",
|
||
|
|
uuid = "8c6e4598-1596-4aa5-a24c-b811914bbbc6")];
|
||
|
|
#[desc = "rustdoc, the Rust documentation extractor"];
|
||
|
|
#[license = "MIT/ASL2"];
|
||
|
|
#[crate_type = "bin"];
|
||
|
|
|
||
|
|
extern mod extra;
|
||
|
|
extern mod rustdoc_ng;
|
||
|
|
|
||
|
|
use rustdoc_ng::*;
|
||
|
|
use std::cell::Cell;
|
||
|
|
|
||
|
|
use extra::serialize::Encodable;
|
||
|
|
|
||
|
|
fn main() {
|
||
|
|
use extra::getopts::*;
|
||
|
|
use extra::getopts::groups::*;
|
||
|
|
|
||
|
|
let args = std::os::args();
|
||
|
|
let opts = ~[
|
||
|
|
optmulti("L", "library-path", "directory to add to crate search path", "DIR"),
|
||
|
|
optmulti("p", "plugin", "plugin to load and run", "NAME"),
|
||
|
|
optmulti("", "plugin-path", "directory to load plugins from", "DIR"),
|
||
|
|
// auxillary pass (defaults to hidden_strip
|
||
|
|
optmulti("a", "pass", "auxillary pass to run", "NAME"),
|
||
|
|
optflag("n", "no-defult-passes", "do not run the default passes"),
|
||
|
|
optflag("h", "help", "show this help message"),
|
||
|
|
];
|
||
|
|
|
||
|
|
let matches = getopts(args.tail(), opts).unwrap();
|
||
|
|
|
||
|
|
if opt_present(&matches, "h") || opt_present(&matches, "help") {
|
||
|
|
println(usage(args[0], opts));
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
let libs = Cell::new(opt_strs(&matches, "L").map(|s| Path(*s)));
|
||
|
|
|
||
|
|
let mut passes = if opt_present(&matches, "n") {
|
||
|
|
~[]
|
||
|
|
} else {
|
||
|
|
~[~"collapse-docs", ~"clean-comments", ~"collapse-privacy" ]
|
||
|
|
};
|
||
|
|
|
||
|
|
opt_strs(&matches, "a").map(|x| passes.push(x.clone()));
|
||
|
|
|
||
|
|
if matches.free.len() != 1 {
|
||
|
|
println(usage(args[0], opts));
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
let cr = Cell::new(Path(matches.free[0]));
|
||
|
|
|
||
|
|
let crate = std::task::try(|| {let cr = cr.take(); core::run_core(libs.take(), &cr)}).unwrap();
|
||
|
|
|
||
|
|
// { "schema": version, "crate": { parsed crate ... }, "plugins": { output of plugins ... }}
|
||
|
|
let mut json = ~extra::treemap::TreeMap::new();
|
||
|
|
json.insert(~"schema", extra::json::String(SCHEMA_VERSION.to_owned()));
|
||
|
|
|
||
|
|
let mut pm = plugins::PluginManager::new(Path("/tmp/rustdoc_ng/plugins"));
|
||
|
|
|
||
|
|
for pass in passes.iter() {
|
||
|
|
pm.add_plugin(match pass.as_slice() {
|
||
|
|
"strip-hidden" => passes::strip_hidden,
|
||
|
|
"clean-comments" => passes::clean_comments,
|
||
|
|
"collapse-docs" => passes::collapse_docs,
|
||
|
|
"collapse-privacy" => passes::collapse_privacy,
|
||
|
|
s => { error!("unknown pass %s, skipping", s); passes::noop },
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
for pname in opt_strs(&matches, "p").move_iter() {
|
||
|
|
pm.load_plugin(pname);
|
||
|
|
}
|
||
|
|
|
||
|
|
let (crate, res) = pm.run_plugins(crate);
|
||
|
|
let plugins_json = ~res.move_iter().filter_map(|opt| opt).collect();
|
||
|
|
|
||
|
|
// FIXME: yuck, Rust -> str -> JSON round trip! No way to .encode
|
||
|
|
// straight to the Rust JSON representation.
|
||
|
|
let crate_json_str = do std::io::with_str_writer |w| {
|
||
|
|
crate.encode(&mut extra::json::Encoder(w));
|
||
|
|
};
|
||
|
|
let crate_json = match extra::json::from_str(crate_json_str) {
|
||
|
|
Ok(j) => j,
|
||
|
|
Err(_) => fail!("Rust generated JSON is invalid??")
|
||
|
|
};
|
||
|
|
|
||
|
|
json.insert(~"crate", crate_json);
|
||
|
|
json.insert(~"plugins", extra::json::Object(plugins_json));
|
||
|
|
|
||
|
|
println(extra::json::Object(json).to_str());
|
||
|
|
}
|