echo chal

1 files changed, 50 insertions, 0 deletions

diff --git a/common/src/msg.rs b/common/src/msg.rs
new file mode 100644
index 0000000..23db171
--- /dev/null
+++ b/common/src/msg.rs
@@ -0,0 +1,50 @@
+use serde::{Deserialize, Serialize};
+
+#[derive(Debug, Serialize, Deserialize)]
+pub struct Message<B> {
+    #[serde(flatten)]
+    pub header: MessageHeader,
+    pub body: MaelstromBodyOr<B>,
+}
+
+#[derive(Debug, Serialize, Deserialize)]
+pub struct MessageHeader {
+    pub src: String,
+    #[serde(rename = "dest")]
+    pub dst: String,
+}
+
+impl MessageHeader {
+    pub fn flip(self) -> Self {
+        Self {
+            src: self.dst,
+            dst: self.src,
+        }
+    }
+}
+
+#[derive(Debug, Serialize, Deserialize)]
+#[serde(untagged)]
+pub enum MaelstromBodyOr<B> {
+    MaelstromBody {
+        #[serde(flatten)]
+        inner: MaelstromBody,
+    },
+    Other {
+        #[serde(flatten)]
+        inner: B,
+    },
+}
+
+#[derive(Debug, Serialize, Deserialize)]
+#[serde(tag = "type")]
+pub enum MaelstromBody {
+    #[serde(rename = "init")]
+    Init {
+        node_id: String,
+        node_ids: Vec<String>,
+        msg_id: usize,
+    },
+    #[serde(rename = "init_ok")]
+    InitOk { msg_id: usize, in_reply_to: usize },
+}