include-crypt-codegen 0.1.1

Codegen helpers for the `include-crypt` crate
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
45
46

use std::{fs::File, io::Read, path::PathBuf};

pub(crate) fn error_mapping<E: std::fmt::Display>(error: E) -> syn::Error {
    syn::Error::new(proc_macro2::Span::mixed_site(), error)
}

/// Opens the specified file and returns the content.
pub(crate) fn read_file<P: Into<PathBuf>>(file_path: P) -> syn::Result<Vec<u8>> {
    let file_path = file_path.into();

    // Create the correct path to the file
    //
    let file_path = if file_path.is_relative() {
        PathBuf::from(std::env::var("CARGO_MANIFEST_DIR").unwrap()).join(file_path)
    } else {
        file_path
    };

    // Check if the file exists
    //
    if !file_path.exists() {
        return Err(error_mapping(format!("File {:?} could not be found.", file_path)));
    };

    // Open and read file.
    //
    let mut file = File::open(&file_path).map_err(error_mapping)?;
    let file_size = file.metadata().map_err(error_mapping)?.len();

    let mut file_bytes = Vec::with_capacity(file_size as usize);
    file.read_to_end(&mut file_bytes).map_err(error_mapping)?;

    // Compress the file if the feature is set
    //
    #[cfg(feature = "compression")]
    {
        use std::io::Write;

        let mut encoder = libflate::deflate::Encoder::new(Vec::with_capacity(file_bytes.len()));
        encoder.write_all(&file_bytes).map_err(error_mapping)?;
        encoder.finish().into_result().map_err(error_mapping)
    }

    #[cfg(not(feature = "compression"))]
    Ok(file_bytes)
}