feat: enhance lsp completion detail (kcl-lang#805)

* feat: enhance lsp completion detail. Add completion kind(icon in vscode), detail and doc in completion item. Enhance completion label for function Signed-off-by: He1pa <[email protected]> * doc: fix doc typo Signed-off-by: He1pa <[email protected]> --------- Signed-off-by: He1pa <[email protected]>
Peefy · Oct 26, 2023 · 1d9eccb · 1d9eccb
1 parent 4eac50d
commit 1d9eccb
Show file tree

Hide file tree

Showing 3 changed files with 356 additions and 215 deletions.
diff --git a/kclvm/sema/src/ty/mod.rs b/kclvm/sema/src/ty/mod.rs
@@ -86,6 +86,21 @@ impl Type {
             TypeKind::Named(name) => name.to_string(),
         }
     }
+
+    pub fn ty_doc(&self) -> Option<String> {
+        match &self.kind {
+            TypeKind::Schema(schema) => Some(schema.doc.clone()),
+            TypeKind::Function(func) => Some(func.doc.clone()),
+            _ => None,
+        }
+    }
+
+    pub fn into_function_ty(&self) -> FunctionType {
+        match &self.kind {
+            TypeKind::Function(func) => func.clone(),
+            _ => panic!("Not a function type"),
+        }
+    }
 }
 
 #[derive(Debug, Clone, PartialEq)]
@@ -399,6 +414,19 @@ impl FunctionType {
             self.return_ty.ty_str()
         )
     }
+
+    pub fn func_signature_str(&self, name: &String) -> String {
+        format!(
+            "function {}({}) -> {}",
+            name,
+            self.params
+                .iter()
+                .map(|param| format!("{}: {}", param.name, param.ty.ty_str()))
+                .collect::<Vec<String>>()
+                .join(", "),
+            self.return_ty.ty_str()
+        )
+    }
 }
 
 impl FunctionType {