lighthouse/beacon_node/rest_api/src/config.rs

use clap::ArgMatches;
use serde::{Deserialize, Serialize};
use std::net::Ipv4Addr;

/// HTTP REST API Configuration
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Config {
    /// Enable the REST API server.
    pub enabled: bool,
    /// The IPv4 address the REST API HTTP server will listen on.
    pub listen_address: Ipv4Addr,
    /// The port the REST API HTTP server will listen on.
    pub port: u16,
}

impl Default for Config {
    fn default() -> Self {
        Config {
            enabled: true, // rest_api enabled by default
            listen_address: Ipv4Addr::new(127, 0, 0, 1),
            port: 5052,
        }
    }
}

impl Config {
    pub fn apply_cli_args(&mut self, args: &ArgMatches) -> Result<(), &'static str> {
        if args.is_present("api") {
            self.enabled = true;
        }

        if let Some(rpc_address) = args.value_of("api-address") {
            self.listen_address = rpc_address
                .parse::<Ipv4Addr>()
                .map_err(|_| "api-address is not a valid IPv4 address.")?;
        }

        if let Some(rpc_port) = args.value_of("api-port") {
            self.port = rpc_port
                .parse::<u16>()
                .map_err(|_| "api-port is not a valid u16.")?;
        }

        Ok(())
    }
}