Refactoring: Extract JetClassOrObject.getOrCreateBody() function

This commit is contained in:
Alexey Sedunov
2015-07-13 15:58:46 +03:00
parent 83e28b360f
commit 3c36795984
4 changed files with 8 additions and 22 deletions
@@ -137,12 +137,7 @@ public abstract class OverrideImplementMethodsHandler : LanguageCodeInsightActio
public fun generateMethods(editor: Editor, classOrObject: JetClassOrObject, selectedElements: List<DescriptorClassMember>) {
runWriteAction {
var body = classOrObject.getBody()
if (body == null) {
val psiFactory = JetPsiFactory(classOrObject)
classOrObject.add(psiFactory.createWhiteSpace())
body = classOrObject.add(psiFactory.createEmptyClassBody()) as JetClassBody
}
val body = classOrObject.getOrCreateBody()
var afterAnchor = findInsertAfterAnchor(editor, body)