css_ast/functions/
attr_function.rs

1use super::prelude::*;
2use crate::Syntax;
3use css_parse::ComponentValues;
4
5/// <https://drafts.csswg.org/css-values-5/#attr-notation>
6///
7/// ```text,ignore
8/// attr() = attr( <attr-name> <attr-type>? , <declaration-value>?)
9/// <attr-type> = type( <syntax> ) | raw-string | <attr-unit>
10/// ```
11#[derive(Parse, Peek, ToSpan, ToCursors, SemanticEq, Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
12#[cfg_attr(feature = "serde", derive(serde::Serialize), serde())]
13#[cfg_attr(feature = "visitable", derive(csskit_derives::Visitable), visit(self))]
14pub struct AttrFunction<'a> {
15	#[atom(CssAtomSet::Attr)]
16	pub name: T![Function],
17	pub params: AttrFunctionParams<'a>,
18	pub close: T![')'],
19}
20
21#[derive(Parse, Peek, ToSpan, ToCursors, SemanticEq, Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
22#[cfg_attr(feature = "serde", derive(serde::Serialize), serde())]
23pub struct AttrFunctionParams<'a>(AttrName, Option<AttrType>, Option<T![,]>, Option<ComponentValues<'a>>);
24
25// <attr-name> = [ <ident-token>? '|' ]? <ident-token>
26#[derive(ToSpan, ToCursors, SemanticEq, Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
27#[cfg_attr(feature = "serde", derive(serde::Serialize), serde())]
28pub struct AttrName(pub Option<T![Ident]>, pub Option<T![|]>, pub Option<T![Ident]>);
29
30impl<'a> Peek<'a> for AttrName {
31	fn peek<I>(p: &Parser<'a, I>, c: Cursor) -> bool
32	where
33		I: Iterator<Item = Cursor> + Clone,
34	{
35		<T![Ident]>::peek(p, c)
36	}
37}
38
39impl<'a> Parse<'a> for AttrName {
40	fn parse<I>(p: &mut Parser<'a, I>) -> ParserResult<Self>
41	where
42		I: Iterator<Item = Cursor> + Clone,
43	{
44		let a = p.parse_if_peek::<T![Ident]>()?;
45		let b = p.parse_if_peek::<T![|]>()?;
46
47		if a.is_some() && b.is_none() {
48			return Ok(Self(None, None, a));
49		}
50
51		if a.is_none() && b.is_some() {
52			return Ok(Self(None, b, Some(p.parse::<T![Ident]>()?)));
53		}
54
55		if a.is_none() && b.is_none() {
56			Err(Diagnostic::new(p.next(), Diagnostic::expected_ident))?
57		}
58
59		debug_assert!(a.is_some() && b.is_some());
60
61		Ok(Self(a, b, Some(p.parse::<T![Ident]>()?)))
62	}
63}
64
65// <attr-type> = type( <syntax> ) | raw-string | <attr-unit>
66#[derive(Parse, Peek, ToSpan, ToCursors, SemanticEq, Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
67#[cfg_attr(feature = "serde", derive(serde::Serialize), serde())]
68pub enum AttrType {
69	#[atom(CssAtomSet::Type)]
70	Type(T![Function], Syntax, T![')']),
71	#[atom(CssAtomSet::RawString)]
72	RawString(T![Ident]),
73	Unit(T![Ident]),
74}
75
76#[cfg(test)]
77mod tests {
78	use super::*;
79	use crate::CssAtomSet;
80	use css_parse::{assert_parse, assert_parse_error};
81
82	#[test]
83	fn size_test() {
84		assert_eq!(std::mem::size_of::<AttrFunction>(), 152);
85	}
86
87	#[test]
88	fn test_writes() {
89		assert_parse!(CssAtomSet::ATOMS, AttrFunction, "attr(foo)");
90		assert_parse!(CssAtomSet::ATOMS, AttrFunction, "attr(foo)");
91		assert_parse!(CssAtomSet::ATOMS, AttrFunction, "attr(bar px)");
92		assert_parse!(CssAtomSet::ATOMS, AttrFunction, "attr(foo|bar px)");
93		assert_parse!(CssAtomSet::ATOMS, AttrFunction, "attr(foo|bar)");
94		assert_parse!(CssAtomSet::ATOMS, AttrFunction, "attr(|bar)");
95		assert_parse!(CssAtomSet::ATOMS, AttrFunction, "attr(|bar px)");
96	}
97
98	#[test]
99	fn test_errors() {
100		assert_parse_error!(CssAtomSet::ATOMS, AttrName, "a|b|c");
101		assert_parse_error!(CssAtomSet::ATOMS, AttrFunction, "attrr(foo)");
102		assert_parse_error!(CssAtomSet::ATOMS, AttrFunction, "attr()");
103		assert_parse_error!(CssAtomSet::ATOMS, AttrFunction, "attr(|)");
104	}
105}
css_ast/functions/attr_function.rs

css_ast/functions/
attr_function.rs
