Don't postpone assigning created body

This commit is contained in:
Nikolay Krasko
2012-11-09 20:59:01 +04:00
parent f7b491042e
commit 5117bf8118
@@ -69,9 +69,10 @@ public abstract class OverrideImplementMethodsHandler implements LanguageCodeIns
List<DescriptorClassMember> selectedElements
) {
JetClassBody body = classOrObject.getBody();
Project project = classOrObject.getProject();
if (body == null) {
body = JetPsiFactory.createEmptyClassBody(project);
Project project = classOrObject.getProject();
classOrObject.add(JetPsiFactory.createWhiteSpace(project));
body = (JetClassBody) classOrObject.add(JetPsiFactory.createEmptyClassBody(project));
}
PsiElement afterAnchor = findInsertAfterAnchor(editor, body);
@@ -87,12 +88,8 @@ public abstract class OverrideImplementMethodsHandler implements LanguageCodeIns
afterAnchor = added;
elementsToCompact.add((JetElement) added);
}
ReferenceToClassesShortening.compactReferenceToClasses(elementsToCompact);
if (!body.isPhysical()) {
classOrObject.add(JetPsiFactory.createWhiteSpace(project));
classOrObject.add(body);
}
ReferenceToClassesShortening.compactReferenceToClasses(elementsToCompact);
}
@Nullable