Support subject variable in specialized code generators for 'when'

2018-06-09 17:00:25 +03:00
parent 3528405666
commit 34b76a3718
17 changed files with 427 additions and 20 deletions
@@ -293,6 +293,16 @@ object WhenChecker {
        }
    }

+    fun whenSubjectTypeWithoutSmartCasts(expression: KtWhenExpression, context: BindingContext): KotlinType? {
+        val subjectVariable = expression.subjectVariable
+        val subjectExpression = expression.subjectExpression
+        return when {
+            subjectVariable != null -> context.get(VARIABLE, subjectVariable)?.type
+            subjectExpression != null -> context.getType(subjectExpression)
+            else -> null
+        }
+    }
+
    @JvmStatic
    fun getEnumMissingCases(
        expression: KtWhenExpression,