From 64c77b259dec7b50e4a005a03f7d8dcc8e685d94 Mon Sep 17 00:00:00 2001
From: Noah Metz <noah@metznet.ca>
Date: Thu, 18 Jan 2024 17:22:41 -0700
Subject: [PATCH] Added callback mechanism with dummy function to retrieve
 notices

---
 src/main.rs | 102 +++++++++++++++++++++++++++++++++++++++++++---------
 1 file changed, 86 insertions(+), 16 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 6b6af20..42bd3b4 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,6 +1,7 @@
 use rumqttc::{MqttOptions, Client, QoS};
 use std::time::Duration;
 use std::thread;
+use std::collections::hash_map::HashMap;
 use prost::Message;
 use std::io::Cursor;
 use serde::{Serialize, Deserialize};
@@ -75,12 +76,12 @@ use serde::{Serialize, Deserialize};
 //   - game/{division_id}/{game_id}/score
 //   - team/{team_string}
 
-pub mod tm_proto {
+pub mod tm {
     include!(concat!(env!("OUT_DIR"), "/tm.rs"));
 }
 
-pub fn deserialize_notice(buf: &[u8]) -> Result<tm_proto::Notice, prost::DecodeError> {
-    tm_proto::Notice::decode(&mut Cursor::new(buf))
+pub fn deserialize_notice(buf: &[u8]) -> Result<tm::Notice, prost::DecodeError> {
+    tm::Notice::decode(&mut Cursor::new(buf))
 }
 
 #[derive(Serialize, Deserialize, Debug)]
@@ -184,29 +185,78 @@ struct ArenaInfo {
     match_tuple: MatchTuple,
 }
 
-fn main() {
-    let score = GameScore{
+struct MQTTMessage {
+    topic: String,
+    payload: String,
+}
+
+fn get_game_score(scores: tm::MatchScore) -> Option<GameScore> {
+    if scores.alliances.len() != 2 {
+        return None;
+    }
+
+    let ref red_score = scores.alliances[0];
+    let ref blue_score = scores.alliances[1];
+
+    // 1) Get the autonomous winner
+    // 2) Get score object and fill AllianceScore struct
+    // 3) Compute total scores
+
+
+    let out = GameScore{
         autonomous_winner: None,
-        red_score: AllianceScore{
+        red_total: 0,
+        blue_total: 0,
+        blue_score: AllianceScore{
             team_goal: 0,
             team_zone: 0,
             green_goal: 0,
             green_zone: 0,
             elevation_tiers: [None, None],
         },
-        blue_score: AllianceScore{
+        red_score : AllianceScore{
             team_goal: 0,
             team_zone: 0,
             green_goal: 0,
             green_zone: 0,
             elevation_tiers: [None, None],
         },
-        red_total: 0,
-        blue_total: 0,
     };
 
-    let serialized = serde_json::to_string(&score).unwrap();
-    println!("Serialized = {}", serialized);
+    return Some(out);
+}
+
+fn on_score_change(notice: tm::Notice) -> Vec<MQTTMessage> {
+    match notice.match_score {
+        None => return Vec::new(),
+        Some(game_scores) => {
+            match get_game_score(game_scores) {
+                Some(score_json) => {
+                    let serialized = serde_json::to_string(&score_json).unwrap();
+                    let arena_topic = String::from("arena/TEST/score");
+                    let mut out = Vec::new();
+                    out.push(MQTTMessage{
+                        topic: arena_topic,
+                        payload: serialized,
+                    });
+                    return out;
+                },
+                None => return Vec::new(),
+            }
+        },
+    }
+}
+
+fn get_next_notice() -> tm::Notice {
+    thread::sleep_ms(1000);
+    return tm::Notice::default();
+}
+
+type NoticeCallback = fn(tm::Notice) -> Vec<MQTTMessage>;
+
+fn main() {
+    let mut callbacks: HashMap<tm::NoticeId, NoticeCallback> = HashMap::new();
+    callbacks.insert(tm::NoticeId::NoticeRealtimeScoreChanged, on_score_change);
 
     // Set client options
     let mut mqttoptions = MqttOptions::new("vex-bridge", "localhost", 1883);
@@ -216,17 +266,37 @@ fn main() {
     let (mut client, mut connection) = Client::new(mqttoptions, 10);
     client.subscribe("bridge", QoS::AtMostOnce).unwrap();
 
+    client.publish("bridge/status", QoS::AtLeastOnce, false, "{\"status\": true}").unwrap();
+
     // Create a thread to own connection and handle it's incoming messages
+    let running = true;
     let mqtt_thread = thread::spawn(move ||
-        for (_, message) in connection.iter().enumerate() {
+        for message in connection.iter() {
             println!("Message = {:?}", message);
         }
     );
 
-    // Publish using the client
-    for _ in 0..10 {
-        client.publish("bridge/status", QoS::AtLeastOnce, false, "{\"status\": true}").unwrap();
-        thread::sleep(Duration::from_millis(100));
+    while running {
+        let notice = get_next_notice();
+        let callback = callbacks.get(&notice.id());
+        match callback {
+            None => {
+                match notice.id {
+                    None => println!("Notice without NoticeId received"),
+                    Some(notice_id) => println!("Unhandled NoticeId: {}", notice_id),
+                }
+            },
+            Some(callback) => {
+                let messages = callback(notice);
+                for message in messages {
+                    let result = client.publish(message.topic, QoS::AtMostOnce, true, message.payload);
+                    match result {
+                        Ok(_) => {},
+                        Err(error) => println!("Publish error: {}", error),
+                    }
+                }
+            },
+        }
     }
 
     mqtt_thread.join().expect("Failed to join mqtt thread");