65 lines
1.7 KiB
Rust
65 lines
1.7 KiB
Rust
|
use std::fmt;
|
||
|
|
|
||
|
|
use iroh::EndpointAddr;
|
||
|
|
|
||
|
|
pub const MVP_IROH_ENDPOINT_ADDR_MASK_ENV: &str = "MVP_IROH_ENDPOINT_ADDR_MASK";
|
||
|
|
|
||
|
|
#[derive(Clone, Copy, Debug, Default, PartialEq, Eq)]
|
||
|
|
pub enum EndpointAddrMask {
|
||
|
|
#[default]
|
||
|
|
Full,
|
||
|
|
RelayOnly,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl EndpointAddrMask {
|
||
|
|
pub fn parse(value: &str) -> Result<Self, String> {
|
||
|
|
match value.trim().to_ascii_lowercase().as_str() {
|
||
|
|
"" | "full" | "none" => Ok(Self::Full),
|
||
|
|
"relay-only" | "relay_only" | "relay" => Ok(Self::RelayOnly),
|
||
|
|
other => Err(format!(
|
||
|
|
"unsupported {MVP_IROH_ENDPOINT_ADDR_MASK_ENV}={other:?}; use full or relay-only"
|
||
|
|
)),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn as_str(self) -> &'static str {
|
||
|
|
match self {
|
||
|
|
Self::Full => "full",
|
||
|
|
Self::RelayOnly => "relay-only",
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn requires_relay(self) -> bool {
|
||
|
|
matches!(self, Self::RelayOnly)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl fmt::Display for EndpointAddrMask {
|
||
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||
|
|
f.write_str(self.as_str())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn advertised_endpoint(
|
||
|
|
endpoint: EndpointAddr,
|
||
|
|
mask: EndpointAddrMask,
|
||
|
|
) -> Result<EndpointAddr, String> {
|
||
|
|
match mask {
|
||
|
|
EndpointAddrMask::Full => Ok(endpoint),
|
||
|
|
EndpointAddrMask::RelayOnly => relay_only_endpoint(endpoint),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
fn relay_only_endpoint(endpoint: EndpointAddr) -> Result<EndpointAddr, String> {
|
||
|
|
let relays = endpoint.relay_urls().cloned().collect::<Vec<_>>();
|
||
|
|
if relays.is_empty() {
|
||
|
|
return Err("relay-only endpoint address mask requires an endpoint relay URL".to_owned());
|
||
|
|
}
|
||
|
|
|
||
|
|
let mut out = EndpointAddr::new(endpoint.id);
|
||
|
|
for relay in relays {
|
||
|
|
out = out.with_relay_url(relay);
|
||
|
|
}
|
||
|
|
Ok(out)
|
||
|
|
}
|