blitz_html/
html_document.rs

1use std::ops::{Deref, DerefMut};
2
3use crate::DocumentHtmlParser;
4
5use blitz_dom::{BaseDocument, DEFAULT_CSS, DocGuard, DocGuardMut, Document, DocumentConfig};
6
7pub struct HtmlDocument {
8    inner: BaseDocument,
9}
10
11impl Deref for HtmlDocument {
12    type Target = BaseDocument;
13    fn deref(&self) -> &BaseDocument {
14        &self.inner
15    }
16}
17impl DerefMut for HtmlDocument {
18    fn deref_mut(&mut self) -> &mut Self::Target {
19        &mut self.inner
20    }
21}
22impl From<HtmlDocument> for BaseDocument {
23    fn from(doc: HtmlDocument) -> BaseDocument {
24        doc.inner
25    }
26}
27impl Document for HtmlDocument {
28    fn inner(&self) -> DocGuard<'_> {
29        DocGuard::Ref(&self.inner)
30    }
31
32    fn inner_mut(&mut self) -> DocGuardMut<'_> {
33        DocGuardMut::Ref(&mut self.inner)
34    }
35}
36
37impl HtmlDocument {
38    /// Parse HTML (or XHTML) into an [`HtmlDocument`]
39    pub fn from_html(html: &str, mut config: DocumentConfig) -> Self {
40        if let Some(ss) = &mut config.ua_stylesheets {
41            if !ss.iter().any(|s| s == DEFAULT_CSS) {
42                ss.push(String::from(DEFAULT_CSS));
43            }
44        }
45        let mut doc = BaseDocument::new(config);
46        let mut mutr = doc.mutate();
47        DocumentHtmlParser::parse_into_mutator(&mut mutr, html);
48        drop(mutr);
49        HtmlDocument { inner: doc }
50    }
51
52    /// Convert the [`HtmlDocument`] into it's inner [`BaseDocument`]
53    pub fn into_inner(self) -> BaseDocument {
54        self.into()
55    }
56}
blitz_html/html_document.rs

blitz_html/
html_document.rs
