hdim-core 0.1.0

Core logic and image manipulation algorithms for HDIM.
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

use anyhow::Result;
use img_parts::ImageEXIF;
use std::path::Path;

/// Extracts EXIF bytes from an image file for use in saving.
///
/// If `strip` is true, sensitive data is removed.
/// Currently, stripping removes the entire EXIF segment, relying on the image
/// encoder to preserve essential dimensions in the format-specific headers.
pub fn get_exif_bytes_for_save(path: &Path, strip: bool) -> Result<Option<Vec<u8>>> {
    if strip {
        // When stripping, we return None to indicate no EXIF segment should be injected.
        // The image crate's encoder will still write the necessary width/height
        // in the standard JPEG/PNG headers.
        return Ok(None);
    }

    let bytes = std::fs::read(path)?;

    // Try to extract EXIF from JPEG
    if let Ok(jpeg) = img_parts::jpeg::Jpeg::from_bytes(bytes.clone().into())
        && let Some(exif) = jpeg.exif()
    {
        let exif_vec: Vec<u8> = exif.to_vec();
        return Ok(Some(exif_vec));
    }

    // Try to extract EXIF from PNG
    if let Ok(png) = img_parts::png::Png::from_bytes(bytes.into())
        && let Some(exif) = png.exif()
    {
        let exif_vec: Vec<u8> = exif.to_vec();
        return Ok(Some(exif_vec));
    }

    Ok(None)
}