private static void addPathCompletions(
@NotNull CompletionResultSet result,
@NotNull BipartiteString caretBipartiteElementText,
@NotNull VirtualFile basePath,
boolean directoryExpected) {
ParentDirWithLastComponentPrefix parentWithLastComponentPrefix =
findParentDirWithLastComponentPrefix(basePath, caretBipartiteElementText.getPrefix());
if (parentWithLastComponentPrefix != null) {
PrefixMatcher matcher =
new StrictPrefixMatcher(parentWithLastComponentPrefix.getLastComponentPrefix(), false);
result = result.withPrefixMatcher(matcher);
VirtualFile parentFile = parentWithLastComponentPrefix.getParent();
VirtualFile[] children = parentFile.getChildren();
Character dirSeparatorSuffix =
extractDirectoryTrailingFileSeparator(caretBipartiteElementText);
if (parentFile.isDirectory()) {
result.addElement(LookupItem.fromString(".."));
}
for (VirtualFile child : children) {
if (child.isDirectory() || !directoryExpected) {
String name = child.getName();
if (child.isDirectory() && dirSeparatorSuffix != null) {
name += dirSeparatorSuffix;
}
result.addElement(LookupItem.fromString(name));
}
}
}
}
public void addCompletions(
@NotNull CompletionParameters parameters,
ProcessingContext context,
@NotNull CompletionResultSet resultSet) {
PsiElement element = parameters.getPosition().getParent();
Project project = element.getProject();
if (!SilexProjectComponent.isEnabled(project)) {
return;
}
if (!(element instanceof StringLiteralExpression)) {
return;
}
Container container =
Utils.findContainerForFirstParameterOfPimpleMethod((StringLiteralExpression) element);
if (container == null) {
return;
}
for (Service service : container.getServices().values()) {
resultSet.addElement(new ServiceLookupElement(service, project));
}
for (Parameter parameter : container.getParameters().values()) {
resultSet.addElement(new ParameterLookupElement(parameter));
}
resultSet.stopHere();
}
@Override
protected void addCompletions(
@NotNull CompletionParameters parameters,
ProcessingContext processingContext,
@NotNull CompletionResultSet completionResultSet) {
PsiElement position = parameters.getPosition();
if (!Symfony2ProjectComponent.isEnabled(position)) {
return;
}
PsiElement prevSiblingOfType =
PsiElementUtils.getPrevSiblingOfType(
position, YamlElementPatternHelper.getPreviousCommaSibling());
if (prevSiblingOfType == null) {
return;
}
String service = PsiElementUtils.trimQuote(prevSiblingOfType.getText());
if (StringUtils.isBlank(service)) {
return;
}
PhpClass phpClass = ServiceUtil.getServiceClass(prevSiblingOfType.getProject(), service);
if (phpClass == null) {
return;
}
for (Method method : phpClass.getMethods()) {
if (method.getAccess().isPublic() && !(method.getName().startsWith("__"))) {
completionResultSet.addElement(new PhpLookupElement(method));
}
}
}
@Override
protected void addCompletions(
@NotNull CompletionParameters completionParameters,
ProcessingContext processingContext,
@NotNull CompletionResultSet completionResultSet) {
PsiElement position = completionParameters.getPosition();
if (!Symfony2ProjectComponent.isEnabled(position)) {
return;
}
YAMLCompoundValue yamlCompoundValue =
PsiTreeUtil.getParentOfType(position, YAMLCompoundValue.class);
if (yamlCompoundValue == null) {
return;
}
String className =
YamlHelper.getYamlKeyValueAsString(yamlCompoundValue, "targetEntity", false);
if (className == null) {
return;
}
PhpClass phpClass = PhpElementsUtil.getClass(position.getProject(), className);
if (phpClass == null) {
return;
}
for (DoctrineModelField field : EntityHelper.getModelFields(phpClass)) {
if (field.getRelation() != null) {
completionResultSet.addElement(new DoctrineModelFieldLookupElement(field));
}
}
}
@Override
protected void addCompletions(
@NotNull CompletionParameters completionParameters,
ProcessingContext processingContext,
@NotNull CompletionResultSet completionResultSet) {
YAMLKeyValue yamlKeyValue =
PsiTreeUtil.getParentOfType(
completionParameters.getOriginalPosition(), YAMLKeyValue.class);
if (yamlKeyValue != null) {
PsiElement compoundValue = yamlKeyValue.getParent();
if (compoundValue instanceof YAMLCompoundValue) {
// path and pattern are valid
String pattern =
YamlHelper.getYamlKeyValueAsString((YAMLCompoundValue) compoundValue, "path", false);
if (pattern == null) {
pattern =
YamlHelper.getYamlKeyValueAsString(
(YAMLCompoundValue) compoundValue, "pattern", false);
}
if (pattern != null) {
Matcher matcher = Pattern.compile("\\{(\\w+)}").matcher(pattern);
while (matcher.find()) {
completionResultSet.addElement(LookupElementBuilder.create(matcher.group(1)));
}
}
}
}
}
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
protected void addCompletions(
@NotNull CompletionParameters parameters,
ProcessingContext context,
@NotNull CompletionResultSet result) {
for (String[] completion :
RegExpLanguageHosts.getInstance().getPosixCharacterClasses(parameters.getPosition())) {
result.addElement(
LookupElementBuilder.create(completion[0])
.withTypeText((completion.length > 1) ? completion[1] : null)
.withIcon(emptyIcon)
.withInsertHandler(
new InsertHandler<LookupElement>() {
@Override
public void handleInsert(InsertionContext context, LookupElement item) {
context.setAddCompletionChar(false);
final Editor editor = context.getEditor();
final Document document = editor.getDocument();
final int tailOffset = context.getTailOffset();
if (document.getTextLength() < tailOffset + 2
|| !document
.getText(new TextRange(tailOffset, tailOffset + 2))
.equals(":]")) {
document.insertString(tailOffset, ":]");
}
editor.getCaretModel().moveCaretRelatively(2, 0, false, false, true);
}
}));
}
}
private void addLanguageKeyword(CompletionResultSet resultSet, String text) {
// TODO:: order most common used first
String keywords =
"|abs|accept|alarm|atan2|AUTOLOAD|BEGIN|bind|binmode|bless|break|caller|chdir|CHECK|chmod|chomp|chop|chown|chr|chroot|close|closedir|connect|cos|crypt|dbmclose|dbmopen|defined|delete|DESTROY|die|dump|each|END|endgrent|endhostent|endnetent|endprotoent|endpwent|endservent|eof|eval|exec|exists|exit|fcntl|fileno|flock|fork|format|formline|getc|getgrent|getgrgid|getgrnam|gethostbyaddr|gethostbyname|gethostent|getlogin|getnetbyaddr|getnetbyname|getnetent|getpeername|getpgrp|getppid|getpriority|getprotobyname|getprotobynumber|getprotoent|getpwent|getpwnam|getpwuid|getservbyname|getservbyport|getservent|getsockname|getsockopt|glob|gmtime|goto|grep|hex|index|INIT|int|ioctl|join|keys|kill|last|lc|each|lcfirst|setnetent|length|link|listen|local|localtime|log|lstat|map|mkdir|msgctl|msgget|msgrcv|msgsnd|next|not|oct|open|opendir|ord|our|pack|pipe|pop|pos|print|printf|prototype|push|quotemeta|rand|read|readdir|readline|readlink|readpipe|recv|redo|ref|rename|require|reset|return|reverse|rewinddir|rindex|rmdir|say|scalar|seek|seekdir|select|semctl|semget|semop|send|setgrent|sethostent|each|lcfirst|setnetent|setpgrp|setpriority|setprotoent|setpwent|setservent|setsockopt|shift|shmctl|shmget|shmread|shmwrite|shutdown|sin|sleep|socket|socketpair|sort|splice|split|sprintf|sqrt|srand|stat|state|study|substr|symlink|syscall|sysopen|sysread|sysseek|system|syswrite|tell|telldir|tie|tied|time|times|truncate|ucfirst|umask|undef|UNITCHECK|unlink|unpack|unshift|untie|use|utime|values|vec|wait|waitpid|wantarray|warn|write|each|lcfirst|setnetent|cmp|continue|CORE|else|elsif|exp|for|foreach|lock|package|unless|until|while|ARGV|ARGVOUT|STDERR|STDIN|STDOUT";
Matcher matcher = Utils.applyRegex("\\|(" + text + "[^\\|]+)", keywords);
while (matcher.find()) {
resultSet.addElement(LookupElementBuilder.create(matcher.group(1)));
}
}
private static void ensureTemplatesShown(
Ref<Boolean> templatesShown, List<TemplateImpl> templates, CompletionResultSet result) {
if (!templatesShown.get()) {
templatesShown.set(true);
for (final TemplateImpl possible : templates) {
result.addElement(new LiveTemplateLookupElement(possible, false));
}
}
}
private static void addCachedVariables(CompletionResultSet resultSet, String str) {
if (str != null && !variablesCache.containsKey(str)) {
if (Utils.verbose) {
Utils.print("cache variable: " + str);
}
variablesCache.put(str, getVariableLookupElementBuilder(str));
}
if (resultSet != null) {
resultSet.addElement(variablesCache.get(str));
}
}
private static void addCachedPackage(CompletionResultSet resultSet, Package packageObj) {
if (packageObj != null && !packagesCache.containsKey(packageObj)) {
if (Utils.verbose) {
Utils.print("cache package: " + packageObj.getQualifiedName());
}
packagesCache.put(packageObj, getPackageLookupElementBuilder(packageObj));
}
if (resultSet != null) {
resultSet.addElement(packagesCache.get(packageObj));
}
}
private static void addResult(
CompletionResultSet result, Collection<Lookup> collection, PsiElement position, int offset) {
result = result.withPrefixMatcher(findPrefixStatic(position, offset));
for (Lookup lookup : collection) {
final LookupItem<Lookup> item = new LookupItem<>(lookup, lookup.toString());
item.setInsertHandler(INSERT_HANDLER);
if (lookup.isKeyword()) {
item.setBold();
}
result.addElement(item);
}
}
@Override
protected void addCompletions(
@NotNull CompletionParameters parameters,
ProcessingContext context,
@NotNull CompletionResultSet result) {
UnicodeCharacterNames.iterate(
name -> {
if (result.getPrefixMatcher().prefixMatches(name)) {
final String type =
new String(new int[] {UnicodeCharacterNames.getCodePoint(name)}, 0, 1);
if (myEmbrace) {
result.addElement(createLookupElement("{" + name + "}", type, emptyIcon));
} else {
result.addElement(
TailTypeDecorator.withTail(
createLookupElement(name, type, emptyIcon),
TailType.createSimpleTailType('}')));
}
}
ProgressManager.checkCanceled();
});
}
@Override
public void addCompletions(
@NotNull final CompletionParameters parameters,
final ProcessingContext context,
@NotNull final CompletionResultSet result) {
for (String[] stringArray :
RegExpLanguageHosts.getInstance().getAllKnownProperties(parameters.getPosition())) {
result.addElement(
TailTypeDecorator.withTail(
createLookupElement(stringArray[0], null, emptyIcon),
TailType.createSimpleTailType('}')));
}
}
private static void fillExpectedTypeArgs(
CompletionResultSet resultSet,
PsiElement context,
final PsiClass actualClass,
final int index,
PsiClassType.ClassResolveResult expectedType,
TailType globalTail) {
final PsiClass expectedClass = expectedType.getElement();
if (!InheritanceUtil.isInheritorOrSelf(actualClass, expectedClass, true)) return;
assert expectedClass != null;
final PsiSubstitutor currentSubstitutor =
TypeConversionUtil.getClassSubstitutor(expectedClass, actualClass, PsiSubstitutor.EMPTY);
assert currentSubstitutor != null;
PsiTypeParameter[] params = actualClass.getTypeParameters();
final List<PsiTypeLookupItem> typeItems = new ArrayList<PsiTypeLookupItem>();
for (int i = index; i < params.length; i++) {
PsiType arg = getExpectedTypeArg(context, i, expectedType, currentSubstitutor, params);
if (arg == null) {
arg = getExpectedTypeArg(context, index, expectedType, currentSubstitutor, params);
if (arg != null) {
resultSet.addElement(
TailTypeDecorator.withTail(
PsiTypeLookupItem.createLookupItem(arg, context),
getTail(index == params.length - 1)));
}
return;
}
typeItems.add(PsiTypeLookupItem.createLookupItem(arg, context));
}
resultSet.addElement(
new TypeArgsLookupElement(
typeItems,
globalTail,
ConstructorInsertHandler.hasConstructorParameters(actualClass, context)));
}
@Override
public boolean process(
@NotNull String name,
@NotNull GoNamedElement element,
@NotNull ExistingImportData importData,
@NotNull CompletionResultSet result) {
GoTypeSpec spec = ((GoTypeSpec) element);
boolean forTypes = myParent instanceof GoTypeReferenceExpression;
double priority;
if (importData.exists) {
priority = forTypes ? GoCompletionUtil.TYPE_PRIORITY : GoCompletionUtil.TYPE_CONVERSION;
} else {
priority =
forTypes
? GoCompletionUtil.NOT_IMPORTED_TYPE_PRIORITY
: GoCompletionUtil.NOT_IMPORTED_TYPE_CONVERSION;
}
String lookupString = replacePackageWithAlias(name, importData.alias);
if (forTypes) {
result.addElement(
GoCompletionUtil.createTypeLookupElement(
spec,
lookupString,
GoAutoImportInsertHandler.SIMPLE_INSERT_HANDLER,
importData.importPath,
priority));
} else {
result.addElement(
GoCompletionUtil.createTypeConversionLookupElement(
spec,
lookupString,
GoAutoImportInsertHandler.TYPE_CONVERSION_INSERT_HANDLER,
importData.importPath,
priority));
}
return true;
}
private static void addCachedSubNoArgs(CompletionResultSet resultSet, Sub sub) {
if (!subsCacheNoArgs.containsKey(sub)) {
if (Utils.verbose) {
Utils.print(
"cache sub(no args): "
+ sub.getName()
+ " , containingPackage:"
+ sub.getPackageObj().getQualifiedName());
}
subsCacheNoArgs.put(sub, getSubLookupElementBuilder(sub, false));
}
if (resultSet != null) {
resultSet.addElement(subsCacheNoArgs.get(sub));
}
}
@Override
public boolean process(
@NotNull String name,
@NotNull GoNamedElement element,
@NotNull ExistingImportData importData,
@NotNull CompletionResultSet result) {
double priority =
importData.exists
? GoCompletionUtil.VAR_PRIORITY
: GoCompletionUtil.NOT_IMPORTED_VAR_PRIORITY;
result.addElement(
GoCompletionUtil.createVariableLikeLookupElement(
element,
replacePackageWithAlias(name, importData.alias),
GoAutoImportInsertHandler.SIMPLE_INSERT_HANDLER,
priority));
return true;
}
private static void addKeywords(CompletionParameters parameters, CompletionResultSet result) {
PsiElement position = parameters.getPosition();
final Set<LookupElement> lookupSet = new LinkedHashSet<LookupElement>();
final Set<CompletionVariant> keywordVariants = new HashSet<CompletionVariant>();
final JavaCompletionData completionData = getCompletionData(PsiUtil.getLanguageLevel(position));
completionData.addKeywordVariants(keywordVariants, position, parameters.getOriginalFile());
completionData.completeKeywordsBySet(
lookupSet,
keywordVariants,
position,
result.getPrefixMatcher(),
parameters.getOriginalFile());
completionData.fillCompletions(parameters, result);
for (final LookupElement item : lookupSet) {
result.addElement(item);
}
}
private static void addSuperSignatureElements(
final PsiClass parent,
boolean implemented,
CompletionResultSet result,
Set<MethodSignature> addedSignatures) {
for (CandidateInfo candidate :
OverrideImplementExploreUtil.getMethodsToOverrideImplement(parent, implemented)) {
PsiMethod baseMethod = (PsiMethod) candidate.getElement();
PsiClass baseClass = baseMethod.getContainingClass();
PsiSubstitutor substitutor = candidate.getSubstitutor();
if (!baseMethod.isConstructor()
&& baseClass != null
&& addedSignatures.add(baseMethod.getSignature(substitutor))) {
result.addElement(
createOverridingLookupElement(implemented, baseMethod, baseClass, substitutor));
}
}
}
@Override
public boolean process(
@NotNull String name,
@NotNull GoNamedElement element,
@NotNull ExistingImportData importData,
@NotNull CompletionResultSet result) {
GoFunctionDeclaration function = ((GoFunctionDeclaration) element);
double priority =
importData.exists
? GoCompletionUtil.FUNCTION_PRIORITY
: GoCompletionUtil.NOT_IMPORTED_FUNCTION_PRIORITY;
result.addElement(
GoCompletionUtil.createFunctionOrMethodLookupElement(
function,
replacePackageWithAlias(name, importData.alias),
GoAutoImportInsertHandler.FUNCTION_INSERT_HANDLER,
priority));
return true;
}
private static void addTopLevelKeysCompletionIfNeeded(
@NotNull CompletionParameters parameters,
@NotNull CompletionResultSet result,
@NotNull BipartiteString caretBipartiteElementText) {
PsiElement element = parameters.getPosition();
YAMLDocument yamlDocument = ObjectUtils.tryCast(element.getParent(), YAMLDocument.class);
if (yamlDocument == null) {
yamlDocument =
JstdConfigFileUtils.getVerifiedHierarchyHead(
element.getParent(), new Class[] {YAMLKeyValue.class}, YAMLDocument.class);
}
if (yamlDocument != null) {
String prefix = caretBipartiteElementText.getPrefix();
result = result.withPrefixMatcher(prefix);
for (String key : JstdConfigFileUtils.VALID_TOP_LEVEL_KEYS) {
if (key.startsWith(prefix)) {
result.addElement(LookupItem.fromString(key + ":"));
}
}
}
}
@Override
protected void addCompletions(
@NotNull CompletionParameters parameters,
ProcessingContext processingContext,
@NotNull CompletionResultSet completionResultSet) {
if (!Symfony2ProjectComponent.isEnabled(parameters.getPosition())) {
return;
}
PhpIndex phpIndex = PhpIndex.getInstance(parameters.getOriginalFile().getProject());
for (PhpClass phpClass : phpIndex.getAllSubclasses("\\Doctrine\\ORM\\EntityRepository")) {
String presentableFQN = phpClass.getPresentableFQN();
if (presentableFQN != null) {
completionResultSet.addElement(
LookupElementBuilder.create(phpClass.getName())
.withTypeText(phpClass.getPresentableFQN(), true)
.withIcon(phpClass.getIcon()));
}
}
}
private static void suggestNonImportedClasses(
CompletionParameters parameters,
CompletionResultSet result,
@Nullable JavaCompletionSession session) {
JavaClassNameCompletionContributor.addAllClasses(
parameters,
true,
result.getPrefixMatcher(),
element -> {
if (session != null && session.alreadyProcessed(element)) {
return;
}
JavaPsiClassReferenceElement classElement =
element.as(JavaPsiClassReferenceElement.CLASS_CONDITION_KEY);
if (classElement != null) {
classElement.setAutoCompletionPolicy(AutoCompletionPolicy.NEVER_AUTOCOMPLETE);
element =
JavaClassNameCompletionContributor.highlightIfNeeded(classElement, parameters);
}
result.addElement(element);
});
}
private static void addGetterSetterElements(
CompletionResultSet result, PsiClass parent, Set<MethodSignature> addedSignatures) {
int count = 0;
for (PsiField field : parent.getFields()) {
if (field instanceof PsiEnumConstant) continue;
List<PsiMethod> prototypes = ContainerUtil.newSmartList();
Collections.addAll(
prototypes, GetterSetterPrototypeProvider.generateGetterSetters(field, true));
Collections.addAll(
prototypes, GetterSetterPrototypeProvider.generateGetterSetters(field, false));
for (final PsiMethod prototype : prototypes) {
if (parent.findMethodBySignature(prototype, false) == null
&& addedSignatures.add(prototype.getSignature(PsiSubstitutor.EMPTY))) {
Icon icon = prototype.getIcon(Iconable.ICON_FLAG_VISIBILITY);
result.addElement(
createGenerateMethodElement(
prototype,
PsiSubstitutor.EMPTY,
icon,
"",
new InsertHandler<LookupElement>() {
@Override
public void handleInsert(InsertionContext context, LookupElement item) {
removeLookupString(context);
insertGenerationInfos(
context,
Collections.singletonList(new PsiGenerationInfo<PsiMethod>(prototype)));
}
}));
if (count++ > 100) return;
}
}
}
}
private static void addCompletionElements(
CompletionResultSet result, final Consumer<Task> consumer, List<Task> items, int index) {
final AutoCompletionPolicy completionPolicy =
ApplicationManager.getApplication().isUnitTestMode()
? AutoCompletionPolicy.ALWAYS_AUTOCOMPLETE
: AutoCompletionPolicy.NEVER_AUTOCOMPLETE;
for (final Task task : items) {
LookupElementBuilder builder =
LookupElementBuilder.create(task, task.getId())
.setIcon(task.getIcon())
.addLookupString(task.getSummary())
.setTailText(" " + task.getSummary(), true)
.setInsertHandler(
new InsertHandler<LookupElement>() {
@Override
public void handleInsert(InsertionContext context, LookupElement item) {
Document document = context.getEditor().getDocument();
String s = task.getId() + ": " + task.getSummary();
document.replaceString(context.getStartOffset(), context.getTailOffset(), s);
context
.getEditor()
.getCaretModel()
.moveToOffset(context.getStartOffset() + s.length());
consumer.consume(task);
}
});
if (task.isClosed()) {
builder = builder.setStrikeout();
}
result.addElement(
PrioritizedLookupElement.withGrouping(
builder.withAutoCompletionPolicy(completionPolicy), index--));
}
}
static void processLabelReference(CompletionResultSet result, PsiLabelReference ref) {
for (String s : ref.getVariants()) {
result.addElement(
TailTypeDecorator.withTail(LookupElementBuilder.create(s), TailType.SEMICOLON));
}
}
public void addCompletions(
@NotNull CompletionParameters parameters,
ProcessingContext context,
@NotNull CompletionResultSet resultSet) {
PsiElement stringLiteralExpression = parameters.getPosition().getParent();
Project project = stringLiteralExpression.getProject();
if (!SilexProjectComponent.isEnabled(project)) {
return;
}
if (!(stringLiteralExpression instanceof StringLiteralExpression)) {
return;
}
PsiElement arrayKeyElement = stringLiteralExpression.getParent();
PsiElement element = arrayKeyElement.getParent();
if (element instanceof ArrayHashElement) {
if (!arrayKeyElement.isEquivalentTo(element.getFirstChild())) {
return;
}
element = element.getParent();
}
if (!(element instanceof ArrayCreationExpression)) {
return;
}
PsiElement parameterList = element.getParent();
if (!(parameterList instanceof ParameterList)) {
return;
}
PsiElement[] params = ((ParameterList) parameterList).getParameters();
if (!(params.length > 1 && params[1].isEquivalentTo(element))) {
return;
}
PsiElement methodReference = parameterList.getParent();
if (!(methodReference instanceof MethodReference)) {
return;
}
String methodReferenceName = ((MethodReference) methodReference).getName();
if ((methodReferenceName == null) || !(methodReferenceName.equals("register"))) {
return;
}
Container container =
Utils.findContainerForMethodReference((MethodReference) methodReference);
if (container == null) {
return;
}
for (Parameter parameter : container.getParameters().values()) {
resultSet.addElement(new ParameterLookupElement(parameter));
}
resultSet.stopHere();
}
private static void completeAnnotationAttributeName(
CompletionResultSet result, PsiElement insertedElement, CompletionParameters parameters) {
PsiNameValuePair pair = PsiTreeUtil.getParentOfType(insertedElement, PsiNameValuePair.class);
PsiAnnotationParameterList parameterList = (PsiAnnotationParameterList) pair.getParent();
PsiAnnotation anno = (PsiAnnotation) parameterList.getParent();
boolean showClasses = psiElement().afterLeaf("(").accepts(insertedElement);
PsiClass annoClass = null;
final PsiJavaCodeReferenceElement referenceElement = anno.getNameReferenceElement();
if (referenceElement != null) {
final PsiElement element = referenceElement.resolve();
if (element instanceof PsiClass) {
annoClass = (PsiClass) element;
if (annoClass.findMethodsByName("value", false).length == 0) {
showClasses = false;
}
}
}
if (showClasses && insertedElement.getParent() instanceof PsiReferenceExpression) {
final Set<LookupElement> set =
JavaCompletionUtil.processJavaReference(
insertedElement,
(PsiJavaReference) insertedElement.getParent(),
new ElementExtractorFilter(createAnnotationFilter(insertedElement)),
JavaCompletionProcessor.Options.DEFAULT_OPTIONS,
result.getPrefixMatcher(),
parameters);
for (final LookupElement element : set) {
result.addElement(element);
}
addAllClasses(parameters, result, new InheritorsHolder(insertedElement, result));
}
if (annoClass != null) {
final PsiNameValuePair[] existingPairs = parameterList.getAttributes();
methods:
for (PsiMethod method : annoClass.getMethods()) {
if (!(method instanceof PsiAnnotationMethod)) continue;
final String attrName = method.getName();
for (PsiNameValuePair existingAttr : existingPairs) {
if (PsiTreeUtil.isAncestor(existingAttr, insertedElement, false)) break;
if (Comparing.equal(existingAttr.getName(), attrName)
|| PsiAnnotation.DEFAULT_REFERENCED_METHOD_NAME.equals(attrName)
&& existingAttr.getName() == null) continue methods;
}
LookupElementBuilder element =
LookupElementBuilder.createWithIcon(method)
.withInsertHandler(
new InsertHandler<LookupElement>() {
@Override
public void handleInsert(InsertionContext context, LookupElement item) {
final Editor editor = context.getEditor();
TailType.EQ.processTail(editor, editor.getCaretModel().getOffset());
context.setAddCompletionChar(false);
context.commitDocument();
PsiAnnotationParameterList paramList =
PsiTreeUtil.findElementOfClassAtOffset(
context.getFile(),
context.getStartOffset(),
PsiAnnotationParameterList.class,
false);
if (paramList != null
&& paramList.getAttributes().length > 0
&& paramList.getAttributes()[0].getName() == null) {
int valueOffset =
paramList.getAttributes()[0].getTextRange().getStartOffset();
context
.getDocument()
.insertString(
valueOffset, PsiAnnotation.DEFAULT_REFERENCED_METHOD_NAME);
TailType.EQ.processTail(
editor,
valueOffset + PsiAnnotation.DEFAULT_REFERENCED_METHOD_NAME.length());
}
}
});
PsiAnnotationMemberValue defaultValue = ((PsiAnnotationMethod) method).getDefaultValue();
if (defaultValue != null) {
Object constant =
JavaPsiFacade.getInstance(method.getProject())
.getConstantEvaluationHelper()
.computeConstantExpression(defaultValue);
if (constant != null) {
element =
element.withTailText(
" default " + (constant instanceof String ? "\"" + constant + "\"" : constant),
true);
}
}
result.addElement(element);
}
}
}
@Override
public void fillCompletionVariants(
final CompletionParameters parameters, final CompletionResultSet _result) {
if (parameters.getCompletionType() != CompletionType.BASIC) {
return;
}
final PsiElement position = parameters.getPosition();
if (!isInJavaContext(position)) {
return;
}
if (AFTER_NUMBER_LITERAL.accepts(position)
|| UNEXPECTED_REFERENCE_AFTER_DOT.accepts(position)) {
_result.stopHere();
return;
}
final CompletionResultSet result = JavaCompletionSorting.addJavaSorting(parameters, _result);
if (ANNOTATION_ATTRIBUTE_NAME.accepts(position)
&& !JavaCompletionData.isAfterPrimitiveOrArrayType(position)) {
JavaCompletionData.addExpectedTypeMembers(parameters, result);
completeAnnotationAttributeName(result, position, parameters);
result.stopHere();
return;
}
final InheritorsHolder inheritors = new InheritorsHolder(position, result);
if (JavaSmartCompletionContributor.IN_TYPE_ARGS.accepts(position)) {
new TypeArgumentCompletionProvider(false, inheritors)
.addCompletions(parameters, new ProcessingContext(), result);
}
PrefixMatcher matcher = result.getPrefixMatcher();
if (JavaSmartCompletionContributor.AFTER_NEW.accepts(position)) {
new JavaInheritorsGetter(ConstructorInsertHandler.BASIC_INSTANCE)
.generateVariants(parameters, matcher, inheritors);
}
if (IMPORT_REFERENCE.accepts(position)) {
result.addElement(LookupElementBuilder.create("*"));
}
addKeywords(parameters, result);
Set<String> usedWords = addReferenceVariants(parameters, result, inheritors);
if (psiElement().inside(PsiLiteralExpression.class).accepts(position)) {
PsiReference reference = position.getContainingFile().findReferenceAt(parameters.getOffset());
if (reference == null || reference.isSoft()) {
WordCompletionContributor.addWordCompletionVariants(result, parameters, usedWords);
}
}
JavaGenerateMemberCompletionContributor.fillCompletionVariants(parameters, result);
addAllClasses(parameters, result, inheritors);
final PsiElement parent = position.getParent();
if (parent instanceof PsiReferenceExpression
&& !((PsiReferenceExpression) parent).isQualified()
&& parameters.isExtendedCompletion()
&& StringUtil.isNotEmpty(matcher.getPrefix())) {
new JavaStaticMemberProcessor(parameters).processStaticMethodsGlobally(matcher, result);
}
result.stopHere();
}
