aldrin_parser/warning/
non_snake_case_function.rs

1use super::Warning;
2use crate::ast::{FunctionDef, Ident};
3use crate::diag::{Diagnostic, DiagnosticKind, Formatted, Formatter};
4use crate::validate::Validate;
5use crate::Parsed;
6use heck::ToSnakeCase;
7
8#[derive(Debug)]
9pub struct NonSnakeCaseFunction {
10    schema_name: String,
11    snake_case: String,
12    ident: Ident,
13}
14
15impl NonSnakeCaseFunction {
16    pub(crate) fn validate(func: &FunctionDef, validate: &mut Validate) {
17        let snake_case = func.name().value().to_snake_case();
18        if func.name().value() != snake_case {
19            validate.add_warning(Self {
20                schema_name: validate.schema_name().to_owned(),
21                snake_case,
22                ident: func.name().clone(),
23            });
24        }
25    }
26
27    pub fn snake_case(&self) -> &str {
28        &self.snake_case
29    }
30
31    pub fn ident(&self) -> &Ident {
32        &self.ident
33    }
34}
35
36impl Diagnostic for NonSnakeCaseFunction {
37    fn kind(&self) -> DiagnosticKind {
38        DiagnosticKind::Warning
39    }
40
41    fn schema_name(&self) -> &str {
42        &self.schema_name
43    }
44
45    fn format<'a>(&'a self, parsed: &'a Parsed) -> Formatted<'a> {
46        let mut fmt = Formatter::new(
47            self,
48            format!(
49                "function `{}` should have a snake-case name",
50                self.ident.value()
51            ),
52        );
53
54        if let Some(schema) = parsed.get_schema(&self.schema_name) {
55            fmt.main_block(schema, self.ident.span().from, self.ident.span(), "");
56        }
57
58        fmt.help(format!(
59            "consider renaming function `{}` to `{}`",
60            self.ident.value(),
61            self.snake_case
62        ));
63        fmt.format()
64    }
65}
66
67impl From<NonSnakeCaseFunction> for Warning {
68    fn from(w: NonSnakeCaseFunction) -> Self {
69        Self::NonSnakeCaseFunction(w)
70    }
71}