Add proc-macro to derive CheckCodePrefix (#1656)

IMO a derive macro is a natural way to generate new code, and it reduces
the chance of merge conflicts.

.github/workflows/ci.yaml

@@ -42,9 +42,8 @@ jobs:
       - run: cargo build --all --release
       - run: ./target/release/ruff_dev generate-all
       - run: git diff --quiet README.md || echo "::error file=README.md::This file is outdated. Run 'cargo +nightly dev generate-all'."
-      - run: git diff --quiet src/registry_gen.rs || echo "::error file=src/registry_gen.rs::This file is outdated. Run 'cargo +nightly dev generate-all'."
       - run: git diff --quiet ruff.schema.json || echo "::error file=ruff.schema.json::This file is outdated. Run 'cargo +nightly dev generate-all'."
-      - run: git diff --exit-code -- README.md src/registry_gen.rs ruff.schema.json
+      - run: git diff --exit-code -- README.md ruff.schema.json
 
   cargo-fmt:
     name: "cargo fmt"

flake8_to_ruff/src/converter.rs

@@ -7,7 +7,7 @@ use ruff::flake8_pytest_style::types::{
 use ruff::flake8_quotes::settings::Quote;
 use ruff::flake8_tidy_imports::settings::Strictness;
 use ruff::pydocstyle::settings::Convention;
-use ruff::registry_gen::CheckCodePrefix;
+use ruff::registry::CheckCodePrefix;
 use ruff::settings::options::Options;
 use ruff::settings::pyproject::Pyproject;
 use ruff::{
@@ -345,7 +345,7 @@ mod tests {
 
     use anyhow::Result;
     use ruff::pydocstyle::settings::Convention;
-    use ruff::registry_gen::CheckCodePrefix;
+    use ruff::registry::CheckCodePrefix;
     use ruff::settings::options::Options;
     use ruff::settings::pyproject::Pyproject;
     use ruff::{flake8_quotes, pydocstyle};

flake8_to_ruff/src/parser.rs

@@ -3,8 +3,7 @@ use std::str::FromStr;
 use anyhow::{bail, Result};
 use once_cell::sync::Lazy;
 use regex::Regex;
-use ruff::registry::PREFIX_REDIRECTS;
-use ruff::registry_gen::CheckCodePrefix;
+use ruff::registry::{CheckCodePrefix, PREFIX_REDIRECTS};
 use ruff::settings::types::PatternPrefixPair;
 use rustc_hash::FxHashMap;
 
@@ -201,7 +200,7 @@ pub fn collect_per_file_ignores(
 #[cfg(test)]
 mod tests {
     use anyhow::Result;
-    use ruff::registry_gen::CheckCodePrefix;
+    use ruff::registry::CheckCodePrefix;
     use ruff::settings::types::PatternPrefixPair;
 
     use crate::parser::{parse_files_to_codes_mapping, parse_prefix_codes, parse_strings};

flake8_to_ruff/src/plugin.rs

@@ -3,7 +3,7 @@ use std::fmt;
 use std::str::FromStr;
 
 use anyhow::anyhow;
-use ruff::registry_gen::CheckCodePrefix;
+use ruff::registry::CheckCodePrefix;
 
 #[derive(Clone, Ord, PartialOrd, Eq, PartialEq)]
 pub enum Plugin {

ruff_dev/Cargo.toml

@@ -6,7 +6,6 @@ edition = "2021"
 [dependencies]
 anyhow = { version = "1.0.66" }
 clap = { version = "4.0.1", features = ["derive"] }
-codegen = { version = "0.2.0" }
 itertools = { version = "0.10.5" }
 libcst = { git = "https://github.com/charliermarsh/LibCST", rev = "f2f0b7a487a8725d161fe8b3ed73a6758b21e177" }
 once_cell = { version = "1.16.0" }

ruff_dev/src/generate_all.rs

@@ -3,10 +3,7 @@
 use anyhow::Result;
 use clap::Args;
 
-use crate::{
-    generate_check_code_prefix, generate_cli_help, generate_json_schema, generate_options,
-    generate_rules_table,
-};
+use crate::{generate_cli_help, generate_json_schema, generate_options, generate_rules_table};
 
 #[derive(Args)]
 pub struct Cli {
@@ -16,9 +13,6 @@ pub struct Cli {
 }
 
 pub fn main(cli: &Cli) -> Result<()> {
-    generate_check_code_prefix::main(&generate_check_code_prefix::Cli {
-        dry_run: cli.dry_run,
-    })?;
     generate_json_schema::main(&generate_json_schema::Cli {
         dry_run: cli.dry_run,
     })?;

ruff_dev/src/lib.rs

@@ -12,7 +12,6 @@
 )]
 
 pub mod generate_all;
-pub mod generate_check_code_prefix;
 pub mod generate_cli_help;
 pub mod generate_json_schema;
 pub mod generate_options;

ruff_dev/src/main.rs

@@ -14,8 +14,8 @@
 use anyhow::Result;
 use clap::{Parser, Subcommand};
 use ruff_dev::{
-    generate_all, generate_check_code_prefix, generate_cli_help, generate_json_schema,
-    generate_options, generate_rules_table, print_ast, print_cst, print_tokens, round_trip,
+    generate_all, generate_cli_help, generate_json_schema, generate_options, generate_rules_table,
+    print_ast, print_cst, print_tokens, round_trip,
 };
 
 #[derive(Parser)]
@@ -30,8 +30,6 @@ struct Cli {
 enum Commands {
     /// Run all code and documentation generation steps.
     GenerateAll(generate_all::Cli),
-    /// Generate the `CheckCodePrefix` enum.
-    GenerateCheckCodePrefix(generate_check_code_prefix::Cli),
     /// Generate JSON schema for the TOML configuration file.
     GenerateJSONSchema(generate_json_schema::Cli),
     /// Generate a Markdown-compatible table of supported lint rules.
@@ -54,7 +52,6 @@ fn main() -> Result<()> {
     let cli = Cli::parse();
     match &cli.command {
         Commands::GenerateAll(args) => generate_all::main(args)?,
-        Commands::GenerateCheckCodePrefix(args) => generate_check_code_prefix::main(args)?,
         Commands::GenerateJSONSchema(args) => generate_json_schema::main(args)?,
         Commands::GenerateRulesTable(args) => generate_rules_table::main(args)?,
         Commands::GenerateOptions(args) => generate_options::main(args)?,

ruff_macros/Cargo.toml

@@ -7,6 +7,7 @@ edition = "2021"
 proc-macro = true
 
 [dependencies]
+once_cell = { version = "1.17.0" }
 proc-macro2 = { version = "1.0.47" }
 quote = { version = "1.0.21" }
 syn = { version = "1.0.103", features = ["derive", "parsing"] }