reoganized command groups on extension call

src/cot/draws/circle.rs

@@ -0,0 +1,126 @@
+use arma_rs::{FromArma, FromArmaError};
+
+pub struct CircleCoTPayload {
+    pub uuid: String,
+    pub center_lat: f64,
+    pub center_lon: f64,
+    pub center_hae: f32,
+    pub major: f64,
+    pub minor: f64,
+    pub angle: f32,
+    pub callsign: String,
+    pub creator_uid: String,
+    pub creator_callsign: String,
+}
+
+impl FromArma for CircleCoTPayload {
+    fn from_arma(data: String) -> Result<Self, FromArmaError> {
+        let (
+            uuid,
+            center_lat,
+            center_lon,
+            center_hae,
+            major,
+            minor,
+            angle,
+            callsign,
+            creator_uid,
+            creator_callsign,
+        ) = <(String, f64, f64, f32, f64, f64, f32, String, String, String)>::from_arma(data)?;
+
+        Ok(Self {
+            uuid,
+            center_lat,
+            center_lon,
+            center_hae,
+            major,
+            minor,
+            angle,
+            callsign,
+            creator_uid,
+            creator_callsign,
+        })
+    }
+}
+
+pub struct ShapeCircleCoT {
+    pub uid: String,
+    pub lat: f64,
+    pub lon: f64,
+    pub hae: f32,
+    pub major: f64,
+    pub minor: f64,
+    pub angle: f32,
+    pub callsign: String,
+    pub creator_uid: String,
+    pub creator_callsign: String,
+}
+
+impl CircleCoTPayload {
+    pub fn to_cot(&self) -> ShapeCircleCoT {
+        ShapeCircleCoT {
+            uid: self.uuid.clone(),
+            lat: self.center_lat,
+            lon: self.center_lon,
+            hae: self.center_hae,
+            major: self.major,
+            minor: self.minor,
+            angle: self.angle,
+            callsign: self.callsign.clone(),
+            creator_uid: self.creator_uid.clone(),
+            creator_callsign: self.creator_callsign.clone(),
+        }
+    }
+}
+
+impl ShapeCircleCoT {
+    pub fn to_xml(&self, now: &str, stale: &str) -> String {
+        format!(
+            r#"<event version="2.0" uid="{uid}" type="u-d-c-c"
+  time="{t}" start="{t}" stale="{stale}"
+  how="h-e" access="Undefined">
+  <point lat="{lat}" lon="{lon}" hae="{hae}" ce="10.9" le="9999999.0" />
+  <detail>
+    <shape>
+      <ellipse major="{major}" minor="{minor}" angle="{angle}" />
+      <link uid="{uid}.Style" type="b-x-KmlStyle" relation="p-c">
+        <Style>
+          <LineStyle>
+            <color>ffffffff</color>
+            <width>3.0</width>
+          </LineStyle>
+          <PolyStyle>
+            <color>96ffffff</color>
+          </PolyStyle>
+        </Style>
+      </link>
+      <link uid="{creator_uid}" type="self" relation="p-p-CenterAnchor" />
+    </shape>
+    <__shapeExtras cpvis="true" editable="true" />
+    <remarks />
+    <contact callsign="{callsign}" />
+    <creator uid="{creator_uid}" callsign="{creator_callsign}" time="{t}" type="a-f-G-U-C" />
+    <archive />
+    <labels_on value="true" />
+    <strokeColor value="-1" />
+    <strokeWeight value="3.0" />
+    <strokeStyle value="solid" />
+    <fillColor value="-1761607681" />
+    <precisionlocation altsrc="GPS" geopointsrc="GPS" />
+  </detail>
+</event>"#,
+            uid = self.uid,
+            t = now,
+            stale = stale,
+            lat = self.lat,
+            lon = self.lon,
+            hae = self.hae,
+            major = self.major,
+            minor = self.minor,
+            angle = self.angle,
+            callsign = self.callsign,
+            creator_uid = self.creator_uid,
+            creator_callsign = self.creator_callsign
+        )
+    }
+}

src/cot/draws/mod.rs

