Operate on tmp files and prompt for changes at the end
parent
efd1d54c2c
commit
c4d2d3c72b
@ -0,0 +1,123 @@
|
||||
use std::{
|
||||
fs::{self, File},
|
||||
io::Write,
|
||||
mem,
|
||||
path::{Path, PathBuf},
|
||||
};
|
||||
|
||||
use chksum::sha2_256::chksum;
|
||||
use dialoguer::Confirm;
|
||||
|
||||
use miette::{Context, IntoDiagnostic, Result};
|
||||
use prettydiff::diff_lines;
|
||||
|
||||
use tempfile::NamedTempFile;
|
||||
|
||||
use super::FsAccess;
|
||||
|
||||
pub struct BufferedFsAccess {
|
||||
mappings: Vec<(NamedTempFile, PathBuf)>,
|
||||
}
|
||||
|
||||
impl BufferedFsAccess {
|
||||
pub fn new() -> Self {
|
||||
Self {
|
||||
mappings: Vec::new(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl FsAccess for BufferedFsAccess {
|
||||
fn write_all(&mut self, dst: &std::path::Path, buf: &[u8]) -> miette::Result<()> {
|
||||
let mut tmp = tmpfile()?;
|
||||
tmp.write_all(buf).into_diagnostic().with_context(|| {
|
||||
format!(
|
||||
"writing {} bytes to file contents {:?}",
|
||||
buf.len(),
|
||||
tmp.path()
|
||||
)
|
||||
})?;
|
||||
self.mappings.push((tmp, dst.to_owned()));
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn copy(&mut self, src: &std::path::Path, dst: &std::path::Path) -> miette::Result<()> {
|
||||
let tmp = tmpfile()?;
|
||||
fs::copy(src, tmp.path())
|
||||
.into_diagnostic()
|
||||
.with_context(|| format!("copying {src:?} to {:?}", tmp.path()))?;
|
||||
self.mappings.push((tmp, dst.to_owned()));
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn persist(&mut self) -> Result<()> {
|
||||
let mappings = mem::take(&mut self.mappings);
|
||||
let mut drop_list = Vec::new();
|
||||
|
||||
for (tmp, dst) in mappings {
|
||||
if confirm_write(tmp.path(), &dst)? {
|
||||
ensure_parent(dst.parent().unwrap())?;
|
||||
fs::copy(tmp.path(), &dst)
|
||||
.into_diagnostic()
|
||||
.with_context(|| format!("copying {:?} to {dst:?}", tmp.path()))?;
|
||||
log::info!("Updated {dst:?}");
|
||||
} else {
|
||||
log::info!("Skipping {dst:?}");
|
||||
}
|
||||
drop_list.push(tmp);
|
||||
}
|
||||
mem::drop(drop_list);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
fn tmpfile() -> Result<NamedTempFile> {
|
||||
NamedTempFile::new()
|
||||
.into_diagnostic()
|
||||
.context("failed to create tmp file")
|
||||
}
|
||||
|
||||
fn confirm_write(new: &Path, old: &Path) -> Result<bool> {
|
||||
if !old.exists() {
|
||||
return Ok(true);
|
||||
}
|
||||
let f1 = File::open(new)
|
||||
.into_diagnostic()
|
||||
.with_context(|| format!("opening file {new:?}"))?;
|
||||
let f2 = File::open(old)
|
||||
.into_diagnostic()
|
||||
.with_context(|| format!("opening file {old:?}"))?;
|
||||
|
||||
if chksum(&f1).into_diagnostic()?.as_bytes() == chksum(&f2).into_diagnostic()?.as_bytes() {
|
||||
return Ok(true);
|
||||
}
|
||||
|
||||
let cont_new = fs::read_to_string(new)
|
||||
.into_diagnostic()
|
||||
.context("reading a")?;
|
||||
let cont_old = fs::read_to_string(old)
|
||||
.into_diagnostic()
|
||||
.context("reading b")?;
|
||||
|
||||
println!(
|
||||
"\n=== Changes to {old:?}\n{}\n=== End of Changes\n",
|
||||
diff_lines(&cont_old, &cont_new)
|
||||
);
|
||||
Confirm::new()
|
||||
.with_prompt("Do you want to apply these changes?")
|
||||
.interact()
|
||||
.into_diagnostic()
|
||||
}
|
||||
|
||||
fn ensure_parent(parent: &Path) -> Result<(), miette::ErrReport> {
|
||||
if parent.exists() {
|
||||
return Ok(());
|
||||
}
|
||||
log::info!("Creating {parent:?}");
|
||||
fs::create_dir_all(&parent)
|
||||
.into_diagnostic()
|
||||
.with_context(|| format!("Creating directory {parent:?}"))
|
||||
}
|
@ -0,0 +1,16 @@
|
||||
use miette::Result;
|
||||
use std::path::Path;
|
||||
|
||||
mod buffered;
|
||||
pub use buffered::BufferedFsAccess;
|
||||
|
||||
pub trait FsAccess {
|
||||
/// Write all bytes to dst
|
||||
fn write_all(&mut self, dst: &Path, buf: &[u8]) -> Result<()>;
|
||||
|
||||
/// Copy src to dst
|
||||
fn copy(&mut self, src: &Path, dst: &Path) -> Result<()>;
|
||||
|
||||
/// Persist the changes if necessary
|
||||
fn persist(&mut self) -> Result<()>;
|
||||
}
|
Loading…
Reference in New Issue