/*
* 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 com.intellij.refactoring.rename;
import com.intellij.ide.TitledHandler;
import com.intellij.openapi.actionSystem.CommonDataKeys;
import com.intellij.openapi.actionSystem.DataContext;
import com.intellij.openapi.application.ApplicationManager;
import com.intellij.openapi.extensions.Extensions;
import com.intellij.openapi.progress.ProcessCanceledException;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.ui.DialogWrapper;
import com.intellij.openapi.util.Key;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.refactoring.RefactoringBundle;
import com.intellij.refactoring.rename.inplace.MemberInplaceRenameHandler;
import com.intellij.refactoring.util.RadioUpDownListener;
import com.intellij.util.ArrayUtil;
import com.intellij.util.containers.HashSet;
import org.jetbrains.annotations.Nullable;
import javax.swing.*;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;
import java.util.TreeMap;
/**
* @author dsl
*/
public class RenameHandlerRegistry {
public static final Key<Boolean> SELECT_ALL = Key.create("rename.selectAll");
private final Set<RenameHandler> myHandlers = new HashSet<>();
private static final RenameHandlerRegistry INSTANCE = new RenameHandlerRegistry();
private final PsiElementRenameHandler myDefaultElementRenameHandler;
public static RenameHandlerRegistry getInstance() {
return INSTANCE;
}
private RenameHandlerRegistry() {
// should be checked last
myDefaultElementRenameHandler = new PsiElementRenameHandler();
}
public boolean hasAvailableHandler(DataContext dataContext) {
for (RenameHandler renameHandler : Extensions.getExtensions(RenameHandler.EP_NAME)) {
if (renameHandler.isAvailableOnDataContext(dataContext)) return true;
}
for (RenameHandler renameHandler : myHandlers) {
if (renameHandler.isAvailableOnDataContext(dataContext)) return true;
}
return myDefaultElementRenameHandler.isAvailableOnDataContext(dataContext);
}
@Nullable
public RenameHandler getRenameHandler(DataContext dataContext) {
final Map<String, RenameHandler> availableHandlers = new TreeMap<>();
for (RenameHandler renameHandler : Extensions.getExtensions(RenameHandler.EP_NAME)) {
checkHandler(renameHandler, dataContext, availableHandlers);
}
for (RenameHandler renameHandler : myHandlers) {
checkHandler(renameHandler, dataContext, availableHandlers);
}
if (availableHandlers.size() == 1) return availableHandlers.values().iterator().next();
for (Iterator<Map.Entry<String, RenameHandler>> iterator = availableHandlers.entrySet().iterator(); iterator.hasNext(); ) {
Map.Entry<String, RenameHandler> entry = iterator.next();
if (entry.getValue() instanceof MemberInplaceRenameHandler) {
iterator.remove();
break;
}
}
if (availableHandlers.size() == 1) return availableHandlers.values().iterator().next();
if (availableHandlers.size() > 1) {
if (ApplicationManager.getApplication().isUnitTestMode()) return availableHandlers.values().iterator().next();
final String[] strings = ArrayUtil.toStringArray(availableHandlers.keySet());
final HandlersChooser chooser = new HandlersChooser(CommonDataKeys.PROJECT.getData(dataContext), strings);
if (chooser.showAndGet()) {
return availableHandlers.get(chooser.getSelection());
}
throw new ProcessCanceledException();
}
return myDefaultElementRenameHandler.isRenaming(dataContext) ? myDefaultElementRenameHandler : null;
}
private static void checkHandler(RenameHandler renameHandler, DataContext dataContext, Map<String, RenameHandler> availableHandlers) {
if (renameHandler.isRenaming(dataContext)) {
availableHandlers.put(getHandlerTitle(renameHandler), renameHandler);
}
}
private static String getHandlerTitle(RenameHandler renameHandler) {
return renameHandler instanceof TitledHandler ? StringUtil.capitalize(((TitledHandler)renameHandler).getActionTitle().toLowerCase()) : renameHandler.toString();
}
/**
* @deprecated
* @see RenameHandler#EP_NAME
*/
public void registerHandler(RenameHandler handler) {
myHandlers.add(handler);
}
private static class HandlersChooser extends DialogWrapper {
private final String[] myRenamers;
private String mySelection;
private final JRadioButton[] myRButtons;
protected HandlersChooser(Project project, String [] renamers) {
super(project);
myRenamers = renamers;
myRButtons = new JRadioButton[myRenamers.length];
mySelection = renamers[0];
setTitle(RefactoringBundle.message("select.refactoring.title"));
init();
}
@Override
protected JComponent createNorthPanel() {
final JPanel radioPanel = new JPanel();
radioPanel.setLayout(new BoxLayout(radioPanel, BoxLayout.Y_AXIS));
final JLabel descriptionLabel = new JLabel(RefactoringBundle.message("what.would.you.like.to.do"));
descriptionLabel.setBorder(BorderFactory.createEmptyBorder(0, 0, 10, 0));
radioPanel.add(descriptionLabel);
final ButtonGroup bg = new ButtonGroup();
boolean selected = true;
int rIdx = 0;
for (final String renamer : myRenamers) {
final JRadioButton rb = new JRadioButton(renamer, selected);
myRButtons[rIdx++] = rb;
final ActionListener listener = new ActionListener() {
@Override
public void actionPerformed(ActionEvent e) {
if (rb.isSelected()) {
mySelection = renamer;
}
}
};
rb.addActionListener(listener);
selected = false;
bg.add(rb);
radioPanel.add(rb);
}
new RadioUpDownListener(myRButtons);
return radioPanel;
}
@Override
public JComponent getPreferredFocusedComponent() {
return myRButtons[0];
}
public String getSelection() {
return mySelection;
}
@Override
protected JComponent createCenterPanel() {
return null;
}
}
}