Add Mysql as a possible database to the Tell plugin

2 files changed, 287 insertions, 0 deletions

diff --git a/src/plugins/tell/database.rs b/src/plugins/tell/database.rs
new file mode 100644
index 0000000..277847e
--- /dev/null
+++ b/src/plugins/tell/database.rs
@@ -0,0 +1,143 @@
+#[cfg(feature = "mysql")]
+extern crate dotenv;
+
+#[cfg(feature = "mysql")]
+use std::sync::Arc;
+use std::collections::HashMap;
+
+#[cfg(feature = "mysql")]
+use diesel::prelude::*;
+#[cfg(feature = "mysql")]
+use diesel::mysql::MysqlConnection;
+#[cfg(feature = "mysql")]
+use r2d2::Pool;
+#[cfg(feature = "mysql")]
+use r2d2_diesel::ConnectionManager;
+
+use chrono::NaiveDateTime;
+
+pub enum DbResponse {
+    Success,
+    Failed(&'static str),
+}
+
+#[cfg_attr(feature = "mysql", derive(Queryable))]
+#[derive(PartialEq, Clone, Debug)]
+pub struct TellMessage {
+    pub id: i64,
+    pub sender: String,
+    pub receiver: String,
+    pub time: NaiveDateTime,
+    pub message: String,
+}
+
+#[cfg_attr(feature = "mysql", derive(Insertable))]
+#[cfg_attr(feature = "mysql", table_name = "tells")]
+pub struct NewTellMessage<'a> {
+    pub sender: &'a str,
+    pub receiver: &'a str,
+    pub time: NaiveDateTime,
+    pub message: &'a str,
+}
+
+pub trait Database: Send {
+    fn insert_tell(&mut self, tell: &NewTellMessage) -> DbResponse;
+    fn get_tells(&self, receiver: &str) -> Option<Vec<TellMessage>>;
+    fn delete_tells(&mut self, receiver: &str) -> DbResponse;
+}
+
+// HashMap
+impl Database for HashMap<String, Vec<TellMessage>> {
+    fn insert_tell(&mut self, tell: &NewTellMessage) -> DbResponse {
+        let tell = TellMessage {
+            id: 0,
+            sender: tell.sender.to_string(),
+            receiver: tell.receiver.to_string(),
+            time: tell.time,
+            message: tell.message.to_string(),
+        };
+
+        let receiver = tell.receiver.clone();
+        let tell_messages = self.entry(receiver)
+            .or_insert_with(|| Vec::with_capacity(3));
+        (*tell_messages).push(tell);
+
+        DbResponse::Success
+    }
+
+    fn get_tells(&self, receiver: &str) -> Option<Vec<TellMessage>> {
+        self.get(receiver).cloned()
+    }
+
+    fn delete_tells(&mut self, receiver: &str) -> DbResponse {
+        match self.remove(receiver) {
+            Some(_) => DbResponse::Success,
+            None => DbResponse::Failed("Tells not found"),
+        }
+    }
+}
+
+// Diesel automatically defines the tells module as public.
+// We create a schema module to keep it private.
+#[cfg(feature = "mysql")]
+mod schema {
+    table! {
+        tells (id) {
+            id -> Bigint,
+            sender -> Varchar,
+            receiver -> Varchar,
+            time -> Timestamp,
+            message -> Varchar,
+        }
+    }
+}
+
+#[cfg(feature = "mysql")]
+use self::schema::tells;
+
+#[cfg(feature = "mysql")]
+impl Database for Arc<Pool<ConnectionManager<MysqlConnection>>> {
+    fn insert_tell(&mut self, tell: &NewTellMessage) -> DbResponse {
+        use diesel;
+
+        let conn = &*self.get().expect("Failed to get connection");
+        match diesel::insert_into(tells::table).values(tell).execute(conn) {
+            Ok(_) => DbResponse::Success,
+            Err(e) => {
+                error!("DB failed to insert tell: {}", e);
+                DbResponse::Failed("Failed to save Tell")
+            }
+        }
+    }
+
+    fn get_tells(&self, receiver: &str) -> Option<Vec<TellMessage>> {
+        use self::tells::columns;
+
+        let conn = &*self.get().expect("Failed to get connection");
+        match tells::table
+            .filter(columns::receiver.eq(receiver))
+            .order(columns::time.asc())
+            .load::<TellMessage>(conn)
+        {
+            Ok(f) => Some(f),
+            Err(e) => {
+                error!("DB failed to get tells: {}", e);
+                None
+            }
+        }
+    }
+
+    fn delete_tells(&mut self, receiver: &str) -> DbResponse {
+        use diesel;
+        use self::tells::columns;
+
+        let conn = &*self.get().expect("Failed to get connection");
+        match diesel::delete(tells::table.filter(columns::receiver.eq(receiver))).execute(conn) {
+            Ok(_) => DbResponse::Success,
+            Err(e) => {
+                error!("DB failed to delete tells: {}", e);
+                DbResponse::Failed("Failed to delete tells")
+            }
+        }
+    }
+}
diff --git a/src/plugins/tell/mod.rs b/src/plugins/tell/mod.rs
new file mode 100644
index 0000000..1a1bef1
--- /dev/null
+++ b/src/plugins/tell/mod.rs
@@ -0,0 +1,144 @@
+use irc::client::prelude::*;
+use irc::error::IrcError;
+
+use time;
+use chrono::NaiveDateTime;
+use std::sync::Mutex;
+
+use plugin::*;
+
+pub mod database;
+use self::database::{Database, DbResponse};
+
+macro_rules! try_lock {
+    ( $m:expr ) => {
+        match $m.lock() {
+            Ok(guard) => guard,
+            Err(poisoned) => poisoned.into_inner(),
+        }
+    }
+}
+
+#[derive(PluginName, Default)]
+pub struct Tell<T: Database> {
+    tells: Mutex<T>,
+}
+
+impl<T: Database> Tell<T> {
+    pub fn new(db: T) -> Tell<T> {
+        Tell {
+            tells: Mutex::new(db),
+        }
+    }
+
+    fn tell_command(&self, client: &IrcClient, command: &PluginCommand) -> Result<&str, String> {
+        if command.tokens.len() < 2 {
+            return Err(self.invalid_command(client));
+        }
+
+        let receiver = command.tokens[0].to_string();
+        let sender = command.source.to_owned();
+
+        if receiver == sender {
+            return Err(String::from("That's your name!"));
+        }
+
+        if command.source != command.target {
+            if let Some(users) = client.list_users(&command.target) {
+                if users.iter().any(|u| u.get_nickname() == receiver) {
+                    return Err(format!("{} is in this channel.", receiver));
+                }
+            }
+        }
+
+        let tm = time::now().to_timespec();
+        let message = command.tokens[1..].join(" ");
+        let tell = database::NewTellMessage {
+            sender: &sender,
+            receiver: &receiver,
+            time: NaiveDateTime::from_timestamp(tm.sec, 0u32),
+            message: &message,
+        };
+
+        match try_lock!(self.tells).insert_tell(&tell) {
+            DbResponse::Success => Ok("Got it!"),
+            DbResponse::Failed(e) => Err(e.to_string()),
+        }
+    }
+
+    fn send_tell(&self, client: &IrcClient, receiver: &str) -> ExecutionStatus {
+        let mut tells = try_lock!(self.tells);
+        if let Some(tell_messages) = tells.get_tells(receiver) {
+            for tell in tell_messages {
+                if let Err(e) = client.send_notice(
+                    receiver,
+                    &format!("Tell from {}: {}", tell.sender, tell.message),
+                ) {
+                    return ExecutionStatus::Err(Box::new(e));
+                }
+                debug!(
+                    "Sent {:?} from {:?} to {:?}",
+                    tell.message, tell.sender, receiver
+                );
+            }
+        }
+        tells.delete_tells(receiver);
+        ExecutionStatus::Done
+    }
+
+    fn invalid_command(&self, client: &IrcClient) -> String {
+        format!(
+            "Incorrect Command. \
+             Send \"{} tell help\" for help.",
+            client.current_nickname()
+        )
+    }
+
+    fn help(&self, client: &IrcClient) -> String {
+        format!(
+            "usage: {} tell user message\r\n\
+             example: {0} tell Foobar Hello!",
+            client.current_nickname()
+        )
+    }
+}
+
+impl<T: Database> Plugin for Tell<T> {
+    fn execute(&self, client: &IrcClient, message: &Message) -> ExecutionStatus {
+        match message.command {
+            Command::JOIN(_, _, _) | Command::PRIVMSG(_, _) => {
+                self.send_tell(client, message.source_nickname().unwrap())
+            }
+            _ => ExecutionStatus::Done,
+        }
+    }
+
+    fn execute_threaded(&self, _: &IrcClient, _: &Message) -> Result<(), IrcError> {
+        panic!("Tell should not use threading")
+    }
+
+    fn command(&self, client: &IrcClient, command: PluginCommand) -> Result<(), IrcError> {
+        if command.tokens.is_empty() {
+            return client.send_notice(&command.source, &self.invalid_command(client));
+        }
+
+        match command.tokens[0].as_ref() {
+            "help" => client.send_notice(&command.source, &self.help(client)),
+            _ => match self.tell_command(client, &command) {
+                Ok(msg) => client.send_notice(&command.source, msg),
+                Err(msg) => client.send_notice(&command.source, &msg),
+            },
+        }
+    }
+
+    fn evaluate(&self, _: &IrcClient, _: PluginCommand) -> Result<String, String> {
+        Err(String::from("This Plugin does not implement any commands."))
+    }
+}
+
+use std::fmt;
+impl<T: Database> fmt::Debug for Tell<T> {
+    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
+        write!(f, "Tell {{ ... }}")
+    }
+}