]> git.r.bdr.sh - rbdr/page/commitdiff
Add gemini parsing
authorRuben Beltran del Rio <redacted>
Sun, 16 Apr 2023 12:44:48 +0000 (14:44 +0200)
committerRuben Beltran del Rio <redacted>
Sun, 16 Apr 2023 12:44:48 +0000 (14:44 +0200)
src/file_handler/file_strategies/file.rs
src/file_handler/file_strategies/gemini.rs
src/file_handler/file_strategies/layout.rs
src/file_handler/mod.rs
src/gemini_parser.rs
src/main.rs

index 42b87d5b692cbb382b8571be9c9f0e35eaaea591..8eafdaa898a431fc6dfc530ea351f75676010957 100644 (file)
@@ -21,7 +21,7 @@ impl FileHandlerStrategy for Strategy {
         }
     }
 
-    fn handle(&self, source: &PathBuf, destination: &PathBuf, file: &File) {
+    fn handle(&self, source: &PathBuf, destination: &PathBuf, file: &File, _l: &String) {
         let relative_path = file.path.strip_prefix(&source).unwrap();
         let complete_destination = destination.join(relative_path);
         let destination_parent = complete_destination.parent().unwrap();
index 2427422834433fbda727f6052f2fbc09784405be..04e2b6aae554e9cf2094bb1cbb1b6407096e2def 100644 (file)
@@ -1,8 +1,29 @@
 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 {
@@ -23,7 +44,42 @@ impl FileHandlerStrategy for Strategy {
         }
     }
 
-    fn handle(&self, source: &PathBuf, destination: &PathBuf, file: &File) {
-        println!("Should parse and copy {}", file.path.display())
+    fn handle(&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 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(generated_html.as_bytes()).unwrap();
     }
 }
index 21ce9ba66aa4f520cfc6c2fd804bf6ef66c5fc40..793b4609da6726db1ae45513a5d5c2ee285cccf0 100644 (file)
@@ -22,5 +22,5 @@ impl FileHandlerStrategy for Strategy {
 
     // We don't implement handling for layout, as we assume there's only one
     // and it got handled before.
-    fn handle(&self, _s: &PathBuf, _d: &PathBuf, _f: &File) {}
+    fn handle(&self, _s: &PathBuf, _d: &PathBuf, _f: &File, _l: &String) {}
 }
index 64224ee4061e896b51358f0d36615d903695e65c..a106c27745013db17b9941e640afee7040d44b86 100644 (file)
@@ -58,7 +58,8 @@ impl FileHandler {
     pub fn handle(&self, source: &PathBuf, destination: &PathBuf, file: &File) {
         for strategy in self.strategies.iter() {
             if strategy.can_handle(&file.file_type) {
-                return strategy.handle(source, destination, file);
+                let layout = self.layout.as_ref().unwrap();
+                return strategy.handle(source, destination, file, layout);
             }
         }
     }
@@ -68,7 +69,7 @@ pub trait FileHandlerStrategy {
     fn is(&self, path: &PathBuf) -> bool;
     fn identify(&self) -> FileType;
     fn can_handle(&self, file_type: &FileType) -> bool;
-    fn handle(&self, source: &PathBuf, destination: &PathBuf, file: &File);
+    fn handle(&self, source: &PathBuf, destination: &PathBuf, file: &File, layout: &String);
 }
 
 pub enum FileType {
index 36abb169ed5788bfc5c1a4cfc82a8cd642ff5d78..5c204263bebede10c4c083ff70399ae1aa11b793 100644 (file)
@@ -7,7 +7,11 @@ pub fn parse(source: &str) -> String {
     let mut current_line_type: Option<LineType> = None;
 
     for line in lines {
-        let line_type = identify_line(&(line[..3]), is_preformatted);
+        let mut line_type = LineType::Text;
+        if line.len() > 2 {
+            let end = line.char_indices().map(|(i, _)| i).nth(2).unwrap();
+            line_type = identify_line(&(line[..end]), is_preformatted);
+        }
         match line_type {
             LineType::PreformattedToggle => is_preformatted = !is_preformatted,
             _ => {
index 7ab009e862e9b7fd6c34d81244076addbfbff2fe..9b823b0e9f822bc346a7866e5366bcba89db8027 100644 (file)
@@ -6,15 +6,10 @@ use std::io::Result;
 use std::env::current_dir;
 use std::fs::{create_dir_all, remove_dir_all};
 
-use crate::gemini_parser::parse;
 use crate::file_finder::find_files;
 use crate::file_handler::FileHandler;
 
 fn main() -> Result<()> {
-    let gemini_source = "# Test\n## 2nd H\na line\n another line\n```\npreformat\n=> preformat\n```\n=> http://lol.com\n=> http://lol.com lol\n* lol\nbla\n* lol\n* lmao\n> blabla\n> blabla";
-    let html = parse(gemini_source);
-    println!("{}", html);
-
     let source = current_dir()?;
     let source_name = source.file_name().unwrap().to_string_lossy();
     let parent = source.parent().unwrap();