support callsnippet

Author: CppCXY

crates/emmylua_ls/src/handlers/completion/add_completions/add_decl_completion.rs

@@ -2,7 +2,8 @@ use emmylua_code_analysis::{DbIndex, LuaDeclId, LuaSemanticDeclId, LuaType};
 use lsp_types::CompletionItem;
 
 use crate::handlers::completion::{
-    completion_builder::CompletionBuilder, completion_data::CompletionData,
+    add_completions::get_function_snippet, completion_builder::CompletionBuilder,
+    completion_data::CompletionData,
 };
 
 use super::{
@@ -19,6 +20,7 @@ pub fn add_decl_completion(
     check_visibility(builder, property_owner.clone())?;
 
     let overload_count = count_function_overloads(builder.semantic_model.get_db(), typ);
+
     let mut completion_item = CompletionItem {
         label: name.to_string(),
         kind: Some(get_completion_kind(typ)),
@@ -34,6 +36,13 @@ pub fn add_decl_completion(
         completion_item.deprecated = Some(true);
     }
 
+    if builder.support_snippets(typ) {
+        if let Some(snippet) = get_function_snippet(builder, name, typ, CallDisplay::None) {
+            completion_item.insert_text = Some(snippet);
+            completion_item.insert_text_format = Some(lsp_types::InsertTextFormat::SNIPPET);
+        }
+    }
+
     builder.add_completion_item(completion_item)?;
     Some(())
 }

crates/emmylua_ls/src/handlers/completion/add_completions/add_member_completion.rs

@@ -9,8 +9,8 @@ use emmylua_parser::{
 use lsp_types::CompletionItem;
 
 use crate::handlers::completion::{
-    completion_builder::CompletionBuilder, completion_data::CompletionData,
-    providers::get_function_remove_nil,
+    add_completions::get_function_snippet, completion_builder::CompletionBuilder,
+    completion_data::CompletionData, providers::get_function_remove_nil,
 };
 
 use super::{
@@ -40,6 +40,7 @@ pub fn add_member_completion(
     }
 
     let member_key = &member_info.key;
+    let mut can_add_snippet = true;
     let label = match status {
         CompletionTriggerStatus::Dot => match member_key {
             LuaMemberKey::Name(name) => name.to_string(),
@@ -70,15 +71,21 @@ pub fn add_member_completion(
             LuaMemberKey::Name(name) => name.to_string(),
             _ => return None,
         },
-        CompletionTriggerStatus::InString => match member_key {
-            LuaMemberKey::Name(name) => name.to_string(),
-            _ => return None,
-        },
-        CompletionTriggerStatus::LeftBracket => match member_key {
-            LuaMemberKey::Name(name) => format!("\"{}\"", name),
-            LuaMemberKey::Integer(index) => format!("{}", index),
-            _ => return None,
-        },
+        CompletionTriggerStatus::InString => {
+            can_add_snippet = false;
+            match member_key {
+                LuaMemberKey::Name(name) => name.to_string(),
+                _ => return None,
+            }
+        }
+        CompletionTriggerStatus::LeftBracket => {
+            can_add_snippet = false;
+            match member_key {
+                LuaMemberKey::Name(name) => format!("\"{}\"", name),
+                LuaMemberKey::Integer(index) => format!("{}", index),
+                _ => return None,
+            }
+        }
     };
 
     let typ = &member_info.typ;
@@ -149,6 +156,13 @@ pub fn add_member_completion(
         );
     }
 
+    if can_add_snippet && builder.support_snippets(typ) {
+        if let Some(snippet) = get_function_snippet(builder, &label, typ, call_display) {
+            completion_item.insert_text = Some(snippet);
+            completion_item.insert_text_format = Some(lsp_types::InsertTextFormat::SNIPPET);
+        }
+    }
+
     // 尝试添加别名补全项, 如果添加成功, 则不添加原来的 `[index]` 补全项
     if !try_add_alias_completion_item_new(builder, &member_info, &completion_item, &label)
         .unwrap_or(false)

crates/emmylua_ls/src/handlers/completion/add_completions/mod.rs

@@ -154,6 +154,83 @@ pub fn get_detail(
     }
 }
 
+pub fn get_function_snippet(
+    builder: &CompletionBuilder,
+    label: &str,
+    typ: &LuaType,
+    display: CallDisplay,
+) -> Option<String> {
+    match typ {
+        LuaType::Signature(signature_id) => {
+            let signature = builder
+                .semantic_model
+                .get_db()
+                .get_signature_index()
+                .get(signature_id)?;
+
+            let mut params_str = signature
+                .get_type_params()
+                .iter()
+                .map(|param| param.0.clone())
+                .collect::<Vec<_>>();
+
+            match display {
+                CallDisplay::AddSelf => {
+                    params_str.insert(0, "self".to_string());
+                }
+                CallDisplay::RemoveFirst => {
+                    if !params_str.is_empty() {
+                        params_str.remove(0);
+                    }
+                }
+                _ => {}
+            }
+
+            Some(format!(
+                "{}({})",
+                label,
+                params_str
+                    .iter()
+                    .enumerate()
+                    .map(|(i, name)| format!("${{{}:{}}}", i + 1, name))
+                    .collect::<Vec<_>>()
+                    .join(", ")
+            ))
+        }
+        LuaType::DocFunction(f) => {
+            let mut params_str = f
+                .get_params()
+                .iter()
+                .map(|param| param.0.clone())
+                .collect::<Vec<_>>();
+
+            match display {
+                CallDisplay::AddSelf => {
+                    params_str.insert(0, "self".to_string());
+                }
+                CallDisplay::RemoveFirst => {
+                    if !params_str.is_empty() {
+                        params_str.remove(0);
+                    }
+                }
+                _ => {}
+            }
+
+            Some(format!(
+                "{}({})",
+                label,
+                params_str
+                    .iter()
+                    .enumerate()
+                    .map(|(i, name)| format!("${{{}:{}}}", i + 1, name))
+                    .collect::<Vec<_>>()
+                    .join(", ")
+            ))
+        }
+        _ => None,
+    }
+}
+
 #[allow(unused)]
 fn truncate_with_ellipsis(s: &str, max_len: usize) -> String {
     if s.chars().count() > max_len {

crates/emmylua_ls/src/handlers/completion/completion_builder.rs

@@ -1,6 +1,6 @@
 use std::collections::HashSet;
 
-use emmylua_code_analysis::SemanticModel;
+use emmylua_code_analysis::{LuaType, SemanticModel};
 use emmylua_parser::LuaSyntaxToken;
 use lsp_types::{CompletionItem, CompletionTriggerKind};
 use rowan::TextSize;
@@ -76,4 +76,14 @@ impl<'a> CompletionBuilder<'a> {
     pub fn is_invoked(&self) -> bool {
         self.trigger_kind == CompletionTriggerKind::INVOKED
     }
+
+    pub fn support_snippets(&self, ty: &LuaType) -> bool {
+        ty.is_function()
+            && self
+                .semantic_model
+                .get_db()
+                .get_emmyrc()
+                .completion
+                .call_snippet
+    }
 }