veilid-igd 0.1.1

use std::net::{IpAddr, Ipv4Addr, Ipv6Addr, SocketAddr};
use std::time::Duration;

/// Gateway search configuration
///
/// SearchOptions::default() should suffice for most situations.
///
/// # Example
/// To customize only a few options you can use `Default::default()` or `SearchOptions::default()` and the
/// [struct update syntax](https://doc.rust-lang.org/book/ch05-01-defining-structs.html#creating-instances-from-other-instances-with-struct-update-syntax).
/// ```
/// # use std::time::Duration;
/// # use veilid_igd::SearchOptions;
/// let opts = SearchOptions {
///     timeout: Some(Duration::from_secs(60)),
///     ..Default::default()
/// };
/// ```
pub struct SearchOptions {
    /// Bind address for UDP socket (defaults to all `0.0.0.0`)
    pub bind_addr: SocketAddr,
    /// Broadcast address for discovery packets (defaults to `239.255.255.250:1900`)
    pub broadcast_address: SocketAddr,
    /// Timeout for a search iteration (defaults to 10s)
    pub timeout: Option<Duration>,
}

impl Default for SearchOptions {
    fn default() -> Self {
        Self::new_v4(10_000)
    }
}

/// IPv6 scope to search for gateway
pub enum Ipv6SearchScope {
    /// Link-local search scope
    LinkLocal,
    /// Subnet search scope
    Subnet,
    /// Administrative search scope
    Admin,
    /// Site-local search scope
    SiteLocal,
}

impl SearchOptions {
    /// Create a new IPv4 search
    pub fn new_v4(timeout_ms: u64) -> Self {
        Self {
            bind_addr: (IpAddr::V4(Ipv4Addr::UNSPECIFIED), 0).into(),
            broadcast_address: "239.255.255.250:1900".parse().unwrap(),
            timeout: Some(Duration::from_millis(timeout_ms)),
        }
    }
    /// Create a new IPv6 search
    pub fn new_v6(scope: Ipv6SearchScope, timeout_ms: u64) -> Self {
        Self {
            bind_addr: (IpAddr::V6(Ipv6Addr::UNSPECIFIED), 0).into(),
            broadcast_address: match scope {
                Ipv6SearchScope::LinkLocal => "[FF02::C]:1900",
                Ipv6SearchScope::Subnet => "[FF03::C]:1900",
                Ipv6SearchScope::Admin => "[FF04::C]:1900",
                Ipv6SearchScope::SiteLocal => "[FF05::C]:1900",
            }
            .parse()
            .unwrap(),
            timeout: Some(Duration::from_millis(timeout_ms)),
        }
    }
}