gix_odb/store_impls/loose/
write.rsuse std::{fs, io, io::Write, path::PathBuf};
use gix_features::{hash, zlib::stream::deflate};
use gix_object::WriteTo;
use tempfile::NamedTempFile;
use super::Store;
use crate::store_impls::loose;
#[derive(thiserror::Error, Debug)]
#[allow(missing_docs)]
pub enum Error {
#[error("Could not {message} '{path}'")]
Io {
source: io::Error,
message: &'static str,
path: PathBuf,
},
#[error("An IO error occurred while writing an object")]
IoRaw(#[from] io::Error),
#[error("Could not turn temporary file into persisted file at '{target}'")]
Persist {
source: tempfile::PersistError,
target: PathBuf,
},
}
impl crate::traits::Write for Store {
fn write(&self, object: &dyn WriteTo) -> Result<gix_hash::ObjectId, crate::write::Error> {
let mut to = self.dest()?;
to.write_all(&object.loose_header()).map_err(|err| Error::Io {
source: err,
message: "write header to tempfile in",
path: self.path.to_owned(),
})?;
object.write_to(&mut to).map_err(|err| Error::Io {
source: err,
message: "stream all data into tempfile in",
path: self.path.to_owned(),
})?;
to.flush().map_err(Box::new)?;
Ok(self.finalize_object(to).map_err(Box::new)?)
}
fn write_buf(&self, kind: gix_object::Kind, from: &[u8]) -> Result<gix_hash::ObjectId, crate::write::Error> {
let mut to = self.dest().map_err(Box::new)?;
to.write_all(&gix_object::encode::loose_header(kind, from.len() as u64))
.map_err(|err| Error::Io {
source: err,
message: "write header to tempfile in",
path: self.path.to_owned(),
})?;
to.write_all(from).map_err(|err| Error::Io {
source: err,
message: "stream all data into tempfile in",
path: self.path.to_owned(),
})?;
to.flush()?;
Ok(self.finalize_object(to)?)
}
fn write_stream(
&self,
kind: gix_object::Kind,
size: u64,
mut from: &mut dyn io::Read,
) -> Result<gix_hash::ObjectId, crate::write::Error> {
let mut to = self.dest().map_err(Box::new)?;
to.write_all(&gix_object::encode::loose_header(kind, size))
.map_err(|err| Error::Io {
source: err,
message: "write header to tempfile in",
path: self.path.to_owned(),
})?;
io::copy(&mut from, &mut to)
.map_err(|err| Error::Io {
source: err,
message: "stream all data into tempfile in",
path: self.path.to_owned(),
})
.map_err(Box::new)?;
to.flush().map_err(Box::new)?;
Ok(self.finalize_object(to)?)
}
}
type CompressedTempfile = deflate::Write<NamedTempFile>;
impl Store {
pub fn object_path(&self, id: &gix_hash::oid) -> PathBuf {
loose::hash_path(id, self.path.clone())
}
}
impl Store {
fn dest(&self) -> Result<hash::Write<CompressedTempfile>, Error> {
#[cfg_attr(not(unix), allow(unused_mut))]
let mut builder = tempfile::Builder::new();
#[cfg(unix)]
{
use std::os::unix::fs::PermissionsExt;
let perms = std::fs::Permissions::from_mode(0o444);
builder.permissions(perms);
}
Ok(hash::Write::new(
deflate::Write::new(builder.tempfile_in(&self.path).map_err(|err| Error::Io {
source: err,
message: "create named temp file in",
path: self.path.to_owned(),
})?),
self.object_hash,
))
}
fn finalize_object(
&self,
hash::Write { hash, inner: file }: hash::Write<CompressedTempfile>,
) -> Result<gix_hash::ObjectId, Error> {
let id = gix_hash::ObjectId::from(hash.digest());
let object_path = loose::hash_path(&id, self.path.clone());
let object_dir = object_path
.parent()
.expect("each object path has a 1 hex-bytes directory");
if let Err(err) = fs::create_dir(object_dir) {
match err.kind() {
io::ErrorKind::AlreadyExists => {}
_ => return Err(err.into()),
}
}
let file = file.into_inner();
let res = file.persist(&object_path);
#[cfg(windows)]
if let Err(err) = &res {
if err.error.kind() == std::io::ErrorKind::PermissionDenied
|| err.error.kind() == std::io::ErrorKind::AlreadyExists
{
return Ok(id);
}
}
res.map_err(|err| Error::Persist {
source: err,
target: object_path,
})?;
Ok(id)
}
}