From 5b8bd04910bea5ce2b70e922b1eb063475d52b5c Mon Sep 17 00:00:00 2001
From: Scott Worley <scottworley@scottworley.com>
Date: Tue, 10 Oct 2023 16:12:31 -0700
Subject: [PATCH 1/1] server: Merge keystroke time ranges

---
 server/src/main.rs | 36 ++++++++++++++++++++++++++++++++++--
 1 file changed, 34 insertions(+), 2 deletions(-)

diff --git a/server/src/main.rs b/server/src/main.rs
index c31f873..94a47d8 100644
--- a/server/src/main.rs
+++ b/server/src/main.rs
@@ -14,12 +14,13 @@
 // You should have received a copy of the GNU General Public License
 // along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
+use std::collections::HashMap;
 use std::net::UdpSocket;
 use std::time::{Duration, SystemTime};
 
 const MESSAGE_SIZE: usize = 12;
 
-#[derive(Debug)]
+#[derive(Eq, Debug, Hash, PartialEq)]
 struct MessageKey {
     epoch: u32,
     device: u16,
@@ -57,7 +58,30 @@ impl TryFrom<&[u8]> for Message {
     }
 }
 
+#[derive(Debug)]
+struct Range {
+    start: SystemTime,
+    end: SystemTime,
+}
+impl Range {
+    fn new(t: &SystemTime) -> Self {
+        Self { start: *t, end: *t }
+    }
+    fn contains(&self, t: &SystemTime) -> bool {
+        t > &self.start && t < &self.end
+    }
+    fn extend(&mut self, t: &SystemTime) {
+        if t < &self.start {
+            self.start = *t;
+        }
+        if t > &self.end {
+            self.end = *t;
+        }
+    }
+}
+
 fn main() {
+    let mut presses = HashMap::<MessageKey, Range>::new();
     let socket = UdpSocket::bind("0.0.0.0:29803").expect("couldn't bind to address");
     loop {
         let mut buf = [0; MESSAGE_SIZE];
@@ -70,7 +94,15 @@ fn main() {
                     continue;
                 }
                 let message = Message::try_from(filled_buf).expect("I can't count");
-                println!("Got packet from {src_addr}: {message:?}");
+                if let Some(r) = presses.get_mut(&message.key) {
+                    if !r.contains(&message.t) {
+                        r.extend(&message.t);
+                        println!("Updated press: {:?}: {r:?}", message.key);
+                    }
+                } else {
+                    println!("Got new press: {:?}: {:?}", message.key, message.t);
+                    presses.insert(message.key, Range::new(&message.t));
+                }
             }
         }
     }
-- 
2.44.1