Excluded lsf files in sources jar from indexing, resolving, errors search

Author: pavel-miniutka

src/com/lsfusion/lang/LSFReferenceAnnotator.java

@@ -88,6 +88,10 @@ public class LSFReferenceAnnotator extends LSFVisitor implements Annotator {
 
     @Override
     public synchronized void annotate(@NotNull PsiElement psiElement, AnnotationHolder holder) {
+        if (psiElement instanceof LSFElement lsfElement && lsfElement.isInLibrarySources()) {
+            return;
+        }
+
         myHolder = holder;
         try {
             psiElement.accept(this);

src/com/lsfusion/lang/psi/LSFElement.java

@@ -1,10 +1,23 @@
 package com.lsfusion.lang.psi;
 
+import com.intellij.openapi.roots.ProjectFileIndex;
+import com.intellij.openapi.vfs.VirtualFile;
 import com.intellij.psi.PsiElement;
 
 public interface LSFElement extends PsiElement {
 
     boolean isCorrect(); // для pin'ов, если pin сработал а не все элементы есть
 
     LSFFile getLSFFile();
+
+    default boolean isInLibrarySources() {
+        LSFFile lsfFile = getLSFFile();
+        if (lsfFile != null) {
+            VirtualFile virtualFile = lsfFile.getVirtualFile();
+            if (virtualFile != null) {
+                return ProjectFileIndex.getInstance(getProject()).isInLibrarySource(virtualFile);
+            }
+        }
+        return false;
+    }
 }

src/com/lsfusion/lang/psi/LSFGlobalResolver.java

@@ -136,7 +136,7 @@ public static <S extends FullNameStubElement<S, T>, T extends LSFFullNameDeclara
     }
 
     public static <G extends LSFStubbedElement> Collection<G> getItemsFromIndex(LSFStringStubIndex<G> index, String name, Project project, GlobalSearchScope scope, LSFLocalSearchScope localScope) {
-        Collection<G> elements = index.get(name, project, scope);
+        Collection<G> elements = index.get(name, project, new LSFSourceFilterScope(scope));
 
         Collection<G> filteredElements = new ArrayList<>();
 

src/com/lsfusion/lang/psi/LSFResolver.java

@@ -23,6 +23,7 @@
 import org.jetbrains.annotations.Nullable;
 
 import java.util.ArrayList;
+import java.util.Collections;
 import java.util.List;
 
 public class LSFResolver implements ResolveCache.AbstractResolver<LSFReference, LSFResolveResult> {
@@ -31,7 +32,10 @@ public class LSFResolver implements ResolveCache.AbstractResolver<LSFReference,
     @Nullable
     @Override
     public LSFResolveResult resolve(@NotNull LSFReference reference, boolean incompleteCode) {
-         return reference.resolveNoCache();
+        if (reference.isInLibrarySources()) {
+            return new LSFResolveResult(Collections.emptyList());
+        }
+        return reference.resolveNoCache();
     }
 
     public static Query<PsiReference> searchWordUsages(Project project, String compoundID) {

src/com/lsfusion/lang/psi/LSFSourceFilterScope.java

@@ -0,0 +1,45 @@
+package com.lsfusion.lang.psi;
+
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.roots.ProjectFileIndex;
+import com.intellij.openapi.vfs.VirtualFile;
+import com.intellij.psi.search.DelegatingGlobalSearchScope;
+import com.intellij.psi.search.GlobalSearchScope;
+import org.jetbrains.annotations.NotNull;
+
+public class LSFSourceFilterScope extends DelegatingGlobalSearchScope {
+    private ProjectFileIndex fileIndex;
+    private boolean includeLibrarySources;
+
+    public LSFSourceFilterScope(@NotNull GlobalSearchScope baseScope) {
+        this(baseScope, false);
+    }
+
+    public LSFSourceFilterScope(@NotNull GlobalSearchScope baseScope, boolean includeLibrarySources) {
+        super(baseScope);
+        Project project = getProject();
+        fileIndex = project == null ? null : ProjectFileIndex.getInstance(project);
+        this.includeLibrarySources = includeLibrarySources;
+    }
+
+    @Override
+    public boolean contains(@NotNull VirtualFile file) {
+        if (!super.contains(file)) {
+            return false;
+        }
+
+        if (fileIndex == null) {
+            return false;
+        }
+
+        if (includeLibrarySources || myBaseScope.isForceSearchingInLibrarySources()) {
+            return true;
+        }
+        return !fileIndex.isInLibrarySource(file);
+    }
+
+    @NotNull
+    public static LSFSourceFilterScope allScope(@NotNull Project project) {
+        return new LSFSourceFilterScope(GlobalSearchScope.allScope(project));
+    }
+}

src/com/lsfusion/lang/psi/indexes/LSFStringStubIndex.java

@@ -5,14 +5,15 @@
 import com.intellij.psi.search.GlobalSearchScope;
 import com.intellij.psi.stubs.StringStubIndexExtension;
 import com.intellij.psi.stubs.StubIndex;
+import com.lsfusion.lang.psi.LSFSourceFilterScope;
 import com.lsfusion.lang.psi.LSFStubbedElement;
 
 import java.util.Collection;
 
 public abstract class LSFStringStubIndex<Psi extends LSFStubbedElement> extends StringStubIndexExtension<Psi> {
     @Override
     public Collection<Psi> get(String s, Project project, GlobalSearchScope scope) {
-        return DumbService.getInstance(project).runReadActionInSmartMode(() -> StubIndex.getElements(getKey(), s, project, scope, getPsiClass()));
+        return DumbService.getInstance(project).runReadActionInSmartMode(() -> StubIndex.getElements(getKey(), s, project, new LSFSourceFilterScope(scope), getPsiClass()));
     }
 
     protected abstract Class<Psi> getPsiClass();

src/com/lsfusion/util/LSFFileUtils.java

@@ -19,6 +19,7 @@
 import com.intellij.psi.search.FilenameIndex;
 import com.intellij.psi.search.GlobalSearchScope;
 import com.lsfusion.lang.psi.LSFGlobalResolver;
+import com.lsfusion.lang.psi.LSFSourceFilterScope;
 import com.lsfusion.lang.psi.Result;
 import com.lsfusion.lang.psi.declarations.LSFModuleDeclaration;
 import org.jetbrains.annotations.NotNull;
@@ -277,7 +278,8 @@ public static GlobalSearchScope getScope(List<String> modulesToInclude, Project
                 }
             }
         } else
-            modulesScope = allScope(project);
+            // all places without source lsf files by default. is used in errors search and enabling/disabling meta for now
+            modulesScope = LSFSourceFilterScope.allScope(project);
         return modulesScope;
     }
 }