Move word_utils to separate file

[learn-rust.git] / src / word_utils.rs

diff --git a/src/word_utils.rs b/src/word_utils.rs
new file mode 100644 (file)
index 0000000..d087820
--- /dev/null
+++ b/src/word_utils.rs
@@ -0,0 +1,19 @@
+pub fn first_word(s: &str) -> &str {
+    for (i, &item) in s.as_bytes().iter().enumerate() {
+        if item == b' ' {
+            return &s[..i];
+        }
+    }
+
+    s
+}
+
+pub fn second_word(s: &str) -> Option<&str> {
+    for (i, &item) in s.as_bytes().iter().enumerate() {
+        if item == b' ' {
+            return Some(first_word(&s[(i + 1)..]));
+        }
+    }
+
+    return None;
+}