use std::convert::{TryFrom, TryInto}; use git_hash::ObjectId; use git_object::bstr::BString; use nom::{ bytes::complete::{tag, take_while}, combinator::{map, opt}, sequence::terminated, IResult, }; use crate::{ parse::{hex_hash, newline}, store_impl::file::loose::Reference, FullName, Target, }; enum MaybeUnsafeState { Id(ObjectId), UnvalidatedPath(BString), } /// The error returned by [`Reference::try_from_path()`]. #[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: git_validate::reference::name::Error, path: BString, }, } impl TryFrom for Target { type Error = Error; fn try_from(v: MaybeUnsafeState) -> Result { Ok(match v { MaybeUnsafeState::Id(id) => Target::Peeled(id), MaybeUnsafeState::UnvalidatedPath(name) => Target::Symbolic(match git_validate::refname(name.as_ref()) { Ok(_) => FullName(name), Err(err) => { return Err(Error::RefnameValidation { source: err, path: name, }) } }), }) } } impl Reference { /// Create a new reference of the given `parent` store with `relative_path` service as unique identifier /// at which the `path_contents` was read to obtain the refs value. pub fn try_from_path(name: FullName, path_contents: &[u8]) -> Result { Ok(Reference { name, target: parse(path_contents) .map_err(|_| Error::Parse { content: path_contents.into(), })? .1 .try_into()?, }) } } fn parse(bytes: &[u8]) -> IResult<&[u8], MaybeUnsafeState> { let is_space = |b: u8| b == b' '; if let (path, Some(_ref_prefix)) = opt(terminated(tag("ref: "), take_while(is_space)))(bytes)? { map( terminated(take_while(|b| b != b'\r' && b != b'\n'), opt(newline)), |path| MaybeUnsafeState::UnvalidatedPath(path.into()), )(path) } else { map(terminated(hex_hash, opt(newline)), |hex| { MaybeUnsafeState::Id(ObjectId::from_hex(hex).expect("prior validation")) })(bytes) } }