cosmic-files/src/operation/reader.rs

use cosmic::iced::futures::{channel::mpsc::Sender, executor, SinkExt};
use std::{fs, io, path::Path, sync::Arc};
use tokio::sync::Mutex;

use super::Controller;
use crate::app::Message;

// Special reader just for operations, handling cancel and progress
pub struct OpReader {
    file: fs::File,
    metadata: fs::Metadata,
    current: u64,
    id: u64,
    msg_tx: Arc<Mutex<Sender<Message>>>,
    controller: Controller,
}

impl OpReader {
    pub fn new<P: AsRef<Path>>(
        path: P,
        id: u64,
        msg_tx: Arc<Mutex<Sender<Message>>>,
        controller: Controller,
    ) -> io::Result<Self> {
        let file = fs::File::open(&path)?;
        let metadata = file.metadata()?;
        Ok(Self {
            file,
            metadata,
            current: 0,
            id,
            msg_tx,
            controller,
        })
    }
}

impl io::Read for OpReader {
    fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
        self.controller
            .check()
            .map_err(|err| io::Error::new(io::ErrorKind::Other, err))?;

        let count = self.file.read(buf)?;
        self.current += count as u64;

        let progress = self.current as f32 / self.metadata.len() as f32;
        executor::block_on(async {
            let _ = self
                .msg_tx
                .lock()
                .await
                .send(Message::PendingProgress(self.id, 100.0 * progress))
                .await;
        });

        Ok(count)
    }
}