feat(lsp): add references code lens

cli/lsp/analysis.rs

@@ -9,6 +9,8 @@ use crate::module_graph::TypeScriptReference;
 use crate::tools::lint::create_linter;
 
 use deno_core::error::AnyError;
+use deno_core::serde::Deserialize;
+use deno_core::serde::Serialize;
 use deno_core::ModuleSpecifier;
 use deno_lint::rules;
 use lspower::lsp;
@@ -249,6 +251,19 @@ pub fn analyze_dependencies(
   }
 }
 
+#[derive(Debug, Deserialize, Serialize)]
+pub enum CodeLensSource {
+  #[serde(rename = "references")]
+  References,
+}
+
+#[derive(Debug, Deserialize)]
+#[serde(rename_all = "camelCase")]
+pub struct CodeLensData {
+  pub source: CodeLensSource,
+  pub specifier: ModuleSpecifier,
+}
+
 #[cfg(test)]
 mod tests {
   use super::*;

cli/lsp/capabilities.rs

@@ -6,6 +6,7 @@
 ///! client.
 ///!
 use lspower::lsp::ClientCapabilities;
+use lspower::lsp::CodeLensOptions;
 use lspower::lsp::CompletionOptions;
 use lspower::lsp::HoverProviderCapability;
 use lspower::lsp::ImplementationProviderCapability;
@@ -59,7 +60,9 @@ pub fn server_capabilities(
     document_symbol_provider: None,
     workspace_symbol_provider: None,
     code_action_provider: None,
-    code_lens_provider: None,
+    code_lens_provider: Some(CodeLensOptions {
+      resolve_provider: Some(true),
+    }),
     document_formatting_provider: Some(OneOf::Left(true)),
     document_range_formatting_provider: None,
     document_on_type_formatting_provider: None,

cli/lsp/config.rs

@@ -15,20 +15,62 @@ pub struct ClientCapabilities {
   pub workspace_did_change_watched_files: bool,
 }
 
-#[derive(Debug, Clone, Default, Deserialize)]
+#[derive(Debug, Default, Deserialize)]
+#[serde(rename_all = "camelCase")]
+pub struct CodeLensSettings {
+  /// Flag for providing reference code lens.
+  #[serde(default)]
+  pub references: bool,
+  /// Flag for providing reference code lens on all functions.  For this to have
+  /// an impact, the `references` flag needs to be `true`.
+  #[serde(default)]
+  pub references_all_functions: bool,
+}
+
+#[derive(Debug, Default, Deserialize)]
 #[serde(rename_all = "camelCase")]
 pub struct WorkspaceSettings {
   pub enable: bool,
   pub config: Option<String>,
   pub import_map: Option<String>,
+  pub code_lens: Option<CodeLensSettings>,
 
   #[serde(default)]
   pub lint: bool,
   #[serde(default)]
   pub unstable: bool,
 }
 
-#[derive(Debug, Clone, Default)]
+impl WorkspaceSettings {
+  /// Determine if any code lenses are enabled at all.  This allows short
+  /// circuiting when there are no code lenses enabled.
+  pub fn enabled_code_lens(&self) -> bool {
+    if let Some(code_lens) = &self.code_lens {
+      // This should contain all the "top level" code lens references
+      code_lens.references
+    } else {
+      false
+    }
+  }
+
+  pub fn enabled_code_lens_references(&self) -> bool {
+    if let Some(code_lens) = &self.code_lens {
+      code_lens.references
+    } else {
+      false
+    }
+  }
+
+  pub fn enabled_code_lens_references_all_functions(&self) -> bool {
+    if let Some(code_lens) = &self.code_lens {
+      code_lens.references_all_functions
+    } else {
+      false
+    }
+  }
+}
+
+#[derive(Debug, Default)]
 pub struct Config {
   pub client_capabilities: ClientCapabilities,
   pub root_uri: Option<Url>,

cli/lsp/documents.rs

@@ -2,6 +2,7 @@
 
 use super::analysis;
 use super::text::LineIndex;
+use super::tsc::NavigationTree;
 
 use crate::import_map::ImportMap;
 use crate::media_type::MediaType;
@@ -33,6 +34,7 @@ impl IndexValid {
 pub struct DocumentData {
   bytes: Option<Vec<u8>>,
   line_index: Option<LineIndex>,
+  navigation_tree: Option<NavigationTree>,
   dependencies: Option<HashMap<String, analysis::Dependency>>,
   version: Option<i32>,
 }
@@ -72,6 +74,7 @@ impl DocumentData {
     } else {
       Some(LineIndex::new(&content))
     };
+    self.navigation_tree = None;
     Ok(())
   }
 
@@ -187,6 +190,14 @@ impl DocumentCache {
     doc.line_index.clone()
   }
 
+  pub fn navigation_tree(
+    &self,
+    specifier: &ModuleSpecifier,
+  ) -> Option<NavigationTree> {
+    let doc = self.docs.get(specifier)?;
+    doc.navigation_tree.clone()
+  }
+
   pub fn open(
     &mut self,
     specifier: ModuleSpecifier,
@@ -218,6 +229,22 @@ impl DocumentCache {
       .collect()
   }
 
+  pub fn set_navigation_tree(
+    &mut self,
+    specifier: &ModuleSpecifier,
+    navigation_tree: NavigationTree,
+  ) -> Result<(), AnyError> {
+    if let Some(mut doc) = self.docs.get_mut(specifier) {
+      doc.navigation_tree = Some(navigation_tree);
+      Ok(())
+    } else {
+      Err(custom_error(
+        "NotFound",
+        "The document \"{}\" was unexpectedly missing.",
+      ))
+    }
+  }
+
   pub fn version(&self, specifier: &ModuleSpecifier) -> Option<i32> {
     self.docs.get(specifier).and_then(|doc| doc.version)
   }