> */ public function getNodeTypes() : array { return [If_::class]; } /** * @param If_ $node * @return Node[]|null */ public function refactor(Node $node) : ?array { if ($this->doesLastStatementBreakFlow($node)) { return null; } if ($node->elseifs !== []) { return $this->handleElseIfs($node); } if ($node->else instanceof Else_) { $stmts = $node->else->stmts; $node->else = null; return \array_merge([$node], $stmts); } return null; } /** * @return Node[] */ private function handleElseIfs(If_ $if) : array { $nodesToReturn = []; $originalIf = clone $if; $firstIf = $this->createIfFromNode($if); $nodesToReturn[] = $firstIf; while ($if->elseifs !== []) { /** @var ElseIf_ $currentElseIf */ $currentElseIf = \array_shift($if->elseifs); // If the last statement in the `elseif` breaks flow, merge it into the original `if` and stop processing if ($this->doesLastStatementBreakFlow($currentElseIf)) { $this->updateIfWithElseIf($if, $currentElseIf); $nodesToReturn = \array_merge(\is_array($nodesToReturn) ? $nodesToReturn : \iterator_to_array($nodesToReturn), [$if], $this->getStatementsElseIfs($if)); break; } $isLastElseIf = $if->elseifs === []; // If it's the last `elseif`, merge it with the original `if` to keep the formatting if ($isLastElseIf) { $this->updateIfWithElseIf($if, $currentElseIf); $nodesToReturn[] = $if; break; } // Otherwise, create a separate `if` node for `elseif` $nodesToReturn[] = $this->createIfFromNode($currentElseIf); } if ($originalIf->else instanceof Else_) { $nodesToReturn[] = $originalIf->else; } return $nodesToReturn; } /** * @return ElseIf_[] */ private function getStatementsElseIfs(If_ $if) : array { $statements = []; foreach ($if->elseifs as $key => $elseif) { if ($this->doesLastStatementBreakFlow($elseif) && $elseif->stmts !== []) { continue; } $statements[] = $elseif; unset($if->elseifs[$key]); } return $statements; } /** * @param \PhpParser\Node\Stmt\If_|\PhpParser\Node\Stmt\ElseIf_|\PhpParser\Node\Stmt\Else_ $node */ private function doesLastStatementBreakFlow($node) : bool { $lastStmt = \end($node->stmts); if ($lastStmt instanceof If_ && $lastStmt->else instanceof Else_) { if ($this->doesLastStatementBreakFlow($lastStmt) || $this->doesLastStatementBreakFlow($lastStmt->else)) { return \true; } foreach ($lastStmt->elseifs as $elseIf) { if ($this->doesLastStatementBreakFlow($elseIf)) { return \true; } } return \false; } return !($lastStmt instanceof Return_ || $lastStmt instanceof Throw_ || $lastStmt instanceof Continue_ || $lastStmt instanceof Expression && $lastStmt->expr instanceof Exit_); } /** * @param \PhpParser\Node\Stmt\If_|\PhpParser\Node\Stmt\ElseIf_ $node */ private function createIfFromNode($node) : If_ { $if = new If_($node->cond); $if->stmts = $node->stmts; $this->mirrorComments($if, $node); return $if; } private function updateIfWithElseIf(If_ $if, ElseIf_ $elseIf) : void { $if->cond = $elseIf->cond; $if->stmts = $elseIf->stmts; $this->mirrorComments($if, $elseIf); $if->else = null; } }