shrupl/src/appstate.rs

use std::{
    cell::{Ref, RefCell},
    fmt, io,
    time::Duration,
};

use console::style;
use indicatif::{ProgressBar, ProgressStyle};
use log::debug;

use super::{
    cachefile::CacheFile,
    cli::Cli,
    file::{self, FileTrait},
    sharry::{self, Client, ClientError},
};

pub struct AppState {
    current_bar: RefCell<Option<ProgressBar>>,
    buffer: Vec<u8>,

    inner: CacheFile,
}

impl fmt::Debug for AppState {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.debug_struct("AppState")
            .field("inner", &self.inner)
            .finish_non_exhaustive()
    }
}

impl AppState {
    fn new(chunk_size: usize, inner: CacheFile) -> Self {
        Self {
            current_bar: None.into(),
            buffer: vec![0; chunk_size * 1024 * 1024],
            inner,
        }
    }

    pub fn try_resume(args: &Cli) -> Option<Self> {
        let inner = CacheFile::try_resume(args)
            .inspect_err(|e| debug!("could not resume from hash {:?}: {e}", args.get_hash()))
            .ok()?;

        Some(Self::new(args.chunk_size, inner))
    }

    pub fn from_args(args: &Cli, http: &impl Client) -> sharry::Result<Self> {
        let share_id = http.share_create(
            &args.get_uri().endpoint("alias/upload/new"),
            &args.alias,
            args.get_share_request(),
        )?;

        Ok(Self::new(
            args.chunk_size,
            CacheFile::from_args(args, share_id),
        ))
    }

    fn get_or_create_progressbar(&self, uploading: &file::Uploading) -> Ref<'_, ProgressBar> {
        let mut slot = self.current_bar.borrow_mut();
        if slot.is_none() {
            let bar = ProgressBar::new(uploading.get_size())
                .with_style(
                    ProgressStyle::with_template(&format!(
                        concat!(
                            "{{msg:.yellow}}: {{bar:50.cyan/blue}} ",
                            "{{binary_bytes:.magenta}}{}{{binary_total_bytes:.magenta}} ",
                            "({{eta}})",
                        ),
                        style("/").magenta(),
                    ))
                    .unwrap(), // safe as long as the style template is valid
                )
                .with_position(uploading.get_offset())
                .with_message(uploading.get_name().to_owned());

            bar.enable_steady_tick(Duration::from_millis(100));
            *slot = Some(bar);
        }
        drop(slot);

        // unwrap is safe: We just made sure it's `Some`.
        Ref::map(self.current_bar.borrow(), |opt| opt.as_ref().unwrap())
    }

    fn finish_bar(&self) {
        let mut slot = self.current_bar.borrow_mut();
        if let Some(bar) = &*slot {
            bar.finish();
            *slot = None;
        }
    }

    pub fn upload_chunk(&mut self, http: &impl Client) -> sharry::Result<Option<()>> {
        let Some(mut uploading) = self.inner.pop_file(http) else {
            self.inner.share_notify(http).unwrap(); // HACK unwrap

            return Ok(None);
        };

        self.get_or_create_progressbar(&uploading);

        debug!("{uploading} chunk {}", self.buffer.len());

        let chunk = uploading
            .read(&mut self.buffer)
            .map_err(ClientError::from)?;

        http.file_patch(
            chunk.get_patch_uri(),
            self.inner.alias_id(),
            chunk.get_offset(),
            chunk.get_data(),
        )?;

        match uploading.check_eof() {
            Ok(uploading) => {
                let bar = self.get_or_create_progressbar(&uploading);
                bar.set_position(uploading.get_offset());
                // BUG in `indicatif` crate?
                // `set_position` does not force immediate redraw, so we also call `inc_length` here
                bar.inc_length(0);
                drop(bar);

                self.inner.push_file(uploading);
                Ok(Some(()))
            }
            Err(path) => {
                debug!("Finished {:?}!", path.display());
                self.finish_bar();

                Ok(self.inner.has_file().then_some(()))
            }
        }
    }

    pub fn file_names(&self) -> Vec<&str> {
        self.inner.file_names()
    }

    pub fn save(&self) -> io::Result<()> {
        self.inner.save()
    }

    pub fn clear(self) -> io::Result<()> {
        self.inner.clear()
    }
}