]> git.r.bdr.sh - rbdr/page/blobdiff - src/file_handler/file_strategies/gemini.rs
Treat preformatted text with alt as an image
[rbdr/page] / src / file_handler / file_strategies / gemini.rs
index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..977464cf8e72490b15ac2c2c11ee68097db340c7 100644 (file)
@@ -0,0 +1,114 @@
+pub struct Strategy {}
+
+use std::path::PathBuf;
+use std::io::Write;
+use std::fs::{create_dir_all, read_to_string, File as IOFile};
+
+use crate::file_handler::{File, FileType, FileHandlerStrategy};
+use crate::gemini_parser::parse;
+
+impl Strategy {
+    fn is_title(&self, line: &str) -> bool {
+        line.starts_with("--- title:")
+    }
+
+    fn is_description(&self, line: &str) -> bool {
+        line.starts_with("--- description:")
+    }
+
+    fn get_title<'a>(&self, line: &'a str) -> &'a str {
+        line.split_once("--- title:").unwrap().1
+    }
+
+    fn get_description<'a>(&self, line: &'a str) -> &'a str {
+        line.split_once("--- description:").unwrap().1
+    }
+}
+
+impl FileHandlerStrategy for Strategy {
+    fn is(&self, path: &PathBuf) -> bool {
+        if let Some(extension) = path.extension() {
+            return !path.is_dir() && extension == "gmi"
+        }
+        false
+    }
+
+    fn identify(&self) -> FileType {
+        FileType::Gemini
+    }
+
+    fn can_handle(&self, file_type: &FileType) -> bool {
+        match file_type {
+            FileType::Gemini => true,
+            _ => false,
+        }
+    }
+
+    fn handle_html(&self, source: &PathBuf, destination: &PathBuf, file: &File, layout: &String) {
+        let gemini_contents = read_to_string(&file.path).unwrap();
+
+        // Front matter extraction
+        let lines: Vec<&str> = gemini_contents.split("\n").collect();
+        let mut lines_found = 0;
+        let mut title = "";
+        let mut description = "";
+        for line in lines[..2].iter() {
+            if self.is_title(&line) {
+                title = self.get_title(&line).trim();
+                lines_found = lines_found + 1;
+                continue;
+            }
+            if self.is_description(&line) {
+                description = self.get_description(&line).trim();
+                lines_found = lines_found + 1;
+                continue;
+            }
+        }
+
+        let gemini_source = lines[lines_found..].join("\n");
+        let content_html = parse(&gemini_source[..]);
+
+        let generated_html = layout
+            .replace("{{ title }}", title)
+            .replace("{{ description }}", description)
+            .replace("{{ content }}", &content_html[..]);
+
+
+        let relative_path = file.path.strip_prefix(&source).unwrap();
+        let mut complete_destination = destination.join(relative_path);
+        complete_destination.set_extension("html");
+        let destination_parent = complete_destination.parent().unwrap();
+        create_dir_all(destination_parent).unwrap();
+
+        let mut destination_file = IOFile::create(&complete_destination).unwrap();
+        destination_file.write_all(generated_html.as_bytes()).unwrap();
+    }
+
+    fn handle_gemini(&self, source: &PathBuf, destination: &PathBuf, file: &File) {
+        let gemini_contents = read_to_string(&file.path).unwrap();
+
+        // Front matter extraction
+        let lines: Vec<&str> = gemini_contents.split("\n").collect();
+        let mut lines_found = 0;
+        for line in lines[..2].iter() {
+            if self.is_title(&line) {
+                lines_found = lines_found + 1;
+                continue;
+            }
+            if self.is_description(&line) {
+                lines_found = lines_found + 1;
+                continue;
+            }
+        }
+
+        let gemini_source = lines[lines_found..].join("\n");
+
+        let relative_path = file.path.strip_prefix(&source).unwrap();
+        let complete_destination = destination.join(relative_path);
+        let destination_parent = complete_destination.parent().unwrap();
+        create_dir_all(destination_parent).unwrap();
+
+        let mut destination_file = IOFile::create(&complete_destination).unwrap();
+        destination_file.write_all(gemini_source.as_bytes()).unwrap();
+    }
+}