feat: updates

This commit is contained in:
2025-01-18 14:37:49 +08:00
parent 18d55793d9
commit 2e4c47e936
2 changed files with 12 additions and 18 deletions

View File

@@ -21,7 +21,7 @@ pub fn install_scripts(arguments: &[String]) {
} }
} }
pub fn install_script(script_name: &str, script_meta_map: &BTreeMap<String, ScriptMeta>) { fn install_script(script_name: &str, script_meta_map: &BTreeMap<String, ScriptMeta>) {
let script_meta = match script_meta_map.get(script_name) { let script_meta = match script_meta_map.get(script_name) {
None => { None => {
failure!("Script not found: {}", script_name); failure!("Script not found: {}", script_name);

View File

@@ -3,25 +3,19 @@ use std::os::unix::fs::PermissionsExt;
pub fn print_template(output: &Option<String>) { pub fn print_template(output: &Option<String>) {
let script = get_script_template(); let script = get_script_template();
match output { if let Some(output_file) = output {
None => { if fs::metadata(output_file).is_ok() {
println!("{}", script); failure!("Output script file exists: {}", output_file);
return;
} }
Some(output_file) => { if let Err(e) = fs::write(output_file, script) {
if fs::metadata(output_file).is_ok() { failure!("Write script file: {}, failed: {}", output_file, e);
failure!("Output script file exists: {}", output_file); } else {
} else { success!("Write script file success: {}", output_file);
match fs::write(output_file, script) { let _ = fs::set_permissions(output_file, PermissionsExt::from_mode(0o755));
Ok(_) => {
success!("Write script file success: {}", output_file);
let _ = fs::set_permissions(output_file, PermissionsExt::from_mode(0o755));
}
Err(e) => {
failure!("Write script file: {}, failed: {}", output_file, e);
}
}
}
} }
} else {
println!("{}", script);
} }
} }