void (@NotNull Project project, @NotNull PsiElement element, @NotNull ModPsiUpdater updater) { if (element instanceof PsiKeyword) { element = element.getParent(); } if (element instanceof PsiSwitchLabeledRuleStatement rule) { PsiSwitchBlock switchBlock = rule.getEnclosingSwitchBlock(); PsiStatement body = rule.getBody(); if (switchBlock instanceof PsiSwitchExpression && body instanceof PsiExpressionStatement statement) { wrapExpression(statement); } else if (body != null) { wrapStatement(body); } } }