@@ -0,0 +1 @@
+pub mod circle;

src/cot/message.rs

@@ -0,0 +1,153 @@
+use arma_rs::{FromArma, FromArmaError};
+use chrono::{Utc, Duration, SecondsFormat};
+use uuid::Uuid;
+
+pub struct MessagePayload {
+    pub sender_callsign: String,
+    pub chatroom: String,
+    pub message_text: String,
+    pub point_lat: f64,
+    pub point_lon: f64,
+    pub point_hae: f32,
+    pub sender_uid: String,
+}
+
+impl FromArma for MessagePayload {
+    fn from_arma(data: String) -> Result<Self, FromArmaError> {
+        let (sender_callsign, chatroom, message_text,
+            point_lat, point_lon, point_hae, sender_uid) =
+            <(String, String, String, f64, f64, f32, String)>::from_arma(data)?;
+
+        Ok(Self {
+            sender_callsign,
+            chatroom,
+            message_text,
+            point_lat,
+            point_lon,
+            point_hae,
+            sender_uid,
+        })
+    }
+}
+
+pub struct MessageCot {
+    pub sender_callsign: String,
+    pub chatroom: String,
+    pub message_text: String,
+    pub point_lat: f64,
+    pub point_lon: f64,
+    pub point_hae: f32,
+    pub sender_uid: String,
+}
+
+impl MessageCot {
+    pub fn from_payload(p: MessagePayload) -> Self {
+        Self {
+            sender_callsign: p.sender_callsign,
+            chatroom: p.chatroom,
+            message_text: p.message_text,
+            point_lat: p.point_lat,
+            point_lon: p.point_lon,
+            point_hae: p.point_hae,
+            sender_uid: p.sender_uid,
+        }
+    }
+
+    pub fn to_xml(&self) -> String {
+        let created_time = Utc::now().to_rfc3339_opts(SecondsFormat::Millis, true);
+        let stale_time = (Utc::now() + Duration::days(1))
+            .to_rfc3339_opts(SecondsFormat::Millis, true);
+
+        // MESSAGE ID (random UUID)
+        let message_uuid = Uuid::new_v4().to_string();
+
+        // FULL EVENT UID
+        // format: GeoChat.{sender}.{chatroom}.{uuid}
+        let event_uid = format!(
+            "GeoChat.{}.{}.{}",
+            self.sender_uid,
+            self.chatroom.replace(" ", "_"),
+            message_uuid,
+        );
+
+        let mut xml = String::new();
+
+        xml.push_str("<?xml version=\"1.0\" encoding=\"UTF-8\"?>");
+
+        xml.push_str(
+            format!(
+                "<event version=\"2.0\" uid=\"{}\" type=\"b-t-f\" time=\"{}\" start=\"{}\" stale=\"{}\" how=\"h-g-i-g-o\" access=\"Undefined\">",
+                event_uid, created_time, created_time, stale_time
+            )
+            .as_str(),
+        );
+
+        xml.push_str(
+            format!(
+                "<point lat=\"{}\" lon=\"{}\" hae=\"{}\" ce=\"10.3\" le=\"9999999.0\"/>",
+                self.point_lat, self.point_lon, self.point_hae
+            )
+            .as_str(),
+        );
+
+        xml.push_str("<detail>");
+
+        // ========== CHAT OBJECT ==========
+
+        xml.push_str(
+            format!(
+                "<__chat parent=\"RootContactGroup\" groupOwner=\"false\" \
+                 messageId=\"{}\" chatroom=\"{}\" id=\"{}\" senderCallsign=\"{}\">",
+                message_uuid,
+                self.chatroom,
+                self.chatroom,
+                self.sender_callsign,
+            )
+            .as_str(),
+        );
+
+        xml.push_str(
+            format!(
+                "<chatgrp uid0=\"{}\" uid1=\"{}\" id=\"{}\" />",
+                self.sender_uid,
+                self.chatroom,
+                self.chatroom
+            )
+            .as_str(),
+        );
+
+        xml.push_str("</__chat>");
+
+        // ========== LINK ELEMENT ==========
+        xml.push_str(
+            format!(
+                "<link uid=\"{}\" type=\"a-f-G-U-C\" relation=\"p-p\" />",
+                self.sender_uid
+            )
+            .as_str(),
+        );
+
+        // ========== SERVER DEST ==========
+        // This is optional — you may remove or customize it
+        xml.push_str(
+            format!(
+                "<__serverdestination destinations=\"0.0.0.0:0:tcp:{}\" />",
+                self.sender_uid
+            )
+            .as_str(),
+        );
+
+        // ========== MESSAGE REMARKS ==========
+        xml.push_str(
+            format!(
+                "<remarks source=\"ARMATAK.{}\" to=\"{}\" time=\"{}\">{}</remarks>",
+                self.sender_uid, self.chatroom, created_time, self.message_text
+            )
+            .as_str(),
+        );
+
+        xml.push_str("</detail></event>");
+
+        xml
+    }
+}

