rimage 0.12.3

Optimize images natively with best-in-class codecs
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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100

use std::path::{Path, PathBuf};

use rayon::prelude::*;

pub fn get_paths(
    files: Vec<PathBuf>,
    out_dir: Option<PathBuf>,
    suffix: Option<String>,
    recursive: bool,
) -> impl ParallelIterator<Item = (PathBuf, PathBuf)> {
    let common_path = if recursive {
        get_common_path(&files)
    } else {
        None
    };

    files
        .into_par_iter()
        .filter_map(move |path| -> Option<(PathBuf, PathBuf)> {
            if !path.is_file() {
                log::warn!("{path:?} is not a file");
                return None;
            }

            let file_name = path
                .file_stem()
                .and_then(|f| f.to_str())
                .unwrap_or("optimized_image");

            let mut out_path = match &out_dir {
                Some(dir) => {
                    if let Some(common) = &common_path {
                        let relative_path =
                            path.parent().unwrap().strip_prefix(common).unwrap_or(&path);
                        dir.join(relative_path)
                    } else {
                        dir.to_owned()
                    }
                }
                None => path.parent().map(|p| p.to_path_buf()).unwrap_or_default(),
            };

            if let Some(s) = &suffix {
                out_path.push(format!("{file_name}@{s}"));
            } else {
                out_path.push(file_name);
            }

            Some((path, out_path))
        })
}

fn get_common_path(paths: &[PathBuf]) -> Option<PathBuf> {
    if paths.is_empty() {
        return None;
    }

    let mut common_path = paths[0].clone();

    for path in paths.iter().skip(1) {
        common_path = common_path
            .iter()
            .zip(path.iter())
            .take_while(|&(a, b)| a == b)
            .map(|(a, _)| a)
            .collect();
    }

    Some(common_path)
}

#[inline]
pub fn collect_files<P: AsRef<Path>>(input: &[P]) -> Vec<PathBuf> {
    #[cfg(windows)]
    {
        input.iter().flat_map(apply_glob_pattern).collect()
    }

    #[cfg(not(windows))]
    {
        input.iter().map(|p| PathBuf::from(p.as_ref())).collect()
    }
}

#[cfg(windows)]
fn apply_glob_pattern<P: AsRef<Path>>(path: P) -> Vec<PathBuf> {
    let matches = path
        .as_ref()
        .to_str()
        .and_then(|pattern| glob::glob(pattern).ok())
        .map(|paths| paths.flatten().collect::<Vec<_>>());

    match matches {
        Some(paths) if !paths.is_empty() => paths,
        _ => vec![PathBuf::from(path.as_ref())],
    }
}

#[cfg(test)]
mod tests;