use std::borrow::Cow;
use bstr::{BStr, BString, ByteSlice};
#[must_use]
pub fn normalize(input: &(impl crate::AsBStr + ?Sized)) -> Cow<'_, BStr> {
normalize_inner(input.as_bstr())
}
fn normalize_inner(mut input: &BStr) -> Cow<'_, BStr> {
if input == "\"\"" {
return Cow::Borrowed(BStr::new(&input[..0]));
}
while input.len() >= 3 && input[0] == b'"' && input[input.len() - 1] == b'"' && input[input.len() - 2] != b'\\' {
input = input[1..input.len() - 1].as_ref();
if input == "\"\"" {
return Cow::Borrowed(BStr::new(&input[..0]));
}
}
if input.find_byteset(br#"\""#).is_none() {
return Cow::Borrowed(input);
}
let mut out: BString = Vec::with_capacity(input.len()).into();
let mut bytes = input.iter().copied();
while let Some(c) = bytes.next() {
match c {
b'\\' => match bytes.next() {
Some(b'n') => out.push(b'\n'),
Some(b't') => out.push(b'\t'),
Some(b'b') => {
out.pop();
}
Some(c) => {
out.push(c);
}
None => break,
},
b'"' => {}
_ => out.push(c),
}
}
Cow::Owned(out)
}