use std::{ fmt, fs, io::{self, Read, Seek, SeekFrom}, path::PathBuf, }; use serde::{Deserialize, Serialize}; use super::{Chunk, FileTrait}; #[derive(Serialize, Deserialize, Debug)] pub struct Uploading { path: PathBuf, size: u64, patch_uri: String, offset: u64, } impl fmt::Display for Uploading { fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { write!( f, "Uploading {:?} ({}/{})", self.path.display(), self.offset, self.size ) } } impl Uploading { pub(super) fn new(path: PathBuf, size: u64, patch_uri: String) -> Self { Self { path, size, patch_uri, offset: 0, } } pub fn get_offset(&self) -> u64 { self.offset } pub fn read<'t>(&'t mut self, buf: &'t mut [u8]) -> io::Result> { let mut f = fs::File::open(&self.path)?; f.seek(SeekFrom::Start(self.offset))?; let read_len = f.read(buf)?; let chunk = Chunk::new(&buf[..read_len], &self.patch_uri, self.offset); self.offset += chunk.get_length(); Ok(chunk) } pub fn check_eof(self) -> Result { if self.offset < self.size { Ok(self) } else { Err(self.path) } } } impl<'t> FileTrait<'t> for Uploading { /// get a reference to the file's name /// /// Uses `SharryFile::extract_file_name`, which may **panic**! fn get_name(&'t self) -> &'t str { ::extract_file_name(&self.path) } fn get_size(&self) -> u64 { self.size } }