Fixed problem when JetTypeElement was replaced by JetTypeReference, resulting in invalid PSI structure.

This commit is contained in:
Evgeny Gerashchenko
2012-05-03 19:12:13 +04:00
parent d228dc5f02
commit daecc45d69
@@ -90,7 +90,10 @@ public class ReferenceToClassesShortening {
parent = parent.getContainingDeclaration(); parent = parent.getContainingDeclaration();
} }
JetTypeArgumentList typeArgumentList = userType.getTypeArgumentList(); JetTypeArgumentList typeArgumentList = userType.getTypeArgumentList();
userType.replace(JetPsiFactory.createType(userType.getProject(), name + (typeArgumentList == null ? "" : typeArgumentList.getText()))); JetTypeElement typeElement = JetPsiFactory.createType(userType.getProject(),
name + (typeArgumentList == null ? "" : typeArgumentList.getText())).getTypeElement();
assert typeElement != null;
userType.replace(typeElement);
} }
}); });
} }