feat: replace js in method name

rome · Nov 5, 2021 · b480bdb · b480bdb
1 parent df33e87
commit b480bdb
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/xtask/src/codegen/kinds_src.rs b/xtask/src/codegen/kinds_src.rs
@@ -3,6 +3,8 @@
 
 use quote::{format_ident, quote};
 
+const LANGUAGE_PREFIXES: [&str; 4] = ["js_", "ts_", "jsx_", "tsx_"];
+
 pub struct KindsSrc<'a> {
 	pub punct: &'a [(&'a str, &'a str)],
 	pub keywords: &'a [&'a str],
@@ -505,10 +507,14 @@ impl Field {
 				if !name_from_label {
 					panic!("The node {} doesn't have a label", ty);
 				}
-				if name == "type" {
+				let mut final_name = name.clone();
+				for prefix in LANGUAGE_PREFIXES {
+					final_name = final_name.replace(prefix, "");
+				}
+				if final_name == "type" {
 					format_ident!("ty")
 				} else {
-					format_ident!("{}", name)
+					format_ident!("{}", final_name)
 				}
 			}
 		}