do smarter IP deserialization

This commit is contained in:
Joe Ardent 2025-08-08 13:38:16 -07:00
parent 940e7984e3
commit cb071f4a7c
2 changed files with 21 additions and 15 deletions

View file

@ -147,7 +147,7 @@ impl Widget for &mut App {
buf,
);
network_info(
&self.service.config.local_ip_addr.unwrap_or([0u8; 4].into()),
&self.service.config.local_ip_addr,
footer_left.inner(footer_margin),
buf,
);

View file

@ -9,7 +9,7 @@ use figment::{
providers::{Format, Serialized, Toml},
};
use local_ip_address::local_ip;
use serde::{Deserialize, Serialize};
use serde::{Deserialize, Deserializer, Serialize};
use crate::{
DEFAULT_PORT, MULTICAST_IP,
@ -22,8 +22,8 @@ pub struct Config {
pub multicast_addr: SocketAddrV4,
pub download_dir: PathBuf,
pub data_dir: PathBuf,
#[serde(skip_serializing)]
pub local_ip_addr: Option<Ipv4Addr>,
#[serde(deserialize_with = "deserialize_local_addr")]
pub local_ip_addr: Ipv4Addr,
pub device: Device,
}
@ -33,7 +33,7 @@ impl Default for Config {
multicast_addr: SocketAddrV4::new(MULTICAST_IP, DEFAULT_PORT),
download_dir: Default::default(),
data_dir: Default::default(),
local_ip_addr: None,
local_ip_addr: Ipv4Addr::from_bits(0),
device: Default::default(),
}
}
@ -47,10 +47,6 @@ impl Config {
let config_file = dirs.config_dir().join("jocalsend.toml");
let data_dir = dirs.data_local_dir().join("jocalsend");
let IpAddr::V4(local_ip_addr) = local_ip()? else {
return Err(LocalSendError::IPv6Unsupported);
};
let key = data_dir.join("key.pem");
let cert = data_dir.join("cert.pem");
let fingerprint = if data_dir.exists() {
@ -67,8 +63,8 @@ impl Config {
let config = Self {
multicast_addr: SocketAddrV4::new(MULTICAST_IP, DEFAULT_PORT),
local_ip_addr: get_local_ip_addr(),
download_dir,
local_ip_addr: Some(local_ip_addr),
data_dir,
device: Device {
alias: rustix::system::uname()
@ -80,7 +76,7 @@ impl Config {
},
};
let mut config = if !config_file.exists() {
let config = if !config_file.exists() {
log::info!("creating config file at {config_file:?}");
std::fs::write(&config_file, toml::to_string(&config)?)?;
config
@ -92,10 +88,6 @@ impl Config {
.map_err(Box::new)? // boxed because the error size from figment is large
};
if config.local_ip_addr.is_none() {
config.local_ip_addr = Some(local_ip_addr);
}
log::info!("using config: {config:?}");
Ok(config)
@ -118,3 +110,17 @@ fn gen_ssl(key: &Path, cert: &Path) -> Result<String> {
std::fs::write(cert, cert_text)?;
Ok(sha256::digest(key_text))
}
fn deserialize_local_addr<'de, D>(_: D) -> std::result::Result<Ipv4Addr, D::Error>
where
D: Deserializer<'de>,
{
Ok(get_local_ip_addr())
}
fn get_local_ip_addr() -> Ipv4Addr {
let IpAddr::V4(local_ip_addr) = local_ip().unwrap_or(IpAddr::V4(Ipv4Addr::from_bits(0))) else {
panic!("IPv6 is unsupported");
};
local_ip_addr
}