emmylua_ls 0.8.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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147

use emmylua_code_analysis::uri_to_file_path;
use lsp_types::{
    DidChangeTextDocumentParams, DidCloseTextDocumentParams, DidOpenTextDocumentParams,
    DidSaveTextDocumentParams,
};
use std::time::Duration;

use crate::context::ServerContextSnapshot;

pub async fn on_did_open_text_document(
    context: ServerContextSnapshot,
    params: DidOpenTextDocumentParams,
) -> Option<()> {
    let mut analysis = context.analysis.write().await;
    let uri = params.text_document.uri;
    let text = params.text_document.text;
    let old_file_id = analysis.get_file_id(&uri);
    // check is filter file
    if old_file_id.is_none() {
        let workspace_manager = context.workspace_manager.read().await;
        if !workspace_manager.is_workspace_file(&uri) {
            return None;
        }
    }

    let file_id = analysis.update_file_by_uri(&uri, Some(text));
    let emmyrc = analysis.get_emmyrc();
    let interval = emmyrc.diagnostics.diagnostic_interval.unwrap_or(500);
    if let Some(file_id) = file_id {
        context
            .file_diagnostic
            .add_diagnostic_task(file_id, interval)
            .await;
    }

    let mut workspace = context.workspace_manager.write().await;
    workspace.current_open_files.insert(uri);
    drop(workspace);

    Some(())
}

pub async fn on_did_save_text_document(
    context: ServerContextSnapshot,
    _: DidSaveTextDocumentParams,
) -> Option<()> {
    let emmyrc = context.analysis.read().await.get_emmyrc();
    if !emmyrc.workspace.enable_reindex {
        return Some(());
    }

    let mut duration = emmyrc.workspace.reindex_duration;
    // if duration is less than 1000ms, set it to 1000ms
    if duration < 1000 {
        duration = 1000;
    }
    let workspace = context.workspace_manager.read().await;
    workspace
        .reindex_workspace(Duration::from_millis(duration))
        .await;
    Some(())
}

pub async fn on_did_change_text_document(
    context: ServerContextSnapshot,
    params: DidChangeTextDocumentParams,
) -> Option<()> {
    let mut analysis = context.analysis.write().await;
    let uri = params.text_document.uri;
    let text = params.content_changes.first()?.text.clone();
    let old_file_id = analysis.get_file_id(&uri);
    // check is filter file
    if old_file_id.is_none() {
        let workspace_manager = context.workspace_manager.read().await;
        if !workspace_manager.is_workspace_file(&uri) {
            return None;
        }
    }

    let file_id = analysis.update_file_by_uri(&uri, Some(text));
    let emmyrc = analysis.get_emmyrc();
    let interval = emmyrc.diagnostics.diagnostic_interval.unwrap_or(500);
    drop(analysis);

    if emmyrc.workspace.enable_reindex {
        let workspace = context.workspace_manager.read().await;
        workspace.extend_reindex_delay().await;
        drop(workspace);
    }
    if let Some(file_id) = file_id {
        context
            .file_diagnostic
            .add_diagnostic_task(file_id, interval)
            .await;
    }

    Some(())
}

pub async fn on_did_close_document(
    context: ServerContextSnapshot,
    params: DidCloseTextDocumentParams,
) -> Option<()> {
    let uri = &params.text_document.uri;
    let mut workspace = context.workspace_manager.write().await;
    workspace
        .current_open_files
        .remove(&params.text_document.uri);
    drop(workspace);

    // 如果关闭后文件不存在, 则移除
    if let Some(file_path) = uri_to_file_path(uri) {
        if !file_path.exists() {
            let mut mut_analysis = context.analysis.write().await;
            mut_analysis.remove_file_by_uri(uri);
            drop(mut_analysis);

            context
                .file_diagnostic
                .clear_file_diagnostics(uri.clone())
                .await;

            return Some(());
        }
    }

    let analysis = context.analysis.read().await;
    let file_id = analysis.get_file_id(uri)?;
    let module_info = analysis
        .compilation
        .get_db()
        .get_module_index()
        .get_module(file_id);
    if module_info.is_none() {
        drop(analysis);
        let mut mut_analysis = context.analysis.write().await;
        mut_analysis.remove_file_by_uri(uri);
        drop(mut_analysis);
        // 发送空诊断消息以清除客户端显示的诊断
        context
            .file_diagnostic
            .clear_file_diagnostics(uri.clone())
            .await;
    }

    Some(())
}