is KtForExpression ->
                parent.loopRange == child

            // While, DoWhile loops use their conditions, not their bodies
            is KtWhileExpressionBase ->
                parent.condition == child

            // Return expressions use the return value
            is KtReturnExpression ->
                parent.returnedExpression == child

    private fun KtType.nonErrorTypeOrNull(): KtType? = takeUnless { it is KtErrorType }

    private fun PsiElement.isWhileLoopCondition() =
        unwrapQualified<KtWhileExpressionBase> { whileExpr, cond -> whileExpr.condition == cond } != null

    private fun PsiElement.isIfCondition() =
        unwrapQualified<KtIfExpression> { ifExpr, cond -> ifExpr.condition == cond } != null