package de.plushnikov.intellij.lombok.util;
import com.intellij.psi.CommonClassNames;
import com.intellij.psi.JavaPsiFacade;
import com.intellij.psi.PsiClass;
import com.intellij.psi.PsiClassType;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiField;
import com.intellij.psi.PsiMethod;
import com.intellij.psi.PsiModifier;
import org.jetbrains.annotations.NotNull;
import javax.annotation.Nonnull;
import java.util.ArrayList;
import java.util.Collection;
/**
* @author Plushnikov Michail
*/
public class PsiClassUtil {
/**
* Workaround to get all of original Methods of the psiClass.
* Normal call to psiClass.getMethods() in PsiAugmentProvider is impossible because of incorrect cache implementation of IntelliJ Idea
*
* @param psiClass psiClass to collect all of methods from
* @return all intern methods of the class
*/
@NotNull
public static PsiMethod[] collectClassMethodsIntern(@NotNull PsiClass psiClass) {
Collection<PsiMethod> result = new ArrayList<PsiMethod>();
for (PsiElement psiElement : psiClass.getChildren()) {
if (psiElement instanceof PsiMethod) {
result.add((PsiMethod) psiElement);
}
}
return result.toArray(new PsiMethod[result.size()]);
//return ((PsiClassImpl) psiClass).getStubOrPsiChildren(Constants.METHOD_BIT_SET, PsiMethod.ARRAY_FACTORY);
}
@NotNull
public static PsiMethod[] collectClassConstructorIntern(@NotNull PsiClass psiClass) {
final PsiMethod[] psiMethods = collectClassMethodsIntern(psiClass);
Collection<PsiMethod> classConstructors = new ArrayList<PsiMethod>(3);
for (PsiMethod psiMethod : psiMethods) {
if (psiMethod.isConstructor()) {
classConstructors.add(psiMethod);
}
}
return classConstructors.toArray(new PsiMethod[classConstructors.size()]);
}
@NotNull
public static PsiMethod[] collectClassStaticMethodsIntern(@NotNull PsiClass psiClass) {
final PsiMethod[] psiMethods = collectClassMethodsIntern(psiClass);
Collection<PsiMethod> staticMethods = new ArrayList<PsiMethod>(5);
for (PsiMethod psiMethod : psiMethods) {
if (psiMethod.hasModifierProperty(PsiModifier.STATIC)) {
staticMethods.add(psiMethod);
}
}
return staticMethods.toArray(new PsiMethod[staticMethods.size()]);
}
/**
* Workaround to get all of original Fields of the psiClass.
* Normal call to psiClass.getFields() in PsiAugmentProvider is impossible because of incorrect cache implementation of IntelliJ Idea
*
* @param psiClass psiClass to collect all of methods from
* @return all intern fields of the class
*/
@NotNull
public static PsiField[] collectClassFieldsIntern(@NotNull PsiClass psiClass) {
Collection<PsiField> result = new ArrayList<PsiField>();
for (PsiElement psiElement : psiClass.getChildren()) {
if (psiElement instanceof PsiField) {
result.add((PsiField) psiElement);
}
}
return result.toArray(new PsiField[result.size()]);
//return ((PsiClassImpl) psiClass).getStubOrPsiChildren(Constants.FIELD_BIT_SET, PsiField.ARRAY_FACTORY);
}
public static PsiClassType getClassType(@NotNull PsiClass psiClass) {
return JavaPsiFacade.getElementFactory(psiClass.getProject()).createType(psiClass);
}
public static boolean hasSuperClass(@Nonnull final PsiClass psiClass) {
final PsiClass superClass = psiClass.getSuperClass();
return null != superClass && !CommonClassNames.JAVA_LANG_OBJECT.equals(superClass.getQualifiedName());
}
public static boolean hasMultiArgumentConstructor(@Nonnull final PsiClass psiClass) {
boolean result = false;
final PsiMethod[] definedConstructors = collectClassConstructorIntern(psiClass);
for (PsiMethod psiMethod : definedConstructors) {
if(psiMethod.getParameterList().getParametersCount() > 0) {
result = true;
break;
}
}
return result;
}
}