From 2236bfa74ad6fd39b2ed88e08d1fe2b598ba9541 Mon Sep 17 00:00:00 2001
From: Erik Eelde <erik.eelde@gmail.com>
Date: Mon, 20 Jun 2022 16:29:51 +0200
Subject: [PATCH] Fix failing tests

---
 .../checks/RetrofitReturnTypeDetector.kt      | 27 ++++++++++++++-----
 1 file changed, 20 insertions(+), 7 deletions(-)

diff --git a/checks/src/main/java/com/kozaxinan/android/checks/RetrofitReturnTypeDetector.kt b/checks/src/main/java/com/kozaxinan/android/checks/RetrofitReturnTypeDetector.kt
index 93b91cb..a54bfc1 100644
--- a/checks/src/main/java/com/kozaxinan/android/checks/RetrofitReturnTypeDetector.kt
+++ b/checks/src/main/java/com/kozaxinan/android/checks/RetrofitReturnTypeDetector.kt
@@ -84,13 +84,19 @@ internal abstract class RetrofitReturnTypeDetector : Detector(), UastScanner {
                 .isNotEmpty()
         }
 
-        private fun findAllInnerFields(typeRef: PsiClassType): Set<UField> {
+        private fun findAllInnerFields(typeRef: PsiClassType, visitedTypes: MutableSet<PsiClassType> = mutableSetOf()): Set<UField> {
             val actualReturnType = findGenericClassType(typeRef)
             val typeClass = actualReturnType
                 .resolve()
                 .toUElement() as? UClass
                 ?: return emptySet()
 
+
+            if (visitedTypes.contains(actualReturnType)) {
+                return setOf()
+            }
+            visitedTypes.add(actualReturnType)
+
             val innerFields: Set<UField> = typeClass
                 .fields
                 .filterNot { it.isStatic && it !is PsiEnumConstant }
@@ -101,7 +107,8 @@ internal abstract class RetrofitReturnTypeDetector : Detector(), UastScanner {
                         .filterNot { it.isStatic }
                         .map { it.type }
                         .filterIsInstance<PsiClassType>()
-                        .map(::findAllInnerFields)
+                        .filterNot { visitedTypes.contains(it) }
+                        .map { innerTypeRef -> findAllInnerFields(innerTypeRef, visitedTypes) }
                         .flatten()
                         .toSet()
         }
@@ -113,15 +120,21 @@ internal abstract class RetrofitReturnTypeDetector : Detector(), UastScanner {
             return if (substitutor == PsiSubstitutor.EMPTY) {
                 returnType
             } else {
-                when (val psiType: PsiType = substitutor.substitutionMap.values.first()) {
+                when (val psiType: PsiType? = substitutor.substitutionMap.values.first()) {
                     is PsiClassReferenceType -> findGenericClassType(psiType)
                     is PsiWildcardType -> {
-                        when (val superBound: PsiType = psiType.superBound) {
-                            is PsiClassType -> findGenericClassType(superBound)
-                            else -> returnType
+                        when {
+                            psiType.isSuper -> {
+                                findGenericClassType(psiType.superBound as PsiClassType)
+                            }
+                            psiType.isExtends -> {
+                                findGenericClassType(psiType.extendsBound as PsiClassType)
+                            }
+                            else -> {
+                                returnType
+                            }
                         }
                     }
-
                     else -> returnType
                 }
             }