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.psi.PsiElement;
import com.intellij.util.ArrayFactory;
import com.intellij.util.containers.ContainerUtil;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
@@ -45,7 +46,7 @@ public class JetPackageDirective extends JetExpressionImplStub<PsiJetPlaceHolder
// This should be either JetSimpleNameExpression, or JetDotQualifiedExpression
@Nullable
public JetExpression getPackageNameExpression() {
return findChildByClass(JetExpression.class);
return JetStubbedPsiUtil.getStubOrPsiChild(this, JetImportDirective.IMPORT_DIRECTIVE_EXPRESSIONS, JetExpression.ARRAY_FACTORY);
}
@NotNull