asp/keep
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: Recalculate digest if digest_type is set but digest_value is missing

Browse Source
This commit is contained in:
Andrew Phillips (aider)
2025-05-13 18:26:42 -03:00
parent 62aeb05e76
commit 9a99bbd8bc
1 changed files with 32 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
src/modes/update.rs
View File

@@ -49,8 +49,8 @@ pub fn mode_update(
let compression_type = CompressionType::from_str(&item.compression)?;
let compression_engine = get_compression_engine(compression_type)
.expect("Unable to get compression engine");
let size = compression_engine.size(item_path)? as i64;
item.size = Some(size);
let size = compression_engine.size(item_path.clone())?;
item.size = Some(size as i64);
db::update_item(conn, item.clone())?;
} else {
debug!(
@@ -60,6 +60,36 @@ pub fn mode_update(
}
}
// If digest is defined and not None, but digest_value is missing, recalculate it
if item.digest_type != "None" && item.digest_value.is_none() {
debug!("MAIN: Recalculating digest for item {:?}", item.id);
// Get the digest type
let digest_type = DigestType::from_str(&item.digest_type)?;
let mut digest_engine = get_digest_engine(digest_type);
// Read the file using the compression engine
let compression_type = CompressionType::from_str(&item.compression)?;
let compression_engine = get_compression_engine(compression_type)
.expect("Unable to get compression engine");
let mut reader = compression_engine.open(item_path.clone())?;
let mut buffer = [0; 4096];
loop {
let n = reader.read(&mut buffer)?;
if n == 0 {
break;
}
digest_engine.update(&buffer[..n]);
}
// Finalize the digest and update the item
let digest_value = digest_engine.finalize()?;
item.digest_value = Some(digest_value);
db::update_item(conn, item.clone())?;
}
if !args.item.meta.is_empty() {
debug!("MAIN: Updating item meta");
for kv in args.item.meta.iter() {