mirror of
https://gitflic.ru/project/openide/openide.git
synced 2026-09-27 10:03:11 +07:00
internal actions go to approprite places
This commit is contained in:
Dmitry Jemerov
@@ -0,0 +1,262 @@
|
||||
/*
|
||||
* User: anna
|
||||
* Date: 28-Jun-2007
|
||||
*/
|
||||
package com.intellij.internal;
|
||||
|
||||
import com.intellij.codeInsight.generation.GenerateMembersUtil;
|
||||
import com.intellij.codeInsight.generation.GenerationInfo;
|
||||
import com.intellij.codeInsight.generation.PsiGenerationInfo;
|
||||
import com.intellij.ide.IdeView;
|
||||
import com.intellij.ide.util.PackageChooserDialog;
|
||||
import com.intellij.ide.util.PackageUtil;
|
||||
import com.intellij.openapi.actionSystem.AnAction;
|
||||
import com.intellij.openapi.actionSystem.AnActionEvent;
|
||||
import com.intellij.openapi.actionSystem.LangDataKeys;
|
||||
import com.intellij.openapi.actionSystem.PlatformDataKeys;
|
||||
import com.intellij.openapi.application.Result;
|
||||
import com.intellij.openapi.command.CommandProcessor;
|
||||
import com.intellij.openapi.command.WriteCommandAction;
|
||||
import com.intellij.openapi.editor.Document;
|
||||
import com.intellij.openapi.fileTypes.StdFileTypes;
|
||||
import com.intellij.openapi.project.Project;
|
||||
import com.intellij.openapi.ui.DialogWrapper;
|
||||
import com.intellij.openapi.util.Condition;
|
||||
import com.intellij.openapi.util.Ref;
|
||||
import com.intellij.psi.*;
|
||||
import com.intellij.psi.search.GlobalSearchScope;
|
||||
import com.intellij.psi.search.PackageScope;
|
||||
import com.intellij.psi.search.searches.ClassInheritorsSearch;
|
||||
import com.intellij.ui.DocumentAdapter;
|
||||
import com.intellij.ui.EditorTextField;
|
||||
import com.intellij.util.IncorrectOperationException;
|
||||
import com.intellij.util.containers.ContainerUtil;
|
||||
import gnu.trove.THashMap;
|
||||
import gnu.trove.THashSet;
|
||||
|
||||
import javax.swing.*;
|
||||
import javax.swing.event.DocumentEvent;
|
||||
import java.awt.*;
|
||||
import java.util.*;
|
||||
import java.util.List;
|
||||
|
||||
public class GenerateVisitorByHierarchyAction extends AnAction {
|
||||
public GenerateVisitorByHierarchyAction() {
|
||||
super("Generate Hierarchy Visitor");
|
||||
}
|
||||
|
||||
public void actionPerformed(AnActionEvent e) {
|
||||
final Ref<String> visitorNameRef = Ref.create("MyVisitor");
|
||||
final Ref<PsiClass> parentClassRef = Ref.create(null);
|
||||
final Project project = e.getData(PlatformDataKeys.PROJECT);
|
||||
assert project != null;
|
||||
final JavaPsiFacade psiFacade = JavaPsiFacade.getInstance(project);
|
||||
final PsiNameHelper helper = psiFacade.getNameHelper();
|
||||
final PackageChooserDialog dialog = new PackageChooserDialog("Choose Visitor Name, Package and Parent Class", project) {
|
||||
{
|
||||
updateOKAction();
|
||||
}
|
||||
protected JComponent createCenterPanel() {
|
||||
final JPanel panel = new JPanel(new BorderLayout());
|
||||
panel.add(super.createCenterPanel(), BorderLayout.CENTER);
|
||||
panel.add(createNamePanel(), BorderLayout.NORTH);
|
||||
panel.add(createBaseClassPanel(), BorderLayout.SOUTH);
|
||||
return panel;
|
||||
}
|
||||
|
||||
private JComponent createNamePanel() {
|
||||
final JPanel panel = new JPanel(new BorderLayout());
|
||||
panel.add(new JLabel("Visitor Name"), BorderLayout.WEST);
|
||||
final JTextField nameField = new JTextField(visitorNameRef.get());
|
||||
nameField.getDocument().addDocumentListener(new DocumentAdapter() {
|
||||
protected void textChanged(final DocumentEvent e) {
|
||||
visitorNameRef.set(nameField.getText());
|
||||
updateOKAction();
|
||||
}
|
||||
});
|
||||
panel.add(nameField, BorderLayout.CENTER);
|
||||
return panel;
|
||||
}
|
||||
|
||||
private JComponent createBaseClassPanel() {
|
||||
final JPanel panel = new JPanel(new BorderLayout());
|
||||
panel.add(new JLabel("Hierarchy Base Class"), BorderLayout.WEST);
|
||||
final PsiElementFactory factory = psiFacade.getElementFactory();
|
||||
final PsiTypeCodeFragment codeFragment = factory.createTypeCodeFragment("", null, true, true);
|
||||
final Document document = PsiDocumentManager.getInstance(project).getDocument(codeFragment);
|
||||
final EditorTextField editorTextField = new EditorTextField(document, project, StdFileTypes.JAVA);
|
||||
editorTextField.addDocumentListener(new com.intellij.openapi.editor.event.DocumentAdapter() {
|
||||
public void documentChanged(final com.intellij.openapi.editor.event.DocumentEvent e) {
|
||||
parentClassRef.set(null);
|
||||
try {
|
||||
final PsiType psiType = codeFragment.getType();
|
||||
final PsiClass psiClass = psiType instanceof PsiClassType ? ((PsiClassType)psiType).resolve() : null;
|
||||
parentClassRef.set(psiClass);
|
||||
}
|
||||
catch (PsiTypeCodeFragment.IncorrectTypeException e1) {
|
||||
// ok
|
||||
}
|
||||
updateOKAction();
|
||||
}
|
||||
});
|
||||
panel.add(editorTextField.getComponent(), BorderLayout.CENTER);
|
||||
return panel;
|
||||
}
|
||||
|
||||
private void updateOKAction() {
|
||||
PsiDocumentManager.getInstance(project).commitAllDocuments();
|
||||
final String message;
|
||||
if (!helper.isQualifiedName(visitorNameRef.get())) {
|
||||
message = "Visitor name is not valid";
|
||||
}
|
||||
else if (parentClassRef.isNull()) {
|
||||
message = "Hierarchy parent should be specified";
|
||||
}
|
||||
else if (parentClassRef.get().isAnnotationType() || parentClassRef.get().isEnum()) {
|
||||
message = "Hierarchy parent should be interface or class";
|
||||
}
|
||||
else message = null;
|
||||
setErrorText(message);
|
||||
setOKActionEnabled(message == null);
|
||||
}
|
||||
|
||||
};
|
||||
final PsiElement element = LangDataKeys.PSI_ELEMENT.getData(e.getDataContext());
|
||||
if (element instanceof PsiPackage) {
|
||||
dialog.selectPackage(((PsiPackage)element).getQualifiedName());
|
||||
}
|
||||
else if (element instanceof PsiDirectory) {
|
||||
final PsiPackage aPackage = JavaDirectoryService.getInstance().getPackage((PsiDirectory)element);
|
||||
if (aPackage != null) {
|
||||
dialog.selectPackage(aPackage.getQualifiedName());
|
||||
}
|
||||
}
|
||||
dialog.show();
|
||||
if (dialog.getExitCode() != DialogWrapper.OK_EXIT_CODE ||
|
||||
dialog.getSelectedPackage() == null ||
|
||||
dialog.getSelectedPackage().getQualifiedName().length() == 0 ||
|
||||
parentClassRef.isNull()) return;
|
||||
final PsiPackage aPackage = dialog.getSelectedPackage();
|
||||
final PsiClass psiClass = parentClassRef.get();
|
||||
final String visitorName = visitorNameRef.get();
|
||||
final String visitorQName = PsiNameHelper.getShortClassName(visitorName).equals(visitorName)? aPackage.getQualifiedName()+"."+visitorName : visitorName;
|
||||
generateVisitorClass(visitorQName, aPackage, psiClass);
|
||||
final IdeView ideView = LangDataKeys.IDE_VIEW.getData(e.getDataContext());
|
||||
final PsiClass visitorClass = JavaPsiFacade.getInstance(project).findClass(visitorQName, GlobalSearchScope.projectScope(project));
|
||||
if (ideView != null && visitorClass != null) {
|
||||
ideView.selectElement(visitorClass);
|
||||
}
|
||||
}
|
||||
|
||||
public void update(final AnActionEvent e) {
|
||||
e.getPresentation().setEnabled(e.getData(PlatformDataKeys.PROJECT) != null);
|
||||
}
|
||||
|
||||
private static void generateVisitorClass(final String visitorName, final PsiPackage aPackage, final PsiClass baseClass) {
|
||||
final THashMap<PsiClass, Set<PsiClass>> classes = new THashMap<PsiClass, Set<PsiClass>>();
|
||||
for (PsiClass aClass : ClassInheritorsSearch.search(baseClass, new PackageScope(aPackage, false, false), true).findAll()) {
|
||||
if (aClass.hasModifierProperty(PsiModifier.ABSTRACT) == baseClass.hasModifierProperty(PsiModifier.ABSTRACT)) {
|
||||
final List<PsiClass> implementors =
|
||||
ContainerUtil.findAll(ClassInheritorsSearch.search(aClass).findAll(), new Condition<PsiClass>() {
|
||||
public boolean value(final PsiClass psiClass) {
|
||||
return !psiClass.hasModifierProperty(PsiModifier.ABSTRACT);
|
||||
}
|
||||
});
|
||||
classes.put(aClass, new THashSet<PsiClass>(implementors));
|
||||
}
|
||||
}
|
||||
final THashMap<PsiClass, Set<PsiClass>> pathMap = new THashMap<PsiClass, Set<PsiClass>>();
|
||||
for (PsiClass aClass : classes.keySet()) {
|
||||
final Set<PsiClass> superClasses = new LinkedHashSet<PsiClass>();
|
||||
for (PsiClass superClass : aClass.getSupers()) {
|
||||
if (superClass.isInheritor(baseClass, true)) {
|
||||
superClasses.add(superClass);
|
||||
final Set<PsiClass> superImplementors = classes.get(superClass);
|
||||
if (superImplementors != null) {
|
||||
superImplementors.removeAll(classes.get(aClass));
|
||||
}
|
||||
}
|
||||
}
|
||||
if (superClasses.isEmpty()) {
|
||||
superClasses.add(baseClass);
|
||||
}
|
||||
pathMap.put(aClass, superClasses);
|
||||
}
|
||||
pathMap.put(baseClass, Collections.<PsiClass>emptySet());
|
||||
final ArrayList<PsiFile> psiFiles = new ArrayList<PsiFile>();
|
||||
for (Set<PsiClass> implementors : classes.values()) {
|
||||
for (PsiClass psiClass : implementors) {
|
||||
psiFiles.add(psiClass.getContainingFile());
|
||||
}
|
||||
}
|
||||
final Project project = baseClass.getProject();
|
||||
final PsiClass visitorClass = JavaPsiFacade.getInstance(project).findClass(visitorName, GlobalSearchScope.projectScope(project));
|
||||
if (visitorClass != null) {
|
||||
psiFiles.add(visitorClass.getContainingFile());
|
||||
}
|
||||
new WriteCommandAction(project, psiFiles.toArray(new PsiFile[psiFiles.size()])) {
|
||||
|
||||
protected void run(final Result result) throws Throwable {
|
||||
CommandProcessor.getInstance().markCurrentCommandAsComplex(project);
|
||||
if (visitorClass == null) {
|
||||
final String shortClassName = PsiNameHelper.getShortClassName(visitorName);
|
||||
final String packageName = visitorName.substring(0, visitorName.length() - shortClassName.length() - 1);
|
||||
final PsiDirectory directory = PackageUtil.findOrCreateDirectoryForPackage(project, packageName, null, false);
|
||||
if (directory != null) {
|
||||
final PsiClass visitorClass = JavaDirectoryService.getInstance().createClass(directory, shortClassName);
|
||||
generateVisitorClass(visitorClass, classes, pathMap);
|
||||
}
|
||||
}
|
||||
else {
|
||||
generateVisitorClass(visitorClass, classes, pathMap);
|
||||
}
|
||||
}
|
||||
}.execute();
|
||||
}
|
||||
|
||||
private static void generateVisitorClass(final PsiClass visitorClass, final Map<PsiClass, Set<PsiClass>> classes,
|
||||
final THashMap<PsiClass, Set<PsiClass>> pathMap) throws Throwable {
|
||||
final PsiElementFactory elementFactory = JavaPsiFacade.getInstance(visitorClass.getProject()).getElementFactory();
|
||||
for (PsiClass psiClass : classes.keySet()) {
|
||||
final PsiMethod method = elementFactory.createMethodFromText(
|
||||
"public void accept(final " + visitorClass.getQualifiedName() + " visitor) { visitor.visit" + psiClass.getName() + "(this); }", psiClass);
|
||||
for (PsiClass implementor : classes.get(psiClass)) {
|
||||
addOrReplaceMethod(method, implementor);
|
||||
}
|
||||
}
|
||||
|
||||
final THashSet<PsiClass> visitedClasses = new THashSet<PsiClass>();
|
||||
final LinkedList<PsiClass> toProcess = new LinkedList<PsiClass>(classes.keySet());
|
||||
while (!toProcess.isEmpty()) {
|
||||
final PsiClass psiClass = toProcess.removeFirst();
|
||||
if (!visitedClasses.add(psiClass)) continue;
|
||||
final Set<PsiClass> pathClasses = pathMap.get(psiClass);
|
||||
toProcess.addAll(pathClasses);
|
||||
final StringBuilder methodText = new StringBuilder();
|
||||
|
||||
methodText.append("public void visit").append(psiClass.getName()).append("(final ").append(psiClass.getQualifiedName()).append(" o) {");
|
||||
boolean first = true;
|
||||
for (PsiClass pathClass : pathClasses) {
|
||||
if (first) first = false;
|
||||
else methodText.append("// ");
|
||||
methodText.append("visit").append(pathClass.getName()).append("(o);\n");
|
||||
}
|
||||
methodText.append("}");
|
||||
final PsiMethod method = elementFactory.createMethodFromText(methodText.toString(), psiClass);
|
||||
addOrReplaceMethod(method, visitorClass);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
private static void addOrReplaceMethod(final PsiMethod method, final PsiClass implementor) throws IncorrectOperationException {
|
||||
final PsiMethod accept = implementor.findMethodBySignature(method, false);
|
||||
if (accept != null) {
|
||||
accept.replace(method);
|
||||
}
|
||||
else {
|
||||
GenerateMembersUtil.insertMembersAtOffset(implementor.getContainingFile(), implementor.getLastChild().getTextOffset(), Collections.<GenerationInfo>singletonList(new PsiGenerationInfo<PsiMethod>(method)));
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user