allow whitespace in spelling codec

elliotcubit · Jul 30, 2021 · 54b0deb · 54b0deb
1 parent 0f50e83
commit 54b0deb
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 14 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "rc"
-version = "0.3.0"
+version = "0.3.1"
 authors = ["Elliot Cubit <[email protected]>"]
 description = "Text encoding swiss army knife"
 edition = "2018"

diff --git a/src/codecs/spelling.rs b/src/codecs/spelling.rs
@@ -23,18 +23,22 @@ impl Codec for SpellingCodec {
 
     fn encode(&self, data: Vec<u8>) -> Result<String, Error> {
         match String::from_utf8(data.clone()) {
-            Ok(s) if s.chars().all(|c| c.is_ascii_alphabetic()) => Ok(s
-                .trim()
-                .to_ascii_lowercase()
-                .split_ascii_whitespace()
-                .map(|word| {
-                    word.chars()
-                        .map(Self::char_to_word)
-                        .collect::<Vec<String>>()
-                        .join(" ")
-                })
-                .collect::<Vec<String>>()
-                .join(" ")),
+            Ok(s)
+                if s.chars()
+                    .all(|c| c.is_ascii_alphabetic() || c.is_ascii_whitespace()) =>
+            {
+                Ok(s.trim()
+                    .to_ascii_lowercase()
+                    .split_ascii_whitespace()
+                    .map(|word| {
+                        word.chars()
+                            .map(Self::char_to_word)
+                            .collect::<Vec<String>>()
+                            .join(" ")
+                    })
+                    .collect::<Vec<String>>()
+                    .join(" "))
+            }
             Ok(_) => Err(Error::new("input data is not ascii".to_string())),
             Err(_) => Err(Error::new("input data is not utf8".to_string())),
         }