From 1a5a628000530638e5e1cbd8fa73bc91f6e6f643 Mon Sep 17 00:00:00 2001
From: Michael Bradley <michaelmbrad@gmail.com>
Date: Wed, 9 Jul 2025 19:37:46 -0400
Subject: [PATCH] Partial: Dedicated Peer distribution system

This is an uncompleted commit to move the work over to my other machine. Should compile though.
---
 src/game/net.rs         |   4 +-
 src/game/ui.rs          |   4 +-
 src/net/distribution.rs |   2 +-
 src/net/io.rs           |  45 ++++++------
 src/net/peer.rs         | 148 +++++++++++++++++++++++++++++++++++-----
 src/net/plugin.rs       |  11 +--
 6 files changed, 161 insertions(+), 53 deletions(-)

diff --git a/src/game/net.rs b/src/game/net.rs
index 2ffa40b..8e43e3f 100644
--- a/src/game/net.rs
+++ b/src/game/net.rs
@@ -4,14 +4,14 @@ use crate::net::prelude::*;
 
 pub fn handle_new_peer(new_peers: Query<&Peer, Added<Peer>>) {
     for peer in new_peers {
-        info!("Peer {} was added", peer.uuid);
+        info!("Peer {} was added", peer.id);
     }
 }
 
 pub fn handle_deleted_peer(mut old_peers: RemovedComponents<Peer>, peers: Query<&Peer>) -> Result {
     for entity in old_peers.read() {
         if let Ok(peer) = peers.get(entity) {
-            info!("Peer {} was removed", peer.uuid);
+            info!("Peer {} was removed", peer.id);
         } else {
             info!("Peer {} was removed", entity);
         }
diff --git a/src/game/ui.rs b/src/game/ui.rs
index 3ee6c50..391a39f 100644
--- a/src/game/ui.rs
+++ b/src/game/ui.rs
@@ -90,8 +90,8 @@ pub fn update_peer_ui_timings(
         if let Ok((peer, recv, send)) = peers.get(id.0) {
             **row = format!(
                 "{} {:.2} {:.2}",
-                peer.uuid,
-                (time.elapsed() - recv.time().unwrap_or(default())).as_secs_f64(),
+                peer.id,
+                (time.elapsed() - recv.time()).as_secs_f64(),
                 (time.elapsed() - send.time()).as_secs_f64()
             )
         }
diff --git a/src/net/distribution.rs b/src/net/distribution.rs
index e5e512d..161ea1a 100644
--- a/src/net/distribution.rs
+++ b/src/net/distribution.rs
@@ -26,7 +26,7 @@ fn sender<T: Into<Vec<u8>> + Component + Clone>(
         for entity in entities {
             outbound.write(OutboundPacket(Packet::new(
                 (*entity).clone().into(),
-                peer.uuid,
+                peer.id,
             )));
         }
     }
diff --git a/src/net/io.rs b/src/net/io.rs
index 7dcceca..25774b6 100644
--- a/src/net/io.rs
+++ b/src/net/io.rs
@@ -3,16 +3,18 @@ use std::time::Duration;
 use bevy::prelude::*;
 use uuid::Uuid;
 
+use crate::net::peer::PotentialPeer;
+
 use super::{
     packet::{InboundPacket, OutboundPacket, Packet},
-    peer::{Peer, PeerChangeEvent, PeerMap, PeerReceiveTiming, PeerSendTiming},
+    peer::{Peer, PeerChangeEvent, PeerData, PeerMap, PeerReceiveTiming, PeerSendTiming},
     queues::{NetworkReceive, NetworkSend},
 };
 
 pub fn handle_network_input(
     from_socket: Res<NetworkReceive>,
     peer_map: Res<PeerMap>,
-    mut peers: Query<(&Peer, &mut PeerReceiveTiming)>,
+    mut peers: Query<(&PeerData, &mut PeerReceiveTiming)>,
     mut to_app: EventWriter<InboundPacket>,
     time: Res<Time>,
     mut change_peer: EventWriter<PeerChangeEvent>,
@@ -44,8 +46,7 @@ pub fn handle_network_input(
 pub fn handle_network_output(
     mut from_app: EventReader<OutboundPacket>,
     peer_map: Res<PeerMap>,
-    mut peers: Query<(&Peer, &mut PeerSendTiming)>,
-    config: Res<Config>,
+    mut peers: Query<(&PeerData, &mut PeerSendTiming)>,
     to_socket: Res<NetworkSend>,
     time: Res<Time>,
 ) -> Result {
@@ -53,8 +54,8 @@ pub fn handle_network_output(
         let peer_id = peer_map.try_get(&packet.0.peer)?;
         let (peer, mut last) = peers.get_mut(*peer_id)?;
         // Append our UUID for client identification
-        let message = [packet.0.message.as_slice(), config.id.as_bytes()].concat();
-        to_socket.send(message, peer.addr)?;
+        let message = [packet.0.message.as_slice(), peer.me.as_bytes()].concat();
+        to_socket.send(message, peer.addr.into())?;
         last.update(&time);
     }
     Ok(())
@@ -63,17 +64,24 @@ pub fn handle_network_output(
 const TIMEOUT: Duration = Duration::from_secs(10);
 
 pub fn heartbeat(
-    peers: Query<(&Peer, &PeerSendTiming)>,
+    peers: Query<(AnyOf<(&Peer, &PotentialPeer)>, &PeerSendTiming)>,
     time: Res<Time>,
     mut outbound: EventWriter<OutboundPacket>,
-) {
+) -> Result {
     for (peer, last) in peers {
         // Allow for 2 consecutive missed heartbeats without timing out
         if last.time() + TIMEOUT / 3 > time.elapsed() {
             continue;
         }
-        outbound.write(OutboundPacket(Packet::new(Vec::new(), peer.uuid)));
+        let id = match peer {
+            (None, None) => return Err("No peer identification".into()),
+            (None, Some(potential)) => potential.id,
+            (Some(actual), None) => actual.id,
+            (Some(_), Some(_)) => return Err("Both a potential and actual peer".into()),
+        };
+        outbound.write(OutboundPacket(Packet::new(Vec::new(), id)));
     }
+    Ok(())
 }
 
 pub fn timeout(
@@ -82,22 +90,9 @@ pub fn timeout(
     mut delete: EventWriter<PeerChangeEvent>,
 ) {
     for (peer, last) in peers {
-        if let Some(previous) = last.time() {
-            if previous + TIMEOUT < time.elapsed() {
-                warn!("Peer {} timed out", peer.uuid);
-                delete.write(PeerChangeEvent::new(peer.uuid, None));
-            }
+        if last.time() + TIMEOUT < time.elapsed() {
+            warn!("Peer {} timed out", peer.id);
+            delete.write(PeerChangeEvent::new(peer.id, None));
         }
     }
 }
-
-#[derive(Debug, Resource)]
-pub struct Config {
-    pub id: Uuid,
-}
-
-impl Config {
-    pub fn new() -> Self {
-        Self { id: Uuid::new_v4() }
-    }
-}
diff --git a/src/net/peer.rs b/src/net/peer.rs
index 097d8b7..8051f40 100644
--- a/src/net/peer.rs
+++ b/src/net/peer.rs
@@ -1,4 +1,9 @@
-use std::{collections::HashMap, net::SocketAddr, time::Duration};
+use std::{
+    array::TryFromSliceError,
+    collections::HashMap,
+    net::{IpAddr, Ipv6Addr, SocketAddr},
+    time::Duration,
+};
 
 use bevy::prelude::*;
 use uuid::Uuid;
@@ -19,32 +24,122 @@ impl PeerSendTiming {
 }
 
 #[derive(Component, Debug, Default)]
-pub struct PeerReceiveTiming(Option<Duration>);
+pub struct PeerReceiveTiming(Duration);
 
 impl PeerReceiveTiming {
     pub fn new(time: &Res<Time>) -> Self {
-        Self(Some(time.elapsed()))
+        Self(time.elapsed())
     }
 
     pub fn update(&mut self, time: &Res<Time>) {
-        self.0 = Some(time.elapsed())
+        self.0 = time.elapsed()
     }
 
-    pub fn time(&self) -> Option<Duration> {
+    pub fn time(&self) -> Duration {
         self.0
     }
 }
 
 #[derive(Component, Debug)]
-#[require(PeerSendTiming, PeerReceiveTiming)]
+#[require(PeerData, PeerReceiveTiming)]
 pub struct Peer {
-    pub addr: SocketAddr,
-    pub uuid: Uuid,
+    pub id: Uuid,
 }
 
-impl Peer {
-    pub fn new(addr: SocketAddr, uuid: Uuid) -> Self {
-        Self { addr, uuid }
+#[derive(Component, Debug)]
+#[require(PeerData)]
+pub struct PotentialPeer {
+    pub id: Uuid,
+}
+
+impl Default for PotentialPeer {
+    fn default() -> Self {
+        Self { id: Uuid::new_v4() }
+    }
+}
+
+#[derive(Clone, Component, Copy, Debug)]
+#[require(PeerSendTiming)]
+pub struct PeerData {
+    pub addr: Address,
+    pub me: Uuid,
+}
+
+impl PeerData {
+    pub fn new(addr: SocketAddr, me: Uuid) -> Self {
+        Self {
+            addr: addr.into(),
+            me,
+        }
+    }
+}
+
+impl Default for PeerData {
+    fn default() -> Self {
+        Self {
+            addr: Default::default(),
+            me: Uuid::new_v4(),
+        }
+    }
+}
+
+#[derive(Clone, Copy, Debug)]
+pub struct Address(SocketAddr);
+
+impl From<Address> for SocketAddr {
+    fn from(value: Address) -> Self {
+        value.0
+    }
+}
+
+impl From<SocketAddr> for Address {
+    fn from(value: SocketAddr) -> Self {
+        Self(value)
+    }
+}
+
+impl PartialEq<SocketAddr> for Address {
+    fn eq(&self, other: &SocketAddr) -> bool {
+        self.0 == *other
+    }
+}
+
+impl PartialEq<Address> for SocketAddr {
+    fn eq(&self, other: &Address) -> bool {
+        other == self
+    }
+}
+
+impl From<Address> for Vec<u8> {
+    fn from(value: Address) -> Self {
+        let mut bytes: Vec<u8> = match value.0.ip() {
+            IpAddr::V4(ipv4_addr) => ipv4_addr.octets().into(),
+            IpAddr::V6(ipv6_addr) => ipv6_addr.octets().into(),
+        };
+        bytes.extend(value.0.port().to_le_bytes());
+        bytes
+    }
+}
+
+impl TryFrom<Vec<u8>> for Address {
+    type Error = TryFromSliceError;
+
+    fn try_from(value: Vec<u8>) -> Result<Self, Self::Error> {
+        let port = u16::from_le_bytes(TryInto::<[u8; 2]>::try_into(
+            value.clone().split_off(value.len() - 2).as_slice(),
+        )?);
+        let addr = if let Ok(bytes) = TryInto::<[u8; 4]>::try_into(value.as_slice()) {
+            SocketAddr::from((bytes, port))
+        } else {
+            SocketAddr::from((TryInto::<[u8; 16]>::try_into(value.as_slice())?, port))
+        };
+        Ok(Address(addr))
+    }
+}
+
+impl Default for Address {
+    fn default() -> Self {
+        Self(SocketAddr::new(IpAddr::V6(Ipv6Addr::UNSPECIFIED), 0))
     }
 }
 
@@ -90,7 +185,7 @@ impl PeerChangeEvent {
 pub fn handle_peer_change(
     mut changes: EventReader<PeerChangeEvent>,
     mut peer_map: ResMut<PeerMap>,
-    mut peers: Query<&mut Peer>,
+    mut peers: Query<&mut PeerData>,
     mut commands: Commands,
     time: Res<Time>,
 ) -> Result {
@@ -98,7 +193,7 @@ pub fn handle_peer_change(
         if let Some(entity) = peer_map.get(&change.peer) {
             if let Some(addr) = change.addr {
                 if let Ok(mut peer) = peers.get_mut(*entity) {
-                    peer.addr = addr;
+                    peer.addr = Address(addr);
                 } else {
                     warn!("Peer {} doesn't exist (just added?)", change.peer);
                 }
@@ -111,7 +206,10 @@ pub fn handle_peer_change(
             peer_map.insert(
                 change.peer,
                 commands
-                    .spawn((Peer::new(addr, change.peer), PeerReceiveTiming::new(&time)))
+                    .spawn((
+                        PeerData::new(addr, Uuid::new_v4()),
+                        PeerReceiveTiming::new(&time),
+                    ))
                     .id(),
             );
         } else {
@@ -126,11 +224,25 @@ pub fn handle_peer_change(
     Ok(())
 }
 
+#[allow(clippy::type_complexity)]
 pub fn handle_new_peer(
-    new_peers: Query<&Peer, Added<Peer>>,
+    new_peers: Query<(Option<Ref<Peer>>, Option<&Peer>, &PeerData)>,
     mut outbound: EventWriter<OutboundPacket>,
-) {
-    for peer in new_peers {
-        outbound.write(OutboundPacket(Packet::new(Vec::new(), peer.uuid)));
+) -> Result {
+    for (change, this, data) in new_peers {
+        if let Some(change) = change {
+            if change.is_added() {
+                if let Some(this) = this {
+                    for (_, _, other) in new_peers {
+                        if data.me != other.me {
+                            outbound.write(OutboundPacket(Packet::new(other.addr.into(), this.id)));
+                        }
+                    }
+                } else {
+                    return Err("Ref<Peer> without Peer".into());
+                }
+            }
+        }
     }
+    Ok(())
 }
diff --git a/src/net/plugin.rs b/src/net/plugin.rs
index 9cf00b1..75b32c8 100644
--- a/src/net/plugin.rs
+++ b/src/net/plugin.rs
@@ -4,10 +4,9 @@ use bevy::prelude::*;
 use uuid::Uuid;
 
 use super::{
-    distribution::Networked,
-    io::{Config, handle_network_input, handle_network_output, heartbeat, timeout},
+    io::{handle_network_input, handle_network_output, heartbeat, timeout},
     packet::{InboundPacket, OutboundPacket},
-    peer::{Peer, PeerChangeEvent, PeerMap, handle_new_peer, handle_peer_change},
+    peer::{PeerChangeEvent, PeerData, PeerMap, handle_new_peer, handle_peer_change},
     queues::{NetworkReceive, NetworkSend},
     socket::bind_socket,
     state::NetworkState,
@@ -41,7 +40,6 @@ impl Plugin for NetIOPlugin {
                 FixedPostUpdate,
                 handle_network_output.run_if(in_state(NetworkState::MultiPlayer)),
             )
-            .insert_resource(Config::new())
             .add_event::<PeerChangeEvent>()
             .add_event::<InboundPacket>()
             .add_event::<OutboundPacket>();
@@ -54,7 +52,10 @@ impl Plugin for NetIOPlugin {
 
                 let mut peer_map = PeerMap::default();
                 if let Some(socket) = self.peer {
-                    let entity = app.world_mut().spawn(Peer::new(socket, Uuid::nil()));
+                    let entity = app.world_mut().spawn(PeerData {
+                        addr: socket.into(),
+                        me: Uuid::nil(),
+                    });
                     peer_map.insert(Uuid::nil(), entity.id());
                 }