postgresql_archive 0.20.2

A library for downloading and extracting PostgreSQL archives
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
40
41
42
43
44

use crate::Result;
use crate::extractor::ExtractDirectories;
use std::fs::create_dir_all;
use std::io::Cursor;
use std::path::PathBuf;
use std::{fs, io};
use tracing::{debug, instrument, warn};
use zip::ZipArchive;

/// Extracts the compressed tar `bytes` to paths defined in `extract_directories`.
///
/// # Errors
/// Returns an error if the extraction fails.
#[instrument(skip(bytes))]
pub fn extract(bytes: &Vec<u8>, extract_directories: &ExtractDirectories) -> Result<Vec<PathBuf>> {
    let mut files = Vec::new();
    let reader = Cursor::new(bytes);
    let mut archive = ZipArchive::new(reader)?;
    let mut extracted_bytes = 0;

    for i in 0..archive.len() {
        let mut file = archive.by_index(i)?;
        let file_path = PathBuf::from(file.name());
        let file_path = PathBuf::from(file_path.file_name().unwrap_or_default());
        let file_name = file_path.to_string_lossy();

        let Ok(extract_dir) = extract_directories.get_path(&file_name) else {
            continue;
        };
        create_dir_all(&extract_dir)?;

        let mut out = Vec::new();
        io::copy(&mut file, &mut out)?;
        extracted_bytes += out.len() as u64;
        let path = PathBuf::from(&extract_dir).join(file_path);
        fs::write(&path, out)?;
        files.push(path);
    }

    let number_of_files = files.len();
    debug!("Extracted {number_of_files} files totalling {extracted_bytes}");

    Ok(files)
}