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
125
126
127
128
129
130
131
132
|
use serde_derive::Deserialize;
use std::collections::HashMap;
use std::fmt;
use super::{parse_to_config_file, ConfigStructure, Flattenable};
use crate::MIMETYPE_FILE;
const fn default_false() -> bool {
false
}
#[derive(Debug, Deserialize)]
pub struct JoshutoMimetypeEntry {
id: usize,
command: String,
#[serde(default)]
args: Vec<String>,
#[serde(default = "default_false")]
fork: bool,
#[serde(default = "default_false")]
silent: bool,
}
impl JoshutoMimetypeEntry {
pub fn get_id(&self) -> usize {
self.id
}
pub fn get_command(&self) -> &str {
&self.command
}
pub fn get_args(&self) -> &[String] {
&self.args
}
pub fn get_fork(&self) -> bool {
self.fork
}
pub fn get_silent(&self) -> bool {
self.silent
}
}
impl std::fmt::Display for JoshutoMimetypeEntry {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
f.write_str(self.get_command()).unwrap();
self.get_args()
.iter()
.for_each(|arg| write!(f, " {}", arg).unwrap());
f.write_str("\t[").unwrap();
if self.get_fork() {
f.write_str("fork,").unwrap();
}
if self.get_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.get_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_to_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(),
}
}
}
|
