void (@NotNull Mover mover, @NotNull ReturnContext context, boolean removeReturn, @NotNull ModPsiUpdater updater) { for (PsiElement anchor : mover.insertBefore) { PsiElement added = anchor.getParent().addBefore(context.returnStatement, anchor); updater.highlight(added); } mover.replaceInline.forEach(e -> { if (e instanceof PsiBreakStatement breakStatement) { replaceStatementKeepComments(breakStatement, context.returnStatement, updater); } else if (e instanceof PsiAssignmentExpression assignment) { inlineAssignment(assignment, context.returnStatement, updater); } }); mover.removeCompletely.forEach(e -> removeElementKeepComments(e)); if (removeReturn) { removeReturn(context); } }