61 lines
1.7 KiB
Rust
61 lines
1.7 KiB
Rust
use std::{net::{IpAddr, Ipv4Addr, SocketAddr, TcpListener}, path::PathBuf};
|
|
#[cfg(feature="desktop")]
|
|
use clap::Args;
|
|
use crate::{config::PartialConfig, utils::{emptiable::Emptiable, mergeable::Mergeable}};
|
|
use libp2p::mdns::Config;
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
use crate::config::error::ConfigError;
|
|
|
|
|
|
pub struct RpcConfig {
|
|
pub listen_address: IpAddr,
|
|
pub port: u16,
|
|
}
|
|
|
|
impl TryFrom<PartialRpcConfig> for RpcConfig {
|
|
type Error = ConfigError;
|
|
fn try_from(config: PartialRpcConfig) -> Result<Self, Self::Error> {
|
|
Ok(Self{
|
|
listen_address: config.listen_address.ok_or(ConfigError::MissingConfig("listen_address".to_string()))?,
|
|
port: config.port.ok_or(ConfigError::MissingConfig("port".to_string()))?,
|
|
})
|
|
}
|
|
}
|
|
|
|
#[derive(Args, Clone, Debug, Deserialize, Emptiable, Mergeable, Serialize)]
|
|
pub struct PartialRpcConfig {
|
|
pub listen_address: Option<IpAddr>,
|
|
pub port: Option<u16>,
|
|
}
|
|
impl PartialRpcConfig {
|
|
pub fn with_unused_port(mut self) -> Self {
|
|
let listneer = if let Some(x) = self.listen_address {
|
|
TcpListener::bind(SocketAddr::new(x,0)).unwrap()
|
|
} else {
|
|
TcpListener::bind("127.0.0.1:0").unwrap()
|
|
};
|
|
self.port = Some(listneer.local_addr().unwrap().port());
|
|
self
|
|
}
|
|
}
|
|
|
|
impl Default for PartialRpcConfig {
|
|
fn default() -> Self {
|
|
Self{
|
|
listen_address: Some(IpAddr::V4(Ipv4Addr::LOCALHOST)),
|
|
port: None,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl From<RpcConfig> for PartialRpcConfig {
|
|
fn from(source: RpcConfig) -> Self {
|
|
Self {
|
|
listen_address: Some(source.listen_address),
|
|
port: Some(source.port),
|
|
}
|
|
}
|
|
}
|
|
|
|
|