50 lines
1.5 KiB
Rust
50 lines
1.5 KiB
Rust
use clap::{App, Arg, ArgMatches, SubCommand};
|
|
use rust_util::util_clap::{Command, CommandError};
|
|
use rust_util::util_msg;
|
|
use std::collections::BTreeMap;
|
|
|
|
use crate::hmacutil;
|
|
|
|
pub struct CommandImpl;
|
|
|
|
impl Command for CommandImpl {
|
|
fn name(&self) -> &str {
|
|
"hmac-encrypt"
|
|
}
|
|
|
|
fn subcommand<'a>(&self) -> App<'a, 'a> {
|
|
SubCommand::with_name(self.name())
|
|
.about("Yubikey HMAC encrypt")
|
|
.arg(
|
|
Arg::with_name("plaintext")
|
|
.long("plaintext")
|
|
.takes_value(true)
|
|
.help("Plaintext"),
|
|
)
|
|
.arg(Arg::with_name("json").long("json").help("JSON output"))
|
|
}
|
|
|
|
fn run(&self, _arg_matches: &ArgMatches, sub_arg_matches: &ArgMatches) -> CommandError {
|
|
let json_output = sub_arg_matches.is_present("json");
|
|
if json_output {
|
|
util_msg::set_logger_std_out(false);
|
|
}
|
|
|
|
let plaintext = sub_arg_matches.value_of("plaintext").unwrap();
|
|
let hmac_encrypt_ciphertext = hmacutil::hmac_encrypt_from_string(plaintext)?;
|
|
|
|
if json_output {
|
|
let mut json = BTreeMap::<&'_ str, String>::new();
|
|
json.insert("ciphertext", hmac_encrypt_ciphertext);
|
|
println!(
|
|
"{}",
|
|
serde_json::to_string_pretty(&json).expect("Convert to JSON failed!")
|
|
);
|
|
} else {
|
|
success!("HMAC encrypt ciphertext: {}", hmac_encrypt_ciphertext);
|
|
}
|
|
|
|
Ok(None)
|
|
}
|
|
}
|