use crate::Error;
use clap::Parser;
use libp2p_identity::Keypair;
use std::{
fs,
io::{self, Read},
path::PathBuf,
};
#[derive(Debug, Parser)]
#[command(
name = "inspect-node-key",
about = "Load a node key from a file or stdin and print the corresponding peer-id."
)]
pub struct InspectNodeKeyCmd {
#[arg(long)]
file: Option<PathBuf>,
#[arg(long)]
bin: bool,
#[deprecated(note = "Network identifier is not used for node-key inspection")]
#[arg(short = 'n', long = "network", value_name = "NETWORK", ignore_case = true)]
pub network_scheme: Option<String>,
}
impl InspectNodeKeyCmd {
pub fn run(&self) -> Result<(), Error> {
let mut file_data = match &self.file {
Some(file) => fs::read(&file)?,
None => {
let mut buf = Vec::with_capacity(64);
io::stdin().lock().read_to_end(&mut buf)?;
buf
},
};
if !self.bin {
let keyhex = String::from_utf8_lossy(&file_data);
file_data = array_bytes::hex2bytes(keyhex.trim())
.map_err(|_| "failed to decode secret as hex")?;
}
let keypair =
Keypair::ed25519_from_bytes(&mut file_data).map_err(|_| "Bad node key file")?;
println!("{}", keypair.public().to_peer_id());
Ok(())
}
}
#[cfg(test)]
mod tests {
use crate::commands::generate_node_key::GenerateNodeKeyCmd;
use super::*;
#[test]
fn inspect_node_key() {
let path = tempfile::tempdir().unwrap().into_path().join("node-id").into_os_string();
let path = path.to_str().unwrap();
let cmd = GenerateNodeKeyCmd::parse_from(&["generate-node-key", "--file", path]);
assert!(cmd.run("test", &String::from("test")).is_ok());
let cmd = InspectNodeKeyCmd::parse_from(&["inspect-node-key", "--file", path]);
assert!(cmd.run().is_ok());
}
}