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

use xmlparser::Token;

use crate::{read_to_end, XmlError, XmlReader, XmlResult};

pub fn read_till_element_start(reader: &mut XmlReader<'_>, tag: &str) -> XmlResult<()> {
    while let Some(token) = reader.next() {
        let token = token?;
        match token {
            Token::ElementStart { span, .. } => {
                if tag == &span.as_str()[1..] {
                    break;
                } else {
                    // log::info!(
                    //     "Unhandled tag: {:?} when reading ParentElement {}. Skipping.",
                    //     __tag, stringify!(#ele_name)
                    // );
                    read_to_end(reader, &span.as_str()[1..])?;
                }
            }
            Token::ElementEnd { .. }
            | Token::Attribute { .. }
            | Token::Text { .. }
            | Token::Cdata { .. } => {
                return Err(XmlError::UnexpectedToken {
                    token: format!("{:?}", token),
                });
            }
            _ => (),
        }
    }

    Ok(())
}

#[test]
fn test_read_till_element_start() -> XmlResult<()> {
    use xmlparser::Tokenizer;

    let mut reader = Tokenizer::from("<tag/>").peekable();

    read_till_element_start(&mut reader, "tag")?;
    assert!(reader.next().is_some()); // "/>"
    assert!(reader.next().is_none());

    let mut reader = Tokenizer::from("<parent><skip/><tag/></parent>").peekable();

    assert!(reader.next().is_some()); // "<parent"
    assert!(reader.next().is_some()); // ">"
    read_till_element_start(&mut reader, "tag")?;
    assert!(reader.next().is_some()); // "/>"
    assert!(reader.next().is_some()); // "</parent>"
    assert!(reader.next().is_none());

    let mut reader = Tokenizer::from("<parent><skip></skip><tag/></parent>").peekable();

    assert!(reader.next().is_some()); // "<parent"
    assert!(reader.next().is_some()); // ">"
    read_till_element_start(&mut reader, "tag")?;
    assert!(reader.next().is_some()); // "/>"
    assert!(reader.next().is_some()); // "</parent>"
    assert!(reader.next().is_none());

    let mut reader = Tokenizer::from("<parent><skip><skip/></skip><tag/></parent>").peekable();

    assert!(reader.next().is_some()); // "<parent"
    assert!(reader.next().is_some()); // ">"
    read_till_element_start(&mut reader, "tag")?;
    assert!(reader.next().is_some()); // "/>"
    assert!(reader.next().is_some()); // "</parent>"
    assert!(reader.next().is_none());

    let mut reader =
        Tokenizer::from("<parent><skip><skip></skip></skip><tag/></parent>").peekable();

    assert!(reader.next().is_some()); // "<parent"
    assert!(reader.next().is_some()); // ">"
    read_till_element_start(&mut reader, "tag")?;
    assert!(reader.next().is_some()); // "/>"
    assert!(reader.next().is_some()); // "</parent>"
    assert!(reader.next().is_none());

    Ok(())
}