fix(server): hoster not identified as server

atoair · Dec 27, 2022 · d464374 · d464374
1 parent 1e4985c
commit d464374
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 8 deletions.
diff --git a/src/yourcontrols-server/src/hoster.rs b/src/yourcontrols-server/src/hoster.rs
@@ -92,7 +92,7 @@ fn process_payload(
                     Payloads::PlayerJoined {
                         name: name.clone(),
                         in_control: state.in_control == *name,
-                        is_server: false,
+                        is_server: info.is_host,
                         is_observer: info.is_observer,
                     },
                     addr,
@@ -101,13 +101,7 @@ fn process_payload(
             }
 
             // Add client
-            state.clients.insert(
-                name.clone(),
-                Client {
-                    addr,
-                    is_observer: false,
-                },
-            );
+            state.clients.insert(name.clone(), Client::new(addr));
 
             // If the client is the first one to connect, give them control and have them "host"
             if state.in_control == SERVER_NAME {
@@ -174,6 +168,7 @@ fn process_payload(
 fn set_host(name: String, state: &mut ServerState, net: &mut SenderReceiver) {
     let client = state.clients.get_mut(&name).expect("always there");
     client.is_observer = false;
+    client.is_host = true;
 
     net.send_message(Payloads::SetHost, client.addr).ok();
     send_to_all(

diff --git a/src/yourcontrols-server/src/servers.rs b/src/yourcontrols-server/src/servers.rs
@@ -9,6 +9,17 @@ use crate::util::{get_random_id, SESSION_ID_LENGTH};
 pub struct Client {
     pub addr: SocketAddr,
     pub is_observer: bool,
+    pub is_host: bool,
+}
+
+impl Client {
+    pub fn new(addr: SocketAddr) -> Self {
+        Self {
+            addr,
+            is_observer: false,
+            is_host: false,
+        }
+    }
 }
 
 pub struct ServerState {