private static String wrapInBlock(PsiStatement body) { if (body instanceof PsiExpressionStatement) { return ((PsiExpressionStatement) body).getExpression().getText(); } final String bodyText = body.getText(); if (!(body instanceof PsiBlockStatement)) { return "{" + bodyText + "}"; } return bodyText; }