feat: added get_servers(), fix: return type of handle_command

Author: Bordi00

src/network.rs

@@ -148,6 +148,8 @@ impl Network {
         }
     }
 
+
+    /// BFS to find path to destination
     #[must_use]
     pub(crate) fn find_path(&self, destination: NodeId) -> Option<Vec<NodeId>> {
         let start = self.nodes[0].id;
@@ -182,6 +184,24 @@ impl Network {
         }
         None
     }
+
+    #[must_use]
+    pub(crate) fn get_servers(&self) -> Option<Vec<NodeId>> {
+        let servers = self.nodes.iter().filter_map(|n| {
+            if n.get_node_type() == NodeType::Server {
+                Some(n.get_id())
+            }
+            else {
+                None
+            }
+        }).collect::<Vec<_>>();
+
+        if servers.is_empty() {
+            None
+        }else {
+            Some(servers)
+        }
+    }
 }
 
 #[cfg(test)]
@@ -267,4 +287,4 @@ mod tests {
 
         assert!(result.is_none());
     }
-}
+}

src/packet_processor.rs

@@ -11,7 +11,7 @@ pub trait Processor {
     fn routing_header(&mut self) -> &mut RoutingHandler;
 
     fn handle_msg(&mut self, msg: Vec<u8>, from: NodeId, session_id: u64);
-    fn handle_command(&mut self, cmd: Box<dyn Any>) -> Result<(), ()>;
+    fn handle_command(&mut self, cmd: Box<dyn Any>) -> bool;
 
     /// Handles a packet in a standard way
     /// # Errors
@@ -53,7 +53,7 @@ pub trait Processor {
             select_biased! {
                 recv(self.controller_recv()) -> cmd => {
                     if let Ok(cmd) = cmd {
-                        if self.handle_command(cmd).is_err() {
+                        if self.handle_command(cmd) {
                             return
                         }
                     }

src/routing_handler.rs

@@ -416,6 +416,11 @@ impl RoutingHandler {
         self.try_send(packet)?;
         Ok(())
     }
+
+    #[must_use]
+    pub fn get_servers(&self) -> Option<Vec<NodeId>> {
+        self.network_view.get_servers()
+    }
 }
 
 #[cfg(test)]

src/types.rs

@@ -50,6 +50,7 @@ pub struct Message {
 }
 
 impl Message {
+    #[must_use]
     pub fn new(from: NodeId, to: NodeId, text: String) -> Self {
         Message { from, to, text }
     }