css_ast/selector/
namespace.rs

1use css_parse::{Cursor, Diagnostic, KindSet, Parse, Parser, Result as ParserResult, T};
2use csskit_derives::{IntoCursor, Peek, SemanticEq, ToCursors, ToSpan};
3
4use super::Tag;
5
6// https://drafts.csswg.org/selectors/#combinators
7#[derive(Peek, ToSpan, ToCursors, SemanticEq, Debug, Copy, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
8#[cfg_attr(feature = "serde", derive(serde::Serialize), serde())]
9#[cfg_attr(feature = "visitable", derive(csskit_derives::Visitable), visit(self))]
10#[derive(csskit_derives::NodeWithMetadata)]
11pub struct Namespace {
12	pub prefix: Option<NamespacePrefix>,
13	pub tag: NamespaceTag,
14}
15
16impl<'a> Parse<'a> for Namespace {
17	fn parse<I>(p: &mut Parser<'a, I>) -> ParserResult<Self>
18	where
19		I: Iterator<Item = Cursor> + Clone,
20	{
21		if p.peek::<T![*|]>() {
22			let prefix = p.parse::<NamespacePrefix>()?;
23			let tag = p.parse::<NamespaceTag>()?;
24			return Ok(Self { prefix: Some(prefix), tag });
25		}
26		if p.peek::<T![|]>() {
27			let prefix = p.parse::<NamespacePrefix>()?;
28			let tag = p.parse::<NamespaceTag>()?;
29			return Ok(Self { prefix: Some(prefix), tag });
30		}
31
32		let ident = p.parse::<T![Ident]>()?;
33		let skip = p.set_skip(KindSet::NONE);
34		if p.peek::<T![|]>() && !p.peek::<T![|=]>() {
35			let pipe = p.parse::<T![|]>();
36			let tag = p.parse::<NamespaceTag>();
37			p.set_skip(skip);
38			let prefix = NamespacePrefix::Name(ident, pipe?);
39			return Ok(Self { prefix: Some(prefix), tag: tag? });
40		}
41		let tag = p.parse::<NamespaceTag>()?;
42		Ok(Self { prefix: None, tag })
43	}
44}
45
46#[derive(Peek, ToSpan, ToCursors, SemanticEq, Debug, Copy, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
47#[cfg_attr(feature = "serde", derive(serde::Serialize), serde())]
48pub enum NamespacePrefix {
49	None(T![|]),
50	Name(T![Ident], T![|]),
51	Wildcard(T![*], T![|]),
52}
53
54impl<'a> Parse<'a> for NamespacePrefix {
55	fn parse<I>(p: &mut Parser<'a, I>) -> ParserResult<Self>
56	where
57		I: Iterator<Item = Cursor> + Clone,
58	{
59		if p.peek::<T![|]>() {
60			let pipe = p.parse::<T![|]>()?;
61			Ok(Self::None(pipe))
62		} else if p.peek::<T![*]>() {
63			let star = p.parse::<T![*]>()?;
64			let skip = p.set_skip(KindSet::NONE);
65			let pipe = p.parse::<T![|]>();
66			p.set_skip(skip);
67			let pipe = pipe?;
68			Ok(Self::Wildcard(star, pipe))
69		} else {
70			let star = p.parse::<T![Ident]>()?;
71			let skip = p.set_skip(KindSet::NONE);
72			let pipe = p.parse::<T![|]>();
73			p.set_skip(skip);
74			let pipe = pipe?;
75			Ok(Self::Name(star, pipe))
76		}
77	}
78}
79
80#[derive(Peek, ToCursors, IntoCursor, Debug, Copy, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
81#[cfg_attr(feature = "serde", derive(serde::Serialize), serde())]
82pub enum NamespaceTag {
83	Wildcard(T![*]),
84	Tag(Tag),
85}
86
87impl<'a> Parse<'a> for NamespaceTag {
88	fn parse<I>(p: &mut Parser<'a, I>) -> ParserResult<Self>
89	where
90		I: Iterator<Item = Cursor> + Clone,
91	{
92		if p.peek::<Self>() {
93			if p.peek::<T![*]>() { Ok(Self::Wildcard(p.parse::<T![*]>()?)) } else { Ok(Self::Tag(p.parse::<Tag>()?)) }
94		} else {
95			Err(Diagnostic::new(p.next(), Diagnostic::unexpected))?
96		}
97	}
98}
99
100#[cfg(test)]
101mod tests {
102	use super::*;
103	use crate::CssAtomSet;
104	use css_parse::{assert_parse, assert_parse_error};
105
106	#[test]
107	fn size_test() {
108		assert_eq!(std::mem::size_of::<Namespace>(), 48);
109	}
110
111	#[test]
112	fn test_writes() {
113		assert_parse!(CssAtomSet::ATOMS, Namespace, "*|a");
114		assert_parse!(CssAtomSet::ATOMS, Namespace, "html|div");
115		assert_parse!(CssAtomSet::ATOMS, Namespace, "|span");
116	}
117
118	#[test]
119	fn test_errors() {
120		assert_parse_error!(CssAtomSet::ATOMS, Namespace, "* | a");
121	}
122}