Skip to content

Commit 0e25693

Browse files
committed
fix cs
1 parent 36ba9ca commit 0e25693

File tree

1 file changed

+15
-12
lines changed

1 file changed

+15
-12
lines changed

src/Rector/AbstractRector.php

Lines changed: 15 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -82,14 +82,10 @@ public function afterTraverse(array $nodes): array
8282

8383
protected function prependNodeAfterNode(Expr $nodeToPrepend, Node $positionNode): void
8484
{
85-
/** @var Node $parentNode */
85+
/** @var Expression $parentNode */
8686
$parentNode = $positionNode->getAttribute(Attribute::PARENT_NODE);
87-
if (! $parentNode instanceof Expression) {
88-
// validate?
89-
return;
90-
}
9187

92-
$expressionToPrepend = new Expression($nodeToPrepend);
88+
$expressionToPrepend = $this->resolveToExpression($nodeToPrepend);
9389

9490
if (isset($this->expressionsToPrependAfter[$parentNode])) {
9591
$this->expressionsToPrependAfter[$parentNode] = array_merge(
@@ -103,14 +99,10 @@ protected function prependNodeAfterNode(Expr $nodeToPrepend, Node $positionNode)
10399

104100
protected function prependNodeBeforeNode(Expr $nodeToPrepend, Node $positionNode): void
105101
{
106-
/** @var Node $parentNode */
102+
/** @var Expression $parentNode */
107103
$parentNode = $positionNode->getAttribute(Attribute::PARENT_NODE);
108-
if (! $parentNode instanceof Expression) {
109-
// validate?
110-
return;
111-
}
112104

113-
$expressionToPrepend = new Expression($nodeToPrepend);
105+
$expressionToPrepend = $this->resolveToExpression($nodeToPrepend);
114106

115107
if (isset($this->expressionsToPrependBefore[$parentNode])) {
116108
$this->expressionsToPrependBefore[$parentNode] = array_merge(
@@ -123,6 +115,8 @@ protected function prependNodeBeforeNode(Expr $nodeToPrepend, Node $positionNode
123115
}
124116

125117
/**
118+
* Adds new nodes before or after particular Expression nodes.
119+
*
126120
* @param Node[] $nodes
127121
* @return Node[] array
128122
*/
@@ -159,4 +153,13 @@ private function prependNodesAfterAndBeforeExpression(array $nodes, Node $node,
159153

160154
return $nodes;
161155
}
156+
157+
private function resolveToExpression(Expr $exprNode): Expression
158+
{
159+
if ($exprNode instanceof Expression) {
160+
return $exprNode;
161+
}
162+
163+
return new Expression($exprNode);
164+
}
162165
}

0 commit comments

Comments
 (0)