feat(generate): Support for --stage=json/parser/lib

crates/cli/src/main.rs

@@ -85,6 +85,17 @@ struct Init {
     pub grammar_path: Option<PathBuf>,
 }
 
+#[derive(Clone, Debug, Default, ValueEnum, PartialEq, Eq)]
+enum GenerationStage {
+    /// Generate `grammar.json` and `node-types.json`
+    Json,
+    /// Generate `parser.c` and related files
+    #[default]
+    Parser,
+    /// Compile to a library
+    Lib,
+}
+
 #[derive(Args)]
 #[command(alias = "gen", alias = "g")]
 struct Generate {
@@ -107,8 +118,12 @@ struct Generate {
                 )
     )]
     pub abi_version: Option<String>,
-    /// Compile all defined languages in the current dir
-    #[arg(long, short = 'b')]
+    /// Which generation stage to end after
+    #[arg(long)]
+    #[clap(value_enum, default_value_t=GenerationStage::Parser)]
+    pub stage: GenerationStage,
+    /// Deprecated: use --stage=lib.
+    #[arg(long, short = 'b', conflicts_with = "stage")]
     pub build: bool,
     /// Compile a parser in debug mode
     #[arg(long, short = '0')]
@@ -133,9 +148,6 @@ struct Generate {
         default_value = "node"
     )]
     pub js_runtime: Option<String>,
-    /// Only generate `grammar.json` by evaluating `grammar.js`, but do not generate `parser.c` and related files afterwards
-    #[arg(long)]
-    pub evaluate_only: bool,
 }
 
 #[derive(Args)]
@@ -803,6 +815,11 @@ impl Generate {
                         version.parse().expect("invalid abi version flag")
                     }
                 });
+        if self.build {
+            // TODO: remove the `--build` argument in 0.27
+            // TODO: migrate to `warn!` once https://github.com/tree-sitter/tree-sitter/pull/4604 is merged
+            eprintln!("Warning: --build is deprecated, use --stage=lib instead");
+        }
         if let Err(err) = tree_sitter_generate::generate_parser_in_directory(
             current_dir,
             self.output.as_deref(),
@@ -810,7 +827,7 @@ impl Generate {
             abi_version,
             self.report_states_for_rule.as_deref(),
             self.js_runtime.as_deref(),
-            self.evaluate_only,
+            self.stage != GenerationStage::Json,
         ) {
             if self.json {
                 eprintln!("{}", serde_json::to_string_pretty(&err)?);
@@ -821,7 +838,7 @@ impl Generate {
                 Err(anyhow!(err.to_string())).with_context(|| "Error when generating parser")?;
             }
         }
-        if self.build {
+        if self.stage == GenerationStage::Lib || self.build {
             if let Some(path) = self.libdir {
                 loader = loader::Loader::with_parser_lib_path(path);
             }