org.jetbrains.kotlin.psi.KtStubbedPsiUtil Maven / Gradle / Ivy
/*
* Copyright 2010-2015 JetBrains s.r.o.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.jetbrains.kotlin.psi;
import com.intellij.psi.PsiElement;
import com.intellij.psi.stubs.StubElement;
import com.intellij.psi.tree.TokenSet;
import com.intellij.psi.util.PsiTreeUtil;
import com.intellij.util.ArrayFactory;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
public final class KtStubbedPsiUtil {
@Nullable
public static KtDeclaration getContainingDeclaration(@NotNull PsiElement element) {
return getPsiOrStubParent(element, KtDeclaration.class, true);
}
@Nullable
public static T getContainingDeclaration(@NotNull PsiElement element, @NotNull Class declarationClass) {
return getPsiOrStubParent(element, declarationClass, true);
}
//TODO: contribute to idea PsiTreeUtil#getPsiOrStubParent
@Nullable
public static T getPsiOrStubParent(
@NotNull PsiElement element,
@NotNull Class declarationClass,
boolean strict
) {
if (!strict && declarationClass.isInstance(element)) {
//noinspection unchecked
return (T) element;
}
if (element instanceof KtElementImplStub) {
StubElement stub = ((KtElementImplStub) element).getStub();
if (stub != null) {
return stub.getParentStubOfType(declarationClass);
}
}
return PsiTreeUtil.getParentOfType(element, declarationClass, strict);
}
@Nullable
public static T getStubOrPsiChild(
@NotNull KtElementImplStub element,
@NotNull TokenSet types,
@NotNull ArrayFactory factory
) {
T[] typeElements = element.getStubOrPsiChildren(types, factory);
if (typeElements.length == 0) {
return null;
}
return typeElements[0];
}
private KtStubbedPsiUtil() {
}
}