Fix rename handle

Author: CppCXY

crates/emmylua_ls/src/handlers/rename/rename_decl.rs

@@ -1,6 +1,10 @@
 use std::collections::HashMap;
 
 use emmylua_code_analysis::{LuaCompilation, LuaDeclId, SemanticModel};
+use emmylua_parser::{
+    LuaAst, LuaAstNode, LuaAstToken, LuaClosureExpr, LuaCommentOwner, LuaDocTagParam, LuaStat,
+    LuaTableField,
+};
 use lsp_types::Uri;
 
 pub fn rename_decl_references(
@@ -37,6 +41,10 @@ pub fn rename_decl_references(
             .or_insert_with(HashMap::new)
             .insert(decl_range, new_name.clone());
 
+        if decl.is_param() {
+            rename_doc_param(semantic_model, decl_id, new_name, result);
+        }
+
         return Some(());
     } else {
         let name = decl.get_name();
@@ -80,3 +88,49 @@ fn get_decl_name_token_lsp_range(
     let document = semantic_model.get_document_by_file_id(decl_id.file_id)?;
     document.to_lsp_range(decl.get_range())
 }
+
+fn rename_doc_param(
+    semantic_model: &SemanticModel,
+    decl_id: LuaDeclId,
+    new_name: String,
+    result: &mut HashMap<Uri, HashMap<lsp_types::Range, String>>,
+) -> Option<()> {
+    let decl = semantic_model
+        .get_db()
+        .get_decl_index()
+        .get_decl(&decl_id)?;
+    let name = decl.get_name();
+    let syntax_id = decl.get_syntax_id();
+    let root = semantic_model.get_root();
+    let param_node = LuaAst::cast(syntax_id.to_node_from_root(root.syntax())?)?;
+    let closure_expr = param_node.ancestors::<LuaClosureExpr>().next()?;
+    let comments = if let Some(table_field) = closure_expr.get_parent::<LuaTableField>() {
+        table_field.get_comments()
+    } else if let Some(stat) = closure_expr.ancestors::<LuaStat>().next() {
+        stat.get_comments()
+    } else {
+        return None;
+    };
+
+    let document = semantic_model.get_document();
+    let uri = document.get_uri();
+    for comment in comments {
+        for tag_doc in comment.get_doc_tags() {
+            if let Some(doc_param) = LuaDocTagParam::cast(tag_doc.syntax().clone()) {
+                if let Some(name_token) = doc_param.get_name_token() {
+                    if name_token.get_text() != name {
+                        continue;
+                    }
+
+                    let range = document.to_lsp_range(name_token.get_range())?;
+                    result
+                        .entry(uri.clone())
+                        .or_insert_with(HashMap::new)
+                        .insert(range, new_name.clone());
+                }
+            }
+        }
+    }
+
+    Some(())
+}

crates/emmylua_parser/src/syntax/node/lua/mod.rs

@@ -6,7 +6,7 @@ mod test;
 use crate::{
     kind::{LuaSyntaxKind, LuaTokenKind},
     syntax::traits::{LuaAstChildren, LuaAstNode, LuaAstToken},
-    LuaSyntaxNode,
+    LuaCommentOwner, LuaSyntaxNode,
 };
 
 pub use expr::*;
@@ -247,6 +247,8 @@ impl LuaAstNode for LuaTableField {
     }
 }
 
+impl LuaCommentOwner for LuaTableField {}
+
 impl LuaTableField {
     pub fn is_assign_field(&self) -> bool {
         self.syntax().kind() == LuaSyntaxKind::TableFieldAssign.into()