Implement new lint API getFields() method

Change-Id: I05f5ca005012feeb11d76ec942711b21f0cab2f0

1 files changed, 16 insertions, 1 deletions

diff --git a/android/src/org/jetbrains/android/inspections/lint/LombokPsiParser.java b/android/src/org/jetbrains/android/inspections/lint/LombokPsiParser.java
index 334a973288f..5be4fdb692e 100644
--- a/android/src/org/jetbrains/android/inspections/lint/LombokPsiParser.java
+++ b/android/src/org/jetbrains/android/inspections/lint/LombokPsiParser.java
@@ -34,7 +34,6 @@ import com.intellij.openapi.util.Computable;
 import com.intellij.psi.*;
 import com.intellij.psi.impl.JavaConstantExpressionEvaluator;
 import com.intellij.psi.search.GlobalSearchScope;
-import com.intellij.psi.util.PsiTreeUtil;
 import com.intellij.psi.util.PsiUtil;
 import com.intellij.psi.util.TypeConversionUtil;
 import com.intellij.util.ArrayUtil;
@@ -955,6 +954,22 @@ public class LombokPsiParser extends JavaParser {
       return Collections.emptyList();
     }
 
+    @NonNull
+    @Override
+    public Iterable<ResolvedField> getFields(boolean includeInherited) {
+      if (myClass != null) {
+        PsiField[] fields = includeInherited ? myClass.getAllFields() : myClass.getFields();
+        if (fields.length > 0) {
+          List<ResolvedField> result = Lists.newArrayListWithExpectedSize(fields.length);
+          for (PsiField field : fields) {
+            result.add(new ResolvedPsiField(field));
+          }
+          return result;
+        }
+      }
+      return Collections.emptyList();
+    }
+
     @Nullable
     @Override
     public ResolvedField getField(@NonNull String name, boolean includeInherited) {