✨ Incluye maud-macros 0.25.0 en código
This commit is contained in:
parent
aa3bd08303
commit
51f33835af
8 changed files with 1374 additions and 0 deletions
|
|
@ -19,5 +19,7 @@ proc-macro = true
|
||||||
[dependencies]
|
[dependencies]
|
||||||
concat-string = "1.0.1"
|
concat-string = "1.0.1"
|
||||||
proc-macro2 = "1.0"
|
proc-macro2 = "1.0"
|
||||||
|
proc-macro-crate = "1.3.1"
|
||||||
|
proc-macro-error = "1.0.4"
|
||||||
quote = "1.0"
|
quote = "1.0"
|
||||||
syn = { version = "1.0", features = ["full"] }
|
syn = { version = "1.0", features = ["full"] }
|
||||||
|
|
|
||||||
|
|
@ -12,6 +12,15 @@ Rendering*).
|
||||||
constantes. No puede considerarse preparado hasta que se libere la versión **0.1.0**.
|
constantes. No puede considerarse preparado hasta que se libere la versión **0.1.0**.
|
||||||
|
|
||||||
|
|
||||||
|
# 🔖 Créditos
|
||||||
|
|
||||||
|
Incluye una versión adaptada del crate [maud-macros](https://crates.io/crates/maud_macros) de
|
||||||
|
[Chris Wong](https://crates.io/users/lambda-fairy) (versión
|
||||||
|
[0.25.0](https://github.com/lambda-fairy/maud/tree/v0.25.0/maud_macros)), para incorporar sus
|
||||||
|
funcionalidades en **PageTop** sin requerir una entrada adicional en el archivo Cargo.toml de cada
|
||||||
|
proyecto.
|
||||||
|
|
||||||
|
|
||||||
# 📜 Licencia
|
# 📜 Licencia
|
||||||
|
|
||||||
Este proyecto tiene licencia, de hecho tiene dos, puedes aplicar cualquiera de las siguientes a tu
|
Este proyecto tiene licencia, de hecho tiene dos, puedes aplicar cualquiera de las siguientes a tu
|
||||||
|
|
|
||||||
|
|
@ -1,8 +1,17 @@
|
||||||
|
mod maud;
|
||||||
|
|
||||||
use concat_string::concat_string;
|
use concat_string::concat_string;
|
||||||
use proc_macro::TokenStream;
|
use proc_macro::TokenStream;
|
||||||
|
use proc_macro_error::proc_macro_error;
|
||||||
use quote::{quote, quote_spanned, ToTokens};
|
use quote::{quote, quote_spanned, ToTokens};
|
||||||
use syn::{parse_macro_input, parse_str, ItemFn};
|
use syn::{parse_macro_input, parse_str, ItemFn};
|
||||||
|
|
||||||
|
#[proc_macro]
|
||||||
|
#[proc_macro_error]
|
||||||
|
pub fn html(input: TokenStream) -> TokenStream {
|
||||||
|
maud::expand(input.into()).into()
|
||||||
|
}
|
||||||
|
|
||||||
#[proc_macro_attribute]
|
#[proc_macro_attribute]
|
||||||
pub fn fn_builder(_attr: TokenStream, item: TokenStream) -> TokenStream {
|
pub fn fn_builder(_attr: TokenStream, item: TokenStream) -> TokenStream {
|
||||||
let fn_item = parse_macro_input!(item as ItemFn);
|
let fn_item = parse_macro_input!(item as ItemFn);
|
||||||
|
|
|
||||||
39
pagetop-macros/src/maud.rs
Normal file
39
pagetop-macros/src/maud.rs
Normal file
|
|
@ -0,0 +1,39 @@
|
||||||
|
// #![doc(html_root_url = "https://docs.rs/maud_macros/0.25.0")]
|
||||||
|
// TokenStream values are reference counted, and the mental overhead of tracking
|
||||||
|
// lifetimes outweighs the marginal gains from explicit borrowing
|
||||||
|
// #![allow(clippy::needless_pass_by_value)]
|
||||||
|
|
||||||
|
mod ast;
|
||||||
|
mod escape;
|
||||||
|
mod generate;
|
||||||
|
mod parse;
|
||||||
|
|
||||||
|
use proc_macro_crate::{crate_name, FoundCrate};
|
||||||
|
use proc_macro2::{Ident, Span, TokenStream, TokenTree};
|
||||||
|
use quote::quote;
|
||||||
|
|
||||||
|
pub fn expand(input: TokenStream) -> TokenStream {
|
||||||
|
let output_ident = TokenTree::Ident(Ident::new("__maud_output", Span::mixed_site()));
|
||||||
|
// Heuristic: the size of the resulting markup tends to correlate with the
|
||||||
|
// code size of the template itself
|
||||||
|
let size_hint = input.to_string().len();
|
||||||
|
let markups = parse::parse(input);
|
||||||
|
let stmts = generate::generate(markups, output_ident.clone());
|
||||||
|
|
||||||
|
let found_crate = crate_name("pagetop").expect("pagetop is present in `Cargo.toml`");
|
||||||
|
let pre_escaped = match found_crate {
|
||||||
|
FoundCrate::Itself => quote!(
|
||||||
|
crate::html::PreEscaped(#output_ident)
|
||||||
|
),
|
||||||
|
_ => quote!(
|
||||||
|
pagetop::html::PreEscaped(#output_ident)
|
||||||
|
)
|
||||||
|
};
|
||||||
|
|
||||||
|
quote!({
|
||||||
|
extern crate alloc;
|
||||||
|
let mut #output_ident = alloc::string::String::with_capacity(#size_hint);
|
||||||
|
#stmts
|
||||||
|
#pre_escaped
|
||||||
|
})
|
||||||
|
}
|
||||||
221
pagetop-macros/src/maud/ast.rs
Normal file
221
pagetop-macros/src/maud/ast.rs
Normal file
|
|
@ -0,0 +1,221 @@
|
||||||
|
use proc_macro2::{TokenStream, TokenTree};
|
||||||
|
use proc_macro_error::SpanRange;
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub enum Markup {
|
||||||
|
/// Used as a placeholder value on parse error.
|
||||||
|
ParseError {
|
||||||
|
span: SpanRange,
|
||||||
|
},
|
||||||
|
Block(Block),
|
||||||
|
Literal {
|
||||||
|
content: String,
|
||||||
|
span: SpanRange,
|
||||||
|
},
|
||||||
|
Symbol {
|
||||||
|
symbol: TokenStream,
|
||||||
|
},
|
||||||
|
Splice {
|
||||||
|
expr: TokenStream,
|
||||||
|
outer_span: SpanRange,
|
||||||
|
},
|
||||||
|
Element {
|
||||||
|
name: TokenStream,
|
||||||
|
attrs: Vec<Attr>,
|
||||||
|
body: ElementBody,
|
||||||
|
},
|
||||||
|
Let {
|
||||||
|
at_span: SpanRange,
|
||||||
|
tokens: TokenStream,
|
||||||
|
},
|
||||||
|
Special {
|
||||||
|
segments: Vec<Special>,
|
||||||
|
},
|
||||||
|
Match {
|
||||||
|
at_span: SpanRange,
|
||||||
|
head: TokenStream,
|
||||||
|
arms: Vec<MatchArm>,
|
||||||
|
arms_span: SpanRange,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Markup {
|
||||||
|
pub fn span(&self) -> SpanRange {
|
||||||
|
match *self {
|
||||||
|
Markup::ParseError { span } => span,
|
||||||
|
Markup::Block(ref block) => block.span(),
|
||||||
|
Markup::Literal { span, .. } => span,
|
||||||
|
Markup::Symbol { ref symbol } => span_tokens(symbol.clone()),
|
||||||
|
Markup::Splice { outer_span, .. } => outer_span,
|
||||||
|
Markup::Element {
|
||||||
|
ref name, ref body, ..
|
||||||
|
} => {
|
||||||
|
let name_span = span_tokens(name.clone());
|
||||||
|
name_span.join_range(body.span())
|
||||||
|
}
|
||||||
|
Markup::Let {
|
||||||
|
at_span,
|
||||||
|
ref tokens,
|
||||||
|
} => at_span.join_range(span_tokens(tokens.clone())),
|
||||||
|
Markup::Special { ref segments } => join_ranges(segments.iter().map(Special::span)),
|
||||||
|
Markup::Match {
|
||||||
|
at_span, arms_span, ..
|
||||||
|
} => at_span.join_range(arms_span),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub enum Attr {
|
||||||
|
Class {
|
||||||
|
dot_span: SpanRange,
|
||||||
|
name: Markup,
|
||||||
|
toggler: Option<Toggler>,
|
||||||
|
},
|
||||||
|
Id {
|
||||||
|
hash_span: SpanRange,
|
||||||
|
name: Markup,
|
||||||
|
},
|
||||||
|
Named {
|
||||||
|
named_attr: NamedAttr,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Attr {
|
||||||
|
pub fn span(&self) -> SpanRange {
|
||||||
|
match *self {
|
||||||
|
Attr::Class {
|
||||||
|
dot_span,
|
||||||
|
ref name,
|
||||||
|
ref toggler,
|
||||||
|
} => {
|
||||||
|
let name_span = name.span();
|
||||||
|
let dot_name_span = dot_span.join_range(name_span);
|
||||||
|
if let Some(toggler) = toggler {
|
||||||
|
dot_name_span.join_range(toggler.cond_span)
|
||||||
|
} else {
|
||||||
|
dot_name_span
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Attr::Id {
|
||||||
|
hash_span,
|
||||||
|
ref name,
|
||||||
|
} => {
|
||||||
|
let name_span = name.span();
|
||||||
|
hash_span.join_range(name_span)
|
||||||
|
}
|
||||||
|
Attr::Named { ref named_attr } => named_attr.span(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub enum ElementBody {
|
||||||
|
Void { semi_span: SpanRange },
|
||||||
|
Block { block: Block },
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ElementBody {
|
||||||
|
pub fn span(&self) -> SpanRange {
|
||||||
|
match *self {
|
||||||
|
ElementBody::Void { semi_span } => semi_span,
|
||||||
|
ElementBody::Block { ref block } => block.span(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct Block {
|
||||||
|
pub markups: Vec<Markup>,
|
||||||
|
pub outer_span: SpanRange,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Block {
|
||||||
|
pub fn span(&self) -> SpanRange {
|
||||||
|
self.outer_span
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct Special {
|
||||||
|
pub at_span: SpanRange,
|
||||||
|
pub head: TokenStream,
|
||||||
|
pub body: Block,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Special {
|
||||||
|
pub fn span(&self) -> SpanRange {
|
||||||
|
let body_span = self.body.span();
|
||||||
|
self.at_span.join_range(body_span)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct NamedAttr {
|
||||||
|
pub name: TokenStream,
|
||||||
|
pub attr_type: AttrType,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl NamedAttr {
|
||||||
|
fn span(&self) -> SpanRange {
|
||||||
|
let name_span = span_tokens(self.name.clone());
|
||||||
|
if let Some(attr_type_span) = self.attr_type.span() {
|
||||||
|
name_span.join_range(attr_type_span)
|
||||||
|
} else {
|
||||||
|
name_span
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub enum AttrType {
|
||||||
|
Normal { value: Markup },
|
||||||
|
Optional { toggler: Toggler },
|
||||||
|
Empty { toggler: Option<Toggler> },
|
||||||
|
}
|
||||||
|
|
||||||
|
impl AttrType {
|
||||||
|
fn span(&self) -> Option<SpanRange> {
|
||||||
|
match *self {
|
||||||
|
AttrType::Normal { ref value } => Some(value.span()),
|
||||||
|
AttrType::Optional { ref toggler } => Some(toggler.span()),
|
||||||
|
AttrType::Empty { ref toggler } => toggler.as_ref().map(Toggler::span),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct Toggler {
|
||||||
|
pub cond: TokenStream,
|
||||||
|
pub cond_span: SpanRange,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Toggler {
|
||||||
|
fn span(&self) -> SpanRange {
|
||||||
|
self.cond_span
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct MatchArm {
|
||||||
|
pub head: TokenStream,
|
||||||
|
pub body: Block,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn span_tokens<I: IntoIterator<Item = TokenTree>>(tokens: I) -> SpanRange {
|
||||||
|
join_ranges(tokens.into_iter().map(|s| SpanRange::single_span(s.span())))
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn join_ranges<I: IntoIterator<Item = SpanRange>>(ranges: I) -> SpanRange {
|
||||||
|
let mut iter = ranges.into_iter();
|
||||||
|
let first = match iter.next() {
|
||||||
|
Some(span) => span,
|
||||||
|
None => return SpanRange::call_site(),
|
||||||
|
};
|
||||||
|
let last = iter.last().unwrap_or(first);
|
||||||
|
first.join_range(last)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn name_to_string(name: TokenStream) -> String {
|
||||||
|
name.into_iter().map(|token| token.to_string()).collect()
|
||||||
|
}
|
||||||
34
pagetop-macros/src/maud/escape.rs
Normal file
34
pagetop-macros/src/maud/escape.rs
Normal file
|
|
@ -0,0 +1,34 @@
|
||||||
|
// !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
|
||||||
|
// !!!!!!!! PLEASE KEEP THIS IN SYNC WITH `maud/src/escape.rs` !!!!!!!!!
|
||||||
|
// !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
|
||||||
|
|
||||||
|
extern crate alloc;
|
||||||
|
|
||||||
|
use alloc::string::String;
|
||||||
|
|
||||||
|
pub fn escape_to_string(input: &str, output: &mut String) {
|
||||||
|
for b in input.bytes() {
|
||||||
|
match b {
|
||||||
|
b'&' => output.push_str("&"),
|
||||||
|
b'<' => output.push_str("<"),
|
||||||
|
b'>' => output.push_str(">"),
|
||||||
|
b'"' => output.push_str("""),
|
||||||
|
_ => unsafe { output.as_mut_vec().push(b) },
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod test {
|
||||||
|
extern crate alloc;
|
||||||
|
|
||||||
|
use super::escape_to_string;
|
||||||
|
use alloc::string::String;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn it_works() {
|
||||||
|
let mut s = String::new();
|
||||||
|
escape_to_string("<script>launchMissiles()</script>", &mut s);
|
||||||
|
assert_eq!(s, "<script>launchMissiles()</script>");
|
||||||
|
}
|
||||||
|
}
|
||||||
308
pagetop-macros/src/maud/generate.rs
Normal file
308
pagetop-macros/src/maud/generate.rs
Normal file
|
|
@ -0,0 +1,308 @@
|
||||||
|
use proc_macro2::{Delimiter, Group, Ident, Literal, Span, TokenStream, TokenTree};
|
||||||
|
use proc_macro_error::SpanRange;
|
||||||
|
use quote::quote;
|
||||||
|
|
||||||
|
use crate::maud::{ast::*, escape};
|
||||||
|
|
||||||
|
use proc_macro_crate::{crate_name, FoundCrate};
|
||||||
|
|
||||||
|
pub fn generate(markups: Vec<Markup>, output_ident: TokenTree) -> TokenStream {
|
||||||
|
let mut build = Builder::new(output_ident.clone());
|
||||||
|
Generator::new(output_ident).markups(markups, &mut build);
|
||||||
|
build.finish()
|
||||||
|
}
|
||||||
|
|
||||||
|
struct Generator {
|
||||||
|
output_ident: TokenTree,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Generator {
|
||||||
|
fn new(output_ident: TokenTree) -> Generator {
|
||||||
|
Generator { output_ident }
|
||||||
|
}
|
||||||
|
|
||||||
|
fn builder(&self) -> Builder {
|
||||||
|
Builder::new(self.output_ident.clone())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn markups(&self, markups: Vec<Markup>, build: &mut Builder) {
|
||||||
|
for markup in markups {
|
||||||
|
self.markup(markup, build);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn markup(&self, markup: Markup, build: &mut Builder) {
|
||||||
|
match markup {
|
||||||
|
Markup::ParseError { .. } => {}
|
||||||
|
Markup::Block(Block {
|
||||||
|
markups,
|
||||||
|
outer_span,
|
||||||
|
}) => {
|
||||||
|
if markups
|
||||||
|
.iter()
|
||||||
|
.any(|markup| matches!(*markup, Markup::Let { .. }))
|
||||||
|
{
|
||||||
|
self.block(
|
||||||
|
Block {
|
||||||
|
markups,
|
||||||
|
outer_span,
|
||||||
|
},
|
||||||
|
build,
|
||||||
|
);
|
||||||
|
} else {
|
||||||
|
self.markups(markups, build);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Markup::Literal { content, .. } => build.push_escaped(&content),
|
||||||
|
Markup::Symbol { symbol } => self.name(symbol, build),
|
||||||
|
Markup::Splice { expr, .. } => self.splice(expr, build),
|
||||||
|
Markup::Element { name, attrs, body } => self.element(name, attrs, body, build),
|
||||||
|
Markup::Let { tokens, .. } => build.push_tokens(tokens),
|
||||||
|
Markup::Special { segments } => {
|
||||||
|
for Special { head, body, .. } in segments {
|
||||||
|
build.push_tokens(head);
|
||||||
|
self.block(body, build);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Markup::Match {
|
||||||
|
head,
|
||||||
|
arms,
|
||||||
|
arms_span,
|
||||||
|
..
|
||||||
|
} => {
|
||||||
|
let body = {
|
||||||
|
let mut build = self.builder();
|
||||||
|
for MatchArm { head, body } in arms {
|
||||||
|
build.push_tokens(head);
|
||||||
|
self.block(body, &mut build);
|
||||||
|
}
|
||||||
|
build.finish()
|
||||||
|
};
|
||||||
|
let mut body = TokenTree::Group(Group::new(Delimiter::Brace, body));
|
||||||
|
body.set_span(arms_span.collapse());
|
||||||
|
build.push_tokens(quote!(#head #body));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn block(
|
||||||
|
&self,
|
||||||
|
Block {
|
||||||
|
markups,
|
||||||
|
outer_span,
|
||||||
|
}: Block,
|
||||||
|
build: &mut Builder,
|
||||||
|
) {
|
||||||
|
let block = {
|
||||||
|
let mut build = self.builder();
|
||||||
|
self.markups(markups, &mut build);
|
||||||
|
build.finish()
|
||||||
|
};
|
||||||
|
let mut block = TokenTree::Group(Group::new(Delimiter::Brace, block));
|
||||||
|
block.set_span(outer_span.collapse());
|
||||||
|
build.push_tokens(TokenStream::from(block));
|
||||||
|
}
|
||||||
|
|
||||||
|
fn splice(&self, expr: TokenStream, build: &mut Builder) {
|
||||||
|
let output_ident = self.output_ident.clone();
|
||||||
|
|
||||||
|
let found_crate = crate_name("pagetop").expect("pagetop is present in `Cargo.toml`");
|
||||||
|
build.push_tokens(match found_crate {
|
||||||
|
FoundCrate::Itself => quote!(
|
||||||
|
crate::html::html_private::render_to!(&#expr, &mut #output_ident);
|
||||||
|
),
|
||||||
|
_ => quote!(
|
||||||
|
pagetop::html::html_private::render_to!(&#expr, &mut #output_ident);
|
||||||
|
)
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
fn element(&self, name: TokenStream, attrs: Vec<Attr>, body: ElementBody, build: &mut Builder) {
|
||||||
|
build.push_str("<");
|
||||||
|
self.name(name.clone(), build);
|
||||||
|
self.attrs(attrs, build);
|
||||||
|
build.push_str(">");
|
||||||
|
if let ElementBody::Block { block } = body {
|
||||||
|
self.markups(block.markups, build);
|
||||||
|
build.push_str("</");
|
||||||
|
self.name(name, build);
|
||||||
|
build.push_str(">");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn name(&self, name: TokenStream, build: &mut Builder) {
|
||||||
|
build.push_escaped(&name_to_string(name));
|
||||||
|
}
|
||||||
|
|
||||||
|
fn attrs(&self, attrs: Vec<Attr>, build: &mut Builder) {
|
||||||
|
for NamedAttr { name, attr_type } in desugar_attrs(attrs) {
|
||||||
|
match attr_type {
|
||||||
|
AttrType::Normal { value } => {
|
||||||
|
build.push_str(" ");
|
||||||
|
self.name(name, build);
|
||||||
|
build.push_str("=\"");
|
||||||
|
self.markup(value, build);
|
||||||
|
build.push_str("\"");
|
||||||
|
}
|
||||||
|
AttrType::Optional {
|
||||||
|
toggler: Toggler { cond, .. },
|
||||||
|
} => {
|
||||||
|
let inner_value = quote!(inner_value);
|
||||||
|
let body = {
|
||||||
|
let mut build = self.builder();
|
||||||
|
build.push_str(" ");
|
||||||
|
self.name(name, &mut build);
|
||||||
|
build.push_str("=\"");
|
||||||
|
self.splice(inner_value.clone(), &mut build);
|
||||||
|
build.push_str("\"");
|
||||||
|
build.finish()
|
||||||
|
};
|
||||||
|
build.push_tokens(quote!(if let Some(#inner_value) = (#cond) { #body }));
|
||||||
|
}
|
||||||
|
AttrType::Empty { toggler: None } => {
|
||||||
|
build.push_str(" ");
|
||||||
|
self.name(name, build);
|
||||||
|
}
|
||||||
|
AttrType::Empty {
|
||||||
|
toggler: Some(Toggler { cond, .. }),
|
||||||
|
} => {
|
||||||
|
let body = {
|
||||||
|
let mut build = self.builder();
|
||||||
|
build.push_str(" ");
|
||||||
|
self.name(name, &mut build);
|
||||||
|
build.finish()
|
||||||
|
};
|
||||||
|
build.push_tokens(quote!(if (#cond) { #body }));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
////////////////////////////////////////////////////////
|
||||||
|
|
||||||
|
fn desugar_attrs(attrs: Vec<Attr>) -> Vec<NamedAttr> {
|
||||||
|
let mut classes_static = vec![];
|
||||||
|
let mut classes_toggled = vec![];
|
||||||
|
let mut ids = vec![];
|
||||||
|
let mut named_attrs = vec![];
|
||||||
|
for attr in attrs {
|
||||||
|
match attr {
|
||||||
|
Attr::Class {
|
||||||
|
name,
|
||||||
|
toggler: Some(toggler),
|
||||||
|
..
|
||||||
|
} => classes_toggled.push((name, toggler)),
|
||||||
|
Attr::Class {
|
||||||
|
name,
|
||||||
|
toggler: None,
|
||||||
|
..
|
||||||
|
} => classes_static.push(name),
|
||||||
|
Attr::Id { name, .. } => ids.push(name),
|
||||||
|
Attr::Named { named_attr } => named_attrs.push(named_attr),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
let classes = desugar_classes_or_ids("class", classes_static, classes_toggled);
|
||||||
|
let ids = desugar_classes_or_ids("id", ids, vec![]);
|
||||||
|
classes.into_iter().chain(ids).chain(named_attrs).collect()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn desugar_classes_or_ids(
|
||||||
|
attr_name: &'static str,
|
||||||
|
values_static: Vec<Markup>,
|
||||||
|
values_toggled: Vec<(Markup, Toggler)>,
|
||||||
|
) -> Option<NamedAttr> {
|
||||||
|
if values_static.is_empty() && values_toggled.is_empty() {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
let mut markups = Vec::new();
|
||||||
|
let mut leading_space = false;
|
||||||
|
for name in values_static {
|
||||||
|
markups.extend(prepend_leading_space(name, &mut leading_space));
|
||||||
|
}
|
||||||
|
for (name, Toggler { cond, cond_span }) in values_toggled {
|
||||||
|
let body = Block {
|
||||||
|
markups: prepend_leading_space(name, &mut leading_space),
|
||||||
|
// TODO: is this correct?
|
||||||
|
outer_span: cond_span,
|
||||||
|
};
|
||||||
|
markups.push(Markup::Special {
|
||||||
|
segments: vec![Special {
|
||||||
|
at_span: SpanRange::call_site(),
|
||||||
|
head: quote!(if (#cond)),
|
||||||
|
body,
|
||||||
|
}],
|
||||||
|
});
|
||||||
|
}
|
||||||
|
Some(NamedAttr {
|
||||||
|
name: TokenStream::from(TokenTree::Ident(Ident::new(attr_name, Span::call_site()))),
|
||||||
|
attr_type: AttrType::Normal {
|
||||||
|
value: Markup::Block(Block {
|
||||||
|
markups,
|
||||||
|
outer_span: SpanRange::call_site(),
|
||||||
|
}),
|
||||||
|
},
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
fn prepend_leading_space(name: Markup, leading_space: &mut bool) -> Vec<Markup> {
|
||||||
|
let mut markups = Vec::new();
|
||||||
|
if *leading_space {
|
||||||
|
markups.push(Markup::Literal {
|
||||||
|
content: " ".to_owned(),
|
||||||
|
span: name.span(),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
*leading_space = true;
|
||||||
|
markups.push(name);
|
||||||
|
markups
|
||||||
|
}
|
||||||
|
|
||||||
|
////////////////////////////////////////////////////////
|
||||||
|
|
||||||
|
struct Builder {
|
||||||
|
output_ident: TokenTree,
|
||||||
|
tokens: Vec<TokenTree>,
|
||||||
|
tail: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Builder {
|
||||||
|
fn new(output_ident: TokenTree) -> Builder {
|
||||||
|
Builder {
|
||||||
|
output_ident,
|
||||||
|
tokens: Vec::new(),
|
||||||
|
tail: String::new(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn push_str(&mut self, string: &str) {
|
||||||
|
self.tail.push_str(string);
|
||||||
|
}
|
||||||
|
|
||||||
|
fn push_escaped(&mut self, string: &str) {
|
||||||
|
escape::escape_to_string(string, &mut self.tail);
|
||||||
|
}
|
||||||
|
|
||||||
|
fn push_tokens(&mut self, tokens: TokenStream) {
|
||||||
|
self.cut();
|
||||||
|
self.tokens.extend(tokens);
|
||||||
|
}
|
||||||
|
|
||||||
|
fn cut(&mut self) {
|
||||||
|
if self.tail.is_empty() {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
let push_str_expr = {
|
||||||
|
let output_ident = self.output_ident.clone();
|
||||||
|
let string = TokenTree::Literal(Literal::string(&self.tail));
|
||||||
|
quote!(#output_ident.push_str(#string);)
|
||||||
|
};
|
||||||
|
self.tail.clear();
|
||||||
|
self.tokens.extend(push_str_expr);
|
||||||
|
}
|
||||||
|
|
||||||
|
fn finish(mut self) -> TokenStream {
|
||||||
|
self.cut();
|
||||||
|
self.tokens.into_iter().collect()
|
||||||
|
}
|
||||||
|
}
|
||||||
752
pagetop-macros/src/maud/parse.rs
Normal file
752
pagetop-macros/src/maud/parse.rs
Normal file
|
|
@ -0,0 +1,752 @@
|
||||||
|
use proc_macro2::{Delimiter, Ident, Literal, Spacing, Span, TokenStream, TokenTree};
|
||||||
|
use proc_macro_error::{abort, abort_call_site, emit_error, SpanRange};
|
||||||
|
use std::collections::HashMap;
|
||||||
|
|
||||||
|
use syn::Lit;
|
||||||
|
|
||||||
|
use crate::maud::ast;
|
||||||
|
|
||||||
|
pub fn parse(input: TokenStream) -> Vec<ast::Markup> {
|
||||||
|
Parser::new(input).markups()
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
struct Parser {
|
||||||
|
/// If we're inside an attribute, then this contains the attribute name.
|
||||||
|
current_attr: Option<String>,
|
||||||
|
input: <TokenStream as IntoIterator>::IntoIter,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Iterator for Parser {
|
||||||
|
type Item = TokenTree;
|
||||||
|
|
||||||
|
fn next(&mut self) -> Option<TokenTree> {
|
||||||
|
self.input.next()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Parser {
|
||||||
|
fn new(input: TokenStream) -> Parser {
|
||||||
|
Parser {
|
||||||
|
current_attr: None,
|
||||||
|
input: input.into_iter(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn with_input(&self, input: TokenStream) -> Parser {
|
||||||
|
Parser {
|
||||||
|
current_attr: self.current_attr.clone(),
|
||||||
|
input: input.into_iter(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Returns the next token in the stream without consuming it.
|
||||||
|
fn peek(&mut self) -> Option<TokenTree> {
|
||||||
|
self.clone().next()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Returns the next two tokens in the stream without consuming them.
|
||||||
|
fn peek2(&mut self) -> Option<(TokenTree, Option<TokenTree>)> {
|
||||||
|
let mut clone = self.clone();
|
||||||
|
clone.next().map(|first| (first, clone.next()))
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Advances the cursor by one step.
|
||||||
|
fn advance(&mut self) {
|
||||||
|
self.next();
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Advances the cursor by two steps.
|
||||||
|
fn advance2(&mut self) {
|
||||||
|
self.next();
|
||||||
|
self.next();
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Parses multiple blocks of markup.
|
||||||
|
fn markups(&mut self) -> Vec<ast::Markup> {
|
||||||
|
let mut result = Vec::new();
|
||||||
|
loop {
|
||||||
|
match self.peek2() {
|
||||||
|
None => break,
|
||||||
|
Some((TokenTree::Punct(ref punct), _)) if punct.as_char() == ';' => self.advance(),
|
||||||
|
Some((TokenTree::Punct(ref punct), Some(TokenTree::Ident(ref ident))))
|
||||||
|
if punct.as_char() == '@' && *ident == "let" =>
|
||||||
|
{
|
||||||
|
self.advance2();
|
||||||
|
let keyword = TokenTree::Ident(ident.clone());
|
||||||
|
result.push(self.let_expr(punct.span(), keyword));
|
||||||
|
}
|
||||||
|
_ => result.push(self.markup()),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
result
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Parses a single block of markup.
|
||||||
|
fn markup(&mut self) -> ast::Markup {
|
||||||
|
let token = match self.peek() {
|
||||||
|
Some(token) => token,
|
||||||
|
None => {
|
||||||
|
abort_call_site!("unexpected end of input");
|
||||||
|
}
|
||||||
|
};
|
||||||
|
let markup = match token {
|
||||||
|
// Literal
|
||||||
|
TokenTree::Literal(literal) => {
|
||||||
|
self.advance();
|
||||||
|
self.literal(literal)
|
||||||
|
}
|
||||||
|
// Special form
|
||||||
|
TokenTree::Punct(ref punct) if punct.as_char() == '@' => {
|
||||||
|
self.advance();
|
||||||
|
let at_span = punct.span();
|
||||||
|
match self.next() {
|
||||||
|
Some(TokenTree::Ident(ident)) => {
|
||||||
|
let keyword = TokenTree::Ident(ident.clone());
|
||||||
|
match ident.to_string().as_str() {
|
||||||
|
"if" => {
|
||||||
|
let mut segments = Vec::new();
|
||||||
|
self.if_expr(at_span, vec![keyword], &mut segments);
|
||||||
|
ast::Markup::Special { segments }
|
||||||
|
}
|
||||||
|
"while" => self.while_expr(at_span, keyword),
|
||||||
|
"for" => self.for_expr(at_span, keyword),
|
||||||
|
"match" => self.match_expr(at_span, keyword),
|
||||||
|
"let" => {
|
||||||
|
let span = SpanRange {
|
||||||
|
first: at_span,
|
||||||
|
last: ident.span(),
|
||||||
|
};
|
||||||
|
abort!(span, "`@let` only works inside a block");
|
||||||
|
}
|
||||||
|
other => {
|
||||||
|
let span = SpanRange {
|
||||||
|
first: at_span,
|
||||||
|
last: ident.span(),
|
||||||
|
};
|
||||||
|
abort!(span, "unknown keyword `@{}`", other);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
abort!(at_span, "expected keyword after `@`");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Element
|
||||||
|
TokenTree::Ident(ident) => {
|
||||||
|
let ident_string = ident.to_string();
|
||||||
|
match ident_string.as_str() {
|
||||||
|
"if" | "while" | "for" | "match" | "let" => {
|
||||||
|
abort!(
|
||||||
|
ident,
|
||||||
|
"found keyword `{}`", ident_string;
|
||||||
|
help = "should this be a `@{}`?", ident_string
|
||||||
|
);
|
||||||
|
}
|
||||||
|
"true" | "false" => {
|
||||||
|
if let Some(attr_name) = &self.current_attr {
|
||||||
|
emit_error!(
|
||||||
|
ident,
|
||||||
|
r#"attribute value must be a string"#;
|
||||||
|
help = "to declare an empty attribute, omit the equals sign: `{}`",
|
||||||
|
attr_name;
|
||||||
|
help = "to toggle the attribute, use square brackets: `{}[some_boolean_flag]`",
|
||||||
|
attr_name;
|
||||||
|
);
|
||||||
|
return ast::Markup::ParseError {
|
||||||
|
span: SpanRange::single_span(ident.span()),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => {}
|
||||||
|
}
|
||||||
|
|
||||||
|
// `.try_namespaced_name()` should never fail as we've
|
||||||
|
// already seen an `Ident`
|
||||||
|
let name = self.try_namespaced_name().expect("identifier");
|
||||||
|
self.element(name)
|
||||||
|
}
|
||||||
|
// Div element shorthand
|
||||||
|
TokenTree::Punct(ref punct) if punct.as_char() == '.' || punct.as_char() == '#' => {
|
||||||
|
let name = TokenTree::Ident(Ident::new("div", punct.span()));
|
||||||
|
self.element(name.into())
|
||||||
|
}
|
||||||
|
// Splice
|
||||||
|
TokenTree::Group(ref group) if group.delimiter() == Delimiter::Parenthesis => {
|
||||||
|
self.advance();
|
||||||
|
ast::Markup::Splice {
|
||||||
|
expr: group.stream(),
|
||||||
|
outer_span: SpanRange::single_span(group.span()),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Block
|
||||||
|
TokenTree::Group(ref group) if group.delimiter() == Delimiter::Brace => {
|
||||||
|
self.advance();
|
||||||
|
ast::Markup::Block(self.block(group.stream(), SpanRange::single_span(group.span())))
|
||||||
|
}
|
||||||
|
// ???
|
||||||
|
token => {
|
||||||
|
abort!(token, "invalid syntax");
|
||||||
|
}
|
||||||
|
};
|
||||||
|
markup
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Parses a literal string.
|
||||||
|
fn literal(&mut self, literal: Literal) -> ast::Markup {
|
||||||
|
match Lit::new(literal.clone()) {
|
||||||
|
Lit::Str(lit_str) => {
|
||||||
|
return ast::Markup::Literal {
|
||||||
|
content: lit_str.value(),
|
||||||
|
span: SpanRange::single_span(literal.span()),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Boolean literals are idents, so `Lit::Bool` is handled in
|
||||||
|
// `markup`, not here.
|
||||||
|
Lit::Int(..) | Lit::Float(..) => {
|
||||||
|
emit_error!(literal, r#"literal must be double-quoted: `"{}"`"#, literal);
|
||||||
|
}
|
||||||
|
Lit::Char(lit_char) => {
|
||||||
|
emit_error!(
|
||||||
|
literal,
|
||||||
|
r#"literal must be double-quoted: `"{}"`"#,
|
||||||
|
lit_char.value(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
emit_error!(literal, "expected string");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
ast::Markup::ParseError {
|
||||||
|
span: SpanRange::single_span(literal.span()),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Parses an `@if` expression.
|
||||||
|
///
|
||||||
|
/// The leading `@if` should already be consumed.
|
||||||
|
fn if_expr(&mut self, at_span: Span, prefix: Vec<TokenTree>, segments: &mut Vec<ast::Special>) {
|
||||||
|
let mut head = prefix;
|
||||||
|
let body = loop {
|
||||||
|
match self.next() {
|
||||||
|
Some(TokenTree::Group(ref block)) if block.delimiter() == Delimiter::Brace => {
|
||||||
|
break self.block(block.stream(), SpanRange::single_span(block.span()));
|
||||||
|
}
|
||||||
|
Some(token) => head.push(token),
|
||||||
|
None => {
|
||||||
|
let mut span = ast::span_tokens(head);
|
||||||
|
span.first = at_span;
|
||||||
|
abort!(span, "expected body for this `@if`");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
segments.push(ast::Special {
|
||||||
|
at_span: SpanRange::single_span(at_span),
|
||||||
|
head: head.into_iter().collect(),
|
||||||
|
body,
|
||||||
|
});
|
||||||
|
self.else_if_expr(segments)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Parses an optional `@else if` or `@else`.
|
||||||
|
///
|
||||||
|
/// The leading `@else if` or `@else` should *not* already be consumed.
|
||||||
|
fn else_if_expr(&mut self, segments: &mut Vec<ast::Special>) {
|
||||||
|
match self.peek2() {
|
||||||
|
Some((TokenTree::Punct(ref punct), Some(TokenTree::Ident(ref else_keyword))))
|
||||||
|
if punct.as_char() == '@' && *else_keyword == "else" =>
|
||||||
|
{
|
||||||
|
self.advance2();
|
||||||
|
let at_span = punct.span();
|
||||||
|
let else_keyword = TokenTree::Ident(else_keyword.clone());
|
||||||
|
match self.peek() {
|
||||||
|
// `@else if`
|
||||||
|
Some(TokenTree::Ident(ref if_keyword)) if *if_keyword == "if" => {
|
||||||
|
self.advance();
|
||||||
|
let if_keyword = TokenTree::Ident(if_keyword.clone());
|
||||||
|
self.if_expr(at_span, vec![else_keyword, if_keyword], segments)
|
||||||
|
}
|
||||||
|
// Just an `@else`
|
||||||
|
_ => match self.next() {
|
||||||
|
Some(TokenTree::Group(ref group))
|
||||||
|
if group.delimiter() == Delimiter::Brace =>
|
||||||
|
{
|
||||||
|
let body =
|
||||||
|
self.block(group.stream(), SpanRange::single_span(group.span()));
|
||||||
|
segments.push(ast::Special {
|
||||||
|
at_span: SpanRange::single_span(at_span),
|
||||||
|
head: vec![else_keyword].into_iter().collect(),
|
||||||
|
body,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
let span = SpanRange {
|
||||||
|
first: at_span,
|
||||||
|
last: else_keyword.span(),
|
||||||
|
};
|
||||||
|
abort!(span, "expected body for this `@else`");
|
||||||
|
}
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// We didn't find an `@else`; stop
|
||||||
|
_ => {}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Parses an `@while` expression.
|
||||||
|
///
|
||||||
|
/// The leading `@while` should already be consumed.
|
||||||
|
fn while_expr(&mut self, at_span: Span, keyword: TokenTree) -> ast::Markup {
|
||||||
|
let keyword_span = keyword.span();
|
||||||
|
let mut head = vec![keyword];
|
||||||
|
let body = loop {
|
||||||
|
match self.next() {
|
||||||
|
Some(TokenTree::Group(ref block)) if block.delimiter() == Delimiter::Brace => {
|
||||||
|
break self.block(block.stream(), SpanRange::single_span(block.span()));
|
||||||
|
}
|
||||||
|
Some(token) => head.push(token),
|
||||||
|
None => {
|
||||||
|
let span = SpanRange {
|
||||||
|
first: at_span,
|
||||||
|
last: keyword_span,
|
||||||
|
};
|
||||||
|
abort!(span, "expected body for this `@while`");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
ast::Markup::Special {
|
||||||
|
segments: vec![ast::Special {
|
||||||
|
at_span: SpanRange::single_span(at_span),
|
||||||
|
head: head.into_iter().collect(),
|
||||||
|
body,
|
||||||
|
}],
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Parses a `@for` expression.
|
||||||
|
///
|
||||||
|
/// The leading `@for` should already be consumed.
|
||||||
|
fn for_expr(&mut self, at_span: Span, keyword: TokenTree) -> ast::Markup {
|
||||||
|
let keyword_span = keyword.span();
|
||||||
|
let mut head = vec![keyword];
|
||||||
|
loop {
|
||||||
|
match self.next() {
|
||||||
|
Some(TokenTree::Ident(ref in_keyword)) if *in_keyword == "in" => {
|
||||||
|
head.push(TokenTree::Ident(in_keyword.clone()));
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
Some(token) => head.push(token),
|
||||||
|
None => {
|
||||||
|
let span = SpanRange {
|
||||||
|
first: at_span,
|
||||||
|
last: keyword_span,
|
||||||
|
};
|
||||||
|
abort!(span, "missing `in` in `@for` loop");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
let body = loop {
|
||||||
|
match self.next() {
|
||||||
|
Some(TokenTree::Group(ref block)) if block.delimiter() == Delimiter::Brace => {
|
||||||
|
break self.block(block.stream(), SpanRange::single_span(block.span()));
|
||||||
|
}
|
||||||
|
Some(token) => head.push(token),
|
||||||
|
None => {
|
||||||
|
let span = SpanRange {
|
||||||
|
first: at_span,
|
||||||
|
last: keyword_span,
|
||||||
|
};
|
||||||
|
abort!(span, "expected body for this `@for`");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
ast::Markup::Special {
|
||||||
|
segments: vec![ast::Special {
|
||||||
|
at_span: SpanRange::single_span(at_span),
|
||||||
|
head: head.into_iter().collect(),
|
||||||
|
body,
|
||||||
|
}],
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Parses a `@match` expression.
|
||||||
|
///
|
||||||
|
/// The leading `@match` should already be consumed.
|
||||||
|
fn match_expr(&mut self, at_span: Span, keyword: TokenTree) -> ast::Markup {
|
||||||
|
let keyword_span = keyword.span();
|
||||||
|
let mut head = vec![keyword];
|
||||||
|
let (arms, arms_span) = loop {
|
||||||
|
match self.next() {
|
||||||
|
Some(TokenTree::Group(ref body)) if body.delimiter() == Delimiter::Brace => {
|
||||||
|
let span = SpanRange::single_span(body.span());
|
||||||
|
break (self.with_input(body.stream()).match_arms(), span);
|
||||||
|
}
|
||||||
|
Some(token) => head.push(token),
|
||||||
|
None => {
|
||||||
|
let span = SpanRange {
|
||||||
|
first: at_span,
|
||||||
|
last: keyword_span,
|
||||||
|
};
|
||||||
|
abort!(span, "expected body for this `@match`");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
ast::Markup::Match {
|
||||||
|
at_span: SpanRange::single_span(at_span),
|
||||||
|
head: head.into_iter().collect(),
|
||||||
|
arms,
|
||||||
|
arms_span,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn match_arms(&mut self) -> Vec<ast::MatchArm> {
|
||||||
|
let mut arms = Vec::new();
|
||||||
|
while let Some(arm) = self.match_arm() {
|
||||||
|
arms.push(arm);
|
||||||
|
}
|
||||||
|
arms
|
||||||
|
}
|
||||||
|
|
||||||
|
fn match_arm(&mut self) -> Option<ast::MatchArm> {
|
||||||
|
let mut head = Vec::new();
|
||||||
|
loop {
|
||||||
|
match self.peek2() {
|
||||||
|
Some((TokenTree::Punct(ref eq), Some(TokenTree::Punct(ref gt))))
|
||||||
|
if eq.as_char() == '='
|
||||||
|
&& gt.as_char() == '>'
|
||||||
|
&& eq.spacing() == Spacing::Joint =>
|
||||||
|
{
|
||||||
|
self.advance2();
|
||||||
|
head.push(TokenTree::Punct(eq.clone()));
|
||||||
|
head.push(TokenTree::Punct(gt.clone()));
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
Some((token, _)) => {
|
||||||
|
self.advance();
|
||||||
|
head.push(token);
|
||||||
|
}
|
||||||
|
None => {
|
||||||
|
if head.is_empty() {
|
||||||
|
return None;
|
||||||
|
} else {
|
||||||
|
let head_span = ast::span_tokens(head);
|
||||||
|
abort!(head_span, "unexpected end of @match pattern");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
let body = match self.next() {
|
||||||
|
// $pat => { $stmts }
|
||||||
|
Some(TokenTree::Group(ref body)) if body.delimiter() == Delimiter::Brace => {
|
||||||
|
let body = self.block(body.stream(), SpanRange::single_span(body.span()));
|
||||||
|
// Trailing commas are optional if the match arm is a braced block
|
||||||
|
if let Some(TokenTree::Punct(ref punct)) = self.peek() {
|
||||||
|
if punct.as_char() == ',' {
|
||||||
|
self.advance();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
body
|
||||||
|
}
|
||||||
|
// $pat => $expr
|
||||||
|
Some(first_token) => {
|
||||||
|
let mut span = SpanRange::single_span(first_token.span());
|
||||||
|
let mut body = vec![first_token];
|
||||||
|
loop {
|
||||||
|
match self.next() {
|
||||||
|
Some(TokenTree::Punct(ref punct)) if punct.as_char() == ',' => break,
|
||||||
|
Some(token) => {
|
||||||
|
span.last = token.span();
|
||||||
|
body.push(token);
|
||||||
|
}
|
||||||
|
None => break,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
self.block(body.into_iter().collect(), span)
|
||||||
|
}
|
||||||
|
None => {
|
||||||
|
let span = ast::span_tokens(head);
|
||||||
|
abort!(span, "unexpected end of @match arm");
|
||||||
|
}
|
||||||
|
};
|
||||||
|
Some(ast::MatchArm {
|
||||||
|
head: head.into_iter().collect(),
|
||||||
|
body,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Parses a `@let` expression.
|
||||||
|
///
|
||||||
|
/// The leading `@let` should already be consumed.
|
||||||
|
fn let_expr(&mut self, at_span: Span, keyword: TokenTree) -> ast::Markup {
|
||||||
|
let mut tokens = vec![keyword];
|
||||||
|
loop {
|
||||||
|
match self.next() {
|
||||||
|
Some(token) => match token {
|
||||||
|
TokenTree::Punct(ref punct) if punct.as_char() == '=' => {
|
||||||
|
tokens.push(token.clone());
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
_ => tokens.push(token),
|
||||||
|
},
|
||||||
|
None => {
|
||||||
|
let mut span = ast::span_tokens(tokens);
|
||||||
|
span.first = at_span;
|
||||||
|
abort!(span, "unexpected end of `@let` expression");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
loop {
|
||||||
|
match self.next() {
|
||||||
|
Some(token) => match token {
|
||||||
|
TokenTree::Punct(ref punct) if punct.as_char() == ';' => {
|
||||||
|
tokens.push(token.clone());
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
_ => tokens.push(token),
|
||||||
|
},
|
||||||
|
None => {
|
||||||
|
let mut span = ast::span_tokens(tokens);
|
||||||
|
span.first = at_span;
|
||||||
|
abort!(
|
||||||
|
span,
|
||||||
|
"unexpected end of `@let` expression";
|
||||||
|
help = "are you missing a semicolon?"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
ast::Markup::Let {
|
||||||
|
at_span: SpanRange::single_span(at_span),
|
||||||
|
tokens: tokens.into_iter().collect(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Parses an element node.
|
||||||
|
///
|
||||||
|
/// The element name should already be consumed.
|
||||||
|
fn element(&mut self, name: TokenStream) -> ast::Markup {
|
||||||
|
if self.current_attr.is_some() {
|
||||||
|
let span = ast::span_tokens(name);
|
||||||
|
abort!(span, "unexpected element");
|
||||||
|
}
|
||||||
|
let attrs = self.attrs();
|
||||||
|
let body = match self.peek() {
|
||||||
|
Some(TokenTree::Punct(ref punct))
|
||||||
|
if punct.as_char() == ';' || punct.as_char() == '/' =>
|
||||||
|
{
|
||||||
|
// Void element
|
||||||
|
self.advance();
|
||||||
|
if punct.as_char() == '/' {
|
||||||
|
emit_error!(
|
||||||
|
punct,
|
||||||
|
"void elements must use `;`, not `/`";
|
||||||
|
help = "change this to `;`";
|
||||||
|
help = "see https://github.com/lambda-fairy/maud/pull/315 for details";
|
||||||
|
);
|
||||||
|
}
|
||||||
|
ast::ElementBody::Void {
|
||||||
|
semi_span: SpanRange::single_span(punct.span()),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Some(_) => match self.markup() {
|
||||||
|
ast::Markup::Block(block) => ast::ElementBody::Block { block },
|
||||||
|
markup => {
|
||||||
|
let markup_span = markup.span();
|
||||||
|
abort!(
|
||||||
|
markup_span,
|
||||||
|
"element body must be wrapped in braces";
|
||||||
|
help = "see https://github.com/lambda-fairy/maud/pull/137 for details"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
None => abort_call_site!("expected `;`, found end of macro"),
|
||||||
|
};
|
||||||
|
ast::Markup::Element { name, attrs, body }
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Parses the attributes of an element.
|
||||||
|
fn attrs(&mut self) -> Vec<ast::Attr> {
|
||||||
|
let mut attrs = Vec::new();
|
||||||
|
loop {
|
||||||
|
if let Some(name) = self.try_namespaced_name() {
|
||||||
|
// Attribute
|
||||||
|
match self.peek() {
|
||||||
|
// Non-empty attribute
|
||||||
|
Some(TokenTree::Punct(ref punct)) if punct.as_char() == '=' => {
|
||||||
|
self.advance();
|
||||||
|
// Parse a value under an attribute context
|
||||||
|
assert!(self.current_attr.is_none());
|
||||||
|
self.current_attr = Some(ast::name_to_string(name.clone()));
|
||||||
|
let attr_type = match self.attr_toggler() {
|
||||||
|
Some(toggler) => ast::AttrType::Optional { toggler },
|
||||||
|
None => {
|
||||||
|
let value = self.markup();
|
||||||
|
ast::AttrType::Normal { value }
|
||||||
|
}
|
||||||
|
};
|
||||||
|
self.current_attr = None;
|
||||||
|
attrs.push(ast::Attr::Named {
|
||||||
|
named_attr: ast::NamedAttr { name, attr_type },
|
||||||
|
});
|
||||||
|
}
|
||||||
|
// Empty attribute (legacy syntax)
|
||||||
|
Some(TokenTree::Punct(ref punct)) if punct.as_char() == '?' => {
|
||||||
|
self.advance();
|
||||||
|
let toggler = self.attr_toggler();
|
||||||
|
attrs.push(ast::Attr::Named {
|
||||||
|
named_attr: ast::NamedAttr {
|
||||||
|
name: name.clone(),
|
||||||
|
attr_type: ast::AttrType::Empty { toggler },
|
||||||
|
},
|
||||||
|
});
|
||||||
|
}
|
||||||
|
// Empty attribute (new syntax)
|
||||||
|
_ => {
|
||||||
|
let toggler = self.attr_toggler();
|
||||||
|
attrs.push(ast::Attr::Named {
|
||||||
|
named_attr: ast::NamedAttr {
|
||||||
|
name: name.clone(),
|
||||||
|
attr_type: ast::AttrType::Empty { toggler },
|
||||||
|
},
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
match self.peek() {
|
||||||
|
// Class shorthand
|
||||||
|
Some(TokenTree::Punct(ref punct)) if punct.as_char() == '.' => {
|
||||||
|
self.advance();
|
||||||
|
let name = self.class_or_id_name();
|
||||||
|
let toggler = self.attr_toggler();
|
||||||
|
attrs.push(ast::Attr::Class {
|
||||||
|
dot_span: SpanRange::single_span(punct.span()),
|
||||||
|
name,
|
||||||
|
toggler,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
// ID shorthand
|
||||||
|
Some(TokenTree::Punct(ref punct)) if punct.as_char() == '#' => {
|
||||||
|
self.advance();
|
||||||
|
let name = self.class_or_id_name();
|
||||||
|
attrs.push(ast::Attr::Id {
|
||||||
|
hash_span: SpanRange::single_span(punct.span()),
|
||||||
|
name,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
// If it's not a valid attribute, backtrack and bail out
|
||||||
|
_ => break,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut attr_map: HashMap<String, Vec<SpanRange>> = HashMap::new();
|
||||||
|
let mut has_class = false;
|
||||||
|
for attr in &attrs {
|
||||||
|
let name = match attr {
|
||||||
|
ast::Attr::Class { .. } => {
|
||||||
|
if has_class {
|
||||||
|
// Only check the first class to avoid spurious duplicates
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
has_class = true;
|
||||||
|
"class".to_string()
|
||||||
|
}
|
||||||
|
ast::Attr::Id { .. } => "id".to_string(),
|
||||||
|
ast::Attr::Named { named_attr } => named_attr
|
||||||
|
.name
|
||||||
|
.clone()
|
||||||
|
.into_iter()
|
||||||
|
.map(|token| token.to_string())
|
||||||
|
.collect(),
|
||||||
|
};
|
||||||
|
let entry = attr_map.entry(name).or_default();
|
||||||
|
entry.push(attr.span());
|
||||||
|
}
|
||||||
|
|
||||||
|
for (name, spans) in attr_map {
|
||||||
|
if spans.len() > 1 {
|
||||||
|
let mut spans = spans.into_iter();
|
||||||
|
let first_span = spans.next().expect("spans should be non-empty");
|
||||||
|
abort!(first_span, "duplicate attribute `{}`", name);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
attrs
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Parses the name of a class or ID.
|
||||||
|
fn class_or_id_name(&mut self) -> ast::Markup {
|
||||||
|
if let Some(symbol) = self.try_name() {
|
||||||
|
ast::Markup::Symbol { symbol }
|
||||||
|
} else {
|
||||||
|
self.markup()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Parses the `[cond]` syntax after an empty attribute or class shorthand.
|
||||||
|
fn attr_toggler(&mut self) -> Option<ast::Toggler> {
|
||||||
|
match self.peek() {
|
||||||
|
Some(TokenTree::Group(ref group)) if group.delimiter() == Delimiter::Bracket => {
|
||||||
|
self.advance();
|
||||||
|
Some(ast::Toggler {
|
||||||
|
cond: group.stream(),
|
||||||
|
cond_span: SpanRange::single_span(group.span()),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
_ => None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Parses an identifier, without dealing with namespaces.
|
||||||
|
fn try_name(&mut self) -> Option<TokenStream> {
|
||||||
|
let mut result = Vec::new();
|
||||||
|
if let Some(token @ TokenTree::Ident(_)) = self.peek() {
|
||||||
|
self.advance();
|
||||||
|
result.push(token);
|
||||||
|
} else {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
let mut expect_ident = false;
|
||||||
|
loop {
|
||||||
|
expect_ident = match self.peek() {
|
||||||
|
Some(TokenTree::Punct(ref punct)) if punct.as_char() == '-' => {
|
||||||
|
self.advance();
|
||||||
|
result.push(TokenTree::Punct(punct.clone()));
|
||||||
|
true
|
||||||
|
}
|
||||||
|
Some(TokenTree::Ident(ref ident)) if expect_ident => {
|
||||||
|
self.advance();
|
||||||
|
result.push(TokenTree::Ident(ident.clone()));
|
||||||
|
false
|
||||||
|
}
|
||||||
|
_ => break,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
Some(result.into_iter().collect())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Parses a HTML element or attribute name, along with a namespace
|
||||||
|
/// if necessary.
|
||||||
|
fn try_namespaced_name(&mut self) -> Option<TokenStream> {
|
||||||
|
let mut result = vec![self.try_name()?];
|
||||||
|
if let Some(TokenTree::Punct(ref punct)) = self.peek() {
|
||||||
|
if punct.as_char() == ':' {
|
||||||
|
self.advance();
|
||||||
|
result.push(TokenStream::from(TokenTree::Punct(punct.clone())));
|
||||||
|
result.push(self.try_name()?);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Some(result.into_iter().collect())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Parses the given token stream as a Maud expression.
|
||||||
|
fn block(&mut self, body: TokenStream, outer_span: SpanRange) -> ast::Block {
|
||||||
|
let markups = self.with_input(body).markups();
|
||||||
|
ast::Block {
|
||||||
|
markups,
|
||||||
|
outer_span,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Add table
Add a link
Reference in a new issue