Initial commit

2025-07-13 22:34:51 -04:00 · 2025-07-13 22:34:51 -04:00 · f665dd032f
commit f665dd032f
6 changed files with 3173 additions and 0 deletions
--- a/src/main.rs
+++ b/src/main.rs
@ -0,0 +1,141 @@
+use typst_as_lib::{typst_kit_options::TypstKitFontOptions, TypstEngine};
+use typst::{ecow::EcoString, html::{HtmlAttr, HtmlDocument, HtmlElement, HtmlNode}};
+use std::fs;
+use std::env;
+use std::path::Path;
+
+static COMMON: &str = include_str!("../common.typ");
+
+fn main() {
+    let args: Vec<String> = env::args().collect();
+
+    match env::current_dir() {
+        Ok(path) => println!("Current working directory: {}", path.display()),
+        Err(e) => eprintln!("Error getting current directory: {}", e),
+    }
+
+    let file_path = Path::new(args.get(1).unwrap());
+    let article_dir: &str = file_path.parent().unwrap().to_str().unwrap();
+    let template_file = format!("{}/article.typ", article_dir);
+    let output = format!("{}/article.html", article_dir);
+    let outline_file = format!("{}/outline.html", article_dir);
+
+    let mut template: EcoString = EcoString::new();
+    template.push_str(COMMON);
+    template.push_str(fs::read_to_string(template_file).unwrap().as_str());
+
+    let template = TypstEngine::builder()
+        .main_file(template.to_string())
+        .search_fonts_with(
+            TypstKitFontOptions::default()
+                .include_system_fonts(false)
+                .include_embedded_fonts(true),
+        )
+        .with_file_system_resolver(article_dir)
+        .build();
+
+    let mut doc: HtmlDocument = template
+        .compile()
+        .output
+        .expect("typst::compile() returned an error!");
+
+    let mut outline = EcoString::new();
+
+    let mut curr_level = 1;
+    parse(&mut doc.root, &mut outline, &mut curr_level);
+    for i in (1..curr_level).rev() {
+        outline.push_str("  ".repeat(i as usize - 1).as_str());
+        outline.push_str("</ul>\n");
+    }
+    fs::write(outline_file, outline.as_bytes()).unwrap();
+
+    let mut body: Option<HtmlElement> = None;
+    for child in &doc.root.children {
+        match child {
+            HtmlNode::Element(e) if e.tag.to_string().as_str() == "<body>" => {
+                body = Some(e.clone());
+            }
+            _ => {}
+        }
+    }
+    doc.root = body.unwrap();
+
+    let mut html: String = typst_html::html(&doc)
+        .expect("Could not generate html.");
+    let lines = html
+        .lines()
+        .map(|line| {
+            if line.len() >= 2 {
+                &line[2..]
+            } else {
+                "" // if line has fewer than 2 chars, drop it
+            }
+        })
+        .collect::<Vec<&str>>();
+
+    if lines.len() > 2 {
+        // Get just the inside of the <body> tag
+        html = lines[2..lines.len()-1].to_vec().join("\n");
+    } else {
+        // File is empty
+        html = "".to_string();
+    }
+
+    fs::write(output, html).expect("Could not write html.");
+}
+
+fn parse(elem: &mut HtmlElement, outline: &mut EcoString, curr_level: &mut u32) {
+    if matches!(elem.tag.to_string().as_str(), "<h2>"|"<h3>"|"<h4>"|"<h5>"|"<h6>") {
+        let mut header_text = EcoString::new();
+
+        // Collate all the children text objects
+        for child in &elem.children {
+            match child {
+                HtmlNode::Text(string, _) => {
+                    header_text.push_str(string);
+                }
+                _ => {}
+            }
+        }
+
+        let slug = header_text.as_str()
+            .to_lowercase()
+            .chars()
+            .filter_map(|c| match c {
+                c if c.is_ascii_alphanumeric() => Some(c),
+                ' ' => Some('-'),
+                '\'' => None,
+                _ => None,
+            })
+            .collect::<String>()
+            .trim_matches('-')
+            .replace("--", "-");
+
+        let level: u32 = elem.tag.to_string().chars().nth(2).unwrap() as u32 - '0' as u32;
+
+        while level > *curr_level {
+            *curr_level += 1;
+            outline.push_str("  ".repeat(*curr_level as usize - 2).as_str());
+            outline.push_str("<ul>\n");
+        }
+        while level < *curr_level {
+            outline.push_str("  ".repeat(*curr_level as usize - 2).as_str());
+            outline.push_str("</ul>\n");
+            *curr_level -= 1;
+        }
+        *curr_level = level;
+
+        outline.push_str("  ".repeat(*curr_level as usize - 1).as_str());
+        outline.push_str(format!("<li><a href=\"#{}\">{}</a></li>\n", slug, header_text).as_str());
+
+        elem.attrs.push(HtmlAttr::intern("id").unwrap(), slug);
+        return;
+    }
+    for child in &mut elem.children {
+        match child {
+            HtmlNode::Element(e) => {parse(e, outline, curr_level)}
+            _ => {}
+        }
+    }
+}
+