lighthouse/src/main.rs

#[macro_use]
extern crate slog;
extern crate slog_term;
extern crate slog_async;
extern crate clap;
extern crate libp2p_peerstore;

pub mod p2p;
pub mod pubkeystore;
pub mod state;
pub mod sync;
pub mod utils;
pub mod config;

use std::path::PathBuf; 

use slog::Drain;
use clap::{ Arg, App };
use config::LighthouseConfig;
use p2p::service::NetworkService;
use p2p::state::NetworkState;
use sync::sync_start;

fn main() {
    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!());

    let matches = App::new("Lighthouse")
        .version("0.0.1")
        .author("Sigma Prime <paul@sigmaprime.io>")
        .about("Eth 2.0 Client")
        .arg(Arg::with_name("datadir")
            .long("datadir")
            .value_name("DIR")
            .help("Data directory for keys and databases.")
            .takes_value(true))
        .arg(Arg::with_name("port")
            .long("port")
            .value_name("PORT")
            .help("Network listen port for p2p connections.")
            .takes_value(true))
        .get_matches();

    let mut config = LighthouseConfig::default();

    // Custom datadir
    if let Some(dir) = matches.value_of("datadir") {
        config.data_dir = PathBuf::from(dir.to_string());
    }

    // Custom p2p listen port
    if let Some(port) = matches.value_of("port") {
        config.p2p_listen_port = port.to_string();
    }

    info!(log, ""; "data_dir" => &config.data_dir.to_str());
    if let Some(_) = matches.subcommand_matches("generate-keys") {
        // keys::generate_keys(&log).expect("Failed to generate keys");
    } else {
        let mut state = NetworkState::new(config, &log).expect("setup failed");
        let (service, net_rx) = NetworkService::new(state, log.new(o!()));
        sync_start(service, net_rx, log.new(o!()));
    }
    info!(log, "Exiting.");
}