-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Define structure: Initial stab at a completion contributor
- Loading branch information
1 parent
c25580d
commit 8e2ccfb
Showing
27 changed files
with
564 additions
and
114 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
134 changes: 134 additions & 0 deletions
134
src/main/java/aldor/editor/completion/AldorCompletionContributor.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,134 @@ | ||
| package aldor.editor.completion; | ||
|
|
||
| import aldor.language.SpadLanguage; | ||
| import aldor.psi.AldorId; | ||
| import aldor.psi.elements.AldorTypes; | ||
| import aldor.spad.SpadLibrary; | ||
| import aldor.spad.SpadLibraryManager; | ||
| import aldor.syntax.Syntax; | ||
| import aldor.syntax.SyntaxUtils; | ||
| import aldor.syntax.components.Apply; | ||
| import aldor.syntax.components.Id; | ||
| import aldor.ui.AldorIcons; | ||
| import com.google.common.annotations.VisibleForTesting; | ||
| import com.intellij.codeInsight.completion.AddSpaceInsertHandler; | ||
| import com.intellij.codeInsight.completion.CompletionContributor; | ||
| import com.intellij.codeInsight.completion.CompletionInitializationContext; | ||
| import com.intellij.codeInsight.completion.CompletionParameters; | ||
| import com.intellij.codeInsight.completion.CompletionProvider; | ||
| import com.intellij.codeInsight.completion.CompletionResultSet; | ||
| import com.intellij.codeInsight.completion.CompletionType; | ||
| import com.intellij.codeInsight.completion.util.ParenthesesInsertHandler; | ||
| import com.intellij.codeInsight.lookup.LookupElement; | ||
| import com.intellij.codeInsight.lookup.LookupElementBuilder; | ||
| import com.intellij.patterns.ElementPattern; | ||
| import com.intellij.patterns.PatternCondition; | ||
| import com.intellij.patterns.StandardPatterns; | ||
| import com.intellij.psi.PsiElement; | ||
| import com.intellij.util.ProcessingContext; | ||
| import org.jetbrains.annotations.NotNull; | ||
|
|
||
| import java.util.List; | ||
| import java.util.Optional; | ||
| import java.util.stream.Collectors; | ||
| import java.util.stream.Stream; | ||
|
|
||
| import static com.intellij.patterns.PlatformPatterns.psiElement; | ||
| import static com.intellij.patterns.StandardPatterns.object; | ||
| import static com.intellij.patterns.StandardPatterns.or; | ||
|
|
||
| public class AldorCompletionContributor extends CompletionContributor { | ||
|
|
||
| public AldorCompletionContributor() { | ||
|
|
||
| ElementPattern<PsiElement> typeElement = psiElement().withElementType(or(object(AldorTypes.TYPE), object(AldorTypes.TYPE_E_12))); | ||
| ElementPattern<PsiElement> typeElementStopPattern = psiElement().withParent(StandardPatterns.not(AldorPatterns.isFirstChild())); | ||
|
|
||
| ElementPattern<? extends PsiElement> insideTypeElementPattern = psiElement(AldorId.class).withLanguage(SpadLanguage.INSTANCE).inside(true, typeElement, typeElementStopPattern); | ||
|
|
||
| extend(CompletionType.BASIC, psiElement(), idCompletion()); | ||
| } | ||
|
|
||
| private CompletionProvider<CompletionParameters> idCompletion() { | ||
| return new CompletionProvider<CompletionParameters>() { | ||
| @Override | ||
| protected void addCompletions(@NotNull CompletionParameters parameters, | ||
| ProcessingContext context, | ||
| @NotNull CompletionResultSet result) { | ||
| List<LookupElement> element = allTypes(parameters); | ||
| result.addAllElements(element); | ||
| } | ||
| }; | ||
| } | ||
|
|
||
| public List<LookupElement> allTypes(CompletionParameters parameters) { | ||
| PsiElement elt = parameters.getPosition(); | ||
| SpadLibrary spadLibrary = SpadLibraryManager.instance().spadLibraryForElement(elt); | ||
| return allTypes(spadLibrary); | ||
| } | ||
|
|
||
| @VisibleForTesting | ||
| public static List<LookupElement> allTypes(SpadLibrary spadLibrary) { | ||
| List<Syntax> allTypes = spadLibrary.allTypes(); | ||
| return allTypes.stream() | ||
| .flatMap(e -> createLookupElement(spadLibrary, e).map(Stream::of).orElse(Stream.empty())) | ||
| .collect(Collectors.toList()); | ||
| } | ||
|
|
||
| private static Optional<LookupElement> createLookupElement(SpadLibrary spadLibrary, Syntax syntax) { | ||
| Optional<Id> id = SyntaxUtils.leadingId(syntax).maybeAs(Id.class); | ||
| if (!id.isPresent()) { | ||
| return Optional.empty(); | ||
| } | ||
| else { | ||
| Optional<Apply> isApply = syntax.maybeAs(Apply.class); | ||
| if (isApply.filter(apply -> apply.arguments().size() > 1).isPresent()) { | ||
| return Optional.of(LookupElementBuilder.create(id.get().symbol()) | ||
| .withInsertHandler(ParenthesesInsertHandler.WITH_PARAMETERS) | ||
| .withIcon(AldorIcons.IDENTIFIER) | ||
| .withTailText(tailTextForElement(spadLibrary, id.get()), true) | ||
| ); | ||
| } | ||
| if (isApply.filter(apply -> apply.arguments().size() == 1).isPresent()) { | ||
| return Optional.of(LookupElementBuilder.create(id.get().symbol()) | ||
| .withInsertHandler(AddSpaceInsertHandler.INSTANCE_WITH_AUTO_POPUP)); | ||
| } | ||
| else { | ||
| return Optional.of(LookupElementBuilder.create(id.get().symbol()) | ||
| .withTailText(tailTextForElement(spadLibrary, id.get()), true) | ||
| ); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| private static String tailTextForElement(SpadLibrary spadLibrary, Id symbol) { | ||
| return " (" + spadLibrary.definingFile(symbol) + ")"; | ||
| } | ||
|
|
||
| @Override | ||
| public void beforeCompletion(@NotNull CompletionInitializationContext context) { | ||
| super.beforeCompletion(context); | ||
| } | ||
|
|
||
| @Override | ||
| public void duringCompletion(@NotNull CompletionInitializationContext context) { | ||
| super.duringCompletion(context); | ||
| } | ||
|
|
||
| @Override | ||
| public void fillCompletionVariants(@NotNull CompletionParameters parameters, @NotNull CompletionResultSet result) { | ||
| super.fillCompletionVariants(parameters, result); | ||
| } | ||
|
|
||
| private static final class AldorPatterns { | ||
| static ElementPattern<PsiElement> isFirstChild() { | ||
| //noinspection InnerClassTooDeeplyNested | ||
| return psiElement().with(new PatternCondition<PsiElement>("firstChild") { | ||
| @Override | ||
| public boolean accepts(@NotNull PsiElement psiElement, ProcessingContext context) { | ||
| return (psiElement.getParent() != null) && psiElement.getParent().getFirstChild().equals(psiElement); | ||
| } | ||
| }); | ||
| } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.