show progress for cleanup taks; make cancelable (IDEA-92533)
[idea/community.git] / platform / lang-impl / src / com / intellij / codeInspection / ex / SequentialCleanupTask.java
1 /*
2  * Copyright 2000-2015 JetBrains s.r.o.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package com.intellij.codeInspection.ex;
17
18 import com.intellij.codeInsight.daemon.impl.HighlightInfo;
19 import com.intellij.openapi.actionSystem.ActionManager;
20 import com.intellij.openapi.application.AccessToken;
21 import com.intellij.openapi.application.ApplicationManager;
22 import com.intellij.openapi.application.WriteAction;
23 import com.intellij.openapi.diagnostic.Logger;
24 import com.intellij.openapi.progress.ProcessCanceledException;
25 import com.intellij.openapi.progress.ProgressIndicator;
26 import com.intellij.openapi.project.Project;
27 import com.intellij.openapi.util.Pair;
28 import com.intellij.openapi.util.TextRange;
29 import com.intellij.psi.PsiFile;
30 import com.intellij.util.SequentialModalProgressTask;
31 import com.intellij.util.SequentialTask;
32
33 import java.util.*;
34
35 class SequentialCleanupTask implements SequentialTask {
36   private static final Logger LOG = Logger.getInstance(SequentialCleanupTask.class);
37
38   private final Project myProject;
39   private final List<Pair<PsiFile, HighlightInfo>> myResults = new ArrayList<Pair<PsiFile, HighlightInfo>>();
40   private final SequentialModalProgressTask myProgressTask;
41   private int myCount = 0;
42   
43   public SequentialCleanupTask(Project project, LinkedHashMap<PsiFile, List<HighlightInfo>> results, SequentialModalProgressTask task) {
44     myProject = project;
45     for (Map.Entry<PsiFile, List<HighlightInfo>> entry : results.entrySet()) {
46       PsiFile file = entry.getKey();
47       List<HighlightInfo> infos = entry.getValue();
48       // sort from bottom to top
49       Collections.sort(infos, new Comparator<HighlightInfo>() {
50         @Override
51         public int compare(HighlightInfo info1, HighlightInfo info2) {
52           return info2.getStartOffset() - info1.getStartOffset();
53         }
54       });
55       for (HighlightInfo info : infos) {
56         myResults.add(Pair.create(file, info));
57       }
58     }
59     myProgressTask = task;
60   }
61
62   @Override
63   public void prepare() {}
64
65   @Override
66   public boolean isDone() {
67     return myCount > myResults.size() - 1;
68   }
69
70   @Override
71   public boolean iteration() {
72     final Pair<PsiFile, HighlightInfo> pair = myResults.get(myCount++);
73     final ProgressIndicator indicator = myProgressTask.getIndicator();
74     if (indicator != null) {
75       indicator.setFraction((double) myCount/myResults.size());
76       indicator.setText("Processing " + pair.first.getName());
77     }
78     for (final Pair<HighlightInfo.IntentionActionDescriptor, TextRange> actionRange : pair.second.quickFixActionRanges) {
79       final AccessToken token = WriteAction.start();
80       try {
81         actionRange.getFirst().getAction().invoke(myProject, null, pair.first);
82       }
83       catch (ProcessCanceledException e) {
84         throw e;
85       }
86       catch (Exception e) {
87         LOG.error(e);
88       }
89       finally {
90         token.finish();
91       }
92     }
93     return true;
94   }
95
96   @Override
97   public void stop() {}
98 }