KT-2178 Remove Java declaration from completion for JS projects

#KT-2178 In Progress

idea/src/org/jetbrains/jet/plugin/caches/JetCacheManager.java

@@ -18,7 +18,9 @@ package org.jetbrains.jet.plugin.caches;
 
 import com.intellij.openapi.components.ProjectComponent;
 import com.intellij.openapi.project.Project;
+import com.intellij.psi.search.PsiShortNamesCache;
 import org.jetbrains.annotations.NotNull;
+import org.jetbrains.jet.plugin.project.JsModuleDetector;
 
 /**
  * @author Nikolay Krasko
@@ -64,4 +66,12 @@ public class JetCacheManager implements ProjectComponent {
     public JetShortNamesCache getNamesCache() {
         return myCache;
     }
+
+    public PsiShortNamesCache getShortNamesCache() {
+        if (JsModuleDetector.isJsProject(myProject)) {
+            return myCache;
+        }
+
+        return PsiShortNamesCache.getInstance(myProject);
+    }
 }

idea/src/org/jetbrains/jet/plugin/completion/JetClassCompletionContributor.java

@@ -22,6 +22,8 @@ import com.intellij.patterns.PlatformPatterns;
 import com.intellij.psi.PsiClass;
 import com.intellij.psi.PsiElement;
 import com.intellij.psi.PsiReference;
+import com.intellij.psi.search.GlobalSearchScope;
+import com.intellij.psi.search.PsiShortNamesCache;
 import com.intellij.util.Consumer;
 import com.intellij.util.ProcessingContext;
 import org.jetbrains.annotations.NotNull;
@@ -30,9 +32,11 @@ import org.jetbrains.jet.lang.descriptors.ClassDescriptor;
 import org.jetbrains.jet.lang.descriptors.DeclarationDescriptor;
 import org.jetbrains.jet.lang.psi.JetFile;
 import org.jetbrains.jet.lang.resolve.BindingContext;
+import org.jetbrains.jet.plugin.caches.JetCacheManager;
 import org.jetbrains.jet.plugin.caches.JetShortNamesCache;
 import org.jetbrains.jet.plugin.completion.handlers.JetJavaClassInsertHandler;
 import org.jetbrains.jet.plugin.completion.weigher.JetCompletionSorting;
+import org.jetbrains.jet.plugin.project.JsModuleDetector;
 import org.jetbrains.jet.plugin.project.WholeProjectAnalyzerFacade;
 import org.jetbrains.jet.plugin.references.JetSimpleNameReference;
 
@@ -90,33 +94,56 @@ public class JetClassCompletionContributor extends CompletionContributor {
             consumer.consume(DescriptorLookupConverter.createLookupElement(bindingContext, jetOnlyClass));
         }
 
-        JavaClassNameCompletionContributor.addAllClasses(
+        if (!JsModuleDetector.isJsProject(parameters.getOriginalFile().getProject())) {
-                parameters,
+            JavaClassNameCompletionContributor.addAllClasses(
-                false,
+                    parameters,
-                JavaCompletionSorting.addJavaSorting(parameters, tempResult).getPrefixMatcher(),
+                    false,
-                new Consumer<LookupElement>() {
+                    JavaCompletionSorting.addJavaSorting(parameters, tempResult).getPrefixMatcher(),
-                    @Override
+                    new Consumer<LookupElement>() {
-                    public void consume(LookupElement lookupElement) {
+                        @Override
-                        if (lookupElement instanceof JavaPsiClassReferenceElement) {
+                        public void consume(LookupElement lookupElement) {
-                            JavaPsiClassReferenceElement javaPsiReferenceElement = (JavaPsiClassReferenceElement) lookupElement;
+                            if (lookupElement instanceof JavaPsiClassReferenceElement) {
+                                JavaPsiClassReferenceElement javaPsiReferenceElement = (JavaPsiClassReferenceElement) lookupElement;
 
-                            PsiClass object = javaPsiReferenceElement.getObject();
+                                PsiClass object = javaPsiReferenceElement.getObject();
-                            if (object instanceof JetLightClass) {
+                                if (addAsJetLookupElement(object, bindingContext, consumer)) {
-                                ClassDescriptor descriptor =
-                                        bindingContext.get(BindingContext.FQNAME_TO_CLASS_DESCRIPTOR, ((JetLightClass)object).getFqName());
-
-                                if (descriptor != null) {
-                                    LookupElement element = DescriptorLookupConverter.createLookupElement(bindingContext, descriptor);
-                                    consumer.consume(element);
                                     return;
                                 }
-                            }
 
-                            // Redefine standard java insert handler which is going to insert fqn
+                                // Redefine standard java insert handler which is going to insert fqn
-                            javaPsiReferenceElement.setInsertHandler(JetJavaClassInsertHandler.JAVA_CLASS_INSERT_HANDLER);
+                                javaPsiReferenceElement.setInsertHandler(JetJavaClassInsertHandler.JAVA_CLASS_INSERT_HANDLER);
-                            consumer.consume(lookupElement);
+                                consumer.consume(lookupElement);
+                            }
+                        }
+                    });
+        }
+        else {
+            GlobalSearchScope globalSearchScope = GlobalSearchScope.allScope(parameters.getOriginalFile().getProject());
+            PsiShortNamesCache cache = JetCacheManager.getInstance(parameters.getOriginalFile().getProject()).getShortNamesCache();
+            for (String className : cache.getAllClassNames()) {
+                if (result.getPrefixMatcher().prefixMatches(className)) {
+                    for (PsiClass aClass : cache.getClassesByName(className, globalSearchScope)) {
+                        if (!addAsJetLookupElement(aClass, bindingContext, consumer)) {
+                            assert false;
                         }
                     }
-                });
+                }
+            }
+        }
+    }
+
+    private static boolean addAsJetLookupElement(PsiClass aClass, BindingContext bindingContext, Consumer<LookupElement> consumer) {
+        if (aClass instanceof JetLightClass) {
+            ClassDescriptor descriptor = bindingContext.get(
+                    BindingContext.FQNAME_TO_CLASS_DESCRIPTOR, ((JetLightClass)aClass).getFqName());
+
+            if (descriptor != null) {
+                LookupElement element = DescriptorLookupConverter.createLookupElement(bindingContext, descriptor);
+                consumer.consume(element);
+                return true;
+            }
+        }
+
+        return false;
     }
 }

idea/src/org/jetbrains/jet/plugin/quickfix/ImportClassAndFunFix.java

@@ -164,7 +164,7 @@ public class ImportClassAndFunFix extends JetHintAction<JetSimpleNameExpression>
     private static Collection<FqName> getJavaClasses(@NotNull final String typeName,
             @NotNull Project project,
             final GlobalSearchScope scope) {
-        PsiShortNamesCache cache = PsiShortNamesCache.getInstance(project);
+        PsiShortNamesCache cache = JetCacheManager.getInstance(project).getShortNamesCache();
 
         PsiClass[] classes = cache.getClassesByName(typeName, scope);