Remove various getDefaultObject() methods

This commit is contained in:
Pavel V. Talanov
2015-03-05 15:58:01 +03:00
parent 101cc9c9ea
commit b73d7e19b7
9 changed files with 17 additions and 29 deletions
@@ -28,6 +28,7 @@ import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiFile;
import com.intellij.psi.util.PsiTreeUtil;
import com.intellij.util.NotNullFunction;
import kotlin.KotlinPackage;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import org.jetbrains.kotlin.descriptors.FunctionDescriptor;
@@ -42,6 +43,8 @@ import org.jetbrains.kotlin.psi.*;
import java.util.List;
import static kotlin.KotlinPackage.singleOrNull;
public class JetRunConfigurationProducer extends RuntimeConfigurationProducer implements Cloneable {
@Nullable
private PsiElement mySourceElement;
@@ -117,7 +120,7 @@ public class JetRunConfigurationProducer extends RuntimeConfigurationProducer im
currentElement = PsiTreeUtil.getParentOfType((PsiElement) currentElement, JetClassOrObject.class, JetFile.class)) {
JetDeclarationContainer entryPointContainer = currentElement;
if (entryPointContainer instanceof JetClass) {
entryPointContainer = ((JetClass) currentElement).getDefaultObject();
entryPointContainer = singleOrNull(((JetClass) currentElement).getDefaultObjects());
}
if (entryPointContainer != null && mainFunctionDetector.hasMain(entryPointContainer.getDeclarations())) return entryPointContainer;
}