]> git.r.bdr.sh - rbdr/page/blob - src/file_handler/mod.rs
a106c27745013db17b9941e640afee7040d44b86
[rbdr/page] / src / file_handler / mod.rs
1 mod file_strategies;
2
3 use file_strategies::file::Strategy as FileStrategy;
4 use file_strategies::gemini::Strategy as GeminiStrategy;
5 use file_strategies::layout::Strategy as LayoutStrategy;
6
7 use std::path::PathBuf;
8 use std::fs::read_to_string;
9
10 pub struct FileHandler {
11 pub strategies: Vec<Box<dyn FileHandlerStrategy>>,
12 pub layout: Option<String>
13 }
14
15 impl Default for FileHandler {
16 fn default() -> FileHandler {
17 FileHandler {
18 strategies: vec![
19 Box::new(GeminiStrategy{}),
20 Box::new(LayoutStrategy{}),
21 Box::new(FileStrategy{}),
22 ],
23 layout: None
24 }
25 }
26 }
27
28 impl FileHandler {
29 pub fn identify(&self, path: &PathBuf) -> FileType {
30 for strategy in self.strategies.iter() {
31 if strategy.is(&path) {
32 return strategy.identify();
33 }
34 }
35 FileType::Unknown
36 }
37
38 pub fn get_layout_or_panic(&mut self, files: &Vec<File>) {
39 for file in files {
40 match file.file_type {
41 FileType::Layout => {
42 let layout_text = read_to_string(&file.path).unwrap();
43 self.layout = Some(layout_text);
44 return;
45 },
46 _ => {}
47 }
48 }
49 panic!("No layout found. Please ensure there's a _layout.html file at the root");
50 }
51
52 pub fn handle_all(&self, source: &PathBuf, destination: &PathBuf, files: &Vec<File>) {
53 for file in files {
54 self.handle(source, destination, file);
55 }
56 }
57
58 pub fn handle(&self, source: &PathBuf, destination: &PathBuf, file: &File) {
59 for strategy in self.strategies.iter() {
60 if strategy.can_handle(&file.file_type) {
61 let layout = self.layout.as_ref().unwrap();
62 return strategy.handle(source, destination, file, layout);
63 }
64 }
65 }
66 }
67
68 pub trait FileHandlerStrategy {
69 fn is(&self, path: &PathBuf) -> bool;
70 fn identify(&self) -> FileType;
71 fn can_handle(&self, file_type: &FileType) -> bool;
72 fn handle(&self, source: &PathBuf, destination: &PathBuf, file: &File, layout: &String);
73 }
74
75 pub enum FileType {
76 Gemini,
77 File,
78 Layout,
79 Unknown,
80 }
81
82 pub struct File {
83 pub path: PathBuf,
84 pub file_type: FileType,
85 }