htmlproc 0.3.3

HTML processors as utils written in Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98

use html5ever::tendril::TendrilSink;
use html5ever::{parse_document, ParseOpts};
use markup5ever_rcdom::{Handle, NodeData, RcDom};

use crate::core::consts::SELF_CLOSING_TAGS;

mod tests;

/// omits specific enclosures of tags in html
///
/// [feature entry point]
///
/// ```rust
/// use htmlproc::omit_enclosure::manipulate;
///
/// let source: &str = "<div>outside <span>inside</span>\n</div>";
/// let target_tags: &[&str;1] = &["span"];
/// let expect: &str = "<div>outside inside\n</div>";
///
/// let result = manipulate(source, target_tags);
/// assert_eq!(result, expect);
/// ```
///
pub fn manipulate(html: &str, omit_tags: &[&str]) -> String {
    let dom = parse_document(RcDom::default(), ParseOpts::default())
        .from_utf8()
        .read_from(&mut html.as_bytes())
        .unwrap();

    let mut omit_tags = Vec::from(omit_tags);
    // omits document outline tags if necessary
    if !html.contains("<html>") {
        omit_tags.push("html")
    };
    if !html.contains("<head>") {
        omit_tags.push("head")
    };
    if !html.contains("<body>") {
        omit_tags.push("body")
    };
    let omit_tags = omit_tags.as_slice();

    let mut output = String::new();
    scan(&dom.document, omit_tags, &mut output);

    output
}

/// scan to manipulate dom recursively
fn scan(handle: &Handle, omit_tags: &[&str], output: &mut String) {
    let node = handle;

    match &node.data {
        NodeData::Document => {
            for child in node.children.borrow().iter() {
                scan(child, omit_tags, output);
            }
        }
        NodeData::Element {
            ref name,
            ref attrs,
            ..
        } => {
            let tag_name = name.local.as_ref();
            if !omit_tags.contains(&tag_name) {
                // start tag
                output.push('<');
                output.push_str(tag_name);

                let attrs = attrs
                    .clone()
                    .into_inner()
                    .iter()
                    .map(|x| format!(" {}=\"{}\"", x.name.local, x.value))
                    .collect::<Vec<String>>()
                    .join("");
                output.push_str(attrs.as_str());

                output.push('>');
            }

            for child in node.children.borrow().iter() {
                scan(child, omit_tags, output);
            }

            if !SELF_CLOSING_TAGS.contains(&tag_name) && !omit_tags.contains(&tag_name) {
                // end tag
                output.push_str("</");
                output.push_str(tag_name);
                output.push('>');
            }
        }
        NodeData::Text { ref contents } => {
            output.push_str(&contents.borrow());
        }
        _ => {}
    }
}