Plugin actions

src/plugin/mod.rs

@@ -4,22 +4,60 @@ pub mod types;
 use std::{
     io::{BufRead, BufReader, Write},
     net::TcpStream,
+    sync::Arc,
 };
 
-use crate::plugin::types::{LoaderMessage, PluginMessage};
+use crate::{
+    plugin::types::{LoaderMessage, PluginMessage},
+    server::Server,
+    types::message::ServerMessage,
+};
 
-pub struct Plugin(TcpStream, BufReader<TcpStream>);
+pub struct Plugin {
+    stream: TcpStream,
+    reader: BufReader<TcpStream>,
+    id: String,
+}
 
 impl Plugin {
-    pub fn send(&mut self, m: &LoaderMessage) {
-        self.0
-            .write(serde_json::to_string(&m).unwrap().as_bytes())
-            .unwrap();
+    pub fn send(&mut self, m: &LoaderMessage) -> crate::Result<()> {
+        self.stream
+            .write(serde_json::to_string(&m).unwrap().as_bytes())?;
+        Ok(())
     }
 
-    pub fn read(&mut self) -> PluginMessage {
+    pub fn read(&mut self) -> crate::Result<PluginMessage> {
         let mut buf = String::new();
-        self.1.read_line(&mut buf).unwrap();
-        serde_json::from_str(&buf).unwrap()
+        self.reader.read_line(&mut buf)?;
+        Ok(serde_json::from_str(&buf)?)
+    }
+
+    pub fn run(&mut self, server: &Arc<Server>) -> crate::Result<()> {
+        loop {
+            match self.read()? {
+                PluginMessage::SendMessage {
+                    channel_id,
+                    contents,
+                } => {
+                    let msg = server.db.insert_message(
+                        &channel_id,
+                        &self.id,
+                        &contents,
+                        chrono::Utc::now().timestamp(),
+                    )?;
+
+                    for c in server.clients.lock().unwrap().iter() {
+                        let c = c.clone();
+                        let server = server.clone();
+                        let msg = msg.clone();
+                        std::thread::spawn(move || {
+                            server
+                                .wrap_err(&c, c.send(ServerMessage::MessageCreate(msg)))
+                                .expect("Failed to broadcast");
+                        });
+                    }
+                }
+            }
+        }
     }
 }