use gix_hash::ObjectId;
use gix_object::bstr::BString;
use crate::{parse::hex_hash, store_impl::file::loose::Reference, FullName, Target};
enum MaybeUnsafeState {
Id(ObjectId),
UnvalidatedPath(BString),
}
#[derive(Debug, thiserror::Error)]
#[allow(missing_docs)]
pub enum Error {
#[error("{content:?} could not be parsed")]
Parse { content: BString },
#[error("The path {path:?} to a symbolic reference within a ref file is invalid")]
RefnameValidation {
source: gix_validate::reference::name::Error,
path: BString,
},
}
impl TryFrom<MaybeUnsafeState> for Target {
type Error = Error;
fn try_from(v: MaybeUnsafeState) -> Result<Self, Self::Error> {
Ok(match v {
MaybeUnsafeState::Id(id) => Target::Object(id),
MaybeUnsafeState::UnvalidatedPath(name) => {
Target::Symbolic(match gix_validate::reference::name(name.as_ref()) {
Ok(_) => FullName(name),
Err(err) => {
return Err(Error::RefnameValidation {
source: err,
path: name,
})
}
})
}
})
}
}
impl Reference {
pub fn try_from_path(name: FullName, path_contents: &[u8], hash_kind: gix_hash::Kind) -> Result<Self, Error> {
Ok(Reference {
name,
target: parse(path_contents, hash_kind)
.map_err(|_| Error::Parse {
content: path_contents.into(),
})?
.try_into()?,
})
}
}
fn parse(mut i: &[u8], hash_kind: gix_hash::Kind) -> Result<MaybeUnsafeState, ()> {
if let Some(rest) = i.strip_prefix(b"ref: ") {
i = rest;
while i.first() == Some(&b' ') {
i = &i[1..];
}
let path_end = i.iter().position(|b| *b == b'\r' || *b == b'\n').unwrap_or(i.len());
let path = i[..path_end].into();
Ok(MaybeUnsafeState::UnvalidatedPath(path))
} else {
let hex = hex_hash(&mut i, hash_kind)?;
if i.first().is_some_and(u8::is_ascii_hexdigit) {
return Err(());
}
Ok(MaybeUnsafeState::Id(ObjectId::from_hex(hex).expect("prior validation")))
}
}