wip

Author: patinthehat

src/Results/Nodes/ClassConstantNode.php

@@ -0,0 +1,25 @@
+<?php
+
+namespace Permafrost\PhpCodeSearch\Results\Nodes;
+
+use Permafrost\PhpCodeSearch\Results\Nodes\Traits\BootsTraits;
+use Permafrost\PhpCodeSearch\Results\Nodes\Traits\HasLocation;
+use Permafrost\PhpCodeSearch\Results\Nodes\Traits\HasName;
+use Permafrost\PhpCodeSearch\Results\Nodes\Traits\HasValue;
+use Permafrost\PhpCodeSearch\Results\Nodes\Traits\HasVisibility;
+use PhpParser\Node;
+
+class ClassConstantNode implements ResultNode
+{
+    use BootsTraits;
+    use HasLocation;
+    use HasName;
+    use HasValue;
+    use HasVisibility;
+
+    public function __construct(Node\Stmt\ClassConst $node)
+    {
+        $this->bootTraits($node);
+        $this->bootHasValue($node->consts[0]);
+    }
+}

src/Results/Nodes/ClassDefinitionNode.php

@@ -5,7 +5,6 @@
 use Permafrost\PhpCodeSearch\Results\Nodes\Traits\BootsTraits;
 use Permafrost\PhpCodeSearch\Results\Nodes\Traits\HasLocation;
 use Permafrost\PhpCodeSearch\Results\Nodes\Traits\HasName;
-use Permafrost\PhpCodeSearch\Support\Collections\Collection;
 use Permafrost\PhpCodeSearch\Support\NameResolver;
 use Permafrost\PhpCodeSearch\Support\StatementTransformer;
 use PhpParser\Node;
@@ -22,10 +21,13 @@ class ClassDefinitionNode implements ResultNode
     /** @var array|ResultNode[]|ValueNode[] */
     public $methods;
 
-    public $implements;
+    public $implements = [];
 
     public $extends;
 
+    /** @var array|ResultNode[]|ValueNode[] */
+    public $constants = [];
+
     public function __construct(Node\Stmt\Class_ $node)
     {
         $this->bootTraits($node);
@@ -34,5 +36,6 @@ public function __construct(Node\Stmt\Class_ $node)
         $this->implements = NameResolver::resolveAll($node->implements);
         $this->properties = StatementTransformer::parserNodesToResultNode($node->getProperties());
         $this->methods = StatementTransformer::parserNodesToResultNode($node->getMethods());
+        $this->constants = StatementTransformer::parserNodesToResultNode($node->getConstants());
     }
 }

src/Results/Nodes/Traits/HasValue.php

@@ -0,0 +1,24 @@
+<?php
+
+namespace Permafrost\PhpCodeSearch\Results\Nodes\Traits;
+
+use Permafrost\PhpCodeSearch\Results\Nodes\Scalar\StringNode;
+use Permafrost\PhpCodeSearch\Support\ExpressionTransformer;
+
+trait HasValue
+{
+    /** @var string|StringNode */
+    public $value;
+
+    public function value()
+    {
+        return $this->value;
+    }
+
+    protected function bootHasValue($node): void
+    {
+        if (property_exists($node, 'value')) {
+            $this->value = ExpressionTransformer::parserNodeToResultNode($node->value);
+        }
+    }
+}

src/Support/NameResolver.php

@@ -24,6 +24,10 @@ public static function resolve($node)
             return $node->props[0]->name;
         }
 
+        if ($node instanceof Node\Stmt\ClassConst) {
+            return $node->consts[0]->name;
+        }
+
         if (self::propertiesExist($node, ['class', 'name'])) {
             $class = static::resolve($node->class);
             $name = static::resolve($node->name);

src/Support/StatementTransformer.php

@@ -2,6 +2,7 @@
 
 namespace Permafrost\PhpCodeSearch\Support;
 
+use Permafrost\PhpCodeSearch\Results\Nodes\ClassConstantNode;
 use Permafrost\PhpCodeSearch\Results\Nodes\ClassMethodNode;
 use Permafrost\PhpCodeSearch\Results\Nodes\ClassPropertyNode;
 use Permafrost\PhpCodeSearch\Results\Nodes\ParameterNode;
@@ -15,6 +16,7 @@ public static function parserNodeToResultNode(Node $node)
             Node\Param::class => ParameterNode::class,
             Node\Stmt\Property::class => ClassPropertyNode::class,
             Node\Stmt\ClassMethod::class => ClassMethodNode::class,
+            Node\Stmt\ClassConst::class => ClassConstantNode::class,
         ];
 
         foreach ($map as $parserNodeClass => $resultNodeClass) {