Kieran/route96
1
0
Fork 0
mirror of https://github.com/v0l/route96.git synced 2025-06-17 08:38:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: probe files for no-compress

Browse Source
This commit is contained in:
kieran
2024-05-29 14:53:11 +01:00
parent 6ed693f13a
commit 7767d21ccf
4 changed files with 119 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
src/filesystem.rs
View File

@ -13,7 +13,7 @@ use tokio::fs::File;
use tokio::io::{AsyncRead, AsyncReadExt, AsyncSeekExt};
use crate::db::{FileLabel, FileUpload};
use crate::processing::{compress_file, FileProcessorResult};
use crate::processing::{compress_file, FileProcessorResult, probe_file, ProbeResult, ProbeStream};
use crate::processing::labeling::label_frame;
use crate::settings::Settings;
@ -148,7 +148,34 @@ impl FileStore {
},
});
}
} else if let FileProcessorResult::Probe(p) = probe_file(tmp_path.clone())? {
let video_stream_size = p.streams.iter().find_map(|s| match s {
ProbeStream::Video { width, height, .. } => Some((width, height)),
_ => None
});
let n = file.metadata().await?.len();
let hash = FileStore::hash_file(&mut file).await?;
return Ok(FileSystemResult {
path: tmp_path,
upload: FileUpload {
id: hash,
name: "".to_string(),
size: n,
created: Utc::now(),
mime_type: mime_type.to_string(),
width: match video_stream_size {
Some((w, _h)) => Some(*w),
_ => None
},
height: match video_stream_size {
Some((_w, h)) => Some(*h),
_ => None
},
..Default::default()
},
});
}
let n = file.metadata().await?.len();
let hash = FileStore::hash_file(&mut file).await?;
Ok(FileSystemResult {
@ -158,6 +185,7 @@ impl FileStore {
name: "".to_string(),
size: n,
created: Utc::now(),
mime_type: mime_type.to_string(),
..Default::default()
},
})