[refactor] move listen function in server module

src/main.rs

@@ -1,53 +1,10 @@
-use chrono;
-use colored::Colorize;
-use std::net::UdpSocket;
-
 use crate::config::Config;
 
 mod config;
 mod vec_to_string;
+mod server;
 
 fn main() {
     let config = Config::from_json("config.json".to_string());
-    listen(&config);
-}
-
-fn listen(config: &Config) {
-    let serv_config = config;
-    let addr = config.get_server_addr();
-
-    println!(
-        "{}: {}",
-        "listening on".green(),
-        serv_config.get_server_addr()
-    );
-
-    let socket: UdpSocket =
-        UdpSocket::bind(serv_config.get_server_addr()).expect("Error while binding the socket.");
-
-    const BUF_SIZE: usize = 1024;
-
-    loop {
-        let mut message: String = String::new();
-        let mut buf: [u8; BUF_SIZE] = [0; BUF_SIZE];
-        let (num_bytes, src_addr) = socket
-            .recv_from(&mut buf)
-            .expect("Error while receiving message");
-
-        if !serv_config.addr_is_blacklisted(&addr) {
-            message.push_str(std::str::from_utf8(&buf[..num_bytes]).expect("Invalid UTF-8 data"));
-
-            let forwarding_addr_opt = serv_config.is_forwarded(src_addr.to_string());
-
-            if forwarding_addr_opt.is_some() {
-                let forwarding_addr = forwarding_addr_opt.unwrap();
-
-                socket
-                    .send_to(message.as_bytes(), forwarding_addr)
-                    .expect("Error forwarding message to another ip.");
-            }
-
-            println!("{}-{}: {}", src_addr.to_string().green(), chrono::Local::now(), message);
-        }
-    }
+    server::listen(&config);
 }

src/server.rs

@@ -0,0 +1,44 @@
+use std::net::UdpSocket;
+use chrono;
+use colored::Colorize;
+use crate::config::Config;
+
+pub fn listen(config: &Config) {
+    let serv_config = config;
+    let addr = config.get_server_addr();
+
+    println!(
+        "{}: {}",
+        "listening on".green(),
+        serv_config.get_server_addr()
+    );
+
+    let socket: UdpSocket =
+        UdpSocket::bind(serv_config.get_server_addr()).expect("Error while binding the socket.");
+
+    const BUF_SIZE: usize = 1024;
+
+    loop {
+        let mut message: String = String::new();
+        let mut buf: [u8; BUF_SIZE] = [0; BUF_SIZE];
+        let (num_bytes, src_addr) = socket
+            .recv_from(&mut buf)
+            .expect("Error while receiving message");
+
+        if !serv_config.addr_is_blacklisted(&addr) {
+            message.push_str(std::str::from_utf8(&buf[..num_bytes]).expect("Invalid UTF-8 data"));
+
+            let forwarding_addr_opt = serv_config.is_forwarded(src_addr.to_string());
+
+            if forwarding_addr_opt.is_some() {
+                let forwarding_addr = forwarding_addr_opt.unwrap();
+
+                socket
+                    .send_to(message.as_bytes(), forwarding_addr)
+                    .expect("Error forwarding message to another ip.");
+            }
+
+            println!("{}-{}: {}", src_addr.to_string().green(), chrono::Local::now(), message);
+        }
+    }
+}