emmylua_ls 0.7.3

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

use emmylua_code_analysis::LuaCompilation;
use lsp_types::{CodeLens, Command, Location, Range, Uri};

use crate::{
    context::ClientId,
    handlers::references::{search_decl_references, search_member_references},
};

use super::CodeLensData;

// VSCode does not support calling editor.action.showReferences directly through LSP,
// it can only be converted through the VSCode plugin
const VSCODE_COMMAND_NAME: &str = "emmy.showReferences";
// In fact, VSCode ultimately uses this command
const OTHER_COMMAND_NAME: &str = "editor.action.showReferences";

pub fn resolve_code_lens(
    compilation: &LuaCompilation,
    code_lens: CodeLens,
    client_id: ClientId,
) -> Option<CodeLens> {
    let data = code_lens.data.as_ref()?;
    let data = serde_json::from_value(data.clone()).ok()?;
    match data {
        CodeLensData::Member(member_id) => {
            let file_id = member_id.file_id;
            let mut semantic_model = compilation.get_semantic_model(file_id)?;
            let mut results = Vec::new();
            search_member_references(&mut semantic_model, compilation, member_id, &mut results);
            let mut ref_count = results.len();
            if ref_count > 0 {
                ref_count -= 1;
            }
            let uri = semantic_model.get_document().get_uri();
            let command = make_usage_command(uri, code_lens.range, ref_count, client_id, results);

            Some(CodeLens {
                range: code_lens.range,
                command: Some(command),
                data: None,
            })
        }
        CodeLensData::DeclId(decl_id) => {
            let file_id = decl_id.file_id;
            let mut semantic_model = compilation.get_semantic_model(file_id)?;
            let mut results = Vec::new();
            search_decl_references(&mut semantic_model, decl_id, &mut results);
            let ref_count = results.len();
            let uri = semantic_model.get_document().get_uri();
            let command = make_usage_command(uri, code_lens.range, ref_count, client_id, results);
            Some(CodeLens {
                range: code_lens.range,
                command: Some(command),
                data: None,
            })
        }
    }
}

fn get_command_name(client_id: ClientId) -> &'static str {
    match client_id {
        ClientId::VSCode => VSCODE_COMMAND_NAME,
        _ => OTHER_COMMAND_NAME,
    }
}

fn make_usage_command(
    uri: Uri,
    range: Range,
    ref_count: usize,
    client_id: ClientId,
    refs: Vec<Location>,
) -> Command {
    let title = format!("{} usage", ref_count);
    let mut args = Vec::new();
    args.push(serde_json::to_value(uri).unwrap());
    args.push(serde_json::to_value(range.start).unwrap());
    args.push(serde_json::to_value(refs).unwrap());

    Command {
        title,
        command: get_command_name(client_id).to_string(),
        arguments: Some(args),
    }
}