> */ 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 !== []) { $originalNode = clone $node; $if = new If_($node->cond); $if->stmts = $node->stmts; $this->mirrorComments($if, $node); /** @var ElseIf_ $firstElseIf */ $firstElseIf = \array_shift($node->elseifs); $node->cond = $firstElseIf->cond; $node->stmts = $firstElseIf->stmts; $this->mirrorComments($node, $firstElseIf); $nodesToReturnAfterNode = $this->getStatementsElseIfs($node); if ($originalNode->else instanceof Else_) { $node->else = null; $nodesToReturnAfterNode = \array_merge($nodesToReturnAfterNode, [$originalNode->else]); } return \array_merge([$if, $node], $nodesToReturnAfterNode); } if ($node->else instanceof Else_) { $stmts = $node->else->stmts; $node->else = null; return \array_merge([$node], $stmts); } return null; } /** * @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_ $node */ private function doesLastStatementBreakFlow($node) : bool { $lastStmt = \end($node->stmts); if ($lastStmt instanceof If_ && $lastStmt->else instanceof Else_) { return $this->doesLastStatementBreakFlow($lastStmt); } return !($lastStmt instanceof Return_ || $lastStmt instanceof Throw_ || $lastStmt instanceof Continue_ || $lastStmt instanceof Expression && $lastStmt->expr instanceof Exit_); } }