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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
use structopt::StructOpt;
use serde::Deserialize;
use std::collections::HashMap;
use std::fs;
use std::io;
use std::path::PathBuf;
#[derive(Debug, StructOpt)]
#[structopt(name = "gallery")]
pub struct Options {
#[structopt(long, help = "Just a thing")]
pub check: bool,
#[structopt(short, long, help = "Config file location")]
pub config: Option<PathBuf>,
#[structopt(help = "Picture location")]
pub load: PathBuf,
#[structopt(short, long, default_value = "png", help = "Image extension to use for converted files")]
pub ext: String,
#[structopt(long, short, default_value = "build", help = "Where to build site")]
pub builddir: PathBuf,
#[structopt(long, default_value = "1080", help = "Scaled size for image")]
pub size_scaled: u32,
#[structopt(long, default_value = "720", help = "Thumbnail size for image")]
pub size_thumb: u32,
}
#[derive(Debug)]
pub enum ConfigError {
Reading(io::Error),
Parsing(serde_yaml::Error),
CompilePattern(glob::PatternError),
}
#[derive(Deserialize, Debug)]
pub struct Config {
pub imageglob: String,
pub info: HashMap<String, String>,
#[serde(skip)]
pub imageglob_compiled: glob::Pattern,
}
#[derive(Debug)]
pub struct Context {
pub options: Options,
pub config: Config,
pub imgdir: PathBuf,
}
impl Context {
pub fn new_with_args() -> Result<Context, ConfigError> {
let opts = Options::from_args();
let config = Config::load_with_options(&opts)?;
let imgdir = opts.builddir.join("imgs");
// Create img dir
if let Err(err) = fs::create_dir(&imgdir) {
if err.kind() != io::ErrorKind::AlreadyExists {
return Err(ConfigError::from(err));
}
}
Ok(Context {
options: opts,
config: config,
imgdir: imgdir,
})
}
pub fn get_image_files(&self) -> Result<Vec<PathBuf>, io::Error> {
let files = fs::read_dir(&self.options.load)?
// Remove errored entries (SILENTLY)
.filter(|entry| entry.is_ok())
.map(|entry| entry.unwrap().path())
// Filter out directories and files that do not match pattern
.filter(|path| path.file_name()
.map_or(false, |file| self.config.imageglob_compiled.matches(file.to_str().unwrap()))
)
.collect();
Ok(files)
}
}
impl Config {
fn load_from_file(path: &PathBuf) -> Result<Config, ConfigError> {
let content = fs::read_to_string(path)?;
let mut config: Config = serde_yaml::from_str(&content)?;
config.imageglob_compiled = glob::Pattern::new(&config.imageglob)?;
Ok(config)
}
pub fn load_with_options(opts: &Options) -> Result<Config, ConfigError> {
let config_path = match &opts.config {
Some(path) => path.to_path_buf(),
None => opts.load.join("imginfo.yml"),
};
Config::load_from_file(&config_path)
}
}
impl From<io::Error> for ConfigError {
fn from(error: io::Error) -> Self {
ConfigError::Reading(error)
}
}
impl From<serde_yaml::Error> for ConfigError {
fn from(error: serde_yaml::Error) -> Self {
ConfigError::Parsing(error)
}
}
impl From<glob::PatternError> for ConfigError {
fn from(error: glob::PatternError) -> Self {
ConfigError::CompilePattern(error)
}
}
|
