Operate on tmp files and prompt for changes at the end
parent
efd1d54c2c
commit
c4d2d3c72b
GPG Key ID:
7E6D18B61C8D2F4B
@ -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