From 57c69c604c0131cc1e29e1487451fb16a56f4270 Mon Sep 17 00:00:00 2001
From: Valmo Trindade <val_mo_jr@hotmail.com>
Date: Tue, 28 Jan 2025 16:25:28 -0300
Subject: [PATCH] added initial version of cot router, already able to connect
 to the TCP port and send Cots (as raw strings)

---
 src/cot_router.rs | 115 ++++++++++++++++++++++++++++++++++++++++++++++
 src/lib.rs        |  23 +++-------
 2 files changed, 121 insertions(+), 17 deletions(-)
 create mode 100644 src/cot_router.rs

diff --git a/src/cot_router.rs b/src/cot_router.rs
new file mode 100644
index 0000000..40ba1df
--- /dev/null
+++ b/src/cot_router.rs
@@ -0,0 +1,115 @@
+use log::info;
+use std::io::Write;
+use std::net::TcpStream;
+use std::sync::mpsc::{self, Receiver, Sender};
+use std::sync::{Arc, Mutex};
+use std::thread;
+use lazy_static::lazy_static;
+
+pub enum TcpCommand {
+    SendMessage(String),
+    Stop,
+}
+
+pub struct TcpClient {
+    pub(crate) tx: Sender<TcpCommand>,
+}
+
+impl TcpClient {
+    pub fn start(&self, address: String, rx: Receiver<TcpCommand>) {
+        if let Some(ref client) = *TCP_CLIENT.lock().unwrap() {
+            client.stop();
+        }
+
+        let connection = Arc::new(Mutex::new(None));
+        let connection_clone = Arc::clone(&connection);
+
+        thread::spawn(move || {
+            let mut running = true;
+
+            // TCP connection thread
+            let tcp_thread = thread::spawn(move || {
+                match TcpStream::connect(&address) {
+                    Ok(stream) => {
+                        info!("Connected to TCP server at {}", address);
+                        *connection_clone.lock().unwrap() = Some(stream);
+                    }
+                    Err(e) => {
+                        info!("Failed to connect to TCP server: {}", e);
+                    }
+                }
+            });
+
+            while running {
+                match rx.recv() {
+                    Ok(TcpCommand::SendMessage(message)) => {
+                        if let Some(mut stream) = connection.lock().unwrap().as_ref() {
+                            if let Err(e) = stream.write_all(message.as_bytes()) {
+                                info!("Failed to send message: {}", e);
+                            }
+                        } else {
+                            info!("No active TCP connection.");
+                        }
+                    }
+                    Ok(TcpCommand::Stop) => {
+                        running = false;
+                        info!("Stopping TCP client.");
+                    }
+                    Err(error) => {
+                        info!("Error receiving command: {}", error.to_string());
+                    }
+                }
+            }
+
+            tcp_thread.join().unwrap();
+        });
+    }
+
+    pub fn send_payload(&self, payload: String) {
+        self.tx.send(TcpCommand::SendMessage(payload)).unwrap();
+    }
+
+    pub fn stop(&self) {
+        self.tx.send(TcpCommand::Stop).unwrap();
+    }
+}
+
+lazy_static! {
+    static ref TCP_CLIENT: Arc<Mutex<Option<TcpClient>>> = Arc::new(Mutex::new(None));
+}
+
+pub fn start(address: String) -> &'static str {
+    let (tx, rx): (Sender<TcpCommand>, Receiver<TcpCommand>) = mpsc::channel();
+
+    let client = TcpClient { tx };
+    client.start(address, rx);
+
+    let mut client_guard = TCP_CLIENT.lock().unwrap();
+    *client_guard = Some(client);
+
+    info!("TCP client started.");
+
+    "Starting TCP Client"
+}
+
+pub fn send_payload(payload: String) -> &'static str {
+    if let Some(ref client) = *TCP_CLIENT.lock().unwrap() {
+        info!("Sending payload: {}", payload);
+        client.send_payload(payload);
+    } else {
+        info!("TCP client is not running.");
+    }
+
+    "Sending payload to TCP server"
+}
+
+pub fn stop() -> &'static str {
+    if let Some(ref client) = *TCP_CLIENT.lock().unwrap() {
+        client.stop();
+        info!("TCP client stopped.");
+    } else {
+        info!("TCP client is not running.");
+    }
+
+    "Stopping TCP Client"
+}
diff --git a/src/lib.rs b/src/lib.rs
index 015fa2c..1c5b0a6 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -2,9 +2,9 @@ use arma_rs::{arma, Extension, Group};
 mod structs;
 mod tests;
 mod websocket;
-mod ots_api;
-mod fts_api;
 mod util;
+mod cot_router;
+mod cot_generator;
 
 #[arma]
 pub fn init() -> Extension {
@@ -40,22 +40,11 @@ pub fn init() -> Extension {
         .command("uuid", util::get_uuid)
         .command("log", util::log_info)
         .group(
-            "ots_api",
+            "cot_router",
             Group::new()
-                .command("get", ots_api::markers::get)
-                .command("get_auth_token", ots_api::get_auth_token)
-                .command("post", ots_api::markers::post)
-                .command("post_debug", ots_api::markers::post_debug)
-                .command("delete", ots_api::markers::delete),
-        )
-        .group(
-            "fts_api",
-            Group::new()
-                .command("get", fts_api::markers::get)
-                .command("get_all", fts_api::markers::get_all)
-                .command("post", fts_api::markers::post)
-                .command("patch", fts_api::markers::patch)
-                .command("delete", fts_api::markers::delete)
+                .command("start", cot_router::start)
+                .command("send_cot", cot_router::send_payload)
+                .command("stop", cot_router::stop)
         )
         .finish()
 }