pub mod undo {
pub type Error = gix_error::Exn<gix_error::ValidationError>;
}
use std::{borrow::Cow, io::Read};
use bstr::{BStr, BString, ByteSlice};
use gix_error::{ErrorExt, OptionExt, ResultExt, ValidationError};
pub fn undo(input: &BStr) -> Result<(Cow<'_, BStr>, usize), undo::Error> {
if !input.starts_with(b"\"") {
return Ok((input.into(), input.len()));
}
if input.len() < 2 {
return Err(ValidationError::new_with_input("Input must be surrounded by double quotes", input).raise());
}
let original = input.as_bstr();
let mut input = &input[1..];
let mut consumed = 1;
let mut out = BString::default();
fn consume_one_past(input: &mut &BStr, position: usize) -> Result<u8, undo::Error> {
use gix_error::{OptionExt, ValidationError};
*input = input
.get(position + 1..)
.ok_or_raise(|| ValidationError::new_with_input("Unexpected end of input", *input))?
.as_bstr();
let next = *input
.first()
.ok_or_raise(|| ValidationError::new_with_input("Unexpected end of input", *input))?;
*input = input.get(1..).unwrap_or_default().as_bstr();
Ok(next)
}
loop {
match input.find_byteset(b"\"\\") {
Some(position) => {
out.extend_from_slice(&input[..position]);
consumed += position + 1;
match input[position] {
b'"' => break,
b'\\' => {
let next = consume_one_past(&mut input, position)?;
consumed += 1;
match next {
b'n' => out.push(b'\n'),
b'r' => out.push(b'\r'),
b't' => out.push(b'\t'),
b'a' => out.push(7),
b'b' => out.push(8),
b'v' => out.push(0xb),
b'f' => out.push(0xc),
b'"' => out.push(b'"'),
b'\\' => out.push(b'\\'),
b'0' | b'1' | b'2' | b'3' => {
let mut buf = [next; 3];
input
.get(..2)
.ok_or_raise(|| {
ValidationError::new_with_input(
"Unexpected end of input when fetching two more octal bytes",
input,
)
})?
.read_exact(&mut buf[1..])
.expect("impossible to fail as numbers match");
let byte = gix_utils::btoi::to_unsigned_with_radix(&buf, 8).or_raise(|| {
ValidationError::new_with_input("Invalid octal escape value", original)
})?;
out.push(byte);
input = &input[2..];
consumed += 2;
}
_ => {
return Err(ValidationError::new_with_input(
format!("Invalid escaped value {next}"),
original,
)
.raise())
}
}
}
_ => unreachable!("cannot find character that we didn't search for"),
}
}
None => {
out.extend_from_slice(input);
consumed += input.len();
break;
}
}
}
Ok((out.into(), consumed))
}