Unnamed repository; edit this file 'description' to name the repository.
Diffstat (limited to 'xtask/src/codegen/grammar.rs')
| -rw-r--r-- | xtask/src/codegen/grammar.rs | 91 |
1 files changed, 84 insertions, 7 deletions
diff --git a/xtask/src/codegen/grammar.rs b/xtask/src/codegen/grammar.rs index e9f3871585..bdba1b348d 100644 --- a/xtask/src/codegen/grammar.rs +++ b/xtask/src/codegen/grammar.rs @@ -67,7 +67,6 @@ fn generate_tokens(grammar: &AstSrc) -> String { let name = format_ident!("{}", token); let kind = format_ident!("{}", to_upper_snake_case(token)); quote! { - #[derive(Debug, Clone, PartialEq, Eq, Hash)] pub struct #name { pub(crate) syntax: SyntaxToken, } @@ -83,6 +82,29 @@ fn generate_tokens(grammar: &AstSrc) -> String { } fn syntax(&self) -> &SyntaxToken { &self.syntax } } + + impl fmt::Debug for #name { + fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { + f.debug_struct(#token).field("syntax", &self.syntax).finish() + } + } + impl Clone for #name { + fn clone(&self) -> Self { + Self { syntax: self.syntax.clone() } + } + } + impl hash::Hash for #name { + fn hash<H: hash::Hasher>(&self, state: &mut H) { + self.syntax.hash(state); + } + } + + impl Eq for #name {} + impl PartialEq for #name { + fn eq(&self, other: &Self) -> bool { + self.syntax == other.syntax + } + } } }); @@ -90,7 +112,10 @@ fn generate_tokens(grammar: &AstSrc) -> String { crate::flags::CodegenType::Grammar, reformat( quote! { + use std::{fmt, hash}; + use crate::{SyntaxKind::{self, *}, SyntaxToken, ast::AstToken}; + #(#tokens)* } .to_string(), @@ -104,6 +129,7 @@ fn generate_nodes(kinds: KindsSrc, grammar: &AstSrc) -> String { .nodes .iter() .map(|node| { + let node_str_name = &node.name; let name = format_ident!("{}", node.name); let kind = format_ident!("{}", to_upper_snake_case(&node.name)); let traits = node @@ -149,7 +175,6 @@ fn generate_nodes(kinds: KindsSrc, grammar: &AstSrc) -> String { ( quote! { #[pretty_doc_comment_placeholder_workaround] - #[derive(Debug, Clone, PartialEq, Eq, Hash)] pub struct #name { pub(crate) syntax: SyntaxNode, } @@ -180,6 +205,31 @@ fn generate_nodes(kinds: KindsSrc, grammar: &AstSrc) -> String { #[inline] fn syntax(&self) -> &SyntaxNode { &self.syntax } } + + impl hash::Hash for #name { + fn hash<H: hash::Hasher>(&self, state: &mut H) { + self.syntax.hash(state); + } + } + + impl Eq for #name {} + impl PartialEq for #name { + fn eq(&self, other: &Self) -> bool { + self.syntax == other.syntax + } + } + + impl Clone for #name { + fn clone(&self) -> Self { + Self { syntax: self.syntax.clone() } + } + } + + impl fmt::Debug for #name { + fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { + f.debug_struct(#node_str_name).field("syntax", &self.syntax).finish() + } + } }, ) }) @@ -265,6 +315,7 @@ fn generate_nodes(kinds: KindsSrc, grammar: &AstSrc) -> String { .sorted_by_key(|(name, _)| *name) .map(|(trait_name, nodes)| { let name = format_ident!("Any{}", trait_name); + let node_str_name = name.to_string(); let trait_name = format_ident!("{}", trait_name); let kinds: Vec<_> = nodes .iter() @@ -274,13 +325,9 @@ fn generate_nodes(kinds: KindsSrc, grammar: &AstSrc) -> String { ( quote! { #[pretty_doc_comment_placeholder_workaround] - #[derive(Debug, Clone, PartialEq, Eq, Hash)] pub struct #name { pub(crate) syntax: SyntaxNode, } - impl ast::#trait_name for #name {} - }, - quote! { impl #name { #[inline] pub fn new<T: ast::#trait_name>(node: T) -> #name { @@ -289,6 +336,9 @@ fn generate_nodes(kinds: KindsSrc, grammar: &AstSrc) -> String { } } } + }, + quote! { + impl ast::#trait_name for #name {} impl AstNode for #name { #[inline] fn can_cast(kind: SyntaxKind) -> bool { @@ -304,6 +354,31 @@ fn generate_nodes(kinds: KindsSrc, grammar: &AstSrc) -> String { } } + impl hash::Hash for #name { + fn hash<H: hash::Hasher>(&self, state: &mut H) { + self.syntax.hash(state); + } + } + + impl Eq for #name {} + impl PartialEq for #name { + fn eq(&self, other: &Self) -> bool { + self.syntax == other.syntax + } + } + + impl Clone for #name { + fn clone(&self) -> Self { + Self { syntax: self.syntax.clone() } + } + } + + impl fmt::Debug for #name { + fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { + f.debug_struct(#node_str_name).field("syntax", &self.syntax).finish() + } + } + #( impl From<#nodes> for #name { #[inline] @@ -346,6 +421,8 @@ fn generate_nodes(kinds: KindsSrc, grammar: &AstSrc) -> String { let ast = quote! { #![allow(non_snake_case)] + use std::{fmt, hash}; + use crate::{ SyntaxNode, SyntaxToken, SyntaxKind::{self, *}, ast::{self, AstNode, AstChildren, support}, @@ -613,7 +690,7 @@ fn generate_syntax_kinds(grammar: KindsSrc) -> String { impl ::core::cmp::PartialOrd for SyntaxKind { #[inline] fn partial_cmp(&self, other: &Self) -> core::option::Option<core::cmp::Ordering> { - (*self as u16).partial_cmp(&(*other as u16)) + Some(self.cmp(other)) } } impl ::core::cmp::Ord for SyntaxKind { |