Feat: lsp support find refs (#721)

* feat: lsp find references Signed-off-by: xiarui.xr <[email protected]> * find refs: add unit test; fix kcl_pos bug Signed-off-by: xiarui.xr <[email protected]> * find references: add more tests Signed-off-by: xiarui.xr <[email protected]> * remove unused comment Signed-off-by: xiarui.xr <[email protected]> * chore: cargo fmt Signed-off-by: xiarui.xr <[email protected]> * find refs: revert bench test temporarily Signed-off-by: xiarui.xr <[email protected]> * minor fix Signed-off-by: xiarui.xr <[email protected]> * minor fix: use if-else instead of match statement Signed-off-by: xiarui.xr <[email protected]> * lsp: cache word index in lsp state Signed-off-by: xiarui.xr <[email protected]> * chore: format code Signed-off-by: xiarui.xr <[email protected]> * find refs: incrementally update word_index after file changes Signed-off-by: xiarui.xr <[email protected]> --------- Signed-off-by: xiarui.xr <[email protected]>
kcl-lang · Oct 12, 2023 · 93eb9d7 · 93eb9d7
1 parent e13ef0e
commit 93eb9d7
Show file tree

Hide file tree

Showing 27 changed files with 1,106 additions and 650 deletions.
diff --git a/kclvm/tools/src/LSP/src/capabilities.rs b/kclvm/tools/src/LSP/src/capabilities.rs
@@ -38,6 +38,7 @@ pub fn server_capabilities(client_caps: &ClientCapabilities) -> ServerCapabiliti
         ),
         document_formatting_provider: Some(OneOf::Left(true)),
         document_range_formatting_provider: Some(OneOf::Left(true)),
+        references_provider: Some(OneOf::Left(true)),
         ..Default::default()
     }
 }
diff --git a/kclvm/tools/src/LSP/src/completion.rs b/kclvm/tools/src/LSP/src/completion.rs
@@ -83,7 +83,7 @@ fn completion_attr(
     if let Some((node, schema_expr)) = is_in_schema(program, pos) {
         let schema_def = find_def(node, &schema_expr.name.get_end_pos(), prog_scope);
         if let Some(schema) = schema_def {
-            if let Definition::Object(obj) = schema {
+            if let Definition::Object(obj, _) = schema {
                 let schema_type = obj.ty.into_schema_type();
                 completions.extend(schema_type.attrs.keys().map(|attr| KCLCompletionItem {
                     label: attr.clone(),
@@ -178,7 +178,7 @@ pub(crate) fn get_completion(
                     let def = find_def(stmt, pos, prog_scope);
                     if let Some(def) = def {
                         match def {
-                            crate::goto_def::Definition::Object(obj) => {
+                            crate::goto_def::Definition::Object(obj, _) => {
                                 match &obj.ty.kind {
                                     // builtin (str) functions
                                     kclvm_sema::ty::TypeKind::Str => {
@@ -226,7 +226,7 @@ pub(crate) fn get_completion(
                                     _ => {}
                                 }
                             }
-                            crate::goto_def::Definition::Scope(s) => {
+                            crate::goto_def::Definition::Scope(s, _) => {
                                 for (name, obj) in &s.elems {
                                     if let ScopeObjectKind::Module(_) = obj.borrow().kind {
                                         continue;
@@ -259,7 +259,7 @@ pub(crate) fn get_completion(
                                 find_def(stmt, &schema_expr.name.get_end_pos(), prog_scope);
                             if let Some(schema) = schema_def {
                                 match schema {
-                                    Definition::Object(obj) => {
+                                    Definition::Object(obj, _) => {
                                         let schema_type = obj.ty.into_schema_type();
                                         items.extend(
                                             schema_type
@@ -271,7 +271,7 @@ pub(crate) fn get_completion(
                                                 .collect::<IndexSet<KCLCompletionItem>>(),
                                         );
                                     }
-                                    Definition::Scope(_) => {}
+                                    Definition::Scope(_, _) => {}
                                 }
                             }
                         }

diff --git a/kclvm/tools/src/LSP/src/find_ref/find_refs.rs b/kclvm/tools/src/LSP/src/find_ref/find_refs.rs
diff --git a/kclvm/tools/src/LSP/src/find_ref/go_to_def.rs b/kclvm/tools/src/LSP/src/find_ref/go_to_def.rs
diff --git a/kclvm/tools/src/LSP/src/find_ref/mod.rs b/kclvm/tools/src/LSP/src/find_ref/mod.rs
diff --git a/kclvm/tools/src/LSP/src/find_ref/test_data/inherit.k b/kclvm/tools/src/LSP/src/find_ref/test_data/inherit.k
diff --git a/kclvm/tools/src/LSP/src/find_ref/test_data/inherit_pkg.k b/kclvm/tools/src/LSP/src/find_ref/test_data/inherit_pkg.k
diff --git a/kclvm/tools/src/LSP/src/find_ref/test_data/test_word_workspace/inherit.k b/kclvm/tools/src/LSP/src/find_ref/test_data/test_word_workspace/inherit.k
diff --git a/kclvm/tools/src/LSP/src/find_ref/test_data/test_word_workspace/inherit_pkg.k b/kclvm/tools/src/LSP/src/find_ref/test_data/test_word_workspace/inherit_pkg.k
diff --git a/kclvm/tools/src/LSP/src/find_ref/test_data/test_word_workspace_map/inherit.k b/kclvm/tools/src/LSP/src/find_ref/test_data/test_word_workspace_map/inherit.k
diff --git a/kclvm/tools/src/LSP/src/find_ref/test_data/test_word_workspace_map/inherit_pkg.k b/kclvm/tools/src/LSP/src/find_ref/test_data/test_word_workspace_map/inherit_pkg.k