From 055e6c3600f59de52ef058be876203a0f51ba306 Mon Sep 17 00:00:00 2001
From: griffi-gh <prasol258@gmail.com>
Date: Fri, 10 Feb 2023 20:36:58 +0100
Subject: [PATCH] ecs server

---
 kubi-server/Cargo.toml  |  1 +
 kubi-server/src/main.rs | 45 +++++++++++++++++++++++++++++++++++------
 kubi-udp/src/server.rs  |  2 +-
 3 files changed, 41 insertions(+), 7 deletions(-)

diff --git a/kubi-server/Cargo.toml b/kubi-server/Cargo.toml
index 5dc3fc5..87c64dd 100644
--- a/kubi-server/Cargo.toml
+++ b/kubi-server/Cargo.toml
@@ -6,3 +6,4 @@ edition = "2021"
 [dependencies]
 kubi-shared = { path = "../kubi-shared" }
 kubi-udp = { path = "../kubi-udp" }
+shipyard = "0.6"
diff --git a/kubi-server/src/main.rs b/kubi-server/src/main.rs
index 9460db3..b82c528 100644
--- a/kubi-server/src/main.rs
+++ b/kubi-server/src/main.rs
@@ -1,14 +1,47 @@
+use shipyard::{World, AllStoragesView, Unique, Workload, IntoWorkload, UniqueView, UniqueViewMut};
 use kubi_udp::server::{Server, ServerConfig};
 use kubi_shared::networking::messages::{ClientToServerMessage, ServerToClientMessage};
 
-fn main() {
-  let mut server: Server<ServerToClientMessage, ClientToServerMessage> = Server::bind(
+#[derive(Unique)]
+#[repr(transparent)]
+pub struct UdpServer(Server<ServerToClientMessage, ClientToServerMessage>);
+
+fn bind_server(
+  storages: AllStoragesView,
+) {
+  let server: Server<ServerToClientMessage, ClientToServerMessage> = Server::bind(
     "0.0.0.0:1234".parse().unwrap(), 
     ServerConfig::default()
   ).unwrap();
-  loop {
-    if let Err(or) = server.update() {
-      println!("Server error: {or:?}")
-    }
+  storages.add_unique(UdpServer(server));
+}
+
+fn update_server(
+  mut server: UniqueViewMut<UdpServer>
+) {
+  if let Err(error) = server.0.update() {
+    println!("Server error: {error:?}")
+  }
+}
+
+fn initialize() -> Workload {
+  (
+    bind_server,
+  ).into_workload()
+}
+
+fn update() -> Workload {
+  (
+    update_server,
+  ).into_workload()
+}
+
+fn main() {
+  let world = World::new();
+  world.add_workload(initialize);
+  world.add_workload(update);
+  world.run_workload(initialize).unwrap();
+  loop {
+    world.run_workload(update).unwrap();
   }
 }
diff --git a/kubi-udp/src/server.rs b/kubi-udp/src/server.rs
index 1a0b537..905aefc 100644
--- a/kubi-udp/src/server.rs
+++ b/kubi-udp/src/server.rs
@@ -48,7 +48,7 @@ pub struct Server<S, R> where S: Encode + Decode, R: Encode + Decode {
   clients: HashMap<ClientId, ConnectedClient, BuildNoHashHasher<u8>>,
   config: ServerConfig,
   event_queue: VecDeque<ServerEvent<R>>,
-  _s: PhantomData<*const S>,
+  _s: PhantomData<S>,
 }
 impl<S, R> Server<S, R> where S: Encode + Decode, R: Encode + Decode {
   fn send_to_addr(&self, addr: SocketAddr, packet: IdServerPacket<S>) -> Result<()> {