mirror of
https://github.com/uutils/coreutils
synced 2024-11-17 02:08:09 +00:00
37 lines
973 B
Rust
37 lines
973 B
Rust
|
use std::env;
|
||
|
|
use std::io::Write;
|
||
|
|
use std::fs::File;
|
||
|
|
use std::path::Path;
|
||
|
|
|
||
|
|
static TEMPLATE: &'static str = "\
|
||
|
|
extern crate uu_@UTIL_CRATE@;
|
||
|
|
extern crate uucore;
|
||
|
|
|
||
|
|
use std::io::Write;
|
||
|
|
use uu_@UTIL_CRATE@::uumain;
|
||
|
|
|
||
|
|
fn main() {
|
||
|
|
uucore::panic::install_sigpipe_hook();
|
||
|
|
|
||
|
|
let code = uumain(std::env::args().collect());
|
||
|
|
// Since stdout is line-buffered by default, we need to ensure any pending
|
||
|
|
// writes are flushed before exiting. Ideally, this should be enforced by
|
||
|
|
// each utility.
|
||
|
|
//
|
||
|
|
// See: https://github.com/rust-lang/rust/issues/23818
|
||
|
|
//
|
||
|
|
std::io::stdout().flush().expect(\"could not flush stdout\");
|
||
|
|
std::process::exit(code);
|
||
|
|
}
|
||
|
|
";
|
||
|
|
|
||
|
|
pub fn main() {
|
||
|
|
let out_dir = env::var("OUT_DIR").unwrap();
|
||
|
|
let pkgname = env::var("CARGO_PKG_NAME").unwrap();
|
||
|
|
|
||
|
|
let main = TEMPLATE.replace("@UTIL_CRATE@", &pkgname);
|
||
|
|
let mut file = File::create(&Path::new(&out_dir).join("main.rs")).unwrap();
|
||
|
|
|
||
|
|
write!(file, "{}", main).unwrap();
|
||
|
|
}
|