177a351462
- The Account Manager has a new subcommand, allowing generation of deterministic keys given a particular validator index. - Split functionality in generate_deterministic_keypair function - Fixed up READMEs to reflect new functionality & correct naming.
92 lines
3.4 KiB
Rust
92 lines
3.4 KiB
Rust
use bls::Keypair;
|
|
use clap::{App, Arg, SubCommand};
|
|
use slog::{debug, info, o, Drain};
|
|
use std::path::PathBuf;
|
|
use validator_client::Config as ValidatorClientConfig;
|
|
use types::test_utils::generate_deterministic_keypair;
|
|
|
|
fn main() {
|
|
// Logging
|
|
let decorator = slog_term::TermDecorator::new().build();
|
|
let drain = slog_term::CompactFormat::new(decorator).build().fuse();
|
|
let drain = slog_async::Async::new(drain).build().fuse();
|
|
let log = slog::Logger::root(drain, o!());
|
|
|
|
// CLI
|
|
let matches = App::new("Lighthouse Accounts Manager")
|
|
.version("0.0.1")
|
|
.author("Sigma Prime <contact@sigmaprime.io>")
|
|
.about("Eth 2.0 Accounts Manager")
|
|
.arg(
|
|
Arg::with_name("datadir")
|
|
.long("datadir")
|
|
.value_name("DIR")
|
|
.help("Data directory for keys and databases.")
|
|
.takes_value(true),
|
|
)
|
|
.subcommand(
|
|
SubCommand::with_name("generate")
|
|
.about("Generates a new validator private key")
|
|
.version("0.0.1")
|
|
.author("Sigma Prime <contact@sigmaprime.io>"),
|
|
)
|
|
.subcommand(
|
|
SubCommand::with_name("generate_deterministic")
|
|
.about("Generates a deterministic validator private key FOR TESTING")
|
|
.version("0.0.1")
|
|
.author("Sigma Prime <contact@sigmaprime.io>")
|
|
.arg(
|
|
Arg::with_name("validator index")
|
|
.long("index")
|
|
.short("i")
|
|
.value_name("index")
|
|
.help("The index of the validator, for which the test key is generated")
|
|
.takes_value(true)
|
|
.required(true)
|
|
)
|
|
)
|
|
.get_matches();
|
|
|
|
let config = ValidatorClientConfig::parse_args(&matches, &log)
|
|
.expect("Unable to build a configuration for the account manager.");
|
|
|
|
// Log configuration
|
|
info!(log, "";
|
|
"data_dir" => &config.data_dir.to_str());
|
|
|
|
match matches.subcommand() {
|
|
("generate", Some(_gen_m)) => {
|
|
let keypair = Keypair::random();
|
|
let key_path: PathBuf = config
|
|
.save_key(&keypair)
|
|
.expect("Unable to save newly generated private key.");
|
|
debug!(
|
|
log,
|
|
"Keypair generated {:?}, saved to: {:?}",
|
|
keypair.identifier(),
|
|
key_path.to_string_lossy()
|
|
);
|
|
},
|
|
("generate_deterministic", Some(gen_d_matches)) => {
|
|
let validator_index = gen_d_matches
|
|
.value_of("validator index")
|
|
.expect("Validator index required.")
|
|
.parse::<u64>()
|
|
.expect("Invalid validator index.") as usize;
|
|
let keypair = generate_deterministic_keypair(validator_index);
|
|
let key_path: PathBuf = config
|
|
.save_key(&keypair)
|
|
.expect("Unable to save newly generated deterministic private key.");
|
|
debug!(
|
|
log,
|
|
"Deterministic Keypair generated {:?}, saved to: {:?}",
|
|
keypair.identifier(),
|
|
key_path.to_string_lossy()
|
|
);
|
|
},
|
|
_ => panic!(
|
|
"The account manager must be run with a subcommand. See help for more information."
|
|
),
|
|
}
|
|
}
|