just-lsp 0.2.8

A language server for just
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60

use super::*;

/// Verifies builtin function calls use a valid argument count and respect
/// variadic constraints.
pub struct FunctionArgumentsRule;

impl Rule for FunctionArgumentsRule {
  fn display_name(&self) -> &'static str {
    "Function Arguments"
  }

  fn id(&self) -> &'static str {
    "function-arguments"
  }

  fn run(&self, context: &RuleContext<'_>) -> Vec<lsp::Diagnostic> {
    let mut diagnostics = Vec::new();

    for function_call in context.function_calls() {
      let function_name = &function_call.name.value;

      let builtin = BUILTINS.iter().find(|f| {
        matches!(f, Builtin::Function { name, .. } if *name == function_name)
      });

      if let Some(Builtin::Function {
        required_args,
        accepts_variadic,
        ..
      }) = builtin
      {
        let arg_count = function_call.arguments.len();

        if arg_count < *required_args {
          diagnostics.push(self.diagnostic(lsp::Diagnostic {
            range: function_call.range,
            severity: Some(lsp::DiagnosticSeverity::ERROR),
            message: format!(
              "Function `{function_name}` requires at least {required_args} {}, but {arg_count} provided",
              Count("argument", *required_args)
            ),
            ..Default::default()
          }));
        } else if !accepts_variadic && arg_count > *required_args {
          diagnostics.push(self.diagnostic(lsp::Diagnostic {
            range: function_call.range,
            severity: Some(lsp::DiagnosticSeverity::ERROR),
            message: format!(
              "Function `{function_name}` accepts {required_args} {}, but {arg_count} provided",
              Count("argument", *required_args)
            ),
            ..Default::default()
          }));
        }
      }
    }

    diagnostics
  }
}