vinezombie 0.3.1

A modular IRCv3 framework
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

//! String transformations, including parsers and casemaps.

mod casemap;
mod escape;
mod trim;

pub use {casemap::*, escape::*, trim::*};

use super::{Transformation, Utf8Policy};

/// Which side of the string to perform an operation on.
#[derive(Clone, Copy, PartialEq, Eq, Debug)]
#[allow(missing_docs)]
pub enum StringSide {
    /// Perform only at the start of the string.
    Start,
    /// Perform only at the end of the string.
    End,
    /// Perform on both ends, starting at the start.
    StartFirst,
    /// Perform on both ends, starting at the end.
    EndFirst,
}

pub(super) unsafe fn map_bytes(
    bytes: &[u8],
    utf8: Utf8Policy,
    mut f: impl FnMut(&u8) -> u8,
) -> Transformation<'_, ()> {
    let mut replace = 0u8;
    let mut idx = 0usize;
    let mut ascii_only = true;
    for byte in bytes {
        let mapped = f(byte);
        ascii_only &= mapped < 128;
        if mapped != *byte {
            replace = mapped;
            break;
        }
        idx += 1;
    }
    let transformed = if idx < bytes.len() {
        let mut new_bytes = Vec::with_capacity(bytes.len());
        new_bytes.extend_from_slice(&bytes[..idx]);
        new_bytes.push(replace);
        idx += 1;
        for byte in &bytes[idx..] {
            let mapped = f(byte);
            ascii_only &= mapped < 128;
            new_bytes.push(mapped);
        }
        new_bytes.into()
    } else {
        bytes.into()
    };
    let utf8 = if ascii_only { Utf8Policy::Valid } else { utf8 };
    Transformation { value: (), transformed, utf8 }
}