From 590b38d9b0953bc3ee4999e25f37296f248738ed Mon Sep 17 00:00:00 2001
From: MeexReay <meexreay@gmail.com>
Date: Sun, 17 Nov 2024 16:15:39 +0300
Subject: [PATCH] events events blah blah

---
 make_event/src/lib.rs    | 56 ++++++++++++++++++++++++++++++++++------
 src/meexprox/meexprox.rs | 17 +++++++++---
 2 files changed, 62 insertions(+), 11 deletions(-)

diff --git a/make_event/src/lib.rs b/make_event/src/lib.rs
index 187dcb1..87f9c16 100644
--- a/make_event/src/lib.rs
+++ b/make_event/src/lib.rs
@@ -20,17 +20,29 @@ pub fn make_event_derive(input: TokenStream) -> TokenStream {
 
     let mut getters = Vec::new();
     let mut setters = Vec::new();
+    let mut new_args = Vec::new();
+    let mut new_inits = Vec::new();
+    let mut has_cancelled_field = false;
 
     if let Data::Struct(data) = input.data {
         if let Fields::Named(fields) = data.fields {
             for field in fields.named.iter() {
                 let field_name = &field.ident;
                 let field_ty = &field.ty;
+
+                if field_name.as_ref().map(|name| name == "cancelled").unwrap_or(false) {
+                    has_cancelled_field = true;
+                } else {
+                    new_args.push(quote! { #field_name: #field_ty });
+                    new_inits.push(quote! { #field_name });
+                }
+
                 getters.push(quote! {
                     pub fn #field_name(&self) -> &#field_ty {
                         &self.#field_name
                     }
                 });
+
                 if field.attrs.iter().any(|attr| attr.path().is_ident("setter")) {
                     let setter_name = format_ident!("set_{}", field_name.as_ref().unwrap());
                     setters.push(quote! {
@@ -45,13 +57,8 @@ pub fn make_event_derive(input: TokenStream) -> TokenStream {
         panic!("MakeEvent can only be derived for structs with named fields");
     }
 
-    let expanded = quote! {
-        impl #struct_name {
-            #(#getters)*
-            #(#setters)*
-        }
-
-        impl Event for #struct_name {
+    let cancel_methods = if has_cancelled_field {
+        quote! {
             fn cancel(&mut self) {
                 self.cancelled = true;
             }
@@ -59,6 +66,39 @@ pub fn make_event_derive(input: TokenStream) -> TokenStream {
             fn is_cancelled(&self) -> bool {
                 self.cancelled
             }
+        }
+    } else {
+        quote! {}
+    };
+
+    let new_method = if has_cancelled_field {
+        quote! {
+            pub fn new(#(#new_args),*) -> Self {
+                Self {
+                    #(#new_inits),*,
+                    cancelled: false,
+                }
+            }
+        }
+    } else {
+        quote! {
+            pub fn new(#(#new_args),*) -> Self {
+                Self {
+                    #(#new_inits),*
+                }
+            }
+        }
+    };
+
+    let expanded = quote! {
+        impl #struct_name {
+            #(#getters)*
+            #(#setters)*
+            #new_method
+        }
+
+        impl Event for #struct_name {
+            #cancel_methods
 
             fn name(&self) -> String {
                 #event_name.to_string()
@@ -67,4 +107,4 @@ pub fn make_event_derive(input: TokenStream) -> TokenStream {
     };
 
     TokenStream::from(expanded)
-}
\ No newline at end of file
+}
diff --git a/src/meexprox/meexprox.rs b/src/meexprox/meexprox.rs
index 8bdc428..3c63e87 100644
--- a/src/meexprox/meexprox.rs
+++ b/src/meexprox/meexprox.rs
@@ -9,7 +9,7 @@ use std::{
     }, thread,
 };
 
-use super::{config::ProxyConfig, connection::Player, error::{AsProxyResult, ProxyError}, event::{Event, EventListener}};
+use super::{config::ProxyConfig, connection::Player, error::{AsProxyResult, ProxyError}, event::{Event, EventListener, StatusEvent}};
 
 
 pub struct MeexProx {
@@ -84,8 +84,19 @@ impl MeexProx {
 
         if next_state == 1 {
             loop {
-                server_conn.write_packet(&client_conn.read_packet().as_proxy()?).as_proxy()?;
-                client_conn.write_packet(&server_conn.read_packet().as_proxy()?).as_proxy()?;
+                let packet = client_conn.read_packet().as_proxy()?;
+                server_conn.write_packet(&packet).as_proxy()?;
+                if packet.id() == 0x00 {
+                    let motd = server_conn.read_packet().as_proxy()?.read_string().as_proxy()?;
+                    
+                    let mut event = StatusEvent::new(addr.clone(), motd, server_address.clone(), server_port, protocol_version);
+                    self.trigger_event(&mut event)?;
+                    let motd = event.motd();
+
+                    client_conn.write_packet(&Packet::build(0x00, |o| o.write_string(&motd)).as_proxy()?).as_proxy()?;
+                } else {
+                    client_conn.write_packet(&server_conn.read_packet().as_proxy()?).as_proxy()?;
+                }
             }
         } else if next_state == 2 {
             self.players.write().unwrap().push(Player::read(