int (PsiElement element) { if (element == null) { return 0; } final PsiElement parent = element.getParent(); if (parent instanceof PsiVariable variable) { final PsiType type = variable.getType(); return type.getArrayDimensions(); } else if (parent instanceof PsiMethod method) { final PsiType type = method.getReturnType(); return (type == null) ? 0 : type.getArrayDimensions(); } else if (element instanceof PsiTypeElement typeElement) { final PsiType type = typeElement.getType(); return type.getArrayDimensions(); } return 0; }