Fix #531

Author: CppCXY

crates/emmylua_code_analysis/src/vfs/mod.rs

@@ -159,4 +159,14 @@ impl Vfs {
             })
             .collect()
     }
+
+    pub fn clear(&mut self) {
+        self.file_id_map.clear();
+        self.file_path_map.clear();
+        self.file_data.clear();
+        self.line_index_map.clear();
+        self.tree_map.clear();
+        self.emmyrc = None;
+        self.node_cache = NodeCache::default();
+    }
 }

crates/emmylua_ls/src/context/client_id.rs

@@ -35,17 +35,13 @@ impl ClientId {
 
 pub fn get_client_id(client_info: &Option<ClientInfo>) -> ClientId {
     match client_info {
-        Some(info) => {
-            match info.name.as_str() {
-                "Visual Studio Code" => ClientId::VSCode,
-                "IntelliJ" => ClientId::Intellij,
-                "Neovim" | "coc.nvim" => ClientId::Neovim,
-                "Cursor" => ClientId::VSCode,
-                "Windsurf" => ClientId::VSCode, // 不确定是不是这个名, 先加上吧
-                "Trae" => ClientId::VSCode,     // 字节的, 但不确定是不是这个名, 先加上吧
-                _ => ClientId::Other,
-            }
-        }
+        Some(info) => match info.name.as_str() {
+            "Visual Studio Code" => ClientId::VSCode,
+            "IntelliJ" => ClientId::Intellij,
+            "Neovim" | "coc.nvim" => ClientId::Neovim,
+            "Cursor" | "Windsurf" | "Trae" => ClientId::VSCode,
+            _ => ClientId::Other,
+        },
         None => ClientId::Other,
     }
 }

crates/emmylua_ls/src/handlers/initialized/mod.rs

@@ -92,6 +92,7 @@ pub async fn init_analysis(
     file_diagnostic: Arc<FileDiagnostic>,
 ) {
     let mut mut_analysis = analysis.write().await;
+    
     // update config
     mut_analysis.update_config(emmyrc.clone());
 
@@ -160,7 +161,7 @@ pub async fn init_analysis(
         .await;
 }
 
-fn get_workspace_folders(params: &InitializeParams) -> Vec<PathBuf> {
+pub fn get_workspace_folders(params: &InitializeParams) -> Vec<PathBuf> {
     let mut workspace_folders = Vec::new();
     if let Some(workspaces) = &params.workspace_folders {
         for workspace in workspaces {

crates/emmylua_ls/src/handlers/mod.rs

@@ -31,8 +31,7 @@ mod test_lib;
 mod text_document;
 mod workspace_symbol;
 
-pub use initialized::initialized_handler;
-pub use initialized::{init_analysis, ClientConfig};
+pub use initialized::{initialized_handler, ClientConfig, init_analysis};
 use lsp_types::{ClientCapabilities, ServerCapabilities};
 pub use notification_handler::on_notification_handler;
 pub use request_handler::on_req_handler;

crates/emmylua_ls/src/handlers/text_document/text_document_handler.rs

@@ -106,5 +106,15 @@ pub async fn on_did_close_document(
         .current_open_files
         .remove(&params.text_document.uri);
     drop(workspace);
+    let analysis = context.analysis.read().await;
+    let uri = &params.text_document.uri;
+    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);
+    }
+
     Some(())
 }