Support separator

Author: hyf0

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "string_wizard"
-version = "0.0.2"
+version = "0.0.3"
 edition = "2021"
 license = "MIT"
 description = "manipulate string like wizards"

src/joiner.rs

@@ -1,12 +1,24 @@
 use crate::{MagicString, CowStr};
 
+pub struct JoinerOptions {
+    pub separator: Option<String>,
+}
+
 #[derive(Default)]
 pub struct Joiner<'s> {
     sources: Vec<MagicString<'s>>,
+    separator: Option<String>,
 }
 
 impl<'s> Joiner<'s> {
     // --- public
+    pub fn with_options(options: JoinerOptions) -> Self {
+        Self {
+            separator: options.separator,
+            ..Default::default()
+        }
+    }
+
     pub fn append(&mut self, source: MagicString<'s>) -> &mut Self {
         self.sources.push(source);
         self
@@ -32,7 +44,15 @@ impl<'s> Joiner<'s> {
     // --- private
 
     fn fragments(&'s self) -> impl Iterator<Item = &'s str> {
-        self.sources.iter().flat_map(|c| c.fragments())
+        let mut iter = self
+            .sources
+            .iter()
+            .flat_map(|c| self.separator.as_ref().map(|s| s.as_str()).into_iter().chain(c.fragments()));
+        // Drop the first separator
+        if self.separator.is_some() {
+            iter.next();
+        }
+        iter
     }
 
 

src/lib.rs

@@ -10,7 +10,7 @@ type CowStr<'s> = Cow<'s, str>;
 use chunk::Chunk;
 use index_vec::IndexVec;
 
-pub use crate::{magic_string::{MagicString, MagicStringOptions}, joiner::Joiner};
+pub use crate::{magic_string::{MagicString, MagicStringOptions}, joiner::{Joiner, JoinerOptions}};
 
 
 index_vec::define_index_type! {
@@ -22,7 +22,3 @@ type ChunkVec<'s> = IndexVec<ChunkIdx, Chunk<'s>>;
 index_vec::define_index_type! {
     struct SourceIdx = u32;
 }
-
-
-
-pub struct Bundle {}

tests/joiner.rs

@@ -1,4 +1,4 @@
-use string_wizard::{Joiner, MagicString};
+use string_wizard::{Joiner, MagicString, JoinerOptions};
 mod append {
     use super::*;
 
@@ -9,4 +9,15 @@ mod append {
         j.append_raw("123").append_raw("456");
         assert_eq!(j.join(), "*123456");
     }
+}
+
+#[test]
+fn separator() {
+    let mut j = Joiner::with_options(JoinerOptions { separator: Some(",".to_string()) });
+    j.append_raw("123");
+    assert_eq!(j.join(), "123");
+    j.append_raw("123");
+    assert_eq!(j.join(), "123,123");
+    j.append_raw("123");
+    assert_eq!(j.join(), "123,123,123");
 }