TLS transport and games framework on client and server

1 files changed, 12 insertions, 2 deletions

diff --git a/src/bin/client/main.rs b/src/bin/client/main.rs
index 2ff6f55..c95880d 100644
--- a/src/bin/client/main.rs
+++ b/src/bin/client/main.rs
@@ -1,12 +1,22 @@
 #[macro_use]
 mod macros;
-mod repl;
-mod database;
 mod connection;
+mod database;
+mod games;
+mod repl;
 
 use repl::ClientState;
 
 fn main() {
+    let args: Vec<String> = std::env::args().collect();
+    let unsafe_mode = args.iter().any(|arg| arg == "--unsafe");
+
+    if unsafe_mode {
+        println!("WARNING: Running in UNSAFE mode - certificate verification disabled!");
+    }
+
+    connection::set_unsafe_mode(unsafe_mode);
+
     let mut state = ClientState::new();
     if let Some(host) = state.addr.clone() {
         state.connect_to(&host);