src/cot/mod.rs

@@ -1,5 +1,7 @@
+pub mod draws;
 pub mod cot;
 pub mod digital_pointer;
 pub mod eud;
 pub mod gps;
+pub mod message;
 pub mod nato;

src/lib.rs

@@ -1,8 +1,8 @@
 use arma_rs::{arma, Extension, Group};
 mod structs;
+mod tcp;
 mod tests;
 mod udp_socket;
-mod tcp_socket;
 mod video_stream;
 
 mod cot;
@@ -35,28 +35,43 @@ pub fn init() -> Extension {
         .command("local_ip", utils::address::get_local_address)
         .command("uuid", utils::uuid::get_uuid)
         .command("log", utils::log::log_info)
-        .group("udp_socket",
+        .group(
+            "udp_socket",
             Group::new()
                 .command("start", udp_socket::start)
                 .command("send_payload", udp_socket::send_payload)
                 .command("send_gps_cot", udp_socket::send_gps_cot)
-                .command("stop", udp_socket::stop)
+                .command("stop", udp_socket::stop),
         )
         .group(
             "tcp_socket",
             Group::new()
-                .command("start", tcp_socket::start)
-                .command("send_payload", tcp_socket::send_payload)
-                .command("send_eud_cot", tcp_socket::send_eud_cot)
-                .command("send_marker_cot", tcp_socket::send_marker_cot)
-                .command("send_digital_pointer_cot", tcp_socket::send_digital_pointer_cot)
-                .command("stop", tcp_socket::stop)
+                .command("start", tcp::start)
+                .command("stop", tcp::stop)
+                .command("send_payload", tcp::send_payload)
+                .group(
+                    "cot",
+                    Group::new()
+                        .command("eud", tcp::cot::send_eud_cot)
+                        .command("marker", tcp::cot::send_marker_cot)
+                        .command("digital_pointer", tcp::cot::send_digital_pointer_cot)
+                        .command("chat", tcp::cot::send_message_cot),
+                )
+                .group(
+                    "draw",
+                    Group::new()
+                        .command("circle", tcp::draw::send_circle_cot)
+                        .command("ellipse", tcp::draw::send_ellipse_cot)
+                        .command("rectangle", tcp::draw::send_rectangle_cot)
+                        .command("free", tcp::draw::send_freedraw_cot)
+                        .command("vector", tcp::draw::send_vectordraw_cot),
+                ),
         )
         .group(
             "video_stream",
             Group::new()
                 .command("start", video_stream::start_stream)
-                .command("stop", video_stream::stop_stream)
+                .command("stop", video_stream::stop_stream),
         )
         .finish()
 }

src/tcp/cot.rs

@@ -0,0 +1,32 @@
+use arma_rs::Context;
+
+use crate::{cot, tcp::send_payload};
+
+pub fn send_eud_cot(ctx: Context, cursor_over_time: cot::eud::EudCoTPayload) -> &'static str {
+    let payload = cursor_over_time.to_cot().convert_to_xml();
+    send_payload(ctx, payload);
+
+    "Sending End User Device Cursor Over Time to TCP server"
+}
+
+pub fn send_marker_cot(ctx: Context, cursor_over_time: cot::nato::MarkerCoTPayload) -> &'static str {
+    let payload = cursor_over_time.to_cot().convert_to_xml();
+    send_payload(ctx, payload);
+
+    "Sending Marker Cursor Over Time to TCP server"
+}
+
+pub fn send_digital_pointer_cot(ctx: Context, cursor_over_time: cot::digital_pointer::DigitalPointerPayload) -> &'static str {
+    let payload = cursor_over_time.to_cot().convert_to_xml();
+    send_payload(ctx, payload);
+
+    "Sending Digital Pointer Cursor Over Time to TCP server"
+}
+
+pub fn send_message_cot(ctx: Context, message_payload: cot::message::MessagePayload) -> &'static str {
+    let message_cot = cot::message::MessageCot::from_payload(message_payload);
+    let payload = message_cot.to_xml();
+    send_payload(ctx, payload);
+
+    "Sending Message CoT to TCP server"
+}

