shrupl/src/main.rs

mod sharry;

use std::{path::PathBuf, time::Duration};

use clap::Parser;
use log::{error, info};
use ureq::Agent;

use sharry::{Alias, File, NewShareRequest, Share, Uri};

#[derive(Parser, Debug)]
#[command(version, about, long_about = None)]
struct Cli {
    /// Protocol for Sharry instance
    #[arg(
        short, long,
        default_value_t = String::from("https"),
        value_parser = clap::builder::PossibleValuesParser::new(["http", "https"]),
    )]
    proto: String,

    /// Name of the new share
    #[arg(
        short, long,
        default_value_t = String::from("shrupl upload"),
    )]
    name: String,

    /// Description of the new share
    #[arg(short, long)]
    desc: Option<String>,

    /// Maximum number of views for the new share
    #[arg(short, long, default_value_t = 10)]
    views: u32,

    /// Chunk size for uploading, in MiB
    #[arg(short, long, default_value_t = 10)]
    chunk: usize,

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

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

    /// Files to upload to the new share
    #[arg(value_name = "FILE")]
    files: Vec<PathBuf>,
}

fn main() {
    env_logger::init();

    let agent: Agent = Agent::config_builder()
        .timeout_global(Some(Duration::from_secs(5)))
        .build()
        .into();

    let args = Cli::parse();

    let files: Vec<File> = (args.files.iter())
        .map(File::new)
        .map(Result::unwrap)
        .collect();

    let alias = Alias::new(Uri::with_protocol(args.proto, args.url), args.alias);

    let share = NewShareRequest::new(args.name, args.desc, 10);
    let share = Share::create(&agent, &alias, share).unwrap();
    info!("share: {share:?}");

    for file in files {
        let file = file.create(&agent, &share).unwrap();
        info!("file: {file:?}");

        for chunk in file.chunked(args.chunk * 1024 * 1024) {
            info!("chunk len: {}", chunk.bytes.len());

            file.upload_chunk(&agent, &alias, &chunk)
                .inspect_err(|e| error!("error: {e}"))
                .unwrap();
        }
    }

    share.notify(&agent).unwrap();
}