utils.rs
83 lines
| 1.9 KiB
| application/rls-services+xml
|
RustLexer
|
r42996 | // utils module | ||
// | ||||
// Copyright 2019 Raphaël Gomès <rgomes@octobus.net> | ||||
// | ||||
// This software may be used and distributed according to the terms of the | ||||
// GNU General Public License version 2 or any later version. | ||||
//! Contains useful functions, traits, structs, etc. for use in core. | ||||
|
r42828 | pub mod files; | ||
|
r42829 | /// Replaces the `from` slice with the `to` slice inside the `buf` slice. | ||
/// | ||||
/// # Examples | ||||
/// | ||||
/// ``` | ||||
/// use crate::hg::utils::replace_slice; | ||||
/// let mut line = b"I hate writing tests!".to_vec(); | ||||
/// replace_slice(&mut line, b"hate", b"love"); | ||||
/// assert_eq!( | ||||
/// line, | ||||
/// b"I love writing tests!".to_vec() | ||||
|
r43109 | /// ); | ||
|
r42829 | /// ``` | ||
|
r42828 | pub fn replace_slice<T>(buf: &mut [T], from: &[T], to: &[T]) | ||
where | ||||
T: Clone + PartialEq, | ||||
{ | ||||
|
r42830 | if buf.len() < from.len() || from.len() != to.len() { | ||
|
r42828 | return; | ||
} | ||||
for i in 0..=buf.len() - from.len() { | ||||
if buf[i..].starts_with(from) { | ||||
buf[i..(i + from.len())].clone_from_slice(to); | ||||
} | ||||
} | ||||
} | ||||
pub trait SliceExt { | ||||
|
r42829 | fn trim_end(&self) -> &Self; | ||
fn trim_start(&self) -> &Self; | ||||
|
r42828 | fn trim(&self) -> &Self; | ||
} | ||||
fn is_not_whitespace(c: &u8) -> bool { | ||||
!(*c as char).is_whitespace() | ||||
} | ||||
impl SliceExt for [u8] { | ||||
fn trim_end(&self) -> &[u8] { | ||||
if let Some(last) = self.iter().rposition(is_not_whitespace) { | ||||
&self[..last + 1] | ||||
} else { | ||||
&[] | ||||
} | ||||
} | ||||
|
r42829 | fn trim_start(&self) -> &[u8] { | ||
if let Some(first) = self.iter().position(is_not_whitespace) { | ||||
&self[first..] | ||||
} else { | ||||
&[] | ||||
} | ||||
} | ||||
/// ``` | ||||
/// use hg::utils::SliceExt; | ||||
/// assert_eq!( | ||||
/// b" to trim ".trim(), | ||||
/// b"to trim" | ||||
/// ); | ||||
/// assert_eq!( | ||||
/// b"to trim ".trim(), | ||||
/// b"to trim" | ||||
/// ); | ||||
/// assert_eq!( | ||||
/// b" to trim".trim(), | ||||
/// b"to trim" | ||||
/// ); | ||||
/// ``` | ||||
fn trim(&self) -> &[u8] { | ||||
self.trim_start().trim_end() | ||||
} | ||||
|
r42828 | } | ||