src/tcp/draw.rs

@@ -0,0 +1,34 @@
+use arma_rs::Context;
+
+use crate::{cot, tcp::send_payload};
+
+pub fn send_circle_cot(ctx: Context, circle_payload: cot::draws::circle::CircleCoTPayload) -> &'static str {
+    let shape_circle_cot = circle_payload.to_cot();
+    let now = chrono::Utc::now().to_rfc3339_opts(chrono::SecondsFormat::Millis, true);
+    let stale = (chrono::Utc::now() + chrono::Duration::days(1))
+        .to_rfc3339_opts(chrono::SecondsFormat::Millis, true);
+    let payload = shape_circle_cot.to_xml(&now, &stale);
+    send_payload(ctx, payload);
+
+    "Sending Circle CoT to TCP server"
+}
+
+pub fn send_ellipse_cot(ctx: Context) -> &'static str {
+  let _ = ctx;
+  "Not implemented: send_ellipse_cot"
+}
+
+pub fn send_rectangle_cot(ctx: Context) -> &'static str {
+  let _ = ctx;
+  "Not implemented: send_ellipse_cot"
+}
+
+pub fn send_freedraw_cot(ctx: Context) -> &'static str {
+  let _ = ctx;
+  "Not implemented: send_ellipse_cot"
+}
+
+pub fn send_vectordraw_cot(ctx: Context) -> &'static str {
+  let _ = ctx;
+  "Not implemented: send_ellipse_cot"
+}

src/tcp/mod.rs

@@ -7,7 +7,8 @@ use std::sync::mpsc::{self, Receiver, Sender};
 use std::sync::{Arc, Mutex};
 use std::thread;
 
-use crate::cot;
+pub mod cot;
+pub mod draw;
 
 pub enum TcpCommand {
     SendMessage(String, Context),
@@ -123,27 +124,6 @@ pub fn send_payload(ctx: Context, payload: String) -> &'static str {
     "Sending payload to TCP server"
 }
 
-pub fn send_eud_cot(ctx: Context, cursor_over_time: cot::eud::EudCoTPayload) -> &'static str {
-    let payload = cursor_over_time.to_cot().convert_to_xml();
-    send_payload(ctx, payload);
-
-    "Sending End User Device Cursor Over Time to TCP server"
-}
-
-pub fn send_marker_cot(ctx: Context, cursor_over_time: cot::nato::MarkerCoTPayload) -> &'static str {
-    let payload = cursor_over_time.to_cot().convert_to_xml();
-    send_payload(ctx, payload);
-
-    "Sending Marker Cursor Over Time to TCP server"
-}
-
-pub fn send_digital_pointer_cot(ctx: Context, cursor_over_time: cot::digital_pointer::DigitalPointerPayload) -> &'static str {
-    let payload = cursor_over_time.to_cot().convert_to_xml();
-    send_payload(ctx, payload);
-
-    "Sending Digital Pointer Cursor Over Time to TCP server"
-}
-
 pub fn stop(ctx: Context) -> &'static str {
     if let Some(ref client) = *TCP_CLIENT.lock().unwrap() {
         client.stop();