brk_string_wizard 1.0.0-rc.2

manipulate string like a wizard
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

use crate::{CowStr, UpdateOptions};

use super::MagicString;

#[derive(Debug)]
pub struct ReplaceOptions {
  /// The maximum number of times to replace the pattern. Default is `1`.
  pub count: usize,
  /// This will store the original content in the `name` field of the generated sourcemap.
  ///
  /// Default is `false`.
  pub store_original_in_sourcemap: bool,
}

impl Default for ReplaceOptions {
  fn default() -> Self {
    Self { count: 1, store_original_in_sourcemap: false }
  }
}

impl<'text> MagicString<'text> {
  pub fn replace(&mut self, from: &str, to: impl Into<CowStr<'text>>) -> Result<&mut Self, String> {
    self.replace_with(from, to, Default::default())
  }

  pub fn replace_all(
    &mut self,
    from: &str,
    to: impl Into<CowStr<'text>>,
  ) -> Result<&mut Self, String> {
    self.replace_with(from, to, ReplaceOptions { count: usize::MAX, ..Default::default() })
  }

  pub fn replace_with(
    &mut self,
    from: &str,
    to: impl Into<CowStr<'text>>,
    options: ReplaceOptions,
  ) -> Result<&mut Self, String> {
    let to: CowStr<'text> = to.into();
    let matches = memchr::memmem::find_iter(self.source.as_bytes(), from.as_bytes())
      .take(options.count)
      .map(|start| (start as u32, (start + from.len()) as u32))
      .collect::<Vec<_>>();
    for (match_start, end) in matches {
      self.update_with(
        match_start,
        end,
        to.clone(),
        UpdateOptions { overwrite: true, keep_original: options.store_original_in_sourcemap },
      )?;
    }

    Ok(self)
  }
}