private void createNameAndReturnTypeEditors() {
myNameCodeFragment = new GroovyCodeFragment(myProject, "");
myNameField =
new EditorTextField(
PsiDocumentManager.getInstance(myProject).getDocument(myNameCodeFragment),
myProject,
myNameCodeFragment.getFileType());
final JavaCodeFragmentFactory factory = JavaCodeFragmentFactory.getInstance(myProject);
myReturnTypeCodeFragment =
factory.createTypeCodeFragment("", myMethod, true, JavaCodeFragmentFactory.ALLOW_VOID);
final Document document =
PsiDocumentManager.getInstance(myProject).getDocument(myReturnTypeCodeFragment);
myReturnTypeField =
new EditorTextField(document, myProject, myReturnTypeCodeFragment.getFileType());
myNameField.setText(myMethod.getName());
final GrTypeElement element = myMethod.getReturnTypeElementGroovy();
if (element != null) {
myReturnTypeField.setText(element.getText());
}
myReturnTypeLabel = new JLabel();
myReturnTypeLabel.setLabelFor(myReturnTypeField);
myNameLabel = new JLabel();
myNameLabel.setLabelFor(myNameField);
}
private static String generateParameterText(GrTableParameterInfo info) {
StringBuilder builder = new StringBuilder();
final PsiTypeCodeFragment typeFragment = info.getTypeFragment();
String typeText = typeFragment != null ? typeFragment.getText().trim() : GrModifier.DEF;
if (typeText.length() == 0) typeText = GrModifier.DEF;
builder.append(typeText).append(' ');
final GroovyCodeFragment nameFragment = info.getNameFragment();
builder.append(nameFragment != null ? nameFragment.getText().trim() : "");
final GroovyCodeFragment defaultInitializer = info.getDefaultInitializerFragment();
final String defaultInitializerText =
defaultInitializer != null ? defaultInitializer.getText().trim() : "";
if (defaultInitializerText.length() > 0) {
builder.append(" = ").append(defaultInitializerText);
}
return builder.toString();
}
private static boolean checkType(PsiTypeCodeFragment typeCodeFragment, boolean allowEllipsis) {
try {
final PsiType type = typeCodeFragment.getType();
return allowEllipsis || !(type instanceof PsiEllipsisType);
} catch (PsiTypeCodeFragment.TypeSyntaxException e) {
return false;
} catch (PsiTypeCodeFragment.NoTypeException e) {
return true; // Groovy accepts methods and parameters without explicit type
}
}
@Override
protected void doAction() {
if (!validateInputData()) {
return;
}
stopEditing();
String modifier = "";
if (myPublicRadioButton.isSelected()) {
modifier = GrModifier.PUBLIC;
} else if (myPrivateRadioButton.isSelected()) {
modifier = GrModifier.PRIVATE;
} else if (myProtectedRadioButton.isSelected()) {
modifier = GrModifier.PROTECTED;
}
PsiType returnType = null;
try {
returnType = myReturnTypeCodeFragment.getType();
} catch (PsiTypeCodeFragment.TypeSyntaxException ignored) {
} catch (PsiTypeCodeFragment.NoTypeException ignored) {
}
String newName = getNewName();
final List<GrTableParameterInfo> tableParameterInfos = myParameterModel.getParameterInfos();
final List<GrParameterInfo> parameterInfos =
ContainerUtil.map(
tableParameterInfos,
new Function<GrTableParameterInfo, GrParameterInfo>() {
public GrParameterInfo fun(GrTableParameterInfo info) {
return info.generateParameterInfo();
}
});
final ThrownExceptionInfo[] exceptionInfos = myExceptionTableModel.getThrownExceptions();
invokeRefactoring(
new GrChangeSignatureProcessor(
myProject,
new GrChangeInfoImpl(
myMethod,
modifier,
returnType == null ? null : CanonicalTypes.createTypeWrapper(returnType),
newName,
parameterInfos,
exceptionInfos,
myDelegateRadioButton.isSelected())));
}
@Override
protected String validateAndCommitData() {
PsiManager manager = PsiManager.getInstance(myProject);
PsiElementFactory factory = JavaPsiFacade.getInstance(manager.getProject()).getElementFactory();
String name = getMethodName();
if (!JavaPsiFacade.getInstance(manager.getProject()).getNameHelper().isIdentifier(name)) {
return RefactoringMessageUtil.getIncorrectIdentifierMessage(name);
}
if (myMethod.canChangeReturnType() == MethodDescriptor.ReadWriteOption.ReadWrite) {
try {
((PsiTypeCodeFragment) myReturnTypeCodeFragment).getType();
} catch (PsiTypeCodeFragment.TypeSyntaxException e) {
myReturnTypeField.requestFocus();
return RefactoringBundle.message(
"changeSignature.wrong.return.type", myReturnTypeCodeFragment.getText());
} catch (PsiTypeCodeFragment.NoTypeException e) {
myReturnTypeField.requestFocus();
return RefactoringBundle.message("changeSignature.no.return.type");
}
}
List<ParameterTableModelItemBase<ParameterInfoImpl>> parameterInfos =
myParametersTableModel.getItems();
final int newParametersNumber = parameterInfos.size();
for (int i = 0; i < newParametersNumber; i++) {
final ParameterTableModelItemBase<ParameterInfoImpl> item = parameterInfos.get(i);
if (!JavaPsiFacade.getInstance(manager.getProject())
.getNameHelper()
.isIdentifier(item.parameter.getName())) {
return RefactoringMessageUtil.getIncorrectIdentifierMessage(item.parameter.getName());
}
final PsiType type;
try {
type = ((PsiTypeCodeFragment) parameterInfos.get(i).typeCodeFragment).getType();
} catch (PsiTypeCodeFragment.TypeSyntaxException e) {
return RefactoringBundle.message(
"changeSignature.wrong.type.for.parameter",
item.typeCodeFragment.getText(),
item.parameter.getName());
} catch (PsiTypeCodeFragment.NoTypeException e) {
return RefactoringBundle.message(
"changeSignature.no.type.for.parameter", item.parameter.getName());
}
item.parameter.setType(type);
if (type instanceof PsiEllipsisType && i != newParametersNumber - 1) {
return RefactoringBundle.message("changeSignature.vararg.not.last");
}
if (item.parameter.oldParameterIndex < 0) {
item.parameter.defaultValue =
ApplicationManager.getApplication()
.runWriteAction(
new Computable<String>() {
@Override
public String compute() {
return JavaCodeStyleManager.getInstance(myProject)
.qualifyClassReferences(item.defaultValueCodeFragment)
.getText();
}
});
String def = item.parameter.defaultValue;
def = def.trim();
if (!(type instanceof PsiEllipsisType)) {
try {
if (!StringUtil.isEmpty(def)) {
factory.createExpressionFromText(def, null);
}
} catch (IncorrectOperationException e) {
return e.getMessage();
}
}
}
}
ThrownExceptionInfo[] exceptionInfos = myExceptionsModel.getThrownExceptions();
PsiTypeCodeFragment[] typeCodeFragments = myExceptionsModel.getTypeCodeFragments();
for (int i = 0; i < exceptionInfos.length; i++) {
ThrownExceptionInfo exceptionInfo = exceptionInfos[i];
PsiTypeCodeFragment typeCodeFragment = typeCodeFragments[i];
try {
PsiType type = typeCodeFragment.getType();
if (!(type instanceof PsiClassType)) {
return RefactoringBundle.message(
"changeSignature.wrong.type.for.exception", typeCodeFragment.getText());
}
PsiClassType throwable =
JavaPsiFacade.getInstance(myProject)
.getElementFactory()
.createTypeByFQClassName("java.lang.Throwable", type.getResolveScope());
if (!throwable.isAssignableFrom(type)) {
return RefactoringBundle.message(
"changeSignature.not.throwable.type", typeCodeFragment.getText());
}
exceptionInfo.setType((PsiClassType) type);
} catch (PsiTypeCodeFragment.TypeSyntaxException e) {
return RefactoringBundle.message(
"changeSignature.wrong.type.for.exception", typeCodeFragment.getText());
} catch (PsiTypeCodeFragment.NoTypeException e) {
return RefactoringBundle.message("changeSignature.no.type.for.exception");
}
}
// warnings
try {
if (myMethod.canChangeReturnType() == MethodDescriptor.ReadWriteOption.ReadWrite) {
if (!RefactoringUtil.isResolvableType(
((PsiTypeCodeFragment) myReturnTypeCodeFragment).getType())) {
if (Messages.showOkCancelDialog(
myProject,
RefactoringBundle.message(
"changeSignature.cannot.resolve.return.type",
myReturnTypeCodeFragment.getText()),
RefactoringBundle.message("changeSignature.refactoring.name"),
Messages.getWarningIcon())
!= 0) {
return EXIT_SILENTLY;
}
}
}
for (ParameterTableModelItemBase<ParameterInfoImpl> item : parameterInfos) {
if (!RefactoringUtil.isResolvableType(
((PsiTypeCodeFragment) item.typeCodeFragment).getType())) {
if (Messages.showOkCancelDialog(
myProject,
RefactoringBundle.message(
"changeSignature.cannot.resolve.parameter.type",
item.typeCodeFragment.getText(),
item.parameter.getName()),
RefactoringBundle.message("changeSignature.refactoring.name"),
Messages.getWarningIcon())
!= 0) {
return EXIT_SILENTLY;
}
}
}
} catch (PsiTypeCodeFragment.IncorrectTypeException ignored) {
}
return null;
}
private boolean validateInputData() {
if (!isGroovyMethodName(getNewName())) {
showErrorHint(message("name.is.wrong", getNewName()));
return false;
}
if (!checkType(myReturnTypeCodeFragment, true)) {
showErrorHint(message("return.type.is.wrong"));
return false;
}
List<GrTableParameterInfo> parameterInfos = myParameterModel.getParameterInfos();
for (int i = 0; i < parameterInfos.size(); i++) {
GrTableParameterInfo info = parameterInfos.get(i);
if (!StringUtil.isJavaIdentifier(info.getName())) {
showErrorHint(message("name.is.wrong", info.getName()));
return false;
}
if (!checkType(info.getTypeFragment(), i == parameterInfos.size() - 1)) {
showErrorHint(message("type.for.parameter.is.incorrect", info.getName()));
return false;
}
String defaultValue = info.getDefaultValue();
final String initializer = info.getDefaultInitializerFragment().getText();
if (info.getOldIndex() < 0
&& defaultValue.trim().length() == 0
&& initializer.trim().length() == 0) {
showErrorHint(message("specify.default.value", info.getName()));
return false;
}
}
ThrownExceptionInfo[] exceptionInfos = myExceptionTableModel.getThrownExceptions();
PsiTypeCodeFragment[] typeCodeFragments = myExceptionTableModel.getTypeCodeFragments();
for (int i = 0; i < exceptionInfos.length; i++) {
ThrownExceptionInfo exceptionInfo = exceptionInfos[i];
PsiTypeCodeFragment typeCodeFragment = typeCodeFragments[i];
try {
PsiType type = typeCodeFragment.getType();
if (!(type instanceof PsiClassType)) {
showErrorHint(
GroovyRefactoringBundle.message(
"changeSignature.wrong.type.for.exception", typeCodeFragment.getText()));
return false;
}
PsiClassType throwable =
JavaPsiFacade.getInstance(myMethod.getProject())
.getElementFactory()
.createTypeByFQClassName("java.lang.Throwable", myMethod.getResolveScope());
if (!throwable.isAssignableFrom(type)) {
showErrorHint(
GroovyRefactoringBundle.message(
"changeSignature.not.throwable.type", typeCodeFragment.getText()));
return false;
}
exceptionInfo.setType((PsiClassType) type);
} catch (PsiTypeCodeFragment.TypeSyntaxException e) {
showErrorHint(
GroovyRefactoringBundle.message(
"changeSignature.wrong.type.for.exception", typeCodeFragment.getText()));
return false;
} catch (PsiTypeCodeFragment.NoTypeException e) {
showErrorHint(GroovyRefactoringBundle.message("changeSignature.no.type.for.exception"));
return false;
}
}
return true;
}