Add scip/lsif flag to exclude vendored libaries

Author: darichey

crates/ide/src/static_index.rs

@@ -230,7 +230,11 @@ impl StaticIndex<'_> {
         self.files.push(result);
     }
 
-    pub fn compute<'a>(analysis: &'a Analysis, workspace_root: &VfsPath) -> StaticIndex<'a> {
+    pub fn compute<'a>(
+        analysis: &'a Analysis,
+        workspace_root: &VfsPath,
+        exclude_vendored_libraries: bool,
+    ) -> StaticIndex<'a> {
         let db = &*analysis.db;
         let work = all_modules(db).into_iter().filter(|module| {
             let file_id = module.definition_source_file_id(db).original_file(db);
@@ -239,7 +243,7 @@ impl StaticIndex<'_> {
             let is_vendored = source_root
                 .path_for_file(&file_id.into())
                 .is_some_and(|module_path| module_path.starts_with(workspace_root));
-            !source_root.is_library || is_vendored
+            !source_root.is_library || (!exclude_vendored_libraries && is_vendored)
         });
         let mut this = StaticIndex {
             files: vec![],
@@ -268,10 +272,13 @@ mod tests {
     use ide_db::{base_db::VfsPath, FileRange, FxHashSet};
     use syntax::TextSize;
 
-    fn check_all_ranges(ra_fixture: &str) {
+    fn check_all_ranges(ra_fixture: &str, exclude_vendored_libraries: bool) {
         let (analysis, ranges) = fixture::annotations_without_marker(ra_fixture);
-        let s =
-            StaticIndex::compute(&analysis, &VfsPath::new_virtual_path("/workspace".to_owned()));
+        let s = StaticIndex::compute(
+            &analysis,
+            &VfsPath::new_virtual_path("/workspace".to_owned()),
+            exclude_vendored_libraries,
+        );
         let mut range_set: FxHashSet<_> = ranges.iter().map(|it| it.0).collect();
         for f in s.files {
             for (range, _) in f.tokens {
@@ -288,10 +295,13 @@ mod tests {
     }
 
     #[track_caller]
-    fn check_definitions(ra_fixture: &str) {
+    fn check_definitions(ra_fixture: &str, exclude_vendored_libraries: bool) {
         let (analysis, ranges) = fixture::annotations_without_marker(ra_fixture);
-        let s =
-            StaticIndex::compute(&analysis, &VfsPath::new_virtual_path("/workspace".to_owned()));
+        let s = StaticIndex::compute(
+            &analysis,
+            &VfsPath::new_virtual_path("/workspace".to_owned()),
+            exclude_vendored_libraries,
+        );
         let mut range_set: FxHashSet<_> = ranges.iter().map(|it| it.0).collect();
         for (_, t) in s.tokens.iter() {
             if let Some(t) = t.definition {
@@ -319,6 +329,7 @@ struct Foo;
 enum E { X(Foo) }
    //^   ^ ^^^
 "#,
+            false,
         );
         check_definitions(
             r#"
@@ -327,6 +338,7 @@ struct Foo;
 enum E { X(Foo) }
    //^   ^
 "#,
+            false,
         );
     }
 
@@ -349,6 +361,7 @@ pub func() {
 
 }
 "#,
+            false,
         );
     }
 
@@ -367,9 +380,28 @@ struct ExternalLibrary(i32);
 struct VendoredLibrary(i32);
      //^^^^^^^^^^^^^^^ ^^^
 "#,
+            false,
         );
     }
 
+    #[test]
+    fn vendored_crate_excluded() {
+        check_all_ranges(
+            r#"
+//- /workspace/main.rs crate:main deps:external,vendored
+struct Main(i32);
+     //^^^^ ^^^
+
+//- /external/lib.rs new_source_root:library crate:[email protected],https://a.b/foo.git library
+struct ExternalLibrary(i32);
+
+//- /workspace/vendored/lib.rs new_source_root:library crate:[email protected],https://a.b/bar.git library
+struct VendoredLibrary(i32);
+"#,
+            true,
+        )
+    }
+
     #[test]
     fn derives() {
         check_all_ranges(
@@ -384,6 +416,7 @@ pub macro Copy {}
 struct Hello(i32);
      //^^^^^ ^^^
 "#,
+            false,
         );
     }
 }

crates/rust-analyzer/src/cli/flags.rs

@@ -138,6 +138,9 @@ xflags::xflags! {
 
         cmd lsif {
             required path: PathBuf
+
+            /// Exclude code from vendored libraries from the resulting index.
+            optional --exclude-vendored-libraries
         }
 
         cmd scip {
@@ -148,6 +151,9 @@ xflags::xflags! {
 
             /// A path to an json configuration file that can be used to customize cargo behavior.
             optional --config-path config_path: PathBuf
+
+            /// Exclude code from vendored libraries from the resulting index.
+            optional --exclude-vendored-libraries
         }
     }
 }
@@ -259,6 +265,8 @@ pub struct Search {
 #[derive(Debug)]
 pub struct Lsif {
     pub path: PathBuf,
+
+    pub exclude_vendored_libraries: bool,
 }
 
 #[derive(Debug)]
@@ -267,6 +275,7 @@ pub struct Scip {
 
     pub output: Option<PathBuf>,
     pub config_path: Option<PathBuf>,
+    pub exclude_vendored_libraries: bool,
 }
 
 impl RustAnalyzer {

crates/rust-analyzer/src/cli/lsif.rs

@@ -296,7 +296,8 @@ impl flags::Lsif {
         let db = host.raw_database();
         let analysis = host.analysis();
 
-        let si = StaticIndex::compute(&analysis, &path.clone().into());
+        let si =
+            StaticIndex::compute(&analysis, &path.clone().into(), self.exclude_vendored_libraries);
 
         let mut lsif = LsifManager::new(&analysis, db, &vfs);
         lsif.add_vertex(lsif::Vertex::MetaData(lsif::MetaData {

crates/rust-analyzer/src/cli/scip.rs

@@ -63,7 +63,8 @@ impl flags::Scip {
         let db = host.raw_database();
         let analysis = host.analysis();
 
-        let si = StaticIndex::compute(&analysis, &root.clone().into());
+        let si =
+            StaticIndex::compute(&analysis, &root.clone().into(), self.exclude_vendored_libraries);
 
         let metadata = scip_types::Metadata {
             version: scip_types::ProtocolVersion::UnspecifiedProtocolVersion.into(),
@@ -352,8 +353,11 @@ mod test {
         let (host, position) = position(ra_fixture);
 
         let analysis = host.analysis();
-        let si =
-            StaticIndex::compute(&analysis, &VfsPath::new_virtual_path("/workspace".to_owned()));
+        let si = StaticIndex::compute(
+            &analysis,
+            &VfsPath::new_virtual_path("/workspace".to_owned()),
+            false,
+        );
 
         let FilePosition { file_id, offset } = position;
 
@@ -617,8 +621,11 @@ pub mod example_mod {
         host.raw_database_mut().apply_change(change_fixture.change);
 
         let analysis = host.analysis();
-        let si =
-            StaticIndex::compute(&analysis, &VfsPath::new_virtual_path("/workspace".to_owned()));
+        let si = StaticIndex::compute(
+            &analysis,
+            &VfsPath::new_virtual_path("/workspace".to_owned()),
+            false,
+        );
 
         let file = si.files.first().unwrap();
         let (_, token_id) = file.tokens.first().unwrap();