#[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());
}