private static void suggestChainedCalls( CompletionParameters parameters, CompletionResultSet result, PsiElement position) { PsiElement parent = position.getParent(); if (!(parent instanceof PsiJavaCodeReferenceElement)) { return; } PsiElement qualifier = ((PsiJavaCodeReferenceElement) parent).getQualifier(); if (!(qualifier instanceof PsiJavaCodeReferenceElement) || ((PsiJavaCodeReferenceElement) qualifier).isQualified() || ((PsiJavaCodeReferenceElement) qualifier).resolve() != null) { return; } String fullPrefix = position .getContainingFile() .getText() .substring(parent.getTextRange().getStartOffset(), parameters.getOffset()); CompletionResultSet qualifiedCollector = result.withPrefixMatcher(fullPrefix); ElementFilter filter = JavaCompletionContributor.getReferenceFilter(position); for (LookupElement base : suggestQualifierItems(parameters, (PsiJavaCodeReferenceElement) qualifier, filter)) { PsiType type = JavaCompletionUtil.getLookupElementType(base); if (type != null && !PsiType.VOID.equals(type)) { PsiReferenceExpression ref = ReferenceExpressionCompletionContributor.createMockReference(position, type, base); for (final LookupElement item : JavaSmartCompletionContributor.completeReference( position, ref, filter, true, true, parameters, result.getPrefixMatcher())) { qualifiedCollector.addElement(new JavaChainLookupElement(base, item)); } } } }
@Override public void fillCompletionVariants( @NotNull final CompletionParameters parameters, @NotNull final CompletionResultSet result) { final JavaCompletionSession session = new JavaCompletionSession(result); ResultTracker tracker = new ResultTracker(result) { @Override public void consume(CompletionResult plainResult) { super.consume(plainResult); LookupElement element = plainResult.getLookupElement(); Object o = element.getObject(); if (o instanceof PsiClass) { session.registerClass((PsiClass) o); } if (element instanceof TypeArgumentCompletionProvider.TypeArgsLookupElement) { ((TypeArgumentCompletionProvider.TypeArgsLookupElement) element) .registerSingleClass(session); } } }; result.runRemainingContributors(parameters, tracker); final boolean empty = tracker.containsOnlyPackages || suggestAllAnnotations(parameters); if (!empty && parameters.getInvocationCount() == 0) { result.restartCompletionWhenNothingMatches(); } if (empty) { delegate(parameters, JavaCompletionSorting.addJavaSorting(parameters, result), session); } else if (Registry.is("ide.completion.show.better.matching.classes")) { if (parameters.getCompletionType() == CompletionType.BASIC && parameters.getInvocationCount() <= 1 && JavaCompletionContributor.mayStartClassName(result) && JavaCompletionContributor.isClassNamePossible(parameters) && !JavaSmartCompletionContributor.AFTER_NEW.accepts(parameters.getPosition())) { suggestNonImportedClasses( parameters, JavaCompletionSorting.addJavaSorting( parameters, result.withPrefixMatcher(tracker.betterMatcher)), session); } } }
private static void delegate( CompletionParameters parameters, CompletionResultSet result, JavaCompletionSession session) { if (parameters.getCompletionType() == CompletionType.BASIC) { PsiElement position = parameters.getPosition(); suggestCollectionUtilities(parameters, result, position); if (parameters.getInvocationCount() <= 1 && (JavaCompletionContributor.mayStartClassName(result) || suggestAllAnnotations(parameters)) && JavaCompletionContributor.isClassNamePossible(parameters)) { suggestNonImportedClasses(parameters, result, session); return; } suggestChainedCalls(parameters, result, position); } if (parameters.getCompletionType() == CompletionType.SMART && parameters.getInvocationCount() == 2) { result.runRemainingContributors(parameters.withInvocationCount(3), true); } }
@Override protected void delegate( CompletionParameters parameters, CompletionResultSet result, Consumer<CompletionResult> passResult) { if (parameters.getCompletionType() == CompletionType.BASIC) { PsiElement position = parameters.getPosition(); if (parameters.getInvocationCount() <= 1 && JavaCompletionContributor.mayStartClassName(result, false) && JavaCompletionContributor.isClassNamePossible(position)) { suggestNonImportedClasses(parameters, result); return; } suggestChainedCalls(parameters, result, position); } if (parameters.getCompletionType() == CompletionType.SMART && parameters.getInvocationCount() == 2) { result.runRemainingContributors(parameters.withInvocationCount(3), passResult); } }