void (@NotNull Project project, @NotNull PsiElement element, @NotNull ModPsiUpdater updater) { PsiIfStatement ifStatement = PsiTreeUtil.getParentOfType(element, PsiIfStatement.class, false); if (ifStatement == null) return; PsiElementFactory factory = JavaPsiFacade.getElementFactory(project); if (!(ifStatement.getParent() instanceof PsiCodeBlock)) { ifStatement = BlockUtils.expandSingleStatementToBlockStatement(ifStatement); } PsiStatement[] thenStatements = unwrap(ifStatement.getThenBranch()); PsiStatement[] elseStatements = unwrap(ifStatement.getElseBranch()); if (tryApplyThenElseFix(ifStatement, factory, thenStatements, elseStatements)) return; applyImplicitElseFix(ifStatement, thenStatements, elseStatements); }