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
|
use serde_derive::Deserialize;
use std::collections::HashMap;
use std::fmt;
use super::{parse_config_file, ConfigStructure, Flattenable};
use crate::MIMETYPE_FILE;
const fn default_false() -> bool {
false
}
#[derive(Debug, Deserialize)]
pub struct JoshutoMimetypeEntry {
pub id: usize,
pub program: String,
pub args: Option<Vec<String>>,
#[serde(default = "default_false")]
pub fork: bool,
#[serde(default = "default_false")]
pub silent: bool,
}
impl std::fmt::Display for JoshutoMimetypeEntry {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
f.write_str(self.program.as_str()).unwrap();
if let Some(s) = self.args.as_ref() {
s.iter().for_each(|arg| write!(f, " {}", arg).unwrap());
}
f.write_str("\t[").unwrap();
if self.fork {
f.write_str("fork,").unwrap();
}
if self.silent {
f.write_str("silent").unwrap();
}
f.write_str("]")
}
}
#[derive(Debug, Deserialize)]
pub struct JoshutoRawMimetype {
#[serde(default)]
entry: Vec<JoshutoMimetypeEntry>,
#[serde(default)]
extension: HashMap<String, Vec<usize>>,
#[serde(default)]
mimetype: HashMap<String, Vec<usize>>,
}
impl Flattenable<JoshutoMimetype> for JoshutoRawMimetype {
fn flatten(self) -> JoshutoMimetype {
let mut entries = HashMap::with_capacity(self.entry.len());
for entry in self.entry {
entries.insert(entry.id, entry);
}
JoshutoMimetype {
entries,
extension: self.extension,
mimetype: self.mimetype,
}
}
}
#[derive(Debug)]
pub struct JoshutoMimetype {
pub entries: HashMap<usize, JoshutoMimetypeEntry>,
pub extension: HashMap<String, Vec<usize>>,
pub mimetype: HashMap<String, Vec<usize>>,
}
impl JoshutoMimetype {
pub fn get_entries_for_ext(&self, extension: &str) -> Vec<&JoshutoMimetypeEntry> {
Self::get_entries(&self.extension, &self.entries, extension)
}
pub fn get_entries_for_mimetype(&self, mimetype: &str) -> Vec<&JoshutoMimetypeEntry> {
Self::get_entries(&self.mimetype, &self.entries, mimetype)
}
fn get_entries<'a>(
map: &HashMap<String, Vec<usize>>,
entry_map: &'a HashMap<usize, JoshutoMimetypeEntry>,
key: &str,
) -> Vec<&'a JoshutoMimetypeEntry> {
match map.get(key) {
Some(entry_ids) => entry_ids
.iter()
.filter_map(|id| entry_map.get(id))
.collect(),
None => Vec::new(),
}
}
}
impl ConfigStructure for JoshutoMimetype {
fn get_config() -> Self {
parse_config_file::<JoshutoRawMimetype, JoshutoMimetype>(MIMETYPE_FILE)
.unwrap_or_else(JoshutoMimetype::default)
}
}
impl std::default::Default for JoshutoMimetype {
fn default() -> Self {
JoshutoMimetype {
entries: HashMap::new(),
mimetype: HashMap::new(),
extension: HashMap::new(),
}
}
}
|