-
-
Notifications
You must be signed in to change notification settings - Fork 117
/
ClassMethodUpdater.php
71 lines (63 loc) · 2.5 KB
/
ClassMethodUpdater.php
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
<?php
namespace Phpactor\CodeBuilder\Adapter\TolerantParser\Updater;
use Microsoft\PhpParser\Node\ClassMembersNode;
use Microsoft\PhpParser\ClassLike;
use Microsoft\PhpParser\Node\EnumMembers;
use Microsoft\PhpParser\Node\Expression\ObjectCreationExpression;
use Microsoft\PhpParser\Node\InterfaceMembers;
use Microsoft\PhpParser\Node\Statement\ClassDeclaration;
use Microsoft\PhpParser\Node\Statement\EnumDeclaration;
use Microsoft\PhpParser\Node\Statement\TraitDeclaration;
use Microsoft\PhpParser\Node\TraitMembers;
use Phpactor\CodeBuilder\Domain\Renderer;
use Phpactor\CodeBuilder\Domain\Prototype\Method;
use RuntimeException;
use Microsoft\PhpParser\Node;
use Webmozart\Assert\Assert;
/**
* @extends AbstractMethodUpdater<ClassMembersNode|TraitMembers>
*/
class ClassMethodUpdater extends AbstractMethodUpdater
{
public function memberDeclarationsNode(ClassLike|ObjectCreationExpression $classNode): ClassMembersNode|TraitMembers|EnumMembers|InterfaceMembers
{
if ($classNode instanceof ClassDeclaration || $classNode instanceof ObjectCreationExpression) {
$classNode = $classNode->classMembers;
Assert::isInstanceOf($classNode, ClassMembersNode::class);
return $classNode;
}
if ($classNode instanceof TraitDeclaration) {
return $classNode->traitMembers;
}
if ($classNode instanceof EnumDeclaration) {
return $classNode->enumMembers;
}
throw new RuntimeException(sprintf(
'Can not get member declarations for "%s"',
get_class($classNode)
));
}
public function renderMethod(Renderer $renderer, Method $method): string
{
return $renderer->render($method) .
"\n" .
$renderer->render($method->body());
}
/** @return array<Node> */
protected function memberDeclarations(ClassLike|ObjectCreationExpression $classNode): array
{
if ($classNode instanceof ClassDeclaration || $classNode instanceof ObjectCreationExpression) {
return $classNode->classMembers->classMemberDeclarations ?? [];
}
if ($classNode instanceof TraitDeclaration) {
return $classNode->traitMembers->traitMemberDeclarations;
}
if ($classNode instanceof EnumDeclaration) {
return $classNode->enumMembers->enumMemberDeclarations;
}
throw new RuntimeException(sprintf(
'Can not get member declarations for "%s"',
get_class($classNode)
));
}
}