rust_assembler/str/
replace_str.rs

1
2/// Replaces multiple `str` at once, avoiding intermediate allocations. It returns a new `String` containing the result. Values passed by value keep ownership.
3/// 
4/// # Examples
5/// 
6/// ```
7/// 
8/// let s = "Ciao bello";
9/// 
10/// let s_replaced = replace_str!( s;
11///     ["ia", "u"],
12///     ["ll", ""]
13/// );
14/// 
15/// assert_eq!(s_replaced, "Cuo beo");
16#[macro_export]
17macro_rules! replace_str {
18    ( $string:expr; $( $pair:expr ),* ) => ({
19        let mut new_string: String = String::new();
20
21        let mut to_skip = 0;
22        let mut next_start = 0;
23        
24        'outer: for i in 0..$string.len() {
25            if to_skip > 0 {
26                to_skip -= 1;
27                continue 'outer
28            }
29
30            $(
31                unsafe {
32                    // It's the element that must be replaced.
33                    let elm_0 = *$pair.get_unchecked(0);
34                    
35                    let elm_0_len = elm_0.len();
36                    // Avoids out of bounds bugs.
37                    if i + elm_0_len > $string.len() { break 'outer }
38
39                    // First compares the first character..
40                    if *elm_0.as_bytes().get_unchecked(0) == *$string.as_bytes().get_unchecked(i) {
41                        // .. then the whole string.
42                        if *elm_0 == *$string.get_unchecked(i..i + elm_0_len) {
43                            extend_string!( new_string;
44                                $string.get_unchecked(next_start..i),
45                                $pair.get_unchecked(1)
46                            );
47                            next_start = i + elm_0_len;
48                            to_skip += elm_0_len - 1;
49                        }
50                    }
51                }
52            )*
53            
54        }
55        
56        unsafe {
57            new_string.push_str($string.get_unchecked(next_start..));
58        }
59
60        new_string
61    });
62}