feat: add argsutil

src/argsutil.rs

@@ -0,0 +1,22 @@
+use std::fs::File;
+use std::io::Read;
+
+use clap::ArgMatches;
+use rust_util::XResult;
+
+use crate::digest::{sha256, sha256_bytes};
+
+pub fn get_sha256_digest_or_hash(sub_arg_matches: &ArgMatches) -> XResult<Vec<u8>> {
+    if let Some(file) = sub_arg_matches.value_of("file") {
+        let mut f = opt_result!(File::open(file), "Open file: {} failed: {}", file);
+        let mut content = vec![];
+        opt_result!(f.read_to_end(&mut content), "Read file: {} failed: {}", file);
+        Ok(sha256_bytes(&content))
+    } else if let Some(input) = sub_arg_matches.value_of("input") {
+        Ok(sha256(input))
+    } else if let Some(hash_hex) = sub_arg_matches.value_of("hash-hex") {
+        Ok(opt_result!(hex::decode(hash_hex), "Parse hash-hex failed: {}"))
+    } else {
+        simple_error!("--file, --input or --hash-hex must assign at least one")
+    }
+}