shrupl/src/sharry/file/checked.rs

use std::{
    ffi::OsStr,
    fs, io,
    path::{Path, PathBuf},
};

use log::debug;
use serde::{Deserialize, Serialize};
use ureq::http::{HeaderValue, StatusCode};

use super::{Alias, FileUploading, Share, SharryAlias, SharryFile};

#[derive(Debug, Clone, Hash, Serialize, Deserialize, PartialEq, Eq, PartialOrd, Ord)]
pub struct FileChecked {
    path: PathBuf,
}

impl FileChecked {
    pub fn new(value: impl AsRef<Path>) -> io::Result<Self> {
        let meta = fs::metadata(&value)?;
        if meta.is_file() {
            Ok(Self {
                path: fs::canonicalize(&value)?,
            })
        } else {
            Err(io::Error::new(
                io::ErrorKind::InvalidInput,
                "Not a regular file",
            ))
        }
    }

    pub fn start_upload(
        self,
        http: &ureq::Agent,
        alias: &Alias,
        share: &Share,
    ) -> io::Result<FileUploading> {
        let size = self.get_size();

        let res = {
            let endpoint = alias.get_endpoint(format!("alias/upload/{}/files/tus", share.id));

            (http.post(endpoint))
                .sharry_header(alias)
                .header("Sharry-File-Name", self.get_name())
                .header("Upload-Length", size)
                .send_empty()
                .map_err(ureq::Error::into_io)?
        };

        if res.status() != StatusCode::CREATED {
            return Err(io::Error::other("unexpected response status"));
        }

        let location = (res.headers().get("Location"))
            .ok_or_else(|| io::Error::other("Location header not found"))?
            .to_str()
            .map_err(|_| io::Error::other("Location header invalid"))?
            .to_string();

        debug!("patch uri: {location}");

        Ok(FileUploading::new(self.path, size, location))
    }
}

impl<'t> SharryFile<'t> for FileChecked {
    /// get a reference to the file's name
    ///
    /// Uses `SharryFile::extract_file_name`, which may **panic**!
    fn get_name(&'t self) -> &'t str {
        <Self as SharryFile>::extract_file_name(&self.path)
    }

    fn get_size(&self) -> u64 {
        fs::metadata(&self.path).unwrap().len()
    }
}