public static void initOffsets(final PsiFile file, final OffsetMap offsetMap) {
int offset =
Math.max(
offsetMap.getOffset(CompletionInitializationContext.SELECTION_END_OFFSET),
offsetMap.getOffset(CompletionInitializationContext.IDENTIFIER_END_OFFSET));
PsiElement element = file.findElementAt(offset);
if (element instanceof PsiWhiteSpace
&& (!element.textContains('\n')
|| CodeStyleSettingsManager.getSettings(file.getProject())
.getCommonSettings(JavaLanguage.INSTANCE)
.METHOD_PARAMETERS_LPAREN_ON_NEXT_LINE)) {
element = file.findElementAt(element.getTextRange().getEndOffset());
}
if (element == null) return;
if (LEFT_PAREN.accepts(element)) {
offsetMap.addOffset(LPAREN_OFFSET, element.getTextRange().getStartOffset());
PsiElement list = element.getParent();
PsiElement last = list.getLastChild();
if (last instanceof PsiJavaToken
&& ((PsiJavaToken) last).getTokenType() == JavaTokenType.RPARENTH) {
offsetMap.addOffset(RPAREN_OFFSET, last.getTextRange().getStartOffset());
}
offsetMap.addOffset(ARG_LIST_END_OFFSET, list.getTextRange().getEndOffset());
}
}
private static OffsetMap translateOffsetMapToInjected(
OffsetMap hostMap, DocumentWindow injectedDocument) {
final OffsetMap map = new OffsetMap(injectedDocument);
for (final OffsetKey key : hostMap.getAllOffsets()) {
map.addOffset(key, injectedDocument.hostToInjected(hostMap.getOffset(key)));
}
return map;
}
private static OffsetMap translateOffsetMapToHost(
CompletionInitializationContext initContext,
PsiFile context,
PsiFile hostFile,
Editor hostEditor) {
final InjectedLanguageManager injectedLanguageManager =
InjectedLanguageManager.getInstance(hostFile.getProject());
final OffsetMap hostMap = new OffsetMap(hostEditor.getDocument());
final OffsetMap original = initContext.getOffsetMap();
for (final OffsetKey key : original.getAllOffsets()) {
hostMap.addOffset(
key, injectedLanguageManager.injectedToHost(context, original.getOffset(key)));
}
return hostMap;
}
void disposeOffsetMaps() {
for (OffsetMap map : myMapsToDispose) {
map.dispose();
}
myMapsToDispose.clear();
}
public int getIdentifierEndOffset() {
return myOffsetMap.getOffset(CompletionInitializationContext.IDENTIFIER_END_OFFSET);
}
private static boolean isInsideIdentifier(final OffsetMap offsetMap) {
return offsetMap.getOffset(CompletionInitializationContext.IDENTIFIER_END_OFFSET)
!= offsetMap.getOffset(CompletionInitializationContext.SELECTION_END_OFFSET);
}
private void doComplete(
CompletionInitializationContext initContext,
boolean hasModifiers,
int invocationCount,
PsiFile hostCopy,
OffsetMap hostMap,
OffsetTranslator translator) {
final Editor editor = initContext.getEditor();
CompletionAssertions.checkEditorValid(editor);
CompletionContext context =
createCompletionContext(
hostCopy,
hostMap.getOffset(CompletionInitializationContext.START_OFFSET),
hostMap,
initContext.getFile());
LookupImpl lookup = obtainLookup(editor);
CompletionParameters parameters = createCompletionParameters(invocationCount, context, editor);
CompletionPhase phase = CompletionServiceImpl.getCompletionPhase();
if (phase instanceof CompletionPhase.CommittingDocuments) {
if (phase.indicator != null) {
phase.indicator.closeAndFinish(false);
}
((CompletionPhase.CommittingDocuments) phase).replaced = true;
} else {
CompletionServiceImpl.assertPhase(CompletionPhase.NoCompletion.getClass());
}
final Semaphore freezeSemaphore = new Semaphore();
freezeSemaphore.down();
final CompletionProgressIndicator indicator =
new CompletionProgressIndicator(
editor,
parameters,
this,
freezeSemaphore,
initContext.getOffsetMap(),
hasModifiers,
lookup);
Disposer.register(indicator, hostMap);
Disposer.register(indicator, context.getOffsetMap());
Disposer.register(indicator, translator);
CompletionServiceImpl.setCompletionPhase(
synchronous
? new CompletionPhase.Synchronous(indicator)
: new CompletionPhase.BgCalculation(indicator));
final AtomicReference<LookupElement[]> data = indicator.startCompletion(initContext);
if (!synchronous) {
return;
}
if (freezeSemaphore.waitFor(2000)) {
final LookupElement[] allItems = data.get();
if (allItems != null
&& !indicator.isRunning()
&& !indicator
.isCanceled()) { // the completion is really finished, now we may auto-insert or show
// lookup
completionFinished(
initContext.getStartOffset(),
initContext.getSelectionEndOffset(),
indicator,
allItems,
hasModifiers);
checkNotSync(indicator, allItems);
return;
}
}
CompletionServiceImpl.setCompletionPhase(new CompletionPhase.BgCalculation(indicator));
indicator.showLookup();
}
public static void resetParensInfo(final OffsetMap offsetMap) {
offsetMap.removeOffset(LPAREN_OFFSET);
offsetMap.removeOffset(RPAREN_OFFSET);
offsetMap.removeOffset(ARG_LIST_END_OFFSET);
offsetMap.removeOffset(CompletionInitializationContext.IDENTIFIER_END_OFFSET);
}
