rift_sdk/

lib.rs

//! Rift SDK: high-level API for embedding Rift VoIP in other applications.
//!
//! This crate wraps mesh, media, discovery, and NAT components into a cohesive
//! runtime with a simpler API surface for native embedding.

use std::collections::HashMap;
use std::fs;
use std::path::{Path, PathBuf};
use std::net::{SocketAddr, ToSocketAddrs};
use std::sync::{
    atomic::{AtomicBool, Ordering},
    Arc, Mutex as StdMutex,
};
use std::time::{Duration, SystemTime, UNIX_EPOCH};

use anyhow::{Context, Result};
use rand::rngs::OsRng;
use rand::RngCore;
use serde::{Deserialize, Serialize};
use serde_json::json;
use tokio::sync::{mpsc, Mutex};
use tokio::time::Instant;

use rift_core::{decode_invite, generate_invite, Identity, Invite, PeerId, KeyStore};
use rift_dht::{DhtConfig as RiftDhtConfig, DhtHandle, PeerEndpointInfo};
use rift_discovery::local_ipv4_addrs;
use rift_media::{
    decode_frame, encode_frame, AudioConfig, AudioIn, AudioMixer, AudioOut, OpusDecoder,
    OpusEncoder,
};
use rift_mesh::{Mesh, MeshConfig, MeshEvent, MeshHandle};
use rift_nat::{
    attempt_hole_punch, gather_local_candidates, gather_public_addrs, parse_turn_server, NatConfig,
    PeerEndpoint,
};
use rift_protocol::{CallState, Capabilities, QosProfile, SessionId};
use rift_rndzv::{
    ChannelKind as RndzvChannelKind, PeerId as RndzvPeerId, RndzvChannel, RndzvConnectTarget,
    RndzvConnector, RndzvListener, Srt as RndzvSrt,
    EscalationPolicy, IdentityConstraints, RendezvousSpaceId, SearchStrategy,
    SemanticRendezvousToken, TimeModel,
};
use hkdf::Hkdf;
use sha2::Sha256;

pub use rift_core::PeerId as RiftPeerId;
pub use rift_protocol::{
    CallState as RiftCallState, ChatMessage, CodecId, FeatureFlag, GroupMode,
    QosProfile as RiftQosProfile, SessionId as RiftSessionId,
};

pub const SDK_VERSION: &str = "0.1.0";
pub const SDK_ABI_VERSION: i32 = 1;

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct RiftConfig {
    /// Optional path to the identity key.
    pub identity_path: Option<PathBuf>,
    /// UDP listen port.
    pub listen_port: u16,
    /// Whether this node can act as a relay.
    pub relay: bool,
    /// Optional display name for UI surfaces.
    pub user_name: Option<String>,
    /// Preferred codecs for negotiation.
    pub preferred_codecs: Vec<CodecId>,
    /// Preferred feature flags for negotiation.
    pub preferred_features: Vec<FeatureFlag>,
    /// QoS tuning parameters.
    #[serde(default)]
    pub qos: QosProfile,
    /// Whether metrics are enabled.
    #[serde(default)]
    pub metrics_enabled: bool,
    /// Security settings (E2EE, auth, etc).
    #[serde(default)]
    pub security: SecurityConfig,
    /// DHT configuration.
    pub dht: DhtConfigSdk,
    /// Audio configuration.
    pub audio: AudioConfigSdk,
    /// Network configuration.
    pub network: NetworkConfigSdk,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct AudioConfigSdk {
    /// Enable audio capture/playback.
    pub enabled: bool,
    /// Optional input device name.
    pub input_device: Option<String>,
    /// Optional output device name.
    pub output_device: Option<String>,
    /// Quality preset identifier.
    pub quality: String,
    /// Push-to-talk enabled.
    pub ptt: bool,
    /// Voice activity detection enabled.
    pub vad: bool,
    /// Mute output playback.
    pub mute_output: bool,
    /// Emit raw voice frames to consumers.
    pub emit_voice_frames: bool,
    /// Allow audio init failures without crashing.
    pub allow_fail: bool,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct NetworkConfigSdk {
    /// Prefer direct P2P routes if possible.
    pub prefer_p2p: bool,
    /// Explicit list of local ports to bind.
    pub local_ports: Option<Vec<u16>>,
    /// Explicit peers to contact on startup.
    pub known_peers: Vec<std::net::SocketAddr>,
    pub invite: Option<String>,
    #[serde(default)]
    pub stun_servers: Vec<String>,
    #[serde(default)]
    pub stun_timeout_ms: Option<u64>,
    #[serde(default)]
    pub enable_turn: bool,
    #[serde(default)]
    pub turn_servers: Vec<String>,
    #[serde(default)]
    pub turn_timeout_ms: Option<u64>,
    #[serde(default)]
    pub turn_keepalive_ms: Option<u64>,
    #[serde(default)]
    pub punch_interval_ms: Option<u64>,
    #[serde(default)]
    pub punch_timeout_ms: Option<u64>,
    #[serde(default)]
    pub max_direct_peers: Option<usize>,
    /// Optional Predictive Rendezvous configuration.
    #[serde(default)]
    pub rndzv: Option<RndzvConfigSdk>,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct RndzvConfigSdk {
    /// SRT URI used for rendezvous.
    pub srt_uri: String,
    /// Role for rendezvous: connector or listener.
    pub role: RndzvRole,
    /// Optional remote address for connector mode.
    pub remote_addr: Option<SocketAddr>,
    /// Optional local bind address for listener mode.
    pub listen_addr: Option<SocketAddr>,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub enum RndzvRole {
    Connector,
    Listener,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct SrtInvite {
    /// Human-readable label for the invite (e.g. "Alice Voice Call").
    pub label: String,
    /// Encoded SRT URI.
    pub uri: String,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct DhtConfigSdk {
    /// Enable DHT discovery.
    pub enabled: bool,
    /// Bootstrap node addresses (string form).
    pub bootstrap_nodes: Vec<String>,
    /// Optional local listen addr override.
    pub listen_addr: Option<String>,
}

impl Default for RiftConfig {
    fn default() -> Self {
        Self {
            identity_path: None,
            listen_port: 7777,
            relay: false,
            user_name: None,
            preferred_codecs: vec![CodecId::Opus, CodecId::PCM16],
            preferred_features: vec![
                FeatureFlag::Voice,
                FeatureFlag::Text,
                FeatureFlag::Relay,
                FeatureFlag::E2EE,
            ],
            qos: QosProfile::default(),
            metrics_enabled: true,
            security: SecurityConfig::default(),
            dht: DhtConfigSdk::default(),
            audio: AudioConfigSdk::default(),
            network: NetworkConfigSdk::default(),
        }
    }
}

/// Create a voice-call SRT invite targeted at a specific peer.
pub fn create_voice_invite(to: PeerId) -> SrtInvite {
    let mut seed = [0u8; 32];
    OsRng.fill_bytes(&mut seed);
    let space = RendezvousSpaceId(*blake3::hash(b"rift-rndzv-voice-call").as_bytes());
    let now = SystemTime::now()
        .duration_since(UNIX_EPOCH)
        .unwrap_or_default()
        .as_secs();
    let token = SemanticRendezvousToken::new(
        space,
        seed,
        IdentityConstraints {
            allowed_fingerprints: vec![to.0],
        },
        TimeModel {
            t0: now.saturating_add(10),
            window_secs: 120,
            slot_ms: 250,
        },
        SearchStrategy::BasicDeterministic,
        EscalationPolicy::None,
    );
    let uri = token
        .to_uri()
        .expect("SRT URI encoding should succeed for valid inputs");
    SrtInvite {
        label: "Voice Call".to_string(),
        uri,
    }
}

/// Parse an SRT invite into a rendezvous token.
pub fn accept_voice_invite(invite: &SrtInvite) -> Result<RndzvSrt, RiftError> {
    RndzvSrt::from_uri(&invite.uri)
        .map_err(|e| RiftError::Other(format!("rndzv srt decode failed: {e}")))
}

#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(default)]
pub struct SecurityConfig {
    /// Trust first-seen identity keys.
    pub trust_on_first_use: bool,
    /// Optional path for known-hosts storage.
    pub known_hosts_path: Option<PathBuf>,
    /// Reject peers on key mismatch.
    pub reject_on_mismatch: bool,
    /// Optional channel shared secret.
    pub channel_shared_secret: Option<String>,
    /// Optional audit log path.
    pub audit_log_path: Option<PathBuf>,
    /// Rekey interval in seconds.
    pub rekey_interval_secs: Option<u64>,
}

impl Default for SecurityConfig {
    fn default() -> Self {
        Self {
            trust_on_first_use: true,
            known_hosts_path: None,
            reject_on_mismatch: false,
            channel_shared_secret: None,
            audit_log_path: None,
            rekey_interval_secs: Some(600),
        }
    }
}

impl Default for AudioConfigSdk {
    fn default() -> Self {
        Self {
            enabled: true,
            input_device: None,
            output_device: None,
            quality: "medium".to_string(),
            ptt: false,
            vad: true,
            mute_output: false,
            emit_voice_frames: false,
            allow_fail: false,
        }
    }
}

impl Default for NetworkConfigSdk {
    fn default() -> Self {
        Self {
            prefer_p2p: true,
            local_ports: None,
            known_peers: Vec::new(),
            invite: None,
            stun_servers: vec![
                "stun.l.google.com:19302".to_string(),
                "stun1.l.google.com:19302".to_string(),
            ],
            stun_timeout_ms: Some(800),
            enable_turn: false,
            turn_servers: Vec::new(),
            turn_timeout_ms: Some(1200),
            turn_keepalive_ms: Some(12000),
            punch_interval_ms: Some(200),
            punch_timeout_ms: Some(5000),
            max_direct_peers: None,
            rndzv: None,
        }
    }
}

impl Default for DhtConfigSdk {
    fn default() -> Self {
        Self {
            enabled: false,
            bootstrap_nodes: Vec::new(),
            listen_addr: None,
        }
    }
}

#[derive(Debug, Clone, Copy)]
pub struct LinkStats {
    /// Round-trip time in milliseconds.
    pub rtt_ms: f32,
    /// Packet loss fraction.
    pub loss: f32,
    /// Jitter in milliseconds.
    pub jitter_ms: f32,
}

#[derive(Debug, Clone, Copy)]
pub struct GlobalStats {
    /// Connected peers.
    pub num_peers: usize,
    /// Active sessions.
    pub num_sessions: usize,
    /// Packets sent.
    pub packets_sent: u64,
    /// Packets received.
    pub packets_received: u64,
    /// Bytes sent.
    pub bytes_sent: u64,
    /// Bytes received.
    pub bytes_received: u64,
}

#[derive(Debug, Clone)]
pub enum RouteKind {
    Direct,
    Relayed { via: PeerId },
}

#[derive(Debug, Clone)]
pub enum RiftEvent {
    /// Incoming chat message.
    IncomingChat(ChatMessage),
    /// Incoming call invitation.
    IncomingCall {
        session: SessionId,
        from: PeerId,
        rndzv_srt_uri: Option<String>,
    },
    /// Call state changes (ringing/active/ended).
    CallStateChanged { session: SessionId, state: CallState },
    /// A peer joined the channel.
    PeerJoinedChannel { peer: PeerId, channel: String },
    /// A peer left the channel.
    PeerLeftChannel { peer: PeerId, channel: String },
    /// Peer capability advertisement.
    PeerCapabilities { peer: PeerId, capabilities: Capabilities },
    /// Audio level update for UI metering.
    AudioLevel { peer: PeerId, level: f32 },
    /// Codec selection update.
    CodecSelected { codec: CodecId },
    /// Audio bitrate update for diagnostics.
    AudioBitrate { bitrate: u32 },
    /// Periodic stats update.
    StatsUpdate { peer: PeerId, stats: LinkStats, global: GlobalStats },
    /// Routing update for a peer.
    RouteUpdated { peer: PeerId, route: RouteKind },
    /// Group topology update.
    GroupTopology { session: SessionId, mode: GroupMode },
    /// Peer fingerprint for trust UX.
    PeerFingerprint { peer: PeerId, fingerprint: String },
    /// Security-related notice (TOFU, mismatch, etc).
    SecurityNotice { message: String },
    /// Raw voice samples (when enabled).
    VoiceFrame { peer: PeerId, samples: Vec<i16> },
}

#[derive(Debug, thiserror::Error)]
/// High-level SDK error type.
pub enum RiftError {
    /// SDK runtime not initialized.
    #[error("not initialized")]
    NotInitialized,
    /// Channel already joined.
    #[error("channel already joined")]
    AlreadyJoined,
    /// Channel not joined.
    #[error("channel not joined")]
    NotJoined,
    /// Mesh subsystem error.
    #[error("mesh error: {0}")]
    Mesh(String),
    /// Audio subsystem error.
    #[error("audio error: {0}")]
    Audio(String),
    /// Generic failure.
    #[error("other: {0}")]
    Other(String),
}

/// Runtime state for audio capture, mixing, and encoding.
struct VoiceRuntime {
    _audio_in: AudioIn,
    mixer: Arc<StdMutex<AudioMixer>>,
    frame_samples: usize,
    emit_voice: bool,
    audio_config: AudioConfig,
    tuning: Arc<StdMutex<AudioTuning>>,
    rndzv_channel: Arc<StdMutex<Option<RndzvChannel>>>,
    rndzv_remote_peer: Arc<StdMutex<Option<PeerId>>>,
}

impl VoiceRuntime {
    fn set_rndzv_channel(&self, channel: RndzvChannel, remote_peer: PeerId) {
        {
            let mut slot = self.rndzv_channel.lock().unwrap();
            *slot = Some(channel.clone());
        }
        {
            let mut peer_slot = self.rndzv_remote_peer.lock().unwrap();
            *peer_slot = Some(remote_peer);
        }

        let mixer = self.mixer.clone();
        let frame_samples = self.frame_samples;
        let emit_voice = self.emit_voice;
        let audio_config = self.audio_config.clone();
        tokio::spawn(async move {
            let mut decoder = match OpusDecoder::new(&audio_config) {
                Ok(decoder) => decoder,
                Err(err) => {
                    tracing::warn!("rndzv opus decoder init failed: {err}");
                    return;
                }
            };
            loop {
                let payload = match channel.recv().await {
                    Ok(Some(payload)) => payload,
                    Ok(None) => continue,
                    Err(err) => {
                        tracing::warn!("rndzv channel recv failed: {err}");
                        return;
                    }
                };
                if let Ok(out) = decode_frame(CodecId::Opus, &payload, &mut decoder, frame_samples) {
                    let mut mixer = mixer.lock().unwrap();
                    mixer.push(peer_to_stream_id(&remote_peer), out.clone());
                    if emit_voice {
                        let _ = out;
                    }
                }
            }
        });
    }

    fn clear_rndzv_channel(&self) {
        {
            let mut slot = self.rndzv_channel.lock().unwrap();
            *slot = None;
        }
        {
            let mut peer_slot = self.rndzv_remote_peer.lock().unwrap();
            *peer_slot = None;
        }
    }
}

/// Current audio tuning parameters.
#[derive(Debug, Clone)]
struct AudioTuning {
    bitrate: u32,
    fec: bool,
    loss_pct: u8,
}

/// QoS state used to adapt audio settings to current network stats.
struct QosState {
    profile: QosProfile,
    peer_stats: HashMap<PeerId, LinkStats>,
    current: AudioTuning,
    last_adjust: Instant,
}

/// Per-session runtime handle for mesh + audio + DHT.
struct SessionRuntime {
    _channel: String,
    handle: MeshHandle,
    _voice: Option<Arc<VoiceRuntime>>,
    _dht: Option<DhtHandle>,
    pending_call_srt: Arc<StdMutex<HashMap<SessionId, String>>>,
}

/// Primary SDK handle exposed to callers.
pub struct RiftHandle {
    /// Persistent identity (if initialized).
    identity: Mutex<Option<Identity>>,
    /// Local peer id (generated at startup).
    _local_peer_id: PeerId,
    /// Effective runtime configuration.
    config: RiftConfig,
    /// In-memory overrides (PTT, mute, etc).
    overrides: Mutex<RiftConfigOverrides>,
    /// Session runtime (mesh/audio/dht).
    runtime: Mutex<Option<SessionRuntime>>,
    /// Event receiver for consumer-facing events.
    event_rx: Mutex<mpsc::UnboundedReceiver<RiftEvent>>,
    event_tx: mpsc::UnboundedSender<RiftEvent>,
    ptt_active: Arc<AtomicBool>,
    mute_active: Arc<AtomicBool>,
}

#[derive(Debug, Default, Clone)]
/// Runtime configuration overrides applied before joining.
struct RiftConfigOverrides {
    dht_enabled: Option<bool>,
    bootstrap_nodes: Option<Vec<String>>,
    invite: Option<String>,
    turn_servers: Option<Vec<String>>,
    audio_quality: Option<String>,
}

impl RiftHandle {
    /// Initialize the SDK runtime with the given config and identity.
    pub async fn new(config: RiftConfig) -> Result<Self, RiftError> {
        rift_metrics::set_enabled(config.metrics_enabled);
        let (event_tx, event_rx) = mpsc::unbounded_channel();
        let identity_path = config
            .identity_path
            .clone()
            .unwrap_or_else(|| Identity::default_path().unwrap_or_else(|_| PathBuf::from("identity.key")));
        let existed = identity_path.exists();
        let identity = KeyStore::load_or_generate(&identity_path)
            .context("identity load failed")
            .map_err(|e| RiftError::Other(format!("{e}")))?;
        if !existed {
            tracing::info!(path = %identity_path.display(), "new identity generated");
        }
        let local_peer_id = identity.peer_id;
        Ok(Self {
            ptt_active: Arc::new(AtomicBool::new(!config.audio.ptt)),
            identity: Mutex::new(Some(identity)),
            _local_peer_id: local_peer_id,
            config,
            overrides: Mutex::new(RiftConfigOverrides::default()),
            runtime: Mutex::new(None),
            event_rx: Mutex::new(event_rx),
            event_tx,
            mute_active: Arc::new(AtomicBool::new(false)),
        })
    }

    /// Enable or disable DHT discovery for subsequent joins.
    pub async fn set_dht_enabled(&self, enabled: bool) {
        let mut overrides = self.overrides.lock().await;
        overrides.dht_enabled = Some(enabled);
    }

    /// Override DHT bootstrap nodes.
    pub async fn set_bootstrap_nodes(&self, nodes: Vec<String>) {
        let mut overrides = self.overrides.lock().await;
        overrides.bootstrap_nodes = Some(nodes);
    }

    /// Provide an invite link override for joining.
    pub async fn set_invite(&self, invite: Option<String>) {
        let mut overrides = self.overrides.lock().await;
        overrides.invite = invite;
    }

    /// Configure TURN servers for relay fallback.
    pub async fn set_turn_servers(&self, servers: Vec<String>) {
        let mut overrides = self.overrides.lock().await;
        overrides.turn_servers = Some(servers);
    }

    /// Override the audio quality preset.
    pub async fn set_audio_quality(&self, quality: Option<String>) {
        let mut overrides = self.overrides.lock().await;
        overrides.audio_quality = quality;
    }

    /// Join a channel by name/password, optionally using internet mode.
    pub async fn join_channel(
        &self,
        name: &str,
        password: Option<&str>,
        internet: bool,
    ) -> Result<(), RiftError> {
        let mut cfg = self.config.clone();
        {
            let overrides = self.overrides.lock().await;
            if let Some(enabled) = overrides.dht_enabled {
                cfg.dht.enabled = enabled;
            }
            if let Some(nodes) = overrides.bootstrap_nodes.clone() {
                cfg.dht.bootstrap_nodes = nodes;
            }
            if let Some(invite) = overrides.invite.clone() {
                cfg.network.invite = Some(invite);
            }
            if let Some(turn_servers) = overrides.turn_servers.clone() {
                cfg.network.turn_servers = turn_servers;
                cfg.network.enable_turn = !cfg.network.turn_servers.is_empty();
            }
            if let Some(quality) = overrides.audio_quality.clone() {
                cfg.audio.quality = quality;
            }
        }
        let mut runtime_guard = self.runtime.lock().await;
        if runtime_guard.is_some() {
            return Err(RiftError::AlreadyJoined);
        }
        let identity = {
            let mut identity_guard = self.identity.lock().await;
            match identity_guard.take() {
                Some(identity) => identity,
                None => Identity::load(cfg.identity_path.as_deref())
                    .context("identity not found")
                    .map_err(|e| RiftError::Other(format!("{e}")))?,
            }
        };

        let auth_token = self
            .config
            .security
            .channel_shared_secret
            .as_deref()
            .map(|secret| derive_auth_token(secret, name));
        let nat_cfg = if internet {
            Some(default_nat_config(
                cfg.listen_port,
                cfg.network.local_ports.clone(),
                cfg.network.stun_servers.clone(),
                cfg.network.stun_timeout_ms,
                cfg.network.punch_interval_ms,
                cfg.network.punch_timeout_ms,
                cfg.network.enable_turn,
                cfg.network.turn_servers.clone(),
                cfg.network.turn_timeout_ms,
                cfg.network.turn_keepalive_ms,
            ))
        } else {
            None
        };
        let mut known_peers = cfg.network.known_peers.clone();
        if internet && known_peers.is_empty() {
            if let Some(nat_cfg) = nat_cfg.as_ref() {
                if !nat_cfg.stun_servers.is_empty() {
                    if let Ok(public_addrs) = gather_public_addrs(nat_cfg).await {
                        if !public_addrs.is_empty() {
                            known_peers = public_addrs;
                        }
                    }
                }
            }
        }
        let invite_for_key = if let Some(invite_str) = &cfg.network.invite {
            Some(decode_invite(invite_str).map_err(|e| RiftError::Other(format!("{e}")))?)
        } else if internet {
            let mut candidates = gather_local_candidates(cfg.listen_port);
            if let Some(nat_cfg) = nat_cfg.as_ref() {
                if !nat_cfg.stun_servers.is_empty() {
                    if let Ok(public_addrs) = gather_public_addrs(nat_cfg).await {
                        candidates.extend(public_addrs);
                    }
                }
            }
            candidates.sort();
            candidates.dedup();
            Some(generate_invite(
                name,
                password,
                known_peers.clone(),
                candidates,
            ))
        } else {
            None
        };
        let e2ee_key = derive_e2ee_key(
            name,
            password,
            invite_for_key.as_ref(),
            cfg.security.channel_shared_secret.as_deref(),
        );
        let config = MeshConfig {
            channel_name: name.to_string(),
            password: password.map(|v| v.to_string()),
            listen_port: cfg.listen_port,
            relay_capable: cfg.relay,
            qos: cfg.qos.clone(),
            auth_token,
            require_auth: cfg.security.channel_shared_secret.is_some(),
            e2ee_key,
            rekey_interval_secs: cfg.security.rekey_interval_secs,
            max_direct_peers: cfg.network.max_direct_peers,
        };
        let mut mesh = Mesh::new(identity, config)
            .await
            .map_err(|e| RiftError::Mesh(format!("{e}")))?;

        let handle = mesh.handle();
        handle
            .set_preferred_codecs(self.config.preferred_codecs.clone())
            .await;
        handle
            .set_preferred_features(self.config.preferred_features.clone())
            .await;
        let security_handle = handle.clone();

        if internet {
            let nat_cfg = nat_cfg.clone().expect("nat cfg");
            mesh.enable_nat(nat_cfg.clone()).await;
            let invite = invite_for_key.clone().unwrap_or_else(|| Invite {
                channel_name: name.to_string(),
                password: password.map(|v| v.to_string()),
                channel_key: [0u8; 32],
                known_peers: Vec::new(),
                candidates: Vec::new(),
                relay_candidates: Vec::new(),
                version: 2,
                created_at: now_timestamp(),
            });
            mesh.join_invite(invite, nat_cfg)
                .await
                .map_err(|e| RiftError::Mesh(format!("{e}")))?;
        } else {
            mesh.start_lan_discovery()
                .map_err(|e| RiftError::Mesh(format!("{e}")))?;
        }

        let event_tx = self.event_tx.clone();
        let channel = name.to_string();
        let channel_for_task = channel.clone();
        let pending_call_srt: Arc<StdMutex<HashMap<SessionId, String>>> =
            Arc::new(StdMutex::new(HashMap::new()));
        let pending_call_srt_task = pending_call_srt.clone();
        let voice = if cfg.audio.enabled {
            match start_audio_pipeline(
                cfg.clone(),
                handle.clone(),
                self.local_peer_id(),
                self.ptt_active.clone(),
                self.mute_active.clone(),
            ) {
                Ok(voice) => Some(Arc::new(voice)),
                Err(err) => {
                    if cfg.audio.allow_fail {
                        tracing::warn!("audio pipeline failed: {err}");
                        None
                    } else {
                        return Err(err);
                    }
                }
            }
        } else {
            None
        };
        if let Some(voice) = voice.as_ref() {
            let _ = event_tx.send(RiftEvent::AudioBitrate {
                bitrate: voice.audio_config.bitrate,
            });
        }

        let dht = if cfg.dht.enabled {
            let dht_config = RiftDhtConfig {
                bootstrap_nodes: parse_socket_addrs(&cfg.dht.bootstrap_nodes),
                listen_addr: cfg
                    .dht
                    .listen_addr
                    .as_deref()
                    .and_then(parse_socket_addr)
                    .unwrap_or_else(|| {
                        SocketAddr::from(([0, 0, 0, 0], cfg.listen_port.saturating_add(100)))
                    }),
            };
            let handle_dht = DhtHandle::new(dht_config)
                .await
                .map_err(|e| RiftError::Other(format!("{e}")))?;

            let channel_id = rift_core::ChannelId::from_channel(&channel, password);
            let nat_cfg = nat_cfg.clone().unwrap_or_else(|| default_nat_config(
                cfg.listen_port,
                cfg.network.local_ports.clone(),
                cfg.network.stun_servers.clone(),
                cfg.network.stun_timeout_ms,
                cfg.network.punch_interval_ms,
                cfg.network.punch_timeout_ms,
                cfg.network.enable_turn,
                cfg.network.turn_servers.clone(),
                cfg.network.turn_timeout_ms,
                cfg.network.turn_keepalive_ms,
            ));
            let addrs = match gather_public_addrs(&nat_cfg).await {
                Ok(public_addrs) if !public_addrs.is_empty() => public_addrs,
                _ => local_ipv4_addrs()
                    .map_err(|e| RiftError::Other(format!("{e}")))?
                    .into_iter()
                    .map(|ip| SocketAddr::new(ip, cfg.listen_port))
                    .collect::<Vec<_>>(),
            };
            let info = PeerEndpointInfo {
                peer_id: self.local_peer_id(),
                addrs,
            };
            let _ = handle_dht.announce(channel_id, info.clone()).await;

            let announce_handle = handle_dht.clone();
            let announce_info = info.clone();
            tokio::spawn(async move {
                let mut tick = tokio::time::interval(Duration::from_secs(30));
                loop {
                    tick.tick().await;
                    let _ = announce_handle
                        .announce(channel_id, announce_info.clone())
                        .await;
                }
            });

            let lookup_handle = handle_dht.clone();
            let mesh_handle = handle.clone();
            let nat_cfg = default_nat_config(
                self.config.listen_port,
                self.config.network.local_ports.clone(),
                self.config.network.stun_servers.clone(),
                self.config.network.stun_timeout_ms,
                self.config.network.punch_interval_ms,
                self.config.network.punch_timeout_ms,
                self.config.network.enable_turn,
                self.config.network.turn_servers.clone(),
                self.config.network.turn_timeout_ms,
                self.config.network.turn_keepalive_ms,
            );
            tokio::spawn(async move {
                let mut tick = tokio::time::interval(Duration::from_secs(12));
                loop {
                    tick.tick().await;
                    if let Ok(peers) = lookup_handle.lookup(channel_id).await {
                        for peer in peers {
                            if peer.peer_id == info.peer_id {
                                continue;
                            }
                            for addr in peer.addrs.iter().copied() {
                                let endpoint = PeerEndpoint {
                                    peer_id: peer.peer_id,
                                    external_addrs: vec![addr],
                                    punch_ports: vec![addr.port()],
                                };
                                if let Ok((socket, remote)) = attempt_hole_punch(&nat_cfg, &endpoint).await {
                                    let _ = mesh_handle.connect_with_socket(socket, remote).await;
                                } else {
                                    let _ = mesh_handle.connect_addr(addr).await;
                                }
                            }
                        }
                    }
                }
            });

            Some(handle_dht)
        } else {
            None
        };

        let voice_state = voice.as_ref().map(|v| VoiceRuntimeRef {
            mixer: v.mixer.clone(),
            frame_samples: v.frame_samples,
            emit_voice: v.emit_voice,
            audio_config: v.audio_config.clone(),
            tuning: v.tuning.clone(),
        });
        let security_cfg = self.config.security.clone();
        let qos_profile = self.config.qos.clone();
        let mut qos_state = voice_state.as_ref().map(|state| QosState {
            profile: qos_profile,
            peer_stats: HashMap::new(),
            current: AudioTuning {
                bitrate: state.audio_config.bitrate,
                fec: false,
                loss_pct: 0,
            },
            last_adjust: Instant::now() - Duration::from_secs(5),
        });

        tokio::spawn(async move {
            let mut mesh = mesh;
            let mut decoder = if voice_state.is_some() {
                Some(
                    OpusDecoder::new(&voice_state.as_ref().unwrap().audio_config)
                        .expect("opus decoder"),
                )
            } else {
                None
            };
            while let Some(event) = mesh.next_event().await {
                match event {
                    MeshEvent::PeerJoined(peer) => {
                        let _ = event_tx.send(RiftEvent::PeerJoinedChannel {
                            peer,
                            channel: channel_for_task.clone(),
                        });
                    }
                    MeshEvent::PeerLeft(peer) => {
                        let _ = event_tx.send(RiftEvent::PeerLeftChannel {
                            peer,
                            channel: channel_for_task.clone(),
                        });
                    }
                    MeshEvent::ChatReceived(chat) => {
                        let _ = event_tx.send(RiftEvent::IncomingChat(chat));
                    }
                    MeshEvent::IncomingCall {
                        session,
                        from,
                        rndzv_srt_uri,
                    } => {
                        if let Some(uri) = rndzv_srt_uri.clone() {
                            let mut map = pending_call_srt_task.lock().unwrap();
                            map.insert(session, uri);
                        }
                        let _ = event_tx.send(RiftEvent::IncomingCall {
                            session,
                            from,
                            rndzv_srt_uri,
                        });
                    }
                    MeshEvent::CallAccepted { session, .. } => {
                        let _ = event_tx.send(RiftEvent::CallStateChanged {
                            session,
                            state: CallState::Active,
                        });
                    }
                    MeshEvent::CallDeclined { session, .. } => {
                        let _ = event_tx.send(RiftEvent::CallStateChanged {
                            session,
                            state: CallState::Ended,
                        });
                    }
                    MeshEvent::CallEnded { session } => {
                        let _ = event_tx.send(RiftEvent::CallStateChanged {
                            session,
                            state: CallState::Ended,
                        });
                    }
                    MeshEvent::VoiceFrame { from, codec, payload, .. } => {
                        if let (Some(state), Some(decoder)) = (voice_state.as_ref(), decoder.as_mut()) {
                            if let Ok(out) = decode_frame(codec, &payload, decoder, state.frame_samples) {
                                let mut mixer = state.mixer.lock().unwrap();
                                mixer.push(peer_to_stream_id(&from), out.clone());
                                let level = audio_level(&out);
                                let _ = event_tx.send(RiftEvent::AudioLevel { peer: from, level });
                                if state.emit_voice {
                                    let _ = event_tx.send(RiftEvent::VoiceFrame { peer: from, samples: out });
                                }
                            }
                        }
                    }
                    MeshEvent::PeerCapabilities { peer_id, capabilities } => {
                        let _ = event_tx.send(RiftEvent::PeerCapabilities { peer: peer_id, capabilities });
                    }
                    MeshEvent::GroupCodec(codec) => {
                        let _ = event_tx.send(RiftEvent::CodecSelected { codec });
                    }
                    MeshEvent::StatsUpdate { peer, stats, global } => {
                        let sdk_stats = LinkStats {
                            rtt_ms: stats.rtt_ms,
                            loss: stats.loss,
                            jitter_ms: stats.jitter_ms,
                        };
                        let sdk_global = GlobalStats {
                            num_peers: global.num_peers,
                            num_sessions: global.num_sessions,
                            packets_sent: global.packets_sent,
                            packets_received: global.packets_received,
                            bytes_sent: global.bytes_sent,
                            bytes_received: global.bytes_received,
                        };
                        let _ = event_tx.send(RiftEvent::StatsUpdate {
                            peer,
                            stats: sdk_stats,
                            global: sdk_global,
                        });
                        if let (Some(state), Some(qos)) = (voice_state.as_ref(), qos_state.as_mut()) {
                            qos.peer_stats.insert(peer, sdk_stats);
                            if let Some(next) = compute_next_tuning(qos) {
                                let mut tuning = state.tuning.lock().unwrap();
                                let bitrate_changed = tuning.bitrate != next.bitrate;
                                *tuning = next.clone();
                                if bitrate_changed {
                                    let _ = event_tx.send(RiftEvent::AudioBitrate { bitrate: next.bitrate });
                                }
                            }
                        }
                    }
                    MeshEvent::RouteUpdated { peer_id, route } => {
                        let route = match route {
                            rift_mesh::PeerRoute::Direct { .. } => RouteKind::Direct,
                            rift_mesh::PeerRoute::Relayed { via } => RouteKind::Relayed { via },
                        };
                        let _ = event_tx.send(RiftEvent::RouteUpdated {
                            peer: peer_id,
                            route,
                        });
                    }
                    MeshEvent::GroupTopology { session, mode } => {
                        let _ = event_tx.send(RiftEvent::GroupTopology { session, mode });
                    }
                    MeshEvent::PeerIdentity { peer_id, public_key } => {
                        if let Err(err) = handle_peer_identity(
                            &event_tx,
                            &security_handle,
                            &security_cfg,
                            peer_id,
                            &public_key,
                        )
                        .await
                        {
                            tracing::warn!("security check failed: {err}");
                        }
                    }
                    MeshEvent::PeerSessionConfig { .. } | MeshEvent::RouteUpgraded(_) => {}
                }
            }
        });

        *runtime_guard = Some(SessionRuntime {
            _channel: channel,
            handle,
            _voice: voice,
            _dht: dht,
            pending_call_srt,
        });
        Ok(())
    }

    /// Leave the currently joined channel and tear down runtime state.
    pub async fn leave_channel(&self, _name: &str) -> Result<(), RiftError> {
        let mut runtime_guard = self.runtime.lock().await;
        if runtime_guard.is_none() {
            return Err(RiftError::NotJoined);
        }
        *runtime_guard = None;
        Ok(())
    }

    /// Send a chat message to all peers.
    pub async fn send_chat(&self, text: &str) -> Result<(), RiftError> {
        let runtime_guard = self.runtime.lock().await;
        let runtime = runtime_guard.as_ref().ok_or(RiftError::NotJoined)?;
        runtime
            .handle
            .broadcast_chat(text.to_string())
            .await
            .map_err(|e| RiftError::Mesh(format!("{e}")))
    }

    /// Start a call with a specific peer.
    pub async fn start_call(&self, peer: PeerId) -> Result<SessionId, RiftError> {
        self.start_call_with_srt(peer, None).await
    }

    /// Start a call with an optional rndzv SRT URI attached to the invite.
    pub async fn start_call_with_srt(
        &self,
        peer: PeerId,
        rndzv_srt_uri: Option<String>,
    ) -> Result<SessionId, RiftError> {
        let parsed_srt = if let Some(uri) = rndzv_srt_uri.as_ref() {
            Some(
                RndzvSrt::from_uri(uri)
                    .map_err(|e| RiftError::Other(format!("rndzv srt decode failed: {e}")))?,
            )
        } else {
            None
        };
        let (handle, voice) = {
            let runtime_guard = self.runtime.lock().await;
            let runtime = runtime_guard.as_ref().ok_or(RiftError::NotJoined)?;
            (runtime.handle.clone(), runtime._voice.clone())
        };

        let session = handle
            .start_call_with_srt(peer, rndzv_srt_uri.clone())
            .await
            .map_err(|e| RiftError::Mesh(format!("{e}")))?;

        if let (Some(srt), Some(voice)) = (parsed_srt, voice) {
            let local_peer = RndzvPeerId(self.local_peer_id().0);
            let target = RndzvConnectTarget::from_srt(srt, local_peer);
            let connector = RndzvConnector::new().with_timeout(Duration::from_secs(5));
            let outcome = connector
                .connect(target)
                .await
                .map_err(|e| RiftError::Other(format!("rndzv connect failed: {e}")))?;
            let session = outcome.session;

            let channel = session
                .open_channel(RndzvChannelKind::UnreliableDatagram)
                .await
                .map_err(|e| RiftError::Other(format!("rndzv channel open failed: {e}")))?;
            let remote_peer = PeerId((session.remote).0);
            voice.set_rndzv_channel(channel, remote_peer);
        }

        Ok(session)
    }

    /// Accept an incoming call.
    pub async fn accept_call(&self, session: SessionId) -> Result<(), RiftError> {
        let (handle, voice, pending_call_srt) = {
            let runtime_guard = self.runtime.lock().await;
            let runtime = runtime_guard.as_ref().ok_or(RiftError::NotJoined)?;
            (
                runtime.handle.clone(),
                runtime._voice.clone(),
                runtime.pending_call_srt.clone(),
            )
        };
        let srt_uri = {
            let mut map = pending_call_srt.lock().unwrap();
            map.remove(&session)
        };
        let parsed_srt = if let Some(uri) = srt_uri.as_ref() {
            Some(
                RndzvSrt::from_uri(uri)
                    .map_err(|e| RiftError::Other(format!("rndzv srt decode failed: {e}")))?,
            )
        } else {
            None
        };

        handle
            .accept_call(session)
            .await
            .map_err(|e| RiftError::Mesh(format!("{e}")))?;

        if let (Some(srt), Some(voice)) = (parsed_srt, voice) {
            let local_peer = RndzvPeerId(self.local_peer_id().0);
            let listener = RndzvListener::new(srt.space, local_peer).with_srt(srt);
            let outcome = listener
                .accept()
                .await
                .map_err(|e| RiftError::Other(format!("rndzv accept failed: {e}")))?;
            let session = outcome.session;

            let channel = session
                .open_channel(RndzvChannelKind::UnreliableDatagram)
                .await
                .map_err(|e| RiftError::Other(format!("rndzv channel open failed: {e}")))?;
            let remote_peer = PeerId((session.remote).0);
            voice.set_rndzv_channel(channel, remote_peer);
        }

        Ok(())
    }

    /// Decline an incoming call with optional reason.
    pub async fn decline_call(&self, session: SessionId, reason: Option<&str>) -> Result<(), RiftError> {
        let (handle, pending_call_srt) = {
            let runtime_guard = self.runtime.lock().await;
            let runtime = runtime_guard.as_ref().ok_or(RiftError::NotJoined)?;
            (runtime.handle.clone(), runtime.pending_call_srt.clone())
        };
        {
            let mut map = pending_call_srt.lock().unwrap();
            map.remove(&session);
        }
        handle
            .decline_call(session, reason.map(|v| v.to_string()))
            .await
            .map_err(|e| RiftError::Mesh(format!("{e}")))
    }

    /// End an active call session.
    pub async fn end_call(&self, session: SessionId) -> Result<(), RiftError> {
        let (handle, voice) = {
            let runtime_guard = self.runtime.lock().await;
            let runtime = runtime_guard.as_ref().ok_or(RiftError::NotJoined)?;
            (runtime.handle.clone(), runtime._voice.clone())
        };
        if let Some(voice) = voice {
            voice.clear_rndzv_channel();
        }
        handle
            .end_call(session)
            .await
            .map_err(|e| RiftError::Mesh(format!("{e}")))
    }

    /// Await the next event from the SDK.
    pub async fn next_event(&self) -> Option<RiftEvent> {
        let mut rx = self.event_rx.lock().await;
        rx.recv().await
    }

    /// Try to fetch the next event without awaiting.
    pub fn try_next_event(&self) -> Option<RiftEvent> {
        let mut rx = self.event_rx.blocking_lock();
        rx.try_recv().ok()
    }

    /// Set push-to-talk active state.
    pub fn set_ptt_active(&self, active: bool) {
        self.ptt_active.store(active, Ordering::Relaxed);
    }

    /// Mute or unmute microphone capture.
    pub fn set_mute(&self, muted: bool) {
        self.mute_active.store(muted, Ordering::Relaxed);
    }

    /// Return the local peer id.
    pub fn local_peer_id(&self) -> PeerId {
        self._local_peer_id
    }
}

struct VoiceRuntimeRef {
    mixer: Arc<StdMutex<AudioMixer>>,
    frame_samples: usize,
    emit_voice: bool,
    audio_config: AudioConfig,
    tuning: Arc<StdMutex<AudioTuning>>,
}

/// Initialize the audio capture/playback pipeline and spawn processing tasks.
fn start_audio_pipeline(
    config: RiftConfig,
    handle: MeshHandle,
    _local_peer_id: PeerId,
    ptt_active: Arc<AtomicBool>,
    mute_active: Arc<AtomicBool>,
) -> Result<VoiceRuntime, RiftError> {
    let mut audio_config = AudioConfig::default();
    let initial_bitrate = map_quality_to_bitrate(Some(&config.audio.quality));
    audio_config.bitrate = initial_bitrate
        .clamp(config.qos.min_bitrate, config.qos.max_bitrate)
        .max(8_000);
    rift_metrics::set_gauge("rift_audio_bitrate", &[], audio_config.bitrate as f64);
    let (audio_in, mut audio_rx) = AudioIn::new_with_device(&audio_config, config.audio.input_device.as_deref())
        .map_err(|e| RiftError::Audio(format!("{e}")))?;
    let mut encoder = OpusEncoder::new(&audio_config).map_err(|e| RiftError::Audio(format!("{e}")))?;
    let output_device = config.audio.output_device.clone();
    let mixer = Arc::new(StdMutex::new(AudioMixer::with_prebuffer(
        audio_config.frame_samples(),
        8,
    )));

    let ptt_enabled = config.audio.ptt;
    let vad_enabled = config.audio.vad;
    let frame_duration = audio_config.frame_duration();
    let tuning = Arc::new(StdMutex::new(AudioTuning {
        bitrate: audio_config.bitrate,
        fec: false,
        loss_pct: 0,
    }));
    let tuning_for_task = tuning.clone();

    let rndzv_channel: Arc<StdMutex<Option<RndzvChannel>>> = Arc::new(StdMutex::new(None));
    let rndzv_remote_peer: Arc<StdMutex<Option<PeerId>>> = Arc::new(StdMutex::new(None));
    let rndzv_channel_for_task = rndzv_channel.clone();
    tokio::spawn(async move {
        let mut seq: u32 = 0;
        let mut hangover: u8 = 0;
        let mut last_applied = AudioTuning {
            bitrate: audio_config.bitrate,
            fec: false,
            loss_pct: 0,
        };
        let rndzv_sender = rndzv_channel_for_task.clone();
        while let Some(frame) = audio_rx.recv().await {
            let next_tuning = {
                let tuning = tuning_for_task.lock().unwrap();
                tuning.clone()
            };
            if next_tuning.bitrate != last_applied.bitrate
                || next_tuning.fec != last_applied.fec
                || next_tuning.loss_pct != last_applied.loss_pct
            {
                if let Err(err) = encoder.set_bitrate(next_tuning.bitrate) {
                    tracing::debug!("opus bitrate update failed: {err}");
                }
                if let Err(err) = encoder.set_fec(next_tuning.fec) {
                    tracing::debug!("opus fec update failed: {err}");
                }
                if let Err(err) = encoder.set_packet_loss(next_tuning.loss_pct) {
                    tracing::debug!("opus loss update failed: {err}");
                }
                last_applied = next_tuning;
            }
            if ptt_enabled && !ptt_active.load(Ordering::Relaxed) {
                continue;
            }
            if mute_active.load(Ordering::Relaxed) {
                continue;
            }
            if !ptt_enabled && vad_enabled {
                let active = is_frame_active(&frame);
                if active {
                    hangover = 4;
                } else if hangover > 0 {
                    hangover -= 1;
                }
                if !active && hangover == 0 {
                    continue;
                }
            }
            if frame_duration > Duration::from_millis(20) {
                tokio::time::sleep(frame_duration - Duration::from_millis(20)).await;
            }
            let codec = handle.group_codec().await;
            let out = match encode_frame(codec, &frame, &mut encoder) {
                Ok(out) => out,
                Err(_) => continue,
            };
            let timestamp = now_timestamp();
            let maybe_channel = rndzv_sender.lock().unwrap().clone();
            if let Some(channel) = maybe_channel {
                let _ = channel.send(&out).await;
            } else {
                let _ = handle.broadcast_voice(seq, timestamp, out).await;
            }
            seq = seq.wrapping_add(1);
        }
    });

    if !config.audio.mute_output {
        let mixer = mixer.clone();
        let audio_config = audio_config.clone();
        std::thread::spawn(move || {
            let audio_out = match AudioOut::new_with_device(&audio_config, output_device.as_deref()) {
                Ok(out) => out,
                Err(err) => {
                    tracing::warn!("audio output init failed: {err}");
                    return;
                }
            };
            let frame_samples = audio_out.frame_samples();
            let frame_duration = audio_config.frame_duration();
            let target_frames = 6usize;
            let mut last_frame = vec![0i16; frame_samples];
            let mut last_active = Instant::now() - Duration::from_secs(1);
            loop {
                std::thread::sleep(frame_duration);
                while audio_out.queued_samples() < target_frames * frame_samples {
                    let (frame, active) = {
                        let mut mixer = mixer.lock().unwrap();
                        mixer.mix_next_with_activity()
                    };
                    let out_frame = if active {
                        last_active = Instant::now();
                        last_frame.clone_from(&frame);
                        frame
                    } else if last_active.elapsed() <= Duration::from_millis(300) {
                        last_frame.clone()
                    } else {
                        frame
                    };
                    if out_frame.len() == frame_samples {
                        audio_out.push_frame(&out_frame);
                    } else {
                        break;
                    }
                }
            }
        });
    }

    Ok(VoiceRuntime {
        _audio_in: audio_in,
        mixer,
        frame_samples: audio_config.frame_samples(),
        emit_voice: config.audio.emit_voice_frames,
        audio_config,
        tuning,
        rndzv_channel,
        rndzv_remote_peer,
    })
}

fn audio_level(frame: &[i16]) -> f32 {
    let mut sum = 0f32;
    for s in frame {
        sum += (*s as f32).abs();
    }
    (sum / frame.len().max(1) as f32) / i16::MAX as f32
}

fn is_frame_active(frame: &[i16]) -> bool {
    let mut sum = 0i64;
    for s in frame {
        sum += (*s as i64).abs();
    }
    let avg = sum / frame.len().max(1) as i64;
    avg > 250
}

/// Compute the next audio tuning parameters based on current QoS stats.
fn compute_next_tuning(qos: &mut QosState) -> Option<AudioTuning> {
    if qos.peer_stats.is_empty() {
        return None;
    }
    let now = Instant::now();
    if now.duration_since(qos.last_adjust) < Duration::from_secs(2) {
        return None;
    }
    let mut worst_rtt = 0.0f32;
    let mut worst_loss = 0.0f32;
    for stats in qos.peer_stats.values() {
        worst_rtt = worst_rtt.max(stats.rtt_ms);
        worst_loss = worst_loss.max(stats.loss);
    }

    let mut bitrate = qos.current.bitrate;
    let max_latency = qos.profile.max_latency_ms as f32;
    let target_latency = qos.profile.target_latency_ms as f32;
    if worst_loss > qos.profile.packet_loss_tolerance || worst_rtt > max_latency {
        bitrate = ((bitrate as f32) * 0.8) as u32;
    } else if worst_loss < qos.profile.packet_loss_tolerance * 0.5
        && worst_rtt < target_latency
    {
        bitrate = ((bitrate as f32) * 1.1) as u32;
    }
    bitrate = bitrate
        .clamp(qos.profile.min_bitrate, qos.profile.max_bitrate)
        .max(8_000);

    let fec = worst_loss > qos.profile.packet_loss_tolerance * 0.5;
    let loss_pct = (worst_loss * 100.0).round().min(100.0) as u8;

    let next = AudioTuning {
        bitrate,
        fec,
        loss_pct,
    };
    if next.bitrate != qos.current.bitrate
        || next.fec != qos.current.fec
        || next.loss_pct != qos.current.loss_pct
    {
        rift_metrics::set_gauge("rift_audio_bitrate", &[], next.bitrate as f64);
        tracing::info!(
            bitrate = next.bitrate,
            fec = next.fec,
            loss_pct = next.loss_pct,
            "qos audio tuning updated"
        );
        qos.current = next.clone();
        qos.last_adjust = now;
        return Some(next);
    }
    None
}

fn map_quality_to_bitrate(quality: Option<&str>) -> u32 {
    match quality.unwrap_or("medium") {
        "low" => 24_000,
        "high" => 96_000,
        _ => 48_000,
    }
}

fn now_timestamp() -> u64 {
    SystemTime::now()
        .duration_since(UNIX_EPOCH)
        .map(|d| d.as_millis() as u64)
        .unwrap_or(0)
}

fn peer_to_stream_id(peer: &PeerId) -> u64 {
    let mut bytes = [0u8; 8];
    bytes.copy_from_slice(&peer.0[..8]);
    u64::from_le_bytes(bytes)
}

fn default_nat_config(
    port: u16,
    ports: Option<Vec<u16>>,
    stun_servers: Vec<String>,
    stun_timeout_ms: Option<u64>,
    punch_interval_ms: Option<u64>,
    punch_timeout_ms: Option<u64>,
    enable_turn: bool,
    turn_servers: Vec<String>,
    turn_timeout_ms: Option<u64>,
    turn_keepalive_ms: Option<u64>,
) -> NatConfig {
    let mut local_ports = ports.unwrap_or_default();
    if local_ports.is_empty() {
        local_ports.push(port);
        local_ports.push(port.saturating_add(1));
        local_ports.push(port.saturating_add(2));
    }
    NatConfig {
        local_ports,
        stun_servers: parse_socket_addrs(&stun_servers),
        stun_timeout_ms: stun_timeout_ms.unwrap_or(800),
        punch_interval_ms: punch_interval_ms.unwrap_or(200),
        punch_timeout_ms: punch_timeout_ms.unwrap_or(5000),
        turn_servers: if enable_turn {
            turn_servers
                .into_iter()
                .filter_map(|s| parse_turn_server(&s).ok())
                .collect()
        } else {
            Vec::new()
        },
        turn_timeout_ms: turn_timeout_ms.unwrap_or(1200),
        turn_keepalive_ms: turn_keepalive_ms.unwrap_or(12000),
    }
}

/// Derive a per-channel auth token from a shared secret.
fn derive_auth_token(secret: &str, channel: &str) -> Vec<u8> {
    let hk = Hkdf::<Sha256>::new(Some(channel.as_bytes()), secret.as_bytes());
    let mut out = [0u8; 32];
    hk.expand(b"rift-auth", &mut out)
        .expect("hkdf expand");
    out.to_vec()
}

fn derive_e2ee_key(
    channel: &str,
    password: Option<&str>,
    invite: Option<&Invite>,
    shared_secret: Option<&str>,
) -> Option<[u8; 32]> {
    if let Some(secret) = shared_secret {
        let hk = Hkdf::<Sha256>::new(Some(channel.as_bytes()), secret.as_bytes());
        let mut out = [0u8; 32];
        hk.expand(b"rift-e2ee", &mut out)
            .expect("hkdf expand");
        return Some(out);
    }
    if let Some(invite) = invite {
        if invite.channel_key.iter().any(|b| *b != 0) {
            return Some(invite.channel_key);
        }
    }
    if let Some(password) = password {
        let mut hasher = blake3::Hasher::new();
        hasher.update(b"rift-e2ee:");
        hasher.update(channel.as_bytes());
        hasher.update(b":");
        hasher.update(password.as_bytes());
        let mut out = [0u8; 32];
        out.copy_from_slice(hasher.finalize().as_bytes());
        return Some(out);
    }
    None
}

fn short_peer(peer_id: &PeerId) -> String {
    let hex = peer_id.to_hex();
    hex.chars().take(8).collect()
}

fn resolve_known_hosts_path(cfg: &SecurityConfig) -> Result<PathBuf, RiftError> {
    if let Some(path) = &cfg.known_hosts_path {
        return Ok(expand_tilde(path));
    }
    let base = dirs::config_dir().ok_or_else(|| RiftError::Other("config dir missing".to_string()))?;
    Ok(base.join("rift").join("known_hosts"))
}

fn expand_tilde(path: &Path) -> PathBuf {
    let path_str = path.to_string_lossy();
    if let Some(rest) = path_str.strip_prefix("~/") {
        if let Some(home) = dirs::home_dir() {
            return home.join(rest);
        }
    }
    path.to_path_buf()
}

fn load_known_hosts(path: &Path) -> HashMap<PeerId, Vec<u8>> {
    let mut map = HashMap::new();
    let Ok(content) = fs::read_to_string(path) else {
        return map;
    };
    for line in content.lines() {
        let line = line.trim();
        if line.is_empty() || line.starts_with('#') {
            continue;
        }
        let mut parts = line.split_whitespace();
        let Some(peer_hex) = parts.next() else { continue; };
        let Some(key_hex) = parts.next() else { continue; };
        let Ok(peer_bytes) = hex::decode(peer_hex) else { continue; };
        let Ok(key_bytes) = hex::decode(key_hex) else { continue; };
        if peer_bytes.len() != 32 {
            continue;
        }
        let mut peer = [0u8; 32];
        peer.copy_from_slice(&peer_bytes);
        map.insert(PeerId(peer), key_bytes);
    }
    map
}

fn append_known_host(path: &Path, peer_id: PeerId, public_key: &[u8]) -> Result<(), RiftError> {
    if let Some(parent) = path.parent() {
        fs::create_dir_all(parent).map_err(|e| RiftError::Other(format!("{e}")))?;
    }
    let line = format!("{} {}\n", peer_id.to_hex(), hex::encode(public_key));
    fs::OpenOptions::new()
        .create(true)
        .append(true)
        .open(path)
        .and_then(|mut file| std::io::Write::write_all(&mut file, line.as_bytes()))
        .map_err(|e| RiftError::Other(format!("{e}")))?;
    Ok(())
}

fn fingerprint_key(public_key: &[u8]) -> String {
    let hash = blake3::hash(public_key);
    let hex = hash.to_hex().to_string();
    hex.chars().take(16).collect()
}

/// Verify and record peer identity for trust-on-first-use and mismatch handling.
async fn handle_peer_identity(
    event_tx: &mpsc::UnboundedSender<RiftEvent>,
    handle: &MeshHandle,
    cfg: &SecurityConfig,
    peer_id: PeerId,
    public_key: &[u8],
) -> Result<(), RiftError> {
    let computed = rift_core::peer_id_from_public_key_bytes(public_key)
        .map_err(|e| RiftError::Other(format!("{e}")))?;
    let fingerprint = fingerprint_key(public_key);
    let _ = event_tx.send(RiftEvent::PeerFingerprint {
        peer: peer_id,
        fingerprint: fingerprint.clone(),
    });
    let known_hosts = resolve_known_hosts_path(cfg)?;
    let mut known = load_known_hosts(&known_hosts);

    if computed != peer_id {
        let msg = format!(
            "peer id mismatch for {} (fingerprint {})",
            short_peer(&peer_id),
            fingerprint
        );
        tracing::warn!(peer = %peer_id, "peer id mismatch");
        let _ = event_tx.send(RiftEvent::SecurityNotice { message: msg.clone() });
        audit_log(cfg, "peer_id_mismatch", &peer_id, Some(&fingerprint), &msg);
        if cfg.reject_on_mismatch {
            handle.disconnect_peer(peer_id).await;
        }
        return Ok(());
    }

    if let Some(existing) = known.get(&peer_id) {
        if existing != public_key {
            let msg = format!(
                "peer key mismatch for {} (fingerprint {})",
                short_peer(&peer_id),
                fingerprint
            );
            tracing::warn!(peer = %peer_id, "peer key mismatch");
            let _ = event_tx.send(RiftEvent::SecurityNotice { message: msg.clone() });
            audit_log(cfg, "peer_key_mismatch", &peer_id, Some(&fingerprint), &msg);
            if cfg.reject_on_mismatch {
                handle.disconnect_peer(peer_id).await;
            }
        }
        return Ok(());
    }

    if cfg.trust_on_first_use {
        append_known_host(&known_hosts, peer_id, public_key)?;
        known.insert(peer_id, public_key.to_vec());
        let msg = format!(
            "new peer: {} fingerprint {} (saved to known_hosts)",
            short_peer(&peer_id),
            fingerprint
        );
        tracing::info!(peer = %peer_id, "new peer key stored");
        let _ = event_tx.send(RiftEvent::SecurityNotice { message: msg.clone() });
        audit_log(cfg, "peer_first_seen", &peer_id, Some(&fingerprint), &msg);
    } else {
        let msg = format!(
            "untrusted peer {} fingerprint {} (TOFU disabled)",
            short_peer(&peer_id),
            fingerprint
        );
        tracing::warn!(peer = %peer_id, "untrusted peer (TOFU disabled)");
        let _ = event_tx.send(RiftEvent::SecurityNotice { message: msg.clone() });
        audit_log(cfg, "peer_untrusted", &peer_id, Some(&fingerprint), &msg);
        handle.disconnect_peer(peer_id).await;
    }
    Ok(())
}

fn audit_log(cfg: &SecurityConfig, event: &str, peer_id: &PeerId, fingerprint: Option<&str>, message: &str) {
    let Some(path) = cfg.audit_log_path.as_ref() else { return; };
    let path = expand_tilde(path);
    if let Some(parent) = path.parent() {
        let _ = fs::create_dir_all(parent);
    }
    let entry = json!({
        "ts": now_timestamp(),
        "event": event,
        "peer_id": peer_id.to_hex(),
        "fingerprint": fingerprint.unwrap_or(""),
        "message": message,
    });
    if let Ok(line) = serde_json::to_string(&entry) {
        if let Ok(mut file) = fs::OpenOptions::new().create(true).append(true).open(&path) {
            let _ = std::io::Write::write_all(&mut file, line.as_bytes());
            let _ = std::io::Write::write_all(&mut file, b"\n");
        }
    }
}

fn parse_socket_addr(input: &str) -> Option<SocketAddr> {
    input.parse::<SocketAddr>().ok()
}

fn parse_socket_addrs(inputs: &[String]) -> Vec<SocketAddr> {
    let mut out = Vec::new();
    for input in inputs {
        if let Ok(addr) = input.parse::<SocketAddr>() {
            out.push(addr);
            continue;
        }
        if let Ok(mut iter) = input.to_socket_addrs() {
            if let Some(addr) = iter.next() {
                out.push(addr);
            }
        }
    }
    out
}

#[cfg(feature = "ffi")]
pub mod ffi;
#[cfg(target_os = "android")]
pub mod android_jni;

#[cfg(test)]
mod tests {
    use super::*;
    use std::io::Write;
    use tempfile::NamedTempFile;

    // ---- Config Default Tests ----

    #[test]
    fn rift_config_default_values() {
        let cfg = RiftConfig::default();
        assert_eq!(cfg.listen_port, 7777);
        assert!(!cfg.relay);
        assert!(cfg.user_name.is_none());
        assert!(cfg.metrics_enabled);
        assert_eq!(cfg.preferred_codecs, vec![CodecId::Opus, CodecId::PCM16]);
        assert!(cfg.preferred_features.contains(&FeatureFlag::Voice));
        assert!(cfg.preferred_features.contains(&FeatureFlag::E2EE));
    }

    #[test]
    fn audio_config_sdk_default_values() {
        let cfg = AudioConfigSdk::default();
        assert!(cfg.enabled);
        assert!(cfg.input_device.is_none());
        assert!(cfg.output_device.is_none());
        assert_eq!(cfg.quality, "medium");
        assert!(!cfg.ptt);
        assert!(cfg.vad);
        assert!(!cfg.mute_output);
        assert!(!cfg.emit_voice_frames);
        assert!(!cfg.allow_fail);
    }

    #[test]
    fn network_config_sdk_default_values() {
        let cfg = NetworkConfigSdk::default();
        assert!(cfg.prefer_p2p);
        assert!(cfg.local_ports.is_none());
        assert!(cfg.known_peers.is_empty());
        assert!(cfg.invite.is_none());
        assert!(!cfg.stun_servers.is_empty());
        assert_eq!(cfg.stun_timeout_ms, Some(800));
        assert!(!cfg.enable_turn);
        assert!(cfg.turn_servers.is_empty());
        assert_eq!(cfg.punch_interval_ms, Some(200));
        assert_eq!(cfg.punch_timeout_ms, Some(5000));
    }

    #[test]
    fn dht_config_sdk_default_values() {
        let cfg = DhtConfigSdk::default();
        assert!(!cfg.enabled);
        assert!(cfg.bootstrap_nodes.is_empty());
        assert!(cfg.listen_addr.is_none());
    }

    #[test]
    fn security_config_default_values() {
        let cfg = SecurityConfig::default();
        assert!(cfg.trust_on_first_use);
        assert!(cfg.known_hosts_path.is_none());
        assert!(!cfg.reject_on_mismatch);
        assert!(cfg.channel_shared_secret.is_none());
        assert!(cfg.audit_log_path.is_none());
        assert_eq!(cfg.rekey_interval_secs, Some(600));
    }

    // ---- Helper Function Tests ----

    #[test]
    fn derive_auth_token_deterministic() {
        let token1 = derive_auth_token("secret", "channel");
        let token2 = derive_auth_token("secret", "channel");
        assert_eq!(token1, token2);
        assert_eq!(token1.len(), 32);
    }

    #[test]
    fn derive_auth_token_different_inputs() {
        let token1 = derive_auth_token("secret1", "channel");
        let token2 = derive_auth_token("secret2", "channel");
        assert_ne!(token1, token2);

        let token3 = derive_auth_token("secret", "channel1");
        let token4 = derive_auth_token("secret", "channel2");
        assert_ne!(token3, token4);
    }

    #[test]
    fn derive_e2ee_key_from_shared_secret() {
        let key = derive_e2ee_key("channel", None, None, Some("shared_secret"));
        assert!(key.is_some());
        let key = key.unwrap();
        assert_eq!(key.len(), 32);

        // Deterministic
        let key2 = derive_e2ee_key("channel", None, None, Some("shared_secret"));
        assert_eq!(key, key2.unwrap());
    }

    #[test]
    fn derive_e2ee_key_from_password() {
        let key = derive_e2ee_key("channel", Some("password"), None, None);
        assert!(key.is_some());
        let key = key.unwrap();
        assert_eq!(key.len(), 32);

        // Different password = different key
        let key2 = derive_e2ee_key("channel", Some("other_password"), None, None);
        assert_ne!(key, key2.unwrap());
    }

    #[test]
    fn derive_e2ee_key_from_invite() {
        let invite = Invite {
            channel_name: "test".to_string(),
            password: None,
            channel_key: [42u8; 32],
            known_peers: Vec::new(),
            candidates: Vec::new(),
            relay_candidates: Vec::new(),
            version: 2,
            created_at: 0,
        };
        let key = derive_e2ee_key("channel", None, Some(&invite), None);
        assert_eq!(key, Some([42u8; 32]));
    }

    #[test]
    fn derive_e2ee_key_none_without_inputs() {
        let key = derive_e2ee_key("channel", None, None, None);
        assert!(key.is_none());
    }

    #[test]
    fn derive_e2ee_key_priority_shared_secret_over_password() {
        // Shared secret takes priority over password
        let key_secret = derive_e2ee_key("channel", Some("password"), None, Some("secret"));
        let key_password = derive_e2ee_key("channel", Some("password"), None, None);
        assert_ne!(key_secret, key_password);
    }

    #[test]
    fn fingerprint_key_returns_16_chars() {
        let key = [1u8; 32];
        let fp = fingerprint_key(&key);
        assert_eq!(fp.len(), 16);
        assert!(fp.chars().all(|c| c.is_ascii_hexdigit()));
    }

    #[test]
    fn fingerprint_key_different_keys() {
        let fp1 = fingerprint_key(&[1u8; 32]);
        let fp2 = fingerprint_key(&[2u8; 32]);
        assert_ne!(fp1, fp2);
    }

    #[test]
    fn short_peer_returns_8_chars() {
        let peer = PeerId([0xab; 32]);
        let short = short_peer(&peer);
        assert_eq!(short.len(), 8);
        assert_eq!(short, "abababab");
    }

    // ---- Audio Helper Tests ----

    #[test]
    fn audio_level_silent_frame() {
        let frame = vec![0i16; 480];
        let level = audio_level(&frame);
        assert_eq!(level, 0.0);
    }

    #[test]
    fn audio_level_loud_frame() {
        let frame = vec![i16::MAX; 480];
        let level = audio_level(&frame);
        assert!((level - 1.0).abs() < 0.01);
    }

    #[test]
    fn audio_level_mixed_frame() {
        let mut frame = vec![0i16; 480];
        for i in 0..240 {
            frame[i] = i16::MAX / 2;
        }
        let level = audio_level(&frame);
        assert!(level > 0.0 && level < 1.0);
    }

    #[test]
    fn is_frame_active_silent() {
        let frame = vec![0i16; 480];
        assert!(!is_frame_active(&frame));
    }

    #[test]
    fn is_frame_active_low_noise() {
        let frame = vec![100i16; 480];
        assert!(!is_frame_active(&frame));
    }

    #[test]
    fn is_frame_active_loud() {
        let frame = vec![1000i16; 480];
        assert!(is_frame_active(&frame));
    }

    // ---- Quality Mapping Tests ----

    #[test]
    fn map_quality_to_bitrate_low() {
        assert_eq!(map_quality_to_bitrate(Some("low")), 24_000);
    }

    #[test]
    fn map_quality_to_bitrate_medium() {
        assert_eq!(map_quality_to_bitrate(Some("medium")), 48_000);
    }

    #[test]
    fn map_quality_to_bitrate_high() {
        assert_eq!(map_quality_to_bitrate(Some("high")), 96_000);
    }

    #[test]
    fn map_quality_to_bitrate_default() {
        assert_eq!(map_quality_to_bitrate(None), 48_000);
        assert_eq!(map_quality_to_bitrate(Some("unknown")), 48_000);
    }

    // ---- Path Expansion Tests ----

    #[test]
    fn expand_tilde_with_tilde() {
        let path = PathBuf::from("~/test/path");
        let expanded = expand_tilde(&path);
        // Should not start with ~ anymore
        assert!(!expanded.to_string_lossy().starts_with("~/"));
    }

    #[test]
    fn expand_tilde_without_tilde() {
        let path = PathBuf::from("/absolute/path");
        let expanded = expand_tilde(&path);
        assert_eq!(expanded, path);
    }

    #[test]
    fn expand_tilde_relative_path() {
        let path = PathBuf::from("relative/path");
        let expanded = expand_tilde(&path);
        assert_eq!(expanded, path);
    }

    // ---- Socket Addr Parsing Tests ----

    #[test]
    fn parse_socket_addr_valid() {
        let addr = parse_socket_addr("127.0.0.1:8080");
        assert!(addr.is_some());
        assert_eq!(addr.unwrap().port(), 8080);
    }

    #[test]
    fn parse_socket_addr_invalid() {
        let addr = parse_socket_addr("invalid");
        assert!(addr.is_none());
    }

    #[test]
    fn parse_socket_addrs_mixed() {
        let inputs = vec![
            "127.0.0.1:8080".to_string(),
            "invalid".to_string(),
            "192.168.1.1:9000".to_string(),
        ];
        let addrs = parse_socket_addrs(&inputs);
        assert_eq!(addrs.len(), 2);
    }

    #[test]
    fn parse_socket_addrs_empty() {
        let addrs = parse_socket_addrs(&[]);
        assert!(addrs.is_empty());
    }

    // ---- Known Hosts Tests ----

    #[test]
    fn load_known_hosts_empty_file() {
        let tmp = NamedTempFile::new().unwrap();
        let hosts = load_known_hosts(tmp.path());
        assert!(hosts.is_empty());
    }

    #[test]
    fn load_known_hosts_with_entries() {
        let mut tmp = NamedTempFile::new().unwrap();
        let peer_hex = hex::encode([0xab; 32]);
        let key_hex = hex::encode([0xcd; 32]);
        writeln!(tmp, "{} {}", peer_hex, key_hex).unwrap();
        tmp.flush().unwrap();

        let hosts = load_known_hosts(tmp.path());
        assert_eq!(hosts.len(), 1);
        let peer = PeerId([0xab; 32]);
        assert!(hosts.contains_key(&peer));
        assert_eq!(hosts[&peer], vec![0xcd; 32]);
    }

    #[test]
    fn load_known_hosts_with_comments() {
        let mut tmp = NamedTempFile::new().unwrap();
        writeln!(tmp, "# This is a comment").unwrap();
        writeln!(tmp, "").unwrap();
        let peer_hex = hex::encode([0xab; 32]);
        let key_hex = hex::encode([0xcd; 32]);
        writeln!(tmp, "{} {}", peer_hex, key_hex).unwrap();
        tmp.flush().unwrap();

        let hosts = load_known_hosts(tmp.path());
        assert_eq!(hosts.len(), 1);
    }

    #[test]
    fn load_known_hosts_nonexistent() {
        let hosts = load_known_hosts(Path::new("/nonexistent/path"));
        assert!(hosts.is_empty());
    }

    // ---- Peer to Stream ID Tests ----

    #[test]
    fn peer_to_stream_id_deterministic() {
        let peer = PeerId([0x12; 32]);
        let id1 = peer_to_stream_id(&peer);
        let id2 = peer_to_stream_id(&peer);
        assert_eq!(id1, id2);
    }

    #[test]
    fn peer_to_stream_id_different_peers() {
        let peer1 = PeerId([0x12; 32]);
        let peer2 = PeerId([0x34; 32]);
        let id1 = peer_to_stream_id(&peer1);
        let id2 = peer_to_stream_id(&peer2);
        assert_ne!(id1, id2);
    }

    // ---- Timestamp Tests ----

    #[test]
    fn now_timestamp_nonzero() {
        let ts = now_timestamp();
        // Should be a reasonable timestamp (after 2020)
        assert!(ts > 1577836800000);
    }

    // ---- Error Display Tests ----

    #[test]
    fn rift_error_display() {
        assert_eq!(format!("{}", RiftError::NotInitialized), "not initialized");
        assert_eq!(format!("{}", RiftError::AlreadyJoined), "channel already joined");
        assert_eq!(format!("{}", RiftError::NotJoined), "channel not joined");
        assert_eq!(format!("{}", RiftError::Mesh("test".to_string())), "mesh error: test");
        assert_eq!(format!("{}", RiftError::Audio("test".to_string())), "audio error: test");
        assert_eq!(format!("{}", RiftError::Other("test".to_string())), "other: test");
    }

    // ---- LinkStats and GlobalStats Tests ----

    #[test]
    fn link_stats_construction() {
        let stats = LinkStats {
            rtt_ms: 50.0,
            loss: 0.01,
            jitter_ms: 5.0,
        };
        assert_eq!(stats.rtt_ms, 50.0);
        assert_eq!(stats.loss, 0.01);
        assert_eq!(stats.jitter_ms, 5.0);
    }

    #[test]
    fn global_stats_construction() {
        let stats = GlobalStats {
            num_peers: 5,
            num_sessions: 2,
            packets_sent: 1000,
            packets_received: 950,
            bytes_sent: 100_000,
            bytes_received: 95_000,
        };
        assert_eq!(stats.num_peers, 5);
        assert_eq!(stats.packets_sent, 1000);
    }

    // ---- Route Kind Tests ----

    #[test]
    fn route_kind_direct() {
        let route = RouteKind::Direct;
        assert!(matches!(route, RouteKind::Direct));
    }

    #[test]
    fn route_kind_relayed() {
        let via = PeerId([0xab; 32]);
        let route = RouteKind::Relayed { via };
        if let RouteKind::Relayed { via: v } = route {
            assert_eq!(v.0, [0xab; 32]);
        } else {
            panic!("expected relayed route");
        }
    }

    // ---- SDK Version Constants ----

    #[test]
    fn sdk_version_defined() {
        assert_eq!(SDK_VERSION, "0.1.0");
        assert_eq!(SDK_ABI_VERSION, 1);
    }

    // ---- NAT Config Tests ----

    #[test]
    fn default_nat_config_basic() {
        let cfg = default_nat_config(
            7777,
            None,
            vec!["stun.example.com:3478".to_string()],
            Some(1000),
            Some(100),
            Some(3000),
            false,
            Vec::new(),
            Some(2000),
            Some(15000),
        );
        assert_eq!(cfg.local_ports, vec![7777, 7778, 7779]);
        assert_eq!(cfg.stun_timeout_ms, 1000);
        assert_eq!(cfg.punch_interval_ms, 100);
        assert_eq!(cfg.punch_timeout_ms, 3000);
        assert!(cfg.turn_servers.is_empty());
    }

    #[test]
    fn default_nat_config_custom_ports() {
        let cfg = default_nat_config(
            7777,
            Some(vec![8000, 8001]),
            Vec::new(),
            None,
            None,
            None,
            false,
            Vec::new(),
            None,
            None,
        );
        assert_eq!(cfg.local_ports, vec![8000, 8001]);
    }

    // ---- Serialization Tests ----

    #[test]
    fn rift_config_serialization() {
        let cfg = RiftConfig::default();
        let json = serde_json::to_string(&cfg).unwrap();
        let parsed: RiftConfig = serde_json::from_str(&json).unwrap();
        assert_eq!(parsed.listen_port, cfg.listen_port);
        assert_eq!(parsed.relay, cfg.relay);
    }

    #[test]
    fn security_config_serialization() {
        let cfg = SecurityConfig {
            trust_on_first_use: false,
            known_hosts_path: Some(PathBuf::from("/test/path")),
            reject_on_mismatch: true,
            channel_shared_secret: Some("secret".to_string()),
            audit_log_path: None,
            rekey_interval_secs: Some(300),
        };
        let json = serde_json::to_string(&cfg).unwrap();
        let parsed: SecurityConfig = serde_json::from_str(&json).unwrap();
        assert!(!parsed.trust_on_first_use);
        assert!(parsed.reject_on_mismatch);
        assert_eq!(parsed.rekey_interval_secs, Some(300));
    }

    #[test]
    fn srt_invite_serialization() {
        let invite = SrtInvite {
            label: "Test Call".to_string(),
            uri: "srt://test".to_string(),
        };
        let json = serde_json::to_string(&invite).unwrap();
        let parsed: SrtInvite = serde_json::from_str(&json).unwrap();
        assert_eq!(parsed.label, "Test Call");
        assert_eq!(parsed.uri, "srt://test");
    }

    // ---- QoS Tuning Tests ----

    #[test]
    fn compute_next_tuning_empty_stats() {
        let mut qos = QosState {
            profile: QosProfile::default(),
            peer_stats: HashMap::new(),
            current: AudioTuning {
                bitrate: 48_000,
                fec: false,
                loss_pct: 0,
            },
            last_adjust: Instant::now() - Duration::from_secs(10),
        };
        let result = compute_next_tuning(&mut qos);
        assert!(result.is_none());
    }

    #[test]
    fn compute_next_tuning_high_loss() {
        let mut qos = QosState {
            profile: QosProfile {
                packet_loss_tolerance: 0.05,
                max_latency_ms: 200,
                target_latency_ms: 100,
                min_bitrate: 16_000,
                max_bitrate: 128_000,
                ..Default::default()
            },
            peer_stats: HashMap::new(),
            current: AudioTuning {
                bitrate: 64_000,
                fec: false,
                loss_pct: 0,
            },
            last_adjust: Instant::now() - Duration::from_secs(10),
        };
        qos.peer_stats.insert(
            PeerId([0; 32]),
            LinkStats {
                rtt_ms: 50.0,
                loss: 0.10,  // 10% loss, above tolerance
                jitter_ms: 5.0,
            },
        );
        let result = compute_next_tuning(&mut qos);
        assert!(result.is_some());
        let tuning = result.unwrap();
        assert!(tuning.bitrate < 64_000); // Should decrease bitrate
        assert!(tuning.fec); // Should enable FEC
    }

    #[test]
    fn compute_next_tuning_good_conditions() {
        let mut qos = QosState {
            profile: QosProfile {
                packet_loss_tolerance: 0.05,
                max_latency_ms: 200,
                target_latency_ms: 100,
                min_bitrate: 16_000,
                max_bitrate: 128_000,
                ..Default::default()
            },
            peer_stats: HashMap::new(),
            current: AudioTuning {
                bitrate: 48_000,
                fec: true,
                loss_pct: 5,
            },
            last_adjust: Instant::now() - Duration::from_secs(10),
        };
        qos.peer_stats.insert(
            PeerId([0; 32]),
            LinkStats {
                rtt_ms: 20.0,
                loss: 0.01,  // 1% loss, below tolerance
                jitter_ms: 2.0,
            },
        );
        let result = compute_next_tuning(&mut qos);
        assert!(result.is_some());
        let tuning = result.unwrap();
        assert!(tuning.bitrate >= 48_000); // Should increase or maintain bitrate
    }

    #[test]
    fn compute_next_tuning_respects_cooldown() {
        let mut qos = QosState {
            profile: QosProfile::default(),
            peer_stats: HashMap::new(),
            current: AudioTuning {
                bitrate: 48_000,
                fec: false,
                loss_pct: 0,
            },
            last_adjust: Instant::now(), // Just adjusted
        };
        qos.peer_stats.insert(
            PeerId([0; 32]),
            LinkStats {
                rtt_ms: 500.0,
                loss: 0.50,
                jitter_ms: 50.0,
            },
        );
        let result = compute_next_tuning(&mut qos);
        assert!(result.is_none()); // Should not adjust due to cooldown
    }
}