JetPackageDirective#getPackageNameExpression() by stub

This commit is contained in:
Pavel V. Talanov
2014-04-17 14:44:20 +04:00
parent b7c1054198
commit e42e923d4e
@@ -18,6 +18,7 @@ package org.jetbrains.jet.lang.psi;
import com.intellij.lang.ASTNode; import com.intellij.lang.ASTNode;
import com.intellij.psi.PsiElement; import com.intellij.psi.PsiElement;
import com.intellij.util.ArrayFactory;
import com.intellij.util.containers.ContainerUtil; import com.intellij.util.containers.ContainerUtil;
import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable; import org.jetbrains.annotations.Nullable;
@@ -45,7 +46,7 @@ public class JetPackageDirective extends JetExpressionImplStub<PsiJetPlaceHolder
// This should be either JetSimpleNameExpression, or JetDotQualifiedExpression // This should be either JetSimpleNameExpression, or JetDotQualifiedExpression
@Nullable @Nullable
public JetExpression getPackageNameExpression() { public JetExpression getPackageNameExpression() {
return findChildByClass(JetExpression.class); return JetStubbedPsiUtil.getStubOrPsiChild(this, JetImportDirective.IMPORT_DIRECTIVE_EXPRESSIONS, JetExpression.ARRAY_FACTORY);
} }
@NotNull @NotNull