ruff/crates/ruff_linter/src/upstream_categories.rs

//! This module should probably not exist in this shape or form.
use crate::codes::Rule;
use crate::registry::Linter;

#[derive(Hash, Eq, PartialEq, Copy, Clone, Debug)]
pub struct UpstreamCategoryAndPrefix {
    pub category: &'static str,
    pub prefix: &'static str,
}

const C: UpstreamCategoryAndPrefix = UpstreamCategoryAndPrefix {
    category: "Convention",
    prefix: "C",
};

const E: UpstreamCategoryAndPrefix = UpstreamCategoryAndPrefix {
    category: "Error",
    prefix: "E",
};

const R: UpstreamCategoryAndPrefix = UpstreamCategoryAndPrefix {
    category: "Refactor",
    prefix: "R",
};

const W: UpstreamCategoryAndPrefix = UpstreamCategoryAndPrefix {
    category: "Warning",
    prefix: "W",
};

impl Rule {
    pub fn upstream_category(&self, linter: &Linter) -> Option<UpstreamCategoryAndPrefix> {
        let code = linter.code_for_rule(*self).unwrap();
        match linter {
            Linter::Pycodestyle => {
                if code.starts_with('E') {
                    Some(E)
                } else if code.starts_with('W') {
                    Some(W)
                } else {
                    None
                }
            }
            Linter::Pylint => {
                if code.starts_with('C') {
                    Some(C)
                } else if code.starts_with('E') {
                    Some(E)
                } else if code.starts_with('R') {
                    Some(R)
                } else if code.starts_with('W') {
                    Some(W)
                } else {
                    None
                }
            }
            _ => None,
        }
    }
}

impl Linter {
    pub const fn upstream_categories(&self) -> Option<&'static [UpstreamCategoryAndPrefix]> {
        match self {
            Linter::Pycodestyle => Some(&[E, W]),
            Linter::Pylint => Some(&[C, E, R, W]),
            _ => None,
        }
    }
}