grace/htmlua
1
0
Fork 0
Code Issues Pull requests Projects Releases 2 Packages Wiki Activity Actions

added template expanding

Browse source
This commit is contained in:
gyoder 2025-06-23 20:04:08 -06:00
parent 013317c13c
commit 0c33e46b30
4 changed files with 82 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

88
htmlua-parser/src/lib.rs
View file

@ -1,5 +1,7 @@
use std::cell::RefCell; use std::cell::RefCell;
use std::fmt::Write; use std::fmt::Write;
use std::fs::read_to_string;
use std::path::PathBuf;
use std::rc::Rc; use std::rc::Rc;
use anyhow::Result; use anyhow::Result;
@ -9,8 +11,7 @@ use kuchikiki::{
traits::TendrilSink, traits::TendrilSink,
}; };
use mlua::{Lua, Table}; use mlua::{Lua, Table};
use tendril::fmt; use tendril::Atomicity;
use tendril::{Atomicity, Tendril};
fn create_luahtml_stdlib(l: &Lua, stdout: &Rc<RefCell<String>>) -> mlua::Result<Table> { fn create_luahtml_stdlib(l: &Lua, stdout: &Rc<RefCell<String>>) -> mlua::Result<Table> {
let t = l.create_table()?; let t = l.create_table()?;
@ -37,14 +38,7 @@ fn create_luahtml_stdlib(l: &Lua, stdout: &Rc<RefCell<String>>) -> mlua::Result<
Ok(t) Ok(t)
} }
pub fn parse<F, A, T>(to_parse: T) -> Result<NodeRef> pub fn execute_lua(document: NodeRef) -> Result<NodeRef> {
where
F: fmt::Format,
A: Atomicity,
T: Into<Tendril<F, A>>,
tendril::Tendril<tendril::fmt::UTF8>: std::convert::From<tendril::Tendril<F, A>>,
{
let document = kuchikiki::parse_html().one(to_parse.into());
let lua = Lua::new(); let lua = Lua::new();
let globals = lua.globals(); let globals = lua.globals();
@ -79,12 +73,32 @@ where
Ok(document) Ok(document)
} }
pub fn expand_template(document: NodeRef, component_path: &PathBuf) -> Result<NodeRef> {
for i in document.select("include").map_err(|()| anyhow!("Error finding include"))? {
let attrs = match i.as_node().as_element() {
Some(e) => e.attributes.borrow(),
None => continue,
};
if let Some(include_path) = attrs.get("path") {
let mut item_path = component_path.clone();
item_path.push(include_path);
println!("{item_path:?}");
let component_text = read_to_string(item_path)?;
let new_node = kuchikiki::parse_html().one(component_text);
let replaced_node = expand_template(new_node, component_path)?;
i.as_node().insert_before(replaced_node);
i.as_node().detach();
}
}
Ok(document)
}
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use super::*; use super::*;
#[test] #[test]
fn base() { fn basic_lua() {
let page = r#" let page = r#"
<!DOCTYPE html> <!DOCTYPE html>
<html> <html>
@ -99,9 +113,59 @@ mod tests {
</lua></span> </lua></span>
</body> </body>
</html>"#; </html>"#;
let d = parse(page).unwrap(); let document = kuchikiki::parse_html().one(page);
let d = execute_lua(document).unwrap();
let text = d.select_first("span").unwrap().as_node().text_contents(); let text = d.select_first("span").unwrap().as_node().text_contents();
assert_eq!(text, "Test from Lua!\n"); assert_eq!(text, "Test from Lua!\n");
assert!(d.select_first("lua").is_err()); assert!(d.select_first("lua").is_err());
} }
#[test]
fn basic_include() {
let page = r#"
<!DOCTYPE html>
<html>
<head>
<title>Basic HTML Page</title>
</head>
<body>
<h1>Hello World</h1>
<p>This is a paragraph.</p>
<include path="comp1.html" />
<span><lua>
htmlua.println("Test from Lua!")
</lua></span>
</body>
</html>"#;
let mut p = PathBuf::from(env!("CARGO_MANIFEST_DIR"));
p.push("tests/components");
let document = kuchikiki::parse_html().one(page);
let d = expand_template(document, &p).unwrap();
let text = d.select_first("span").unwrap().as_node().text_contents();
assert_eq!(text, "included1");
assert!(d.select_first("include").is_err());
}
#[test]
fn recursive_include() {
let page = r#"
<!DOCTYPE html>
<html>
<head>
<title>Basic HTML Page</title>
</head>
<body>
<h1>Hello World</h1>
<include path="comp2_level1.html" />
</body>
</html>"#;
let mut p = PathBuf::from(env!("CARGO_MANIFEST_DIR"));
p.push("tests/components");
let document = kuchikiki::parse_html().one(page);
let d = expand_template(document, &p).unwrap();
let text = d.select_first("span").unwrap().as_node().text_contents();
assert_eq!(text, "included_twice");
assert!(d.select_first("include").is_err());
}
} }

1
htmlua-parser/tests/components/comp1.html Normal file
View file

@ -0,0 +1 @@
<span>included1</span>

4
htmlua-parser/tests/components/comp2_level1.html Normal file
View file

@ -0,0 +1,4 @@
<div>
<p>test</p>
<include path="comp2_level2.html"/>
</div>

1
htmlua-parser/tests/components/comp2_level2.html Normal file
View file

@ -0,0 +1 @@
<span>included_twice</span>