mega 0.8.0

An API client library for interacting with MEGA
Documentation
//!
//! Example program that simply uploads a file to MEGA
//! with progress reporting.
//!

use std::env;
use std::path::Path;
use std::sync::Arc;
use std::time::Duration;

use async_read_progress::TokioAsyncReadProgressExt;
use console::style;
use indicatif::{ProgressBar, ProgressStyle};
use tokio::fs::File;
use tokio_util::compat::TokioAsyncReadCompatExt;

async fn run(mega: &mut mega::Client, file: &str, folder: &str) -> mega::Result<()> {
    let file_name = Path::new(file).file_name().unwrap().to_str().unwrap();

    let nodes = mega.fetch_own_nodes().await?;

    let node = nodes
        .get_node_by_path(folder)
        .expect("could not find node by path");

    let file = File::open(file).await?;
    let size = file.metadata().await?.len();

    let bar = ProgressBar::new(size);
    bar.set_style(progress_bar_style());
    bar.set_message(format!("uploading {0}...", node.name()));
    let bar = Arc::new(bar);

    let reader = {
        let bar = bar.clone();
        file.report_progress(Duration::from_millis(100), move |bytes_read| {
            bar.set_position(bytes_read as u64);
        })
    };

    mega.upload_node(
        &node,
        file_name,
        size,
        reader.compat(),
        mega::LastModified::Now,
    )
    .await?;

    bar.finish_with_message(format!("{0} uploaded !", node.name()));

    Ok(())
}

#[tokio::main(flavor = "current_thread")]
async fn main() {
    let email = env::var("MEGA_EMAIL").expect("missing MEGA_EMAIL environment variable");
    let password = env::var("MEGA_PASSWORD").expect("missing MEGA_PASSWORD environment variable");
    let mfa = env::var("MEGA_MFA").ok();

    let args: Vec<String> = std::env::args().skip(1).collect();
    let [file, folder] = args.as_slice() else {
        panic!("expected 2 command-line arguments: {{source_file}} {{destination_folder}}");
    };

    let http_client = reqwest::Client::new();
    let mut mega = mega::Client::builder().build(http_client).unwrap();

    mega.login(&email, &password, mfa.as_deref()).await.unwrap();
    let result = run(&mut mega, file, folder).await;
    mega.logout().await.unwrap();

    result.unwrap();
}

pub fn progress_bar_style() -> ProgressStyle {
    let template = format!(
        "{}{{bar:30.magenta.bold/magenta/bold}}{} {{percent}} % (ETA {{eta}}): {{msg}}",
        style("").bold().magenta(),
        style("").bold().magenta(),
    );

    ProgressStyle::default_bar()
        .progress_chars("▨▨╌")
        .template(template.as_str())
        .unwrap()
}