Allow extraneous whitespace at the end of command input.

Cargo.lock

@@ -167,7 +167,7 @@ checksum = "4785bdd1c96b2a846b2bd7cc02e86b6b3dbf14e7e53446c4f54c92a361040822"
 
 [[package]]
 name = "chronicle-dicebot"
-version = "0.4.0"
+version = "0.5.0"
 dependencies = [
  "async-trait",
  "dirs",

src/commands.rs

@@ -1,5 +1,6 @@
 use crate::cofd::dice::DicePool;
 use crate::dice::ElementExpression;
+use crate::parser::trim;
 use crate::roll::Roll;
 
 pub mod parser;
@@ -65,12 +66,17 @@ impl Command for PoolRollCommand {
 /// command was recognized.
 pub fn parse_command(s: &str) -> Result<Option<Box<dyn Command>>, String> {
     match parser::parse_command(s) {
-        Ok((input, result)) => match (input, &result) {
+        Ok((input, command)) => match (input, &command) {
             //This clause prevents bot from spamming messages to itself
             //after executing a previous command.
-            ("", Some(_)) | (_, None) => Ok(result),
+            ("", Some(_)) | (_, None) => Ok(command),
+
+            //Any unconsumed input (except whitespace) is considered a parsing error.
+            (extra, _) => match trim(extra).as_str() {
+                "" => Ok(command),
                 _ => Err(format!("{}: malformed dice expression", s)),
             },
+        },
         Err(err) => Err(err.to_string()),
     }
 }
@@ -87,10 +93,38 @@ mod tests {
     #[test]
     fn roll_malformed_expression_test() {
         assert!(parse_command("!roll 1d20asdlfkj").is_err());
+        assert!(parse_command("!roll 1d20asdlfkj   ").is_err());
     }
 
     #[test]
-    fn roll_dice_pool_expression_test() {
+    fn roll_dice_pool_malformed_expression_test() {
+        assert!(parse_command("!pool 8abc").is_err());
         assert!(parse_command("!pool 8abc    ").is_err());
     }
+
+    #[test]
+    fn pool_whitespace_test() {
+        assert!(parse_command("!pool 8ns3   ")
+            .map(|p| p.is_some())
+            .expect("was error"));
+        assert!(parse_command("   !pool 8ns3")
+            .map(|p| p.is_some())
+            .expect("was error"));
+        assert!(parse_command("   !pool 8ns3   ")
+            .map(|p| p.is_some())
+            .expect("was error"));
+    }
+
+    #[test]
+    fn roll_whitespace_test() {
+        assert!(parse_command("!roll 1d4 + 5d6 -3   ")
+            .map(|p| p.is_some())
+            .expect("was error"));
+        assert!(parse_command("!roll 1d4 + 5d6 -3   ")
+            .map(|p| p.is_some())
+            .expect("was error"));
+        assert!(parse_command("   !roll 1d4 + 5d6 -3   ")
+            .map(|p| p.is_some())
+            .expect("was error"));
+    }
 }

src/parser.rs

@@ -9,3 +9,7 @@ pub fn eat_whitespace(input: &str) -> IResult<&str, &str> {
     let (input, whitespace) = take_while(is_whitespace)(input)?;
     Ok((input, whitespace))
 }
+
+pub fn trim(input: &str) -> String {
+    input.chars().filter(|c| !c.is_whitespace()).collect()
+}