emmylua_ls 0.4.2

A language server for emmylua.
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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92

use std::str::FromStr;

use emmylua_code_analysis::{DiagnosticCode, FileId, SemanticModel};
use lsp_types::{CodeActionOrCommand, CodeActionResponse, Diagnostic, NumberOrString, Range};

use crate::handlers::command::{make_disable_code_command, DisableAction};

pub fn build_actions(
    semantic_model: &mut SemanticModel,
    diagnostics: Vec<Diagnostic>,
) -> Option<CodeActionResponse> {
    let mut actions = Vec::new();
    let file_id = semantic_model.get_file_id();
    for diagnostic in diagnostics {
        if diagnostic.source.is_none() {
            continue;
        }

        let source = diagnostic.source.unwrap();
        if source != "EmmyLua" {
            continue;
        }

        if let Some(code) = diagnostic.code {
            if let NumberOrString::String(action_string) = code {
                if let Some(diagnostic_code) = DiagnosticCode::from_str(&action_string).ok() {
                    add_fix_code_action(&mut actions, diagnostic_code, file_id, diagnostic.range);
                    add_disable_code_action(
                        &mut actions,
                        diagnostic_code,
                        file_id,
                        diagnostic.range,
                    );
                }
            }
        }
    }

    Some(actions)
}

#[allow(unused_variables)]
fn add_fix_code_action(
    actions: &mut Vec<CodeActionOrCommand>,
    diagnostic_code: DiagnosticCode,
    file_id: FileId,
    range: Range,
) -> Option<()> {
    Some(())
}

fn add_disable_code_action(
    actions: &mut Vec<CodeActionOrCommand>,
    diagnostic_code: DiagnosticCode,
    file_id: FileId,
    range: Range,
) -> Option<()> {
    actions.push(CodeActionOrCommand::Command(make_disable_code_command(
        &format!(
            "Disable current line diagnostic ({})",
            diagnostic_code.get_name()
        ),
        DisableAction::DisableLine,
        diagnostic_code,
        file_id,
        range,
    )));

    actions.push(CodeActionOrCommand::Command(make_disable_code_command(
        &format!(
            "Disable all diagnostics in current file ({})",
            diagnostic_code.get_name()
        ),
        DisableAction::DisableFile,
        diagnostic_code,
        file_id,
        range,
    )));

    actions.push(CodeActionOrCommand::Command(make_disable_code_command(
        &format!(
            "Disable all diagnostics in current project ({})",
            diagnostic_code.get_name()
        ),
        DisableAction::DisableProject,
        diagnostic_code,
        file_id,
        range,
    )));

    Some(())
}