brk_string_wizard 0.4.0

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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71

use crate::{CowStr, MagicString, chunk::EditOptions};

#[derive(Debug, Default, Clone)]
pub struct UpdateOptions {
  /// `true` will store the original content in the `name` field of the generated sourcemap.
  pub keep_original: bool,

  /// `true` will clear the `intro` and `outro` for the corresponding range.
  pub overwrite: bool,
}

impl<'text> MagicString<'text> {
  /// A shorthand for `update_with(start, end, content, Default::default())`;
  pub fn update(
    &mut self,
    start: usize,
    end: usize,
    content: impl Into<CowStr<'text>>,
  ) -> &mut Self {
    self.update_with(start, end, content, Default::default())
  }

  pub fn update_with(
    &mut self,
    start: usize,
    end: usize,
    content: impl Into<CowStr<'text>>,
    opts: UpdateOptions,
  ) -> &mut Self {
    self.inner_update_with(start, end, content.into(), opts, true);
    self
  }

  // --- private

  pub(super) fn inner_update_with(
    &mut self,
    start: usize,
    end: usize,
    content: CowStr<'text>,
    opts: UpdateOptions,
    panic_if_start_equal_end: bool,
  ) -> &mut Self {
    if panic_if_start_equal_end && start == end {
      panic!("Cannot overwrite a zero-length range – use append_left or prepend_right instead")
    }
    assert!(start < end);
    self.split_at(start);
    self.split_at(end);

    let start_idx = self.chunk_by_start.get(&start).copied().unwrap();
    let end_idx = self.chunk_by_end.get(&end).copied().unwrap();

    let start_chunk = &mut self.chunks[start_idx];
    start_chunk
      .edit(content, EditOptions { overwrite: opts.overwrite, store_name: opts.keep_original });

    let mut rest_chunk_idx = if start_idx != end_idx {
      start_chunk.next.unwrap()
    } else {
      return self;
    };

    while rest_chunk_idx != end_idx {
      let rest_chunk = &mut self.chunks[rest_chunk_idx];
      rest_chunk.edit("".into(), Default::default());
      rest_chunk_idx = rest_chunk.next.unwrap();
    }
    self
  }
}