]> git.r.bdr.sh - rbdr/page/blame_incremental - src/file_handler/mod.rs
Handle layout error better
[rbdr/page] / src / file_handler / mod.rs
... / ...
CommitLineData
1mod file_strategies;
2
3use file_strategies::file::Strategy as FileStrategy;
4use file_strategies::gemini::Strategy as GeminiStrategy;
5use file_strategies::layout::Strategy as LayoutStrategy;
6
7use std::path::PathBuf;
8use std::fs::read_to_string;
9
10pub struct FileHandler {
11 pub strategies: Vec<Box<dyn FileHandlerStrategy>>,
12 pub layout: Option<String>
13}
14
15impl 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
28impl 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>) -> Result<(), &str> {
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 Ok(());
45 },
46 _ => {}
47 }
48 }
49 Err("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
68pub 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
75pub enum FileType {
76 Gemini,
77 File,
78 Layout,
79 Unknown,
80}
81
82pub struct File {
83 pub path: PathBuf,
84 pub file_type: FileType,
85}