Cleanup: NotNull/Nullable
[idea/community.git] / java / java-impl / src / com / intellij / lang / java / JavaModuleRenameListener.java
1 // Copyright 2000-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
2 package com.intellij.lang.java;
3
4 import com.intellij.ProjectTopics;
5 import com.intellij.codeInsight.daemon.impl.analysis.JavaModuleGraphUtil;
6 import com.intellij.openapi.application.AppUIExecutor;
7 import com.intellij.openapi.application.ModalityState;
8 import com.intellij.openapi.components.ProjectComponent;
9 import com.intellij.openapi.module.JavaModuleType;
10 import com.intellij.openapi.module.Module;
11 import com.intellij.openapi.module.ModuleType;
12 import com.intellij.openapi.project.ModuleListener;
13 import com.intellij.openapi.project.Project;
14 import com.intellij.openapi.roots.ModuleRootManager;
15 import com.intellij.openapi.util.Pair;
16 import com.intellij.psi.PsiJavaModule;
17 import com.intellij.psi.PsiNamedElement;
18 import com.intellij.psi.SmartPointerManager;
19 import com.intellij.psi.SmartPsiElementPointer;
20 import com.intellij.psi.impl.light.LightJavaModule;
21 import com.intellij.refactoring.RefactoringBundle;
22 import com.intellij.refactoring.rename.AutomaticRenamingDialog;
23 import com.intellij.refactoring.rename.RenameProcessor;
24 import com.intellij.refactoring.rename.naming.AutomaticRenamer;
25 import com.intellij.util.Function;
26 import org.jetbrains.annotations.Nls;
27 import org.jetbrains.annotations.NotNull;
28
29 import java.util.ArrayList;
30 import java.util.List;
31 import java.util.Map;
32 import java.util.Objects;
33 import java.util.stream.Stream;
34
35 import static com.intellij.openapi.util.Pair.pair;
36
37 public class JavaModuleRenameListener implements ProjectComponent, ModuleListener {
38   public JavaModuleRenameListener(@NotNull Project project) {
39     project.getMessageBus().connect().subscribe(ProjectTopics.MODULES, this);
40   }
41
42   @Override
43   public void modulesRenamed(@NotNull Project project, @NotNull List<Module> modules, @NotNull Function<Module, String> oldNameProvider) {
44     List<Pair<SmartPsiElementPointer<PsiJavaModule>, String>> suggestions = new ArrayList<>();
45
46     for (Module module : modules) {
47       if (ModuleType.get(module) == JavaModuleType.getModuleType()) {
48         PsiJavaModule javaModule = Stream.of(ModuleRootManager.getInstance(module).getSourceRoots(false))
49             .map(root -> JavaModuleGraphUtil.findDescriptorByFile(root, project))
50             .filter(Objects::nonNull)
51             .findFirst().orElse(null);
52         if (javaModule != null && javaModule.getName().equals(LightJavaModule.moduleName(oldNameProvider.fun(module)))) {
53           suggestions.add(pair(SmartPointerManager.getInstance(project).createSmartPsiElementPointer(javaModule),
54                            LightJavaModule.moduleName(module.getName())));
55         }
56       }
57     }
58
59     if (!suggestions.isEmpty()) {
60       AppUIExecutor.onUiThread(ModalityState.NON_MODAL)
61           .later()
62           .inSmartMode(project)
63           .inTransaction(project)
64           .execute(() -> renameModules(project, suggestions));
65     }
66   }
67
68   private static void renameModules(Project project, List<Pair<SmartPsiElementPointer<PsiJavaModule>, String>> suggestions) {
69     MyAutomaticRenamer renamer = new MyAutomaticRenamer();
70     for (Pair<SmartPsiElementPointer<PsiJavaModule>, String> rename : suggestions) {
71       PsiJavaModule javaModule = rename.first.getElement();
72       if (javaModule != null) {
73         renamer.addElement(javaModule, rename.second);
74       }
75     }
76
77     if (!renamer.getElements().isEmpty()) {
78       AutomaticRenamingDialog dialog = new AutomaticRenamingDialog(project, renamer);
79       dialog.showOptionsPanel();
80
81       if (dialog.showAndGet()) {
82         RenameProcessor processor = null;
83
84         for (Map.Entry<PsiNamedElement, String> entry : renamer.getRenames().entrySet()) {
85           String newName = entry.getValue();
86           if (newName != null) {
87             if (processor == null) {
88               processor = new RenameProcessor(project, entry.getKey(), newName, dialog.isSearchInComments(), dialog.isSearchTextOccurrences());
89             }
90             else {
91               processor.addElement(entry.getKey(), newName);
92             }
93           }
94         }
95
96         if (processor != null) {
97           processor.run();
98         }
99       }
100     }
101   }
102
103   private static class MyAutomaticRenamer extends AutomaticRenamer {
104     private void addElement(PsiJavaModule module, String newName) {
105       myElements.add(module);
106       suggestAllNames(module.getName(), newName);
107     }
108
109     @Nls(capitalization = Nls.Capitalization.Title)
110     @Override
111     public String getDialogTitle() {
112       return RefactoringBundle.message("auto.rename.module.dialog.title");
113     }
114
115     @Nls(capitalization = Nls.Capitalization.Sentence)
116     @Override
117     public String getDialogDescription() {
118       return RefactoringBundle.message("auto.rename.module.dialog.description");
119     }
120
121     @Override
122     public String entityName() {
123       return RefactoringBundle.message("auto.rename.module.entity");
124     }
125   }
126 }