overhaul random module to use regex instead of prefix match

src/modules/random.rs

@@ -5,20 +5,27 @@ use std::sync::Arc;
 use std::sync::mpsc::Receiver;
 use transformable_channels::mpsc::ExtSender;
 
-use helpers::command::split_command;
+use regex;
+use regex::Regex;
+
 use event::{Event, Envelope};
 use rand;
 
 pub struct RandomModule {
-    prefix: String,
+    pattern: Regex,
     responses: Vec<String>
 }
 
 impl RandomModule {
     pub fn new (_: &Table, configuration: &Table) -> Box<EventLoop> {
-        let prefix = configuration.get("prefix")
+        let pattern = configuration.get("pattern")
                                    .and_then(|value| value.as_str())
-                                  .unwrap_or("?random");
+                                   .map(String::from)
+                                   .or_else(|| configuration.get("prefix")
+                                                            .and_then(|value| value.as_str())
+                                                            .map(|value| format!("^{}", regex::escape(value))))
+                                   .and_then(|value| Regex::new(&value).ok())
+                                   .expect("Invalid value for pattern");
 
         let responses = configuration.get("responses")
                                      .and_then(|value| value.as_array())
@@ -29,7 +36,7 @@ impl RandomModule {
                                      .collect();
 
         Box::new(RandomModule {
-            prefix: String::from(prefix),
+            pattern: pattern,
             responses: responses
         })
     }
@@ -42,20 +49,13 @@ impl EventLoop for RandomModule {
         loop {
             match receiver.recv() {
                 Ok(envelope) => {
-                    match envelope.event {
+                    if let Event::Message { ref message } = envelope.event {
-                        Event::Message { ref message } => {
                         debug!("Received message from module {:?}... {:?}", envelope.from, message.content);
-                            match split_command(&message.content) {
+                        if let Some(captures) = self.pattern.captures(&message.content) {
-                                Some((command, _)) => {
+                            let mut response = String::new();
-                                    if command == self.prefix {
+                            captures.expand(&rand::sample(&mut rng, &self.responses, 1)[0], &mut response);
-                                        let response = &rand::sample(&mut rng, &self.responses, 1)[0][..];
+                            message.reply(&response);
-                                        message.reply(response);
                         }
-                                },
-                                _ => {}
-                            }
-                        }
-                        _ => ()
                     }
                 }
                 Err(error) => { error!("Error {:?}", error) }

tenquestionmarks.toml

@@ -96,3 +96,9 @@ parents = ["stdin", "discord"]
 [logger]
 parents = ["stdin", "discord"]
 filters = [{ username = "Dave" }, { username = "Kevin" }]
+
+[icced]
+type = "random"
+parents = ["stdin", "discord"]
+pattern = "\\bicc?ed?\\b"
+responses = ["Did some carbon-based lifeform just say **I C E**?"]