From 977dadaa07f86a3ebabedb425cc6f68b2e14d569 Mon Sep 17 00:00:00 2001
From: Katharina Fey <kookie@spacekookie.de>
Date: Wed, 29 Jan 2020 07:56:26 +0000
Subject: Adding some more stat functions to handle incoming packets

---
 src/gens.rs | 47 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100644 src/gens.rs

(limited to 'src/gens.rs')

diff --git a/src/gens.rs b/src/gens.rs
new file mode 100644
index 000000000000..cd0af16754f3
--- /dev/null
+++ b/src/gens.rs
@@ -0,0 +1,47 @@
+//! Helpers to determine if a node can send a particular packet.
+
+/// These functions are unfortunately all a bit stringly typed atm
+pub mod can_send {
+    use crate::data::{Level, PacketType, Upgrade};
+    use Level::*;
+
+    #[inline]
+    pub fn base() -> Vec<&'static str> {
+        vec!["ping", "capture"]
+    }
+
+    /// Guard nodes are the most versatile
+    #[inline]
+    pub fn guard(lvl: Level) -> Vec<&'static str> {
+        match lvl {
+            // This is just kinda gross
+            One => {
+                let mut p = base();
+                p.append(&mut vec!["payload", "cns", "reset"]);
+                p
+            }
+            Two => {
+                let mut p = guard(One);
+                p.append(&mut vec!["nitm", "virus"]);
+                p
+            }
+            Three => {
+                let mut p = guard(Two);
+                p.append(&mut vec!["takeover"]);
+                p
+            }
+        }
+    }
+
+    /// Compute nodes can sennd compute packets
+    pub fn compute() -> Vec<&'static str> {
+        let mut p = base();
+        p.append(&mut vec!["compute"]);
+        p
+    }
+
+    /// Relays only relay!
+    pub fn relay() -> Vec<&'static str> {
+        vec![]
+    }
+}
-- 
cgit v1.2.3