From f59511fa4677d99455b9f47da9064a7ed94333aa Mon Sep 17 00:00:00 2001
From: Carson McManus <carson.mcmanus1@gmail.com>
Date: Mon, 8 Jul 2024 08:17:25 -0400
Subject: [PATCH] refactor(codegen): generate most functions on `LanguageKind`
 with a macro

---
 xtask/codegen/src/language_kind.rs | 122 +++++++----------------------
 1 file changed, 29 insertions(+), 93 deletions(-)

diff --git a/xtask/codegen/src/language_kind.rs b/xtask/codegen/src/language_kind.rs
index 79c3437e3868..63da4c4d3307 100644
--- a/xtask/codegen/src/language_kind.rs
+++ b/xtask/codegen/src/language_kind.rs
@@ -7,7 +7,7 @@ use crate::json_kinds_src::JSON_KINDS_SRC;
 use crate::kind_src::KindsSrc;
 use crate::yaml_kinds_src::YAML_KINDS_SRC;
 use proc_macro2::{Ident, Span, TokenStream};
-use quote::quote;
+use quote::{format_ident, quote};
 use std::str::FromStr;
 
 pub const LANGUAGE_PREFIXES: [&str; 9] = [
@@ -68,104 +68,40 @@ impl FromStr for LanguageKind {
     }
 }
 
