void (@NotNull ActionContext context, @NotNull PsiExpressionStatement expressionStatement, @NotNull ModPsiUpdater updater) { String valueToSwitch = expressionStatement.getExpression().getText(); PsiSwitchStatement switchStatement = (PsiSwitchStatement)new CommentTracker().replaceAndRestoreComments( expressionStatement, "switch (" + valueToSwitch + ") {}"); CodeStyleManager.getInstance(context.project()).reformat(switchStatement); PsiCodeBlock body = switchStatement.getBody(); PsiJavaToken rBrace = body == null ? null : body.getRBrace(); if (rBrace != null) { updater.moveTo(rBrace); updater.moveToPrevious('\n'); } }