use crate::Error;
use clap::Parser;
use libp2p_identity::{ed25519, Keypair};
use std::{
fs,
io::{self, Write},
path::PathBuf,
};
#[derive(Debug, Parser)]
#[command(
name = "generate-node-key",
about = "Generate a random node key, write it to a file or stdout \
and write the corresponding peer-id to stderr"
)]
pub struct GenerateNodeKeyCmd {
#[arg(long)]
file: Option<PathBuf>,
#[arg(long)]
bin: bool,
}
impl GenerateNodeKeyCmd {
pub fn run(&self) -> Result<(), Error> {
let keypair = ed25519::Keypair::generate();
let secret = keypair.secret();
let file_data = if self.bin {
secret.as_ref().to_owned()
} else {
array_bytes::bytes2hex("", secret).into_bytes()
};
match &self.file {
Some(file) => fs::write(file, file_data)?,
None => io::stdout().lock().write_all(&file_data)?,
}
eprintln!("{}", Keypair::from(keypair).public().to_peer_id());
Ok(())
}
}
#[cfg(test)]
mod tests {
use super::*;
use std::io::Read;
use tempfile::Builder;
#[test]
fn generate_node_key() {
let mut file = Builder::new().prefix("keyfile").tempfile().unwrap();
let file_path = file.path().display().to_string();
let generate = GenerateNodeKeyCmd::parse_from(&["generate-node-key", "--file", &file_path]);
assert!(generate.run().is_ok());
let mut buf = String::new();
assert!(file.read_to_string(&mut buf).is_ok());
assert!(array_bytes::hex2bytes(&buf).is_ok());
}
}