理解Rector中如何获取MethodCall的父节点

在Rector项目中，开发者经常需要处理AST(抽象语法树)中的节点关系。一个常见需求是获取某个方法调用(MethodCall)节点的父节点，以便进行更复杂的代码重构操作。

问题背景

当我们需要检查一个方法调用是否已经被某种断言(如assertTrue)包裹时，就需要访问该节点的父节点。例如，在测试代码中，我们可能希望确保所有save()方法调用都被assertTrue()包裹。

解决方案分析

在Rector中直接通过getAttribute('parent')获取父节点可能不会返回预期结果。这是因为Rector处理节点的方式需要更精确的节点类型定义。

正确的方法

1. 注册正确的节点类型：不应该只注册MethodCall::class，而应该注册Expression::class，因为方法调用通常作为表达式的一部分存在。

2. 检查表达式内容：在refactor()方法中，首先检查表达式的内容是否是我们感兴趣的方法调用。

3. 重构逻辑：确认表达式内容后，再进行相应的重构操作。

实现示例

public function getNodeTypes(): array
{
    return [Expression::class];
}

public function refactor(Node $node): ?Node
{
    if (!$node->expr instanceof MethodCall) {
        return null;
    }

    $methodCall = $node->expr;
    
    if (!$this->isName($methodCall->name, 'save')) {
        return null;
    }

    // 检查是否已经被assertTrue包裹
    if ($this->isAssertTrueCall($node)) {
        return null;
    }

    // 创建新的assertTrue调用
    $newAssertCall = new StaticCall(
        new Name('self'),
        'assertTrue',
        [new Arg($methodCall)]
    );

    // 返回新的表达式节点
    return new Expression($newAssertCall);
}

关键点说明

1. 节点类型选择：选择Expression而非MethodCall可以让我们访问到完整的表达式上下文。

2. 节点检查顺序：先检查节点类型，再检查方法名，最后检查是否已被处理。

3. 重构安全性：确保不重复处理已经被正确包裹的方法调用。

最佳实践

1. 在Rector规则开发中，选择正确的节点类型作为入口点非常重要。

2. 对于方法调用相关的重构，通常需要检查表达式级别的节点而非直接的方法调用节点。

3. 在重构前总是先检查是否已经符合目标状态，避免无限循环或不必要的修改。

通过这种方式，我们可以更可靠地处理AST节点间的关系，实现复杂的代码重构需求。