arloader 0.1.63

Command line application and library for uploading files to Arweave.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

//! Async [`TempDir`] for testing.

use crate::error::Error;
use base64::{self, encode_config};
use ring::rand::{SecureRandom, SystemRandom};
use std::{fs as fsstd, path::PathBuf};
use tokio::fs;

/// Tuple struct with a [`PathBuf`] in it.
pub struct TempDir(pub PathBuf);

/// Implemented to create a temporary directory with a random 8 byte
/// base64 url string as a name. Drop implemented to remove directory
/// when [`TempDir`] goes out of scope.
impl TempDir {
    pub async fn from_str(path_str: &str) -> Result<Self, Error> {
        if path_str.chars().last().unwrap() != '/' {
            return Err(Error::MissingTrailingSlash);
        }
        let rng = SystemRandom::new();
        let mut rand_bytes: [u8; 8] = [0; 8];
        let _ = rng.fill(&mut rand_bytes)?;
        let temp_stem = encode_config(rand_bytes, base64::URL_SAFE_NO_PAD);
        let path = PathBuf::from(path_str).join(temp_stem);
        fs::create_dir(&path).await?;
        Ok(Self(path))
    }
}

impl Drop for TempDir {
    fn drop(&mut self) {
        match fsstd::remove_dir_all(&self.0) {
            Ok(_) => {}
            Err(e) => {
                eprintln!("{}", e);
            }
        }
    }
}