basekit 0.4.0

Universal library for encoding in various bases
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

use super::super::config::Base64EncodeConfig;
use super::encode_into_slice;
use super::output::Base64EncodeOutput;
use crate::utils::init_vec_with;

#[inline]
pub fn encode(config: &Base64EncodeConfig, data: impl AsRef<[u8]>) -> Base64EncodeOutput {
    let data = data.as_ref();

    if data.is_empty() {
        return Base64EncodeOutput { inner: Vec::new() };
    }

    let full_groups_count = data.len() / 3;
    let remainder = data.len() % 3;
    let output_len = full_groups_count * 4
        + match (remainder, config.padding.is_some()) {
            (0, _) => 0,
            (1, true) => 4,
            (1, false) => 2,
            (2, true) => 4,
            (2, false) => 3,
            _ => unreachable!(),
        };

    let full_groups_src = if full_groups_count > 0 {
        Some(&data[..full_groups_count * 3])
    } else {
        None
    };
    let tail_src = if remainder > 0 {
        Some(&data[full_groups_count * 3..])
    } else {
        None
    };

    let output = unsafe {
        init_vec_with(output_len, |buf| {
            encode_into_slice(config, buf, full_groups_src, tail_src)
        })
        .unwrap()
    };

    Base64EncodeOutput { inner: output }
}