| /* |
| * Copyright 2000-2009 JetBrains s.r.o. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package com.intellij.codeInsight.daemon.impl; |
| |
| import com.intellij.codeHighlighting.TextEditorHighlightingPass; |
| import com.intellij.codeInsight.CodeInsightSettings; |
| import com.intellij.codeInsight.daemon.*; |
| import com.intellij.codeInsight.intention.IntentionAction; |
| import com.intellij.codeInspection.HintAction; |
| import com.intellij.lang.annotation.HighlightSeverity; |
| import com.intellij.openapi.actionSystem.ActionManager; |
| import com.intellij.openapi.actionSystem.IdeActions; |
| import com.intellij.openapi.application.Application; |
| import com.intellij.openapi.application.ApplicationManager; |
| import com.intellij.openapi.editor.Document; |
| import com.intellij.openapi.editor.Editor; |
| import com.intellij.openapi.extensions.Extensions; |
| import com.intellij.openapi.keymap.KeymapUtil; |
| import com.intellij.openapi.progress.ProgressIndicator; |
| import com.intellij.openapi.project.Project; |
| import com.intellij.openapi.util.Pair; |
| import com.intellij.openapi.util.TextRange; |
| import com.intellij.psi.PsiElement; |
| import com.intellij.psi.PsiFile; |
| import com.intellij.psi.PsiReference; |
| import com.intellij.util.Processor; |
| import org.jetbrains.annotations.NotNull; |
| |
| import java.util.ArrayList; |
| import java.util.List; |
| |
| public class ShowAutoImportPass extends TextEditorHighlightingPass { |
| private final Editor myEditor; |
| |
| private final PsiFile myFile; |
| |
| private final int myStartOffset; |
| private final int myEndOffset; |
| |
| public ShowAutoImportPass(@NotNull Project project, @NotNull final PsiFile file, @NotNull Editor editor) { |
| super(project, editor.getDocument(), false); |
| ApplicationManager.getApplication().assertIsDispatchThread(); |
| |
| myEditor = editor; |
| |
| TextRange range = VisibleHighlightingPassFactory.calculateVisibleRange(myEditor); |
| myStartOffset = range.getStartOffset(); |
| myEndOffset = range.getEndOffset(); |
| |
| myFile = file; |
| } |
| |
| @Override |
| public void doCollectInformation(@NotNull ProgressIndicator progress) { |
| } |
| |
| @Override |
| public void doApplyInformationToEditor() { |
| Application application = ApplicationManager.getApplication(); |
| application.assertIsDispatchThread(); |
| if (!application.isUnitTestMode() && !myEditor.getContentComponent().hasFocus()) return; |
| int caretOffset = myEditor.getCaretModel().getOffset(); |
| importUnambiguousImports(caretOffset); |
| List<HighlightInfo> visibleHighlights = getVisibleHighlights(myStartOffset, myEndOffset, myProject, myEditor); |
| |
| for (int i = visibleHighlights.size() - 1; i >= 0; i--) { |
| HighlightInfo info = visibleHighlights.get(i); |
| if (info.startOffset <= caretOffset && showAddImportHint(info)) return; |
| } |
| |
| for (HighlightInfo visibleHighlight : visibleHighlights) { |
| if (visibleHighlight.startOffset > caretOffset && showAddImportHint(visibleHighlight)) return; |
| } |
| } |
| |
| private void importUnambiguousImports(final int caretOffset) { |
| if (!DaemonCodeAnalyzerSettings.getInstance().isImportHintEnabled()) return; |
| if (!DaemonCodeAnalyzer.getInstance(myProject).isImportHintsEnabled(myFile)) return; |
| if (!CodeInsightSettings.getInstance().ADD_UNAMBIGIOUS_IMPORTS_ON_THE_FLY) return; |
| |
| Document document = getDocument(); |
| final List<HighlightInfo> infos = new ArrayList<HighlightInfo>(); |
| DaemonCodeAnalyzerEx.processHighlights(document, myProject, null, 0, document.getTextLength(), new Processor<HighlightInfo>() { |
| @Override |
| public boolean process(HighlightInfo info) { |
| if (!info.hasHint() || info.getSeverity() != HighlightSeverity.ERROR) { |
| return true; |
| } |
| PsiReference reference = myFile.findReferenceAt(info.getActualStartOffset()); |
| if (reference != null && reference.getElement().getTextRange().containsOffset(caretOffset)) return true; |
| infos.add(info); |
| return true; |
| } |
| }); |
| |
| ReferenceImporter[] importers = Extensions.getExtensions(ReferenceImporter.EP_NAME); |
| for (HighlightInfo info : infos) { |
| for(ReferenceImporter importer: importers) { |
| if (importer.autoImportReferenceAt(myEditor, myFile, info.getActualStartOffset())) break; |
| } |
| } |
| } |
| |
| @NotNull |
| private static List<HighlightInfo> getVisibleHighlights(final int startOffset, final int endOffset, Project project, final Editor editor) { |
| final List<HighlightInfo> highlights = new ArrayList<HighlightInfo>(); |
| DaemonCodeAnalyzerEx.processHighlights(editor.getDocument(), project, null, startOffset, endOffset, new Processor<HighlightInfo>() { |
| @Override |
| public boolean process(HighlightInfo info) { |
| if (info.hasHint() && !editor.getFoldingModel().isOffsetCollapsed(info.startOffset)) { |
| highlights.add(info); |
| } |
| return true; |
| } |
| }); |
| return highlights; |
| } |
| |
| private boolean showAddImportHint(HighlightInfo info) { |
| if (!DaemonCodeAnalyzerSettings.getInstance().isImportHintEnabled()) return false; |
| if (!DaemonCodeAnalyzer.getInstance(myProject).isImportHintsEnabled(myFile)) return false; |
| PsiElement element = myFile.findElementAt(info.startOffset); |
| if (element == null || !element.isValid()) return false; |
| |
| final List<Pair<HighlightInfo.IntentionActionDescriptor, TextRange>> list = info.quickFixActionRanges; |
| for (Pair<HighlightInfo.IntentionActionDescriptor, TextRange> pair : list) { |
| final IntentionAction action = pair.getFirst().getAction(); |
| if (action instanceof HintAction && action.isAvailable(myProject, myEditor, myFile)) { |
| return ((HintAction)action).showHint(myEditor); |
| } |
| } |
| return false; |
| } |
| |
| public static String getMessage(final boolean multiple, final String name) { |
| final String messageKey = multiple ? "import.popup.multiple" : "import.popup.text"; |
| String hintText = DaemonBundle.message(messageKey, name); |
| hintText += " " + KeymapUtil.getFirstKeyboardShortcutText(ActionManager.getInstance().getAction(IdeActions.ACTION_SHOW_INTENTION_ACTIONS)); |
| return hintText; |
| } |
| } |