From f27df56c475eacc2de12ac089c16dcf65c77a722 Mon Sep 17 00:00:00 2001
From: MedzikUser <medzik@duck.com>
Date: Sat, 13 Aug 2022 12:06:53 +0200
Subject: [PATCH] feat(plugins): implement events

- Events: onConnect and onSend now works
---
 plugin_test/src/lib.rs | 12 ++++++++++++
 rustfmt.toml           |  1 +
 src/plugins/types.rs   |  3 ++-
 src/server/client.rs   | 20 +++++++++++++++-----
 src/server/run.rs      |  8 +++++++-
 5 files changed, 37 insertions(+), 7 deletions(-)

diff --git a/plugin_test/src/lib.rs b/plugin_test/src/lib.rs
index 8622dcf..ab1f4f7 100644
--- a/plugin_test/src/lib.rs
+++ b/plugin_test/src/lib.rs
@@ -36,8 +36,20 @@ impl Command for PluginTest {
     }
 }
 
+#[async_trait]
+impl Event for PluginTest {
+    fn event(&self) -> EventType {
+        EventType::OnConnect
+    }
+
+    async fn execute(&self, client: &Client) -> anyhow::Result<()> {
+        client.send("Hello!")
+    }
+}
+
 #[no_mangle]
 pub fn plugin_entry(registrar: &mut dyn Registrar) {
     registrar.register_plugins(Box::new(PluginTest));
     registrar.register_commands(Box::new(PluginTest));
+    registrar.register_events(Box::new(PluginTest));
 }
diff --git a/rustfmt.toml b/rustfmt.toml
index 3bf4091..03f625c 100644
--- a/rustfmt.toml
+++ b/rustfmt.toml
@@ -3,6 +3,7 @@
 # stable
 edition = "2021"
 newline_style = "Unix"
+match_block_trailing_comma = true
 
 # nightly
 group_imports = "StdExternalCrate"
diff --git a/src/plugins/types.rs b/src/plugins/types.rs
index d9a094f..4014acc 100644
--- a/src/plugins/types.rs
+++ b/src/plugins/types.rs
@@ -29,6 +29,7 @@ pub trait Command: Any + Send + Sync {
 }
 
 /// All possible to run events.
+#[derive(Debug, PartialEq, Eq)]
 pub enum EventType {
     /// On client connected.
     OnConnect,
@@ -42,7 +43,7 @@ pub trait Event: Any + Send + Sync {
     /// Type of the event.
     fn event(&self) -> EventType;
     /// Event function.
-    async fn execute(&self) -> anyhow::Result<()>;
+    async fn execute(&self, client: &Client) -> anyhow::Result<()>;
 }
 
 pub trait Registrar {
diff --git a/src/server/client.rs b/src/server/client.rs
index 7e65a7f..e65ed9e 100644
--- a/src/server/client.rs
+++ b/src/server/client.rs
@@ -9,7 +9,7 @@ use std::{
 use tungstenite::{accept, Message, WebSocket};
 
 use super::run::PLUGINS_MANAGER;
-use crate::plugins::manager::PluginsManagerType;
+use crate::plugins::{manager::PluginsManagerType, prelude::EventType};
 
 /// Max length of a TCP and UDP packet
 pub const MAX_PACKET_LEN: usize = 65536;
@@ -110,14 +110,14 @@ impl Client {
 
                 // decode buffer (&[u8]) to a String
                 String::from_utf8(buf.to_vec())?
-            }
+            },
             ClientStream::WebSocket(stream) => {
                 // read the message from the stream
                 let msg = stream.lock().unwrap().read_message()?;
 
                 // decode message to a String
                 msg.to_string()
-            }
+            },
         };
 
         // remove new line characters
@@ -145,7 +145,7 @@ impl Client {
             ClientStream::TCP(stream) => stream.as_ref().write_all(buf)?,
             ClientStream::WebSocket(stream) => {
                 stream.lock().unwrap().write_message(Message::from(msg))?
-            }
+            },
         }
 
         Ok(())
@@ -165,7 +165,7 @@ impl Client {
     pub fn flush(&self) -> anyhow::Result<()> {
         match &self.stream {
             ClientStream::TCP(stream) => stream.as_ref().flush()?,
-            ClientStream::WebSocket(_stream) => {}
+            ClientStream::WebSocket(_stream) => {},
         }
 
         Ok(())
@@ -180,4 +180,14 @@ impl Client {
 
         Ok(())
     }
+
+    pub async fn run_events(&self, event_type: EventType) -> anyhow::Result<()> {
+        for event in self.plugins_manager.events.iter() {
+            if event.event() == event_type {
+                event.execute(self).await?;
+            }
+        }
+
+        Ok(())
+    }
 }
diff --git a/src/server/run.rs b/src/server/run.rs
index 7d861e4..38c0ff9 100644
--- a/src/server/run.rs
+++ b/src/server/run.rs
@@ -6,7 +6,7 @@ use lazy_static::lazy_static;
 use tracing::{error, info};
 
 use crate::{
-    plugins::{self, manager::PluginsManagerType},
+    plugins::{self, manager::PluginsManagerType, prelude::EventType},
     server::Client,
     CLIENTS, CLIENT_NEXT,
 };
@@ -46,9 +46,15 @@ async fn process(client: Client) -> anyhow::Result<()> {
 
     info!("Processing client connection: {}", client_addr);
 
+    // run `onConnect` events
+    client.run_events(EventType::OnConnect).await?;
+
     loop {
         let buf = client.read()?;
 
+        // run `onSend` events
+        client.run_events(EventType::OnSend).await?;
+
         let mut args: Vec<&str> = buf.split_ascii_whitespace().collect();
 
         // if client sent an empty buffer