Remove whitespace in result text

1 files changed, 21 insertions, 1 deletions

diff --git a/src/hooks/wolfram_alpha.rs b/src/hooks/wolfram_alpha.rs
index 5584935..1e7b6d8 100644
--- a/src/hooks/wolfram_alpha.rs
+++ b/src/hooks/wolfram_alpha.rs
@@ -32,6 +32,16 @@ struct SubPod {
     plaintext: String,
 }
 
+fn clean_result_text(text: &str) -> String {
+    text
+        // Remove newlines
+        .replace("\n", "; ")
+        // Remove multiple whitespace
+        .split_whitespace()
+        .collect::<Vec<&str>>()
+        .join(" ")
+}
+
 /// Reduces all 'pod' plaintexts to a single string.
 /// Same as gonzobot does it.
 fn to_single_string(wa_res: WaResponse) -> String {
@@ -44,7 +54,7 @@ fn to_single_string(wa_res: WaResponse) -> String {
             let subpod_texts = pod
                 .subpods
                 .iter()
-                .map(|subpod| subpod.plaintext.clone())
+                .map(|subpod| clean_result_text(&subpod.plaintext))
                 .collect::<Vec<String>>()
                 .join(", ");
 
@@ -143,6 +153,8 @@ pub async fn wa(bot: &crate::Bot, msg: Message) -> Result<()> {
 #[cfg(test)]
 mod tests {
 
+    use crate::hooks::wolfram_alpha::clean_result_text;
+
     use super::{get_input_query, get_wa_user_short_url, wa_query};
     use anyhow::{Error, Result};
     use mockito::{self, Matcher};
@@ -221,6 +233,14 @@ mod tests {
         Ok(())
     }
 
+    #[test]
+    fn test_clean_result_text() {
+        assert_eq!(
+            clean_result_text("Newlines\nand  multiple\n\n whitespace   is removed."),
+            "Newlines; and multiple; ; whitespace is removed.",
+        )
+    }
+
     #[tokio::test]
     async fn test_query_result_parsing() -> Result<(), Error> {
         let body = include_str!("../../tests/resources/wolfram_alpha_api_response.json");