fix: lsp completion depend on func return type for func call expr (#1472

)

Signed-off-by: he1pa <[email protected]>

kclvm/sema/src/advanced_resolver/node.rs

@@ -910,7 +910,16 @@ impl<'ctx> AdvancedResolver<'ctx> {
                         end,
                         None,
                     );
-                    expr_symbol.sema_info.ty = Some(ty.clone());
+                    expr_symbol.sema_info.ty = if matches!(&expr.node, | ast::Expr::Call(_)) {
+                        if let TypeKind::Function(func_ty) = &ty.kind {
+                            Some(func_ty.return_ty.clone())
+                        } else {
+                            Some(ty.clone())
+                        }
+                    } else {
+                        Some(ty.clone())
+                    };
+
                     Ok(self.gs.get_symbols_mut().alloc_expression_symbol(
                         expr_symbol,
                         self.ctx.get_node_key(&expr.id),

kclvm/tools/src/LSP/src/completion.rs

@@ -2024,4 +2024,28 @@ mod tests {
         8,
         None
     );
+
+    completion_label_without_builtin_func_test_snapshot!(
+        func_return_ty_1,
+        "src/test_data/completion_test/dot/func_return/func_return_1/func_return_1.k",
+        4,
+        8,
+        Some('.')
+    );
+
+    completion_label_without_builtin_func_test_snapshot!(
+        func_return_ty_2,
+        "src/test_data/completion_test/dot/func_return/func_return_2/func_return_2.k",
+        8,
+        12,
+        Some('.')
+    );
+
+    completion_label_without_builtin_func_test_snapshot!(
+        func_return_ty_3,
+        "src/test_data/completion_test/dot/func_return/func_return_3/func_return_3.k",
+        3,
+        2,
+        Some('.')
+    );
 }

kclvm/tools/src/LSP/src/snapshots/kcl_language_server__completion__tests__func_return_ty_1.snap

@@ -0,0 +1,5 @@
+---
+source: tools/src/LSP/src/completion.rs
+expression: "format!(\"{:?}\", got_labels)"
+---
+["capitalize(…)", "count(…)", "endswith(…)", "find(…)", "format(…)", "index(…)", "isalnum(…)", "isalpha(…)", "isdigit(…)", "islower(…)", "isspace(…)", "istitle(…)", "isupper(…)", "join(…)", "lower(…)", "lstrip(…)", "removeprefix(…)", "removesuffix(…)", "replace(…)", "rfind(…)", "rindex(…)", "rsplit(…)", "rstrip(…)", "split(…)", "splitlines(…)", "startswith(…)", "strip(…)", "title(…)", "upper(…)"]

kclvm/tools/src/LSP/src/snapshots/kcl_language_server__completion__tests__func_return_ty_2.snap

@@ -0,0 +1,5 @@
+---
+source: tools/src/LSP/src/completion.rs
+expression: "format!(\"{:?}\", got_labels)"
+---
+["name"]

kclvm/tools/src/LSP/src/snapshots/kcl_language_server__completion__tests__func_return_ty_3.snap

@@ -0,0 +1,5 @@
+---
+source: tools/src/LSP/src/completion.rs
+expression: "format!(\"{:?}\", got_labels)"
+---
+[]

kclvm/tools/src/LSP/src/test_data/completion_test/dot/func_return/func_return_1/func_return_1.k

@@ -0,0 +1,4 @@
+f: () -> str = lambda {
+    ""
+}
+a = f()

kclvm/tools/src/LSP/src/test_data/completion_test/dot/func_return/func_return_2/func_return_2.k

@@ -0,0 +1,8 @@
+schema Name:
+    name: str
+
+f1: (str) -> Name = lambda s: str -> Name {
+    Name {name: s}
+}
+
+b = f1("a")

kclvm/tools/src/LSP/src/test_data/completion_test/dot/func_return/func_return_3/func_return_3.k

@@ -0,0 +1,3 @@
+f: () -> str = lambda {
+    ""
+}