68 lines
2.0 KiB
Rust
68 lines
2.0 KiB
Rust
|
use clap::ArgMatches;
|
||
|
use http_server::HttpServerConfig;
|
||
|
use network::NetworkConfig;
|
||
|
use serde_derive::{Deserialize, Serialize};
|
||
|
use std::fs;
|
||
|
use std::path::PathBuf;
|
||
|
|
||
|
/// The core configuration of a Lighthouse beacon node.
|
||
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||
|
pub struct Config {
|
||
|
pub data_dir: PathBuf,
|
||
|
pub db_type: String,
|
||
|
db_name: String,
|
||
|
pub network: network::NetworkConfig,
|
||
|
pub rpc: rpc::RPCConfig,
|
||
|
pub http: HttpServerConfig,
|
||
|
}
|
||
|
|
||
|
impl Default for Config {
|
||
|
fn default() -> Self {
|
||
|
Self {
|
||
|
data_dir: PathBuf::from(".lighthouse"),
|
||
|
db_type: "disk".to_string(),
|
||
|
db_name: "chain_db".to_string(),
|
||
|
// Note: there are no default bootnodes specified.
|
||
|
// Once bootnodes are established, add them here.
|
||
|
network: NetworkConfig::new(),
|
||
|
rpc: rpc::RPCConfig::default(),
|
||
|
http: HttpServerConfig::default(),
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
impl Config {
|
||
|
/// Returns the path to which the client may initialize an on-disk database.
|
||
|
pub fn db_path(&self) -> Option<PathBuf> {
|
||
|
self.data_dir()
|
||
|
.and_then(|path| Some(path.join(&self.db_name)))
|
||
|
}
|
||
|
|
||
|
/// Returns the core path for the client.
|
||
|
pub fn data_dir(&self) -> Option<PathBuf> {
|
||
|
let path = dirs::home_dir()?.join(&self.data_dir);
|
||
|
fs::create_dir_all(&path).ok()?;
|
||
|
Some(path)
|
||
|
}
|
||
|
|
||
|
/// Apply the following arguments to `self`, replacing values if they are specified in `args`.
|
||
|
///
|
||
|
/// Returns an error if arguments are obviously invalid. May succeed even if some values are
|
||
|
/// invalid.
|
||
|
pub fn apply_cli_args(&mut self, args: &ArgMatches) -> Result<(), String> {
|
||
|
if let Some(dir) = args.value_of("datadir") {
|
||
|
self.data_dir = PathBuf::from(dir);
|
||
|
};
|
||
|
|
||
|
if let Some(dir) = args.value_of("db") {
|
||
|
self.db_type = dir.to_string();
|
||
|
}
|
||
|
|
||
|
self.network.apply_cli_args(args)?;
|
||
|
self.rpc.apply_cli_args(args)?;
|
||
|
self.http.apply_cli_args(args)?;
|
||
|
|
||
|
Ok(())
|
||
|
}
|
||
|
}
|