pep/xmpp-rs-mirror
1
0
Fork 0
mirror of https://gitlab.com/xmpp-rs/xmpp-rs.git synced 2024-07-12 22:21:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

token: use many0 combinator for StartTag parsing

Browse source
This commit is contained in:
Astro 2022-03-23 19:30:53 +01:00
parent 5c5e32e7ff
commit ce1d95710b
1 changed files with 22 additions and 27 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

49
minidom/src/token.rs
View file

@ -81,6 +81,7 @@ impl Token {
fn parse_tag(s: &[u8]) -> IResult<&[u8], Token> {
let (s, _) = tag("<")(s)?;
alt((|s| -> IResult<&[u8], Token> {
// CDATA
let (s, _) = tag("![CDATA[")(s)?;
let mut end = None;
for i in 0..s.len() - 2 {
@ -96,6 +97,7 @@ impl Token {
Err(nom::Err::Incomplete(nom::Needed::Unknown))
}
}, |s| {
// EndTag
let (s, _) = tag("/")(s)?;
let (s, _) = space0(s)?;
let (s, name) = take_while1(|b| !(is_space(b) || b == b'>'))(s)?;
@ -104,34 +106,27 @@ impl Token {
let name = Self::str_from_utf8(name)?;
Ok((s, Token::EndTag { name: name.into() }))
}, |s| {
// StartTag
let (s, _) = space0(s)?;
let (s, name) = take_while1(|b| !(is_space(b) || b == b'>' || b == b'/'))(s)?;
let mut attrs = vec![];
let mut self_closing = false;
let mut s_ = s;
loop {
let (s, _) = space0(s_)?;
let (s, attr) = alt((|s| {
let (s, _) = tag("/")(s)?;
let (s, _) = space0(s)?;
let (s, _) = tag(">")(s)?;
self_closing = true;
Ok((s, None))
}, |s| {
let (s, _) = tag(">")(s)?;
Ok((s, None))
}, |s| {
let (s, (name, value)) = Self::parse_attr(s)?;
Ok((s, Some((name, value))))
}))(s)?;
s_ = s;
if let Some(attr) = attr {
attrs.push(attr);
} else {
break;
}
}
Ok((s_, Token::StartTag {
let (s, _) = space0(s)?;
let (s, attrs) = many0(|s| {
let (s, (name, value)) = Self::parse_attr(s)?;
let (s, _) = space0(s)?;
Ok((s, (name, value)))
})(s)?;
let (s, self_closing) = alt((|s| {
let (s, _) = tag("/")(s)?;
let (s, _) = space0(s)?;
let (s, _) = tag(">")(s)?;
Ok((s, true))
}, |s| {
let (s, _) = tag(">")(s)?;
Ok((s, false))
}))(s)?;
Ok((s, Token::StartTag {
name: Self::str_from_utf8(name)?
.into(),
attrs: attrs.into_iter()
@ -143,7 +138,7 @@ impl Token {
}
fn parse_attr(s: &[u8]) -> IResult<&[u8], (&str, String)> {
let (s, name) = take_while1(|b| !(is_space(b) || b == b'='))(s)?;
let (s, name) = take_while1(|b| !(is_space(b) || b == b'=' || b == b'/' || b == b'>'))(s)?;
let name = Self::str_from_utf8(name)?;
let (s, _) = space0(s)?;
let (s, _) = tag("=")(s)?;