-impl LanguageKind {
-    pub(crate) fn syntax_crate_ident(&self) -> Ident {
-        Ident::new(self.syntax_crate_name().as_str(), Span::call_site())
-    }
-
-    pub(crate) fn syntax_kind(&self) -> TokenStream {
-        match self {
-            LanguageKind::Js => quote! { JsSyntaxKind },
-            LanguageKind::Css => quote! { CssSyntaxKind },
-            LanguageKind::Json => quote! { JsonSyntaxKind },
-            LanguageKind::Graphql => quote! { GraphqlSyntaxKind },
-            LanguageKind::Grit => quote! { GritSyntaxKind },
-            LanguageKind::Html => quote! { HtmlSyntaxKind },
-            LanguageKind::Yaml => quote! { YamlSyntaxKind },
-        }
-    }
-    pub(crate) fn syntax_factory(&self) -> TokenStream {
-        match self {
-            LanguageKind::Js => quote! { JsSyntaxFactory },
-            LanguageKind::Css => quote! { CssSyntaxFactory },
-            LanguageKind::Json => quote! { JsonSyntaxFactory },
-            LanguageKind::Graphql => quote! { GraphqlSyntaxFactory },
-            LanguageKind::Grit => quote! { GritSyntaxFactory },
-            LanguageKind::Html => quote! { HtmlSyntaxFactory },
-            LanguageKind::Yaml => quote! { YamlSyntaxFactory },
-        }
-    }
-
-    pub(crate) fn syntax_node(&self) -> TokenStream {
-        match self {
-            LanguageKind::Js => quote! { JsSyntaxNode },
-            LanguageKind::Css => quote! { CssSyntaxNode },
-            LanguageKind::Json => quote! { JsonSyntaxNode },
-            LanguageKind::Graphql => quote! { GraphqlSyntaxNode },
-            LanguageKind::Grit => quote! { GritSyntaxNode },
-            LanguageKind::Html => quote! { HtmlSyntaxNode },
-            LanguageKind::Yaml => quote! { YamlSyntaxNode },
-        }
-    }
-
-    pub(crate) fn syntax_element(&self) -> TokenStream {
-        match self {
-            LanguageKind::Js => quote! { JsSyntaxElement },
-            LanguageKind::Css => quote! { CssSyntaxElement },
-            LanguageKind::Json => quote! { JsonSyntaxElement },
-            LanguageKind::Graphql => quote! { GraphqlSyntaxElement },
-            LanguageKind::Grit => quote! { GritSyntaxElement },
-            LanguageKind::Html => quote! { HtmlSyntaxElement },
-            LanguageKind::Yaml => quote! { YamlSyntaxElement },
-        }
-    }
-
-    pub(crate) fn syntax_token(&self) -> TokenStream {
-        match self {
-            LanguageKind::Js => quote! { JsSyntaxToken },
-            LanguageKind::Css => quote! { CssSyntaxToken },
-            LanguageKind::Json => quote! { JsonSyntaxToken },
-            LanguageKind::Graphql => quote! { GraphqlSyntaxToken },
-            LanguageKind::Grit => quote! { GritSyntaxToken },
-            LanguageKind::Html => quote! { HtmlSyntaxToken },
-            LanguageKind::Yaml => quote! { YamlSyntaxToken },
+/// A helper macro to make it easier to define functions that return tokens for a specific language kind.
+macro_rules! define_language_kind_function {
+    ([$($kind:ident),*],$func:ident,$out:expr) => {
+        pub(crate) fn $func(&self) -> TokenStream {
+            match self {
+                $( LanguageKind::$kind => {
+                    // HACK: workaround for $kind$out adding an extra space between the two
+                    let ident = format_ident!("{}{}", stringify!($kind), stringify!($out));
+                    quote! { #ident }
+                },)*
+            }
         }
     }
+}
 
-    pub(crate) fn syntax_element_children(&self) -> TokenStream {
-        match self {
-            LanguageKind::Js => quote! { JsSyntaxElementChildren },
-            LanguageKind::Css => quote! { CssSyntaxElementChildren },
-            LanguageKind::Json => quote! { JsonSyntaxElementChildren },
-            LanguageKind::Graphql => quote! { GraphqlSyntaxElementChildren },
-            LanguageKind::Grit => quote! { GritSyntaxElementChildren },
-            LanguageKind::Html => quote! { HtmlSyntaxElementChildren },
-            LanguageKind::Yaml => quote! { YamlSyntaxElementChildren },
-        }
+/// A helper macro to define functions for each language kind to make it slightly less tedious to add new languages.
+macro_rules! define_language_kind_functions {
+    ([$($kind:ident),*]) => {
+        define_language_kind_function!([$($kind),*], syntax_kind, SyntaxKind);
+        define_language_kind_function!([$($kind),*], syntax_factory, SyntaxFactory);
+        define_language_kind_function!([$($kind),*], syntax_node, SyntaxNode);
+        define_language_kind_function!([$($kind),*], syntax_element, SyntaxElement);
+        define_language_kind_function!([$($kind),*], syntax_token, SyntaxToken);
+        define_language_kind_function!([$($kind),*], syntax_element_children, SyntaxElementChildren);
+        define_language_kind_function!([$($kind),*], syntax_list, SyntaxList);
+        define_language_kind_function!([$($kind),*], language, Language);
     }
+}
 
-    pub(crate) fn syntax_list(&self) -> TokenStream {
-        match self {
-            LanguageKind::Js => quote! { JsSyntaxList },
-            LanguageKind::Css => quote! { CssSyntaxList },
-            LanguageKind::Json => quote! { JsonSyntaxList },
-            LanguageKind::Graphql => quote! { GraphqlSyntaxList },
-            LanguageKind::Grit => quote! { GritSyntaxList },
-            LanguageKind::Html => quote! { HtmlSyntaxList },
-            LanguageKind::Yaml => quote! { YamlSyntaxList },
-        }
-    }
+impl LanguageKind {
+    define_language_kind_functions!([Js, Css, Json, Graphql, Grit, Html, Yaml]);
 
-    pub(crate) fn language(&self) -> TokenStream {
-        match self {
-            LanguageKind::Js => quote! { JsLanguage },
-            LanguageKind::Css => quote! { CssLanguage },
-            LanguageKind::Json => quote! { JsonLanguage },
-            LanguageKind::Graphql => quote! { GraphqlLanguage },
-            LanguageKind::Grit => quote! { GritLanguage },
-            LanguageKind::Html => quote! { HtmlLanguage },
-            LanguageKind::Yaml => quote! { YamlLanguage },
-        }
+    pub(crate) fn syntax_crate_ident(&self) -> Ident {
+        Ident::new(self.syntax_crate_name().as_str(), Span::call_site())
     }
 
     pub fn formatter_crate_name(&self) -> String {