hatter/runrs
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

refactor: refactor main.rs

Browse Source
This commit is contained in:
Hatter Jiang
2022-08-06 13:15:08 +08:00
parent d2249b6cac
commit 20287855b4
1 changed files with 52 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

86
src/main.rs
View File

@@ -9,21 +9,9 @@ use rust_util::{util_cmd, util_msg};
use rust_util::util_msg::MessageType;
fn main() {
let home = env::var("HOME").unwrap_or_else(|_|
failure_and_exit!("$HOME not found!")
);
let rust_script = PathBuf::from(format!("{}/.cargo/bin/rust-script", home));
if !rust_script.exists() {
warning!("rust-script not found, install it...");
let mut cargo_install_rust_script = Command::new("cargo");
cargo_install_rust_script.args(&["install", "--git", "https://git.hatter.ink/hatter/runrs", "rust-script"]);
debugging!("Run command: {:?}", cargo_install_rust_script);
match util_cmd::run_command_and_wait(&mut cargo_install_rust_script) {
Err(e) => failure_and_exit!("Install rust-script failed: {}", e),
Ok(exist_status) if !exist_status.success() => failure!("Install rust-script not success: {}", exist_status),
Ok(_) => (),
}
}
let home = get_user_home();
let rust_script = get_run_script_bin_name(&home);
let args = env::args().skip(1).collect::<Vec<_>>();
if args.is_empty() {
failure_and_exit!("runrs v{}, need arguments, e.g.\n\nrunrs <script.rs> [arguments]\n",
@@ -33,18 +21,7 @@ fn main() {
let first_argument = args.get(0).unwrap_or_else(|| failure_and_exit!("Must assign file name"));
if first_argument == "--help" {
println!(r##"{} v{} - {}
Help:
runrs --help
Run Rust Script:
runrs <script.rs> [arguments]
"##,
env!("CARGO_PKG_NAME"),
env!("CARGO_PKG_VERSION"),
env!("CARGO_PKG_DESCRIPTION")
);
print_help();
return;
}
@@ -76,14 +53,55 @@ runrs <script.rs> [arguments]
match util_cmd::run_command_and_wait(&mut run_script_cmd) {
Err(e) => failure_and_exit!("Run rust-script failed: {}", e),
Ok(exit_status) => {
if let Ok(Some(canonicalized_script_file)) = PathBuf::from(script_file)
.canonicalize().map(|f| f.to_str().map(|f| f.to_string())) {
let cache_script_bin_name_src = format!("{}.src", cache_script_bin_name);
if let Ok(_) = fs::write(&cache_script_bin_name_src, &format!("{}\n", canonicalized_script_file)) {
debugging!("Write {} to {}", canonicalized_script_file, cache_script_bin_name_src);
}
}
write_script_file_to_src(script_file, cache_script_bin_name);
process::exit(exit_status.code().unwrap_or_else(|| 0))
}
}
}
fn write_script_file_to_src(script_file: &String, cache_script_bin_name: String) {
if let Ok(Some(canonicalized_script_file)) = PathBuf::from(script_file)
.canonicalize().map(|f| f.to_str().map(|f| f.to_string())) {
let cache_script_bin_name_src = format!("{}.src", cache_script_bin_name);
if let Ok(_) = fs::write(&cache_script_bin_name_src, &format!("{}\n", canonicalized_script_file)) {
debugging!("Write {} to {}", canonicalized_script_file, cache_script_bin_name_src);
}
}
}
fn print_help() {
println!(r##"{} v{} - {}
Help:
runrs --help
Run Rust Script:
runrs <script.rs> [arguments]
"##,
env!("CARGO_PKG_NAME"),
env!("CARGO_PKG_VERSION"),
env!("CARGO_PKG_DESCRIPTION")
);
}
fn get_run_script_bin_name(home: &str) -> PathBuf {
let rust_script = PathBuf::from(format!("{}/.cargo/bin/rust-script", home));
if !rust_script.exists() {
warning!("rust-script not found, install it...");
let mut cargo_install_rust_script = Command::new("cargo");
cargo_install_rust_script.args(&["install", "--git", "https://git.hatter.ink/hatter/runrs", "rust-script"]);
debugging!("Run command: {:?}", cargo_install_rust_script);
match util_cmd::run_command_and_wait(&mut cargo_install_rust_script) {
Err(e) => failure_and_exit!("Install rust-script failed: {}", e),
Ok(exist_status) if !exist_status.success() => failure!("Install rust-script not success: {}", exist_status),
Ok(_) => (),
}
}
rust_script
}
fn get_user_home() -> String {
env::var("HOME").unwrap_or_else(|_|
failure_and_exit!("$HOME not found!")
)
}