]> git.r.bdr.sh - rbdr/page/blame - src/file_handler/mod.rs
Add static file copying
[rbdr/page] / src / file_handler / mod.rs
CommitLineData
1e2d00b6
RBR
1mod file_strategies;
2
3use file_strategies::file::Strategy as FileStrategy;
102a4884
RBR
4use file_strategies::gemini::Strategy as GeminiStrategy;
5use file_strategies::layout::Strategy as LayoutStrategy;
6
1e2d00b6 7use std::path::PathBuf;
4fd89b80 8use std::fs::read_to_string;
1e2d00b6
RBR
9
10pub struct FileHandler {
4fd89b80
RBR
11 pub strategies: Vec<Box<dyn FileHandlerStrategy>>,
12 pub layout: Option<String>
1e2d00b6
RBR
13}
14
15impl Default for FileHandler {
16 fn default() -> FileHandler {
17 FileHandler {
102a4884
RBR
18 strategies: vec![
19 Box::new(GeminiStrategy{}),
20 Box::new(LayoutStrategy{}),
21 Box::new(FileStrategy{}),
4fd89b80
RBR
22 ],
23 layout: None
1e2d00b6
RBR
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
4fd89b80
RBR
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) {
1e2d00b6 59 for strategy in self.strategies.iter() {
4fd89b80 60 if strategy.can_handle(&file.file_type) {
ea529736 61 return strategy.handle(source, destination, file);
1e2d00b6
RBR
62 }
63 }
64 }
65}
66
67pub trait FileHandlerStrategy {
68 fn is(&self, path: &PathBuf) -> bool;
69 fn identify(&self) -> FileType;
4fd89b80 70 fn can_handle(&self, file_type: &FileType) -> bool;
ea529736 71 fn handle(&self, source: &PathBuf, destination: &PathBuf, file: &File);
1e2d00b6
RBR
72}
73
74pub enum FileType {
75 Gemini,
76 File,
77 Layout,
78 Unknown,
79}
80
81pub struct File {
82 pub path: PathBuf,
83 pub file_type: FileType,
84}