@Nullable private static JetCallExpression getContainingCallExpression( JetFunctionLiteralExpression expression) { PsiElement parent = expression.getParent(); if (parent instanceof JetCallExpression) { // f {} return (JetCallExpression) parent; } if (parent instanceof JetValueArgument) { // f ({}) or f(p = {}) JetValueArgument argument = (JetValueArgument) parent; PsiElement argList = argument.getParent(); if (argList == null) return null; PsiElement call = argList.getParent(); if (call instanceof JetCallExpression) { return (JetCallExpression) call; } } return null; }