refactor(crate): Introduce Word scruct to handle more info about words

src/content.rs

@@ -6,17 +6,23 @@ use std::{borrow::Cow, fmt, fmt::Display, str::FromStr};
 
 use snafu::prelude::*;
 
+#[derive(Clone, Debug)]
+#[non_exhaustive]
+pub struct Chunk {
+    pub segments: Vec<Segment>,
+}
+
 #[derive(Clone, Debug, PartialEq)]
 #[non_exhaustive]
 pub enum Segment {
     Separator(String),
-    Word(String),
+    Word(Word),
 }
 
-#[derive(Clone, Debug)]
+#[derive(Clone, Debug, PartialEq)]
 #[non_exhaustive]
-pub struct Chunk {
-    pub segments: Vec<Segment>,
+pub struct Word {
+    word: String,
 }
 
 #[derive(Snafu)]
@@ -42,7 +48,9 @@ fn split_chunk(s: &str) -> Chunk {
         if let Some(m) = capture.name("separator") {
             segments.push(Segment::Separator(m.as_str().to_string()));
         } else if let Some(m) = capture.name("word") {
-            segments.push(Segment::Word(m.as_str().to_string()));
+            segments.push(Segment::Word(Word {
+                word: m.as_str().to_owned(),
+            }));
         }
     }
     Chunk { segments }
@@ -79,21 +87,28 @@ impl FromStr for Chunk {
     }
 }
 
+impl Display for Chunk {
+    fn fmt(&self, fmt: &mut fmt::Formatter) -> fmt::Result {
+        for segment in &self.segments {
+            fmt.write_str(segment.to_string().as_ref())?;
+        }
+        Ok(())
+    }
+}
+
 impl Display for Segment {
     fn fmt(&self, fmt: &mut fmt::Formatter) -> fmt::Result {
         let _ = match self {
             Segment::Separator(string) => fmt.write_str(string),
-            Segment::Word(string) => fmt.write_str(string),
+            Segment::Word(word) => fmt.write_str(&word.to_string()),
         };
         Ok(())
     }
 }
 
-impl Display for Chunk {
+impl Display for Word {
     fn fmt(&self, fmt: &mut fmt::Formatter) -> fmt::Result {
-        for segment in &self.segments {
-            fmt.write_str(segment.to_string().as_ref())?;
-        }
+        fmt.write_str(self.word.as_ref());
         Ok(())
     }
 }