public void doSurround(ActionContext context, PsiElement element, ModPsiUpdater updater) {
    if (element instanceof PsiExpression expression) {
      CodeBlockSurrounder surrounder = CodeBlockSurrounder.forExpression(ExpressionUtils.getTopLevelExpression(expression));
      if (surrounder == null) return;
      element = surrounder.surround().getAnchor();
    } else {
      element = CommonJavaRefactoringUtil.getParentStatement(element, false);
      if (element == null) return;
    }
    surroundStatements(context, element.getParent(), new PsiElement[]{element}, updater);
  }
public void doSurround(ActionContext context, PsiElement element, ModPsiUpdater updater)