archive
/
md-pdf-rs
Archived
1
Fork 0
This repository has been archived on 2023-03-26. You can view files and clone it, but cannot push or open issues or pull requests.
md-pdf-rs/src/html.rs

54 lines
1.5 KiB
Rust

use sciter::{Window, Element};
use std::ops::Deref;
pub fn get_root(frame: Window) -> Element {
return frame.get_host().deref().get_root().expect("Failed to get root of window");
}
pub fn sciter_start(source: String) -> Element {
let mut frame = Window::new();
frame.load_html(&source.as_bytes(), None);
return get_root(frame);
}
pub fn get_html(element: Element) -> String {
element.update(true).expect("Failed to update");
return String::from_utf8(element.get_html(true)).expect(&format!(
"Failed to get HTML from {}.",
element.get_tag()
));
}
pub fn find_all(root: &mut Element, selector: &str) -> Vec<Element> {
let elements = root.find_all(selector).expect(&format!("Failed to get {}.", selector));
if elements.is_none() {
return Vec::new();
}
return elements.unwrap();
}
pub fn find_first(root: &mut Element, selector: &str) -> Element {
let mut all_matches = find_all(root, selector);
all_matches.reverse();
return all_matches.pop().expect(&format!("Failed to find {}.", selector));
}
pub fn destroy_matching(root: &mut Element, selector: &str) {
let matches = find_all(root, selector);
if matches.is_empty() {
return;
}
for mut ele in matches {
ele.destroy().expect("Failed to delete");
}
}
pub fn get_head(root: &mut Element) -> Element {
return find_first(root, "head");
}
pub fn get_body(root: &mut Element) -> Element {
return find_first(root, "body");
}