210530da9c
When the CRETONNE_DBG environment variable is set, send debug messages to a file named cretonne.dbg.*. The trace facility is only enabled when debug assertions are on.
81 lines
2.0 KiB
Rust
81 lines
2.0 KiB
Rust
#[macro_use(dbg)]
|
|
extern crate cretonne;
|
|
extern crate cton_reader;
|
|
extern crate docopt;
|
|
extern crate rustc_serialize;
|
|
extern crate filecheck;
|
|
extern crate num_cpus;
|
|
|
|
use cretonne::VERSION;
|
|
use docopt::Docopt;
|
|
use std::io::{self, Write};
|
|
use std::process;
|
|
|
|
mod utils;
|
|
mod filetest;
|
|
mod cat;
|
|
mod print_cfg;
|
|
mod rsfilecheck;
|
|
|
|
const USAGE: &'static str = "
|
|
Cretonne code generator utility
|
|
|
|
Usage:
|
|
cton-util test [-v] <file>...
|
|
cton-util cat <file>...
|
|
cton-util filecheck [-v] <file>
|
|
cton-util print-cfg <file>...
|
|
cton-util --help | --version
|
|
|
|
Options:
|
|
-v, --verbose be more verbose
|
|
-h, --help print this help message
|
|
--version print the Cretonne version
|
|
|
|
";
|
|
|
|
#[derive(RustcDecodable, Debug)]
|
|
struct Args {
|
|
cmd_test: bool,
|
|
cmd_cat: bool,
|
|
cmd_filecheck: bool,
|
|
cmd_print_cfg: bool,
|
|
arg_file: Vec<String>,
|
|
flag_verbose: bool,
|
|
}
|
|
|
|
/// A command either succeeds or fails with an error message.
|
|
pub type CommandResult = Result<(), String>;
|
|
|
|
/// Parse the command line arguments and run the requested command.
|
|
fn cton_util() -> CommandResult {
|
|
// Parse comand line arguments.
|
|
let args: Args = Docopt::new(USAGE)
|
|
.and_then(|d| d.help(true).version(Some(format!("Cretonne {}", VERSION))).decode())
|
|
.unwrap_or_else(|e| e.exit());
|
|
|
|
// Find the sub-command to execute.
|
|
if args.cmd_test {
|
|
filetest::run(args.flag_verbose, args.arg_file)
|
|
} else if args.cmd_cat {
|
|
cat::run(args.arg_file)
|
|
} else if args.cmd_filecheck {
|
|
rsfilecheck::run(args.arg_file, args.flag_verbose)
|
|
} else if args.cmd_print_cfg {
|
|
print_cfg::run(args.arg_file)
|
|
} else {
|
|
// Debugging / shouldn't happen with proper command line handling above.
|
|
Err(format!("Unhandled args: {:?}", args))
|
|
}
|
|
}
|
|
|
|
fn main() {
|
|
if let Err(mut msg) = cton_util() {
|
|
if !msg.ends_with('\n') {
|
|
msg.push('\n');
|
|
}
|
|
io::stderr().write(msg.as_bytes()).unwrap();
|
|
process::exit(1);
|
|
}
|
|
}
|