| /* |
| * Copyright 2000-2014 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 org.jetbrains.plugins.groovy.codeInspection; |
| |
| import com.intellij.codeInsight.CodeInsightSettings; |
| import com.intellij.codeInsight.daemon.DaemonCodeAnalyzer; |
| import com.intellij.codeInsight.daemon.impl.*; |
| import com.intellij.codeInsight.intention.IntentionAction; |
| import com.intellij.lang.annotation.HighlightSeverity; |
| import com.intellij.openapi.application.ApplicationManager; |
| import com.intellij.openapi.command.undo.UndoManager; |
| import com.intellij.openapi.diagnostic.Attachment; |
| import com.intellij.openapi.diagnostic.Logger; |
| import com.intellij.openapi.editor.Document; |
| import com.intellij.openapi.editor.Editor; |
| import com.intellij.openapi.project.Project; |
| import com.intellij.openapi.roots.ProjectRootManager; |
| import com.intellij.openapi.util.Comparing; |
| import com.intellij.openapi.util.TextRange; |
| import com.intellij.openapi.vfs.VirtualFile; |
| import com.intellij.psi.PsiDocumentManager; |
| import com.intellij.psi.PsiFile; |
| import com.intellij.util.DocumentUtil; |
| import com.intellij.util.IncorrectOperationException; |
| import com.intellij.util.Processor; |
| import org.jetbrains.annotations.NotNull; |
| import org.jetbrains.plugins.groovy.editor.GroovyImportOptimizer; |
| import org.jetbrains.plugins.groovy.lang.psi.GroovyFile; |
| import org.jetbrains.plugins.groovy.lang.psi.api.toplevel.imports.GrImportStatement; |
| |
| public class GroovyOptimizeImportsFix implements IntentionAction { |
| private static final Logger LOG = Logger.getInstance("#org.jetbrains.plugins.groovy.codeInspection.local.GroovyPostHighlightingPass"); |
| private final boolean onTheFly; |
| |
| public GroovyOptimizeImportsFix(boolean onTheFly) { |
| this.onTheFly = onTheFly; |
| } |
| |
| @Override |
| public void invoke(@NotNull Project project, Editor editor, PsiFile file) throws IncorrectOperationException { |
| final Runnable optimize = new GroovyImportOptimizer().processFile(file); |
| GroovyOptimizeImportsFix.invokeOnTheFlyImportOptimizer(optimize, file, editor); |
| } |
| |
| @Override |
| @NotNull |
| public String getText() { |
| return GroovyInspectionBundle.message("optimize.all.imports"); |
| } |
| |
| @Override |
| @NotNull |
| public String getFamilyName() { |
| return GroovyInspectionBundle.message("optimize.imports"); |
| } |
| |
| @Override |
| public boolean startInWriteAction() { |
| return false; |
| } |
| |
| @Override |
| public boolean isAvailable(@NotNull Project project, Editor editor, PsiFile file) { |
| return file instanceof GroovyFile && (!onTheFly || timeToOptimizeImports((GroovyFile)file, editor)); |
| } |
| |
| private boolean timeToOptimizeImports(GroovyFile myFile, Editor editor) { |
| if (!CodeInsightSettings.getInstance().OPTIMIZE_IMPORTS_ON_THE_FLY) return false; |
| if (onTheFly && editor != null) { |
| // if we stand inside import statements, do not optimize |
| final VirtualFile vfile = myFile.getVirtualFile(); |
| if (vfile != null && ProjectRootManager.getInstance(myFile.getProject()).getFileIndex().isInSource(vfile)) { |
| final GrImportStatement[] imports = myFile.getImportStatements(); |
| if (imports.length > 0) { |
| final int offset = editor.getCaretModel().getOffset(); |
| if (imports[0].getTextRange().getStartOffset() <= offset && offset <= imports[imports.length - 1].getTextRange().getEndOffset()) { |
| return false; |
| } |
| } |
| } |
| } |
| |
| DaemonCodeAnalyzerImpl codeAnalyzer = (DaemonCodeAnalyzerImpl)DaemonCodeAnalyzer.getInstance(myFile.getProject()); |
| if (!codeAnalyzer.isHighlightingAvailable(myFile)) return false; |
| |
| if (!codeAnalyzer.isErrorAnalyzingFinished(myFile)) return false; |
| Document myDocument = PsiDocumentManager.getInstance(myFile.getProject()).getDocument(myFile); |
| boolean errors = containsErrorsPreventingOptimize(myFile, myDocument); |
| |
| return !errors && DaemonListeners.canChangeFileSilently(myFile); |
| } |
| |
| private boolean containsErrorsPreventingOptimize(GroovyFile myFile, Document myDocument) { |
| // ignore unresolved imports errors |
| final TextRange ignoreRange; |
| final GrImportStatement[] imports = myFile.getImportStatements(); |
| if (imports.length != 0) { |
| final int start = imports[0].getTextRange().getStartOffset(); |
| final int end = imports[imports.length - 1].getTextRange().getEndOffset(); |
| ignoreRange = new TextRange(start, end); |
| } else { |
| ignoreRange = TextRange.EMPTY_RANGE; |
| } |
| |
| return !DaemonCodeAnalyzerEx |
| .processHighlights(myDocument, myFile.getProject(), HighlightSeverity.ERROR, 0, myDocument.getTextLength(), new Processor<HighlightInfo>() { |
| @Override |
| public boolean process(HighlightInfo error) { |
| int infoStart = error.getActualStartOffset(); |
| int infoEnd = error.getActualEndOffset(); |
| |
| return ignoreRange.containsRange(infoStart, infoEnd) && error.type.equals(HighlightInfoType.WRONG_REF); |
| } |
| }); |
| } |
| |
| public static void invokeOnTheFlyImportOptimizer(@NotNull final Runnable runnable, |
| @NotNull final PsiFile file, |
| @NotNull final Editor editor) { |
| final long stamp = editor.getDocument().getModificationStamp(); |
| ApplicationManager.getApplication().invokeLater(new Runnable() { |
| @Override |
| public void run() { |
| if (file.getProject().isDisposed() || editor.isDisposed() || editor.getDocument().getModificationStamp() != stamp) return; |
| //no need to optimize imports on the fly during undo/redo |
| final UndoManager undoManager = UndoManager.getInstance(editor.getProject()); |
| if (undoManager.isUndoInProgress() || undoManager.isRedoInProgress()) return; |
| PsiDocumentManager.getInstance(file.getProject()).commitAllDocuments(); |
| String beforeText = file.getText(); |
| final long oldStamp = editor.getDocument().getModificationStamp(); |
| DocumentUtil.writeInRunUndoTransparentAction(runnable); |
| if (oldStamp != editor.getDocument().getModificationStamp()) { |
| String afterText = file.getText(); |
| if (Comparing.strEqual(beforeText, afterText)) { |
| String path = file.getViewProvider().getVirtualFile().getPath(); |
| LOG.error("Import optimizer hasn't optimized any imports", new Attachment(path, afterText)); |
| } |
| } |
| } |
| }); |
| } |
| } |