J2K: JetChangeSignatureProcessor
This commit is contained in:
Alexey Sedunov
-138
@@ -1,138 +0,0 @@
|
|||||||
/*
|
|
||||||
* Copyright 2010-2015 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.kotlin.idea.refactoring.changeSignature;
|
|
||||||
|
|
||||||
import com.intellij.openapi.application.ApplicationManager;
|
|
||||||
import com.intellij.openapi.project.Project;
|
|
||||||
import com.intellij.openapi.util.Ref;
|
|
||||||
import com.intellij.psi.PsiElement;
|
|
||||||
import com.intellij.refactoring.RefactoringBundle;
|
|
||||||
import com.intellij.refactoring.changeSignature.ChangeSignatureProcessorBase;
|
|
||||||
import com.intellij.refactoring.changeSignature.ChangeSignatureUsageProcessor;
|
|
||||||
import com.intellij.refactoring.changeSignature.JavaChangeInfo;
|
|
||||||
import com.intellij.refactoring.changeSignature.JavaChangeSignatureUsageProcessor;
|
|
||||||
import com.intellij.refactoring.rename.RenameUtil;
|
|
||||||
import com.intellij.refactoring.ui.ConflictsDialog;
|
|
||||||
import com.intellij.usageView.UsageInfo;
|
|
||||||
import com.intellij.usageView.UsageViewDescriptor;
|
|
||||||
import com.intellij.util.containers.HashSet;
|
|
||||||
import com.intellij.util.containers.MultiMap;
|
|
||||||
import kotlin.KotlinPackage;
|
|
||||||
import org.jetbrains.annotations.NotNull;
|
|
||||||
import org.jetbrains.kotlin.idea.refactoring.changeSignature.usages.JetUsageInfo;
|
|
||||||
import org.jetbrains.kotlin.idea.refactoring.changeSignature.usages.KotlinWrapperForJavaUsageInfos;
|
|
||||||
|
|
||||||
import java.util.*;
|
|
||||||
|
|
||||||
public class JetChangeSignatureProcessor extends ChangeSignatureProcessorBase {
|
|
||||||
private final String commandName;
|
|
||||||
|
|
||||||
public JetChangeSignatureProcessor(Project project, JetChangeInfo changeInfo, String commandName) {
|
|
||||||
super(project, changeInfo);
|
|
||||||
this.commandName = commandName;
|
|
||||||
}
|
|
||||||
|
|
||||||
@NotNull
|
|
||||||
@Override
|
|
||||||
protected UsageViewDescriptor createUsageViewDescriptor(UsageInfo[] usages) {
|
|
||||||
String subject = ChangeSignaturePackage.getKind(getChangeInfo()).getIsConstructor() ? "constructor" : "function";
|
|
||||||
return new JetUsagesViewDescriptor(myChangeInfo.getMethod(), RefactoringBundle.message("0.to.change.signature", subject));
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public JetChangeInfo getChangeInfo() {
|
|
||||||
return (JetChangeInfo) super.getChangeInfo();
|
|
||||||
}
|
|
||||||
|
|
||||||
@NotNull
|
|
||||||
@Override
|
|
||||||
protected UsageInfo[] findUsages() {
|
|
||||||
List<UsageInfo> allUsages = new ArrayList<UsageInfo>();
|
|
||||||
|
|
||||||
List<JavaChangeInfo> javaChangeInfos = getChangeInfo().getOrCreateJavaChangeInfos();
|
|
||||||
if (javaChangeInfos != null) {
|
|
||||||
JavaChangeSignatureUsageProcessor javaProcessor = new JavaChangeSignatureUsageProcessor();
|
|
||||||
for (JavaChangeInfo javaChangeInfo : javaChangeInfos) {
|
|
||||||
UsageInfo[] javaUsages = javaProcessor.findUsages(javaChangeInfo);
|
|
||||||
allUsages.add(new KotlinWrapperForJavaUsageInfos(javaChangeInfo, javaUsages, getChangeInfo().getMethod()));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
KotlinPackage.filterIsInstanceTo(super.findUsages(), allUsages, JetUsageInfo.class);
|
|
||||||
|
|
||||||
return allUsages.toArray(new UsageInfo[allUsages.size()]);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
protected boolean preprocessUsages(Ref<UsageInfo[]> refUsages) {
|
|
||||||
for (ChangeSignatureUsageProcessor processor : ChangeSignatureUsageProcessor.EP_NAME.getExtensions()) {
|
|
||||||
if (!processor.setupDefaultValues(myChangeInfo, refUsages, myProject)) return false;
|
|
||||||
}
|
|
||||||
MultiMap<PsiElement, String> conflictDescriptions = new MultiMap<PsiElement, String>();
|
|
||||||
for (ChangeSignatureUsageProcessor usageProcessor : ChangeSignatureUsageProcessor.EP_NAME.getExtensions()) {
|
|
||||||
MultiMap<PsiElement, String> conflicts = usageProcessor.findConflicts(myChangeInfo, refUsages);
|
|
||||||
for (PsiElement key : conflicts.keySet()) {
|
|
||||||
Collection<String> collection = conflictDescriptions.get(key);
|
|
||||||
if (collection.size() == 0) collection = new HashSet<String>();
|
|
||||||
collection.addAll(conflicts.get(key));
|
|
||||||
conflictDescriptions.put(key, collection);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
UsageInfo[] usagesIn = refUsages.get();
|
|
||||||
RenameUtil.addConflictDescriptions(usagesIn, conflictDescriptions);
|
|
||||||
Set<UsageInfo> usagesSet = new HashSet<UsageInfo>(Arrays.asList(usagesIn));
|
|
||||||
RenameUtil.removeConflictUsages(usagesSet);
|
|
||||||
if (!conflictDescriptions.isEmpty()) {
|
|
||||||
if (ApplicationManager.getApplication().isUnitTestMode()) {
|
|
||||||
throw new ConflictsInTestsException(conflictDescriptions.values());
|
|
||||||
}
|
|
||||||
if (myPrepareSuccessfulSwingThreadCallback != null) {
|
|
||||||
ConflictsDialog dialog = prepareConflictsDialog(conflictDescriptions, usagesIn);
|
|
||||||
dialog.show();
|
|
||||||
if (!dialog.isOK()) {
|
|
||||||
if (dialog.isShowConflicts()) prepareSuccessful();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
UsageInfo[] array = usagesSet.toArray(new UsageInfo[usagesSet.size()]);
|
|
||||||
Arrays.sort(array, new Comparator<UsageInfo>() {
|
|
||||||
@Override
|
|
||||||
public int compare(@NotNull UsageInfo u1, @NotNull UsageInfo u2) {
|
|
||||||
PsiElement element1 = u1.getElement();
|
|
||||||
PsiElement element2 = u2.getElement();
|
|
||||||
int rank1 = element1 != null ? element1.getTextOffset() : -1;
|
|
||||||
int rank2 = element2 != null ? element2.getTextOffset() : -1;
|
|
||||||
return rank2 - rank1; // Reverse order
|
|
||||||
}
|
|
||||||
});
|
|
||||||
refUsages.set(array);
|
|
||||||
prepareSuccessful();
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
protected boolean isPreviewUsages(UsageInfo[] usages) {
|
|
||||||
return isPreviewUsages();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
protected String getCommandName() {
|
|
||||||
return commandName;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
+107
@@ -0,0 +1,107 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2010-2015 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.kotlin.idea.refactoring.changeSignature
|
||||||
|
|
||||||
|
import com.intellij.openapi.application.ApplicationManager
|
||||||
|
import com.intellij.openapi.project.Project
|
||||||
|
import com.intellij.openapi.util.Ref
|
||||||
|
import com.intellij.psi.PsiElement
|
||||||
|
import com.intellij.refactoring.BaseRefactoringProcessor
|
||||||
|
import com.intellij.refactoring.RefactoringBundle
|
||||||
|
import com.intellij.refactoring.changeSignature.ChangeSignatureProcessorBase
|
||||||
|
import com.intellij.refactoring.changeSignature.ChangeSignatureUsageProcessor
|
||||||
|
import com.intellij.refactoring.changeSignature.JavaChangeSignatureUsageProcessor
|
||||||
|
import com.intellij.refactoring.rename.RenameUtil
|
||||||
|
import com.intellij.usageView.UsageInfo
|
||||||
|
import com.intellij.usageView.UsageViewDescriptor
|
||||||
|
import com.intellij.util.containers.MultiMap
|
||||||
|
import org.jetbrains.kotlin.idea.refactoring.changeSignature.usages.JetUsageInfo
|
||||||
|
import org.jetbrains.kotlin.idea.refactoring.changeSignature.usages.KotlinWrapperForJavaUsageInfos
|
||||||
|
import java.util.*
|
||||||
|
|
||||||
|
public class JetChangeSignatureProcessor(project: Project,
|
||||||
|
changeInfo: JetChangeInfo,
|
||||||
|
private val commandName: String) : ChangeSignatureProcessorBase(project, changeInfo) {
|
||||||
|
override fun createUsageViewDescriptor(usages: Array<UsageInfo>): UsageViewDescriptor {
|
||||||
|
val subject = if (getChangeInfo().kind.isConstructor) "constructor" else "function"
|
||||||
|
return JetUsagesViewDescriptor(myChangeInfo.getMethod(), RefactoringBundle.message("0.to.change.signature", subject))
|
||||||
|
}
|
||||||
|
|
||||||
|
override fun getChangeInfo() = super.getChangeInfo() as JetChangeInfo
|
||||||
|
|
||||||
|
override fun findUsages(): Array<UsageInfo> {
|
||||||
|
val allUsages = ArrayList<UsageInfo>()
|
||||||
|
getChangeInfo().getOrCreateJavaChangeInfos()?.let { javaChangeInfos ->
|
||||||
|
val javaProcessor = JavaChangeSignatureUsageProcessor()
|
||||||
|
javaChangeInfos.mapTo(allUsages) {
|
||||||
|
KotlinWrapperForJavaUsageInfos(it, javaProcessor.findUsages(it), getChangeInfo().getMethod())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
super.findUsages().filterIsInstanceTo(allUsages, javaClass<JetUsageInfo<PsiElement>>())
|
||||||
|
|
||||||
|
return allUsages.toTypedArray()
|
||||||
|
}
|
||||||
|
|
||||||
|
override fun preprocessUsages(refUsages: Ref<Array<UsageInfo>>): Boolean {
|
||||||
|
val usageProcessors = ChangeSignatureUsageProcessor.EP_NAME.getExtensions()
|
||||||
|
|
||||||
|
if (!usageProcessors.all { it.setupDefaultValues(myChangeInfo, refUsages, myProject) }) return false
|
||||||
|
|
||||||
|
val conflictDescriptions = object: MultiMap<PsiElement, String>() {
|
||||||
|
override fun createCollection() = LinkedHashSet<String>()
|
||||||
|
}
|
||||||
|
usageProcessors.forEach { conflictDescriptions.putAllValues(it.findConflicts(myChangeInfo, refUsages)) }
|
||||||
|
|
||||||
|
val usages = refUsages.get()
|
||||||
|
val usagesSet = usages.toHashSet()
|
||||||
|
|
||||||
|
RenameUtil.addConflictDescriptions(usages, conflictDescriptions)
|
||||||
|
RenameUtil.removeConflictUsages(usagesSet)
|
||||||
|
if (!conflictDescriptions.isEmpty()) {
|
||||||
|
if (ApplicationManager.getApplication().isUnitTestMode()) {
|
||||||
|
throw BaseRefactoringProcessor.ConflictsInTestsException(conflictDescriptions.values())
|
||||||
|
}
|
||||||
|
|
||||||
|
myPrepareSuccessfulSwingThreadCallback?.let {
|
||||||
|
val dialog = prepareConflictsDialog(conflictDescriptions, usages)
|
||||||
|
dialog.show()
|
||||||
|
if (!dialog.isOK()) {
|
||||||
|
if (dialog.isShowConflicts()) prepareSuccessful()
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
val usageArray = usagesSet.toTypedArray()
|
||||||
|
Arrays.sort(usageArray) { u1, u2 ->
|
||||||
|
val element1 = u1.getElement()
|
||||||
|
val element2 = u2.getElement()
|
||||||
|
val rank1 = if (element1 != null) element1.getTextOffset() else -1
|
||||||
|
val rank2 = if (element2 != null) element2.getTextOffset() else -1
|
||||||
|
rank2 - rank1 // Reverse order
|
||||||
|
}
|
||||||
|
refUsages.set(usageArray)
|
||||||
|
|
||||||
|
prepareSuccessful()
|
||||||
|
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
override fun isPreviewUsages(usages: Array<UsageInfo>?) = isPreviewUsages()
|
||||||
|
|
||||||
|
override fun getCommandName() = commandName
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user