shrupl/src/cli.rs

use std::{
    fmt,
    hash::{DefaultHasher, Hash, Hasher},
    time::Duration,
};

use clap::{
    Parser,
    builder::{PossibleValuesParser, TypedValueParser},
    value_parser,
};

use super::{
    file::Checked,
    sharry::{NewShareRequest, Uri},
};

#[derive(Parser, Hash)]
#[command(version, about, long_about = None)]
pub struct Cli {
    /// Timeout in seconds for HTTP actions (set 0 or invalid to disable)
    #[arg(
        short, long,
        default_value = "10", value_name = "SECS",
        value_parser = parse_seconds,
    )]
    timeout: Duration,

    /// Protocol for Sharry instance
    #[arg(
        short, long,
        default_value = "https", value_name = "VARIANT",
        value_parser = PossibleValuesParser::new(["http", "https"]),
    )]
    protocol: String,

    /// Name of the new share
    #[arg(short, long, default_value = "ShrUpl Upload", value_name = "TEXT")]
    name: String,

    /// Description of the new share
    #[arg(short, long, value_name = "TEXT")]
    description: Option<String>,

    /// Maximum number of views for the new share
    #[arg(short, long, default_value_t = 100, value_name = "N")]
    max_views: u32,

    /// Chunk size for uploading, in MiB
    #[arg(
        short, long,
        default_value_t = 10, value_name = "M",
        value_parser = value_parser!(u32).range(1..).map(|s| s as usize),
    )]
    pub chunk_size: usize,

    /// Base URL for Sharry Instance
    url: String,

    /// ID of a public alias to use
    pub alias: String,

    /// Files to upload to the new share
    #[arg(value_name = "FILE", required = true, value_parser = parse_sharry_file)]
    pub files: Vec<Checked>,
}

impl fmt::Debug for Cli {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.debug_struct("Cli")
            .field("uri", &self.get_uri())
            .field("alias", &self.alias)
            .field("timeout", &self.get_timeout())
            .field("chunk_size", &self.chunk_size)
            .field("share_request", &self.get_share_request())
            .field("files", &self.files)
            .field("hash", &self.get_hash())
            .finish_non_exhaustive()
    }
}

fn parse_seconds(data: &str) -> Result<Duration, String> {
    data.parse().or(Ok(0)).map(Duration::from_secs)
}

fn parse_sharry_file(data: &str) -> Result<Checked, String> {
    Checked::new(data).map_err(|e| e.to_string())
}

impl Cli {
    pub fn get_timeout(&self) -> Option<Duration> {
        (!self.timeout.is_zero()).then_some(self.timeout)
    }

    pub fn get_uri(&self) -> Uri {
        Uri::with_protocol(&self.protocol, &self.url)
    }

    pub fn get_share_request(&self) -> NewShareRequest {
        NewShareRequest::new(&self.name, self.description.as_ref(), self.max_views)
    }

    pub fn get_hash(&self) -> String {
        let file_refs = {
            let mut refs: Vec<_> = self.files.iter().collect();
            refs.sort_unstable();

            refs
        };

        let mut hasher = DefaultHasher::new();
        (self.get_uri(), &self.alias, file_refs).hash(&mut hasher);

        format!("{:x}", hasher.finish())
    }
}