use crate::Error;
use structopt::StructOpt;
use std::{path::PathBuf, fs};
use tetsy_libp2p::identity::{ed25519 as tetsy_libp2p_ed25519, PublicKey};
#[derive(Debug, StructOpt)]
#[structopt(
name = "generate-node-key",
about = "Generate a random node libp2p key, save it to \
file or print it to stdout and print its peer ID to stderr"
)]
pub struct GenerateNodeKeyCmd {
#[structopt(long)]
file: Option<PathBuf>,
}
impl GenerateNodeKeyCmd {
pub fn run(&self) -> Result<(), Error> {
let keypair = tetsy_libp2p_ed25519::Keypair::generate();
let secret = keypair.secret();
let peer_id = PublicKey::Ed25519(keypair.public()).into_peer_id();
let secret_hex = hex::encode(secret.as_ref());
match &self.file {
Some(file) => fs::write(file, secret_hex)?,
None => print!("{}", secret_hex),
}
eprintln!("{}", peer_id);
Ok(())
}
}
#[cfg(test)]
mod tests {
use super::*;
use tempfile::Builder;
use std::io::Read;
#[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::from_iter(&["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!(hex::decode(buf).is_ok());
}
}