[Gedmo to Knp] Add TreeBehaviorRector

config/set/knplabs/doctrine-gedmo-to-knplabs.yaml

@@ -2,3 +2,4 @@
 services:
     Rector\DoctrineGedmoToKnplabs\Rector\Class_\TimestampableBehaviorRector: null
     Rector\DoctrineGedmoToKnplabs\Rector\Class_\SluggableBehaviorRector: null
+    Rector\DoctrineGedmoToKnplabs\Rector\Class_\TreeBehaviorRector: ~

packages/BetterPhpDocParser/src/PhpDocNode/Gedmo/TreeLeftTagValueNode.php

@@ -0,0 +1,21 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\BetterPhpDocParser\PhpDocNode\Gedmo;
+
+use Gedmo\Mapping\Annotation\TreeLeft;
+use Rector\BetterPhpDocParser\PhpDocNode\AbstractTagValueNode;
+
+final class TreeLeftTagValueNode extends AbstractTagValueNode
+{
+    /**
+     * @var string
+     */
+    public const CLASS_NAME = TreeLeft::class;
+
+    public function __toString(): string
+    {
+        return '';
+    }
+}

packages/BetterPhpDocParser/src/PhpDocNode/Gedmo/TreeLevelTagValueNode.php

@@ -0,0 +1,21 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\BetterPhpDocParser\PhpDocNode\Gedmo;
+
+use Gedmo\Mapping\Annotation\TreeLevel;
+use Rector\BetterPhpDocParser\PhpDocNode\AbstractTagValueNode;
+
+final class TreeLevelTagValueNode extends AbstractTagValueNode
+{
+    /**
+     * @var string
+     */
+    public const CLASS_NAME = TreeLevel::class;
+
+    public function __toString(): string
+    {
+        return '';
+    }
+}

packages/BetterPhpDocParser/src/PhpDocNode/Gedmo/TreeParentTagValueNode.php

@@ -0,0 +1,21 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\BetterPhpDocParser\PhpDocNode\Gedmo;
+
+use Gedmo\Mapping\Annotation\TreeParent;
+use Rector\BetterPhpDocParser\PhpDocNode\AbstractTagValueNode;
+
+final class TreeParentTagValueNode extends AbstractTagValueNode
+{
+    /**
+     * @var string
+     */
+    public const CLASS_NAME = TreeParent::class;
+
+    public function __toString(): string
+    {
+        return '';
+    }
+}

packages/BetterPhpDocParser/src/PhpDocNode/Gedmo/TreeRightTagValueNode.php

@@ -0,0 +1,21 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\BetterPhpDocParser\PhpDocNode\Gedmo;
+
+use Gedmo\Mapping\Annotation\TreeRight;
+use Rector\BetterPhpDocParser\PhpDocNode\AbstractTagValueNode;
+
+final class TreeRightTagValueNode extends AbstractTagValueNode
+{
+    /**
+     * @var string
+     */
+    public const CLASS_NAME = TreeRight::class;
+
+    public function __toString(): string
+    {
+        return '';
+    }
+}

packages/BetterPhpDocParser/src/PhpDocNode/Gedmo/TreeRootTagValueNode.php

@@ -0,0 +1,21 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\BetterPhpDocParser\PhpDocNode\Gedmo;
+
+use Gedmo\Mapping\Annotation\TreeRoot;
+use Rector\BetterPhpDocParser\PhpDocNode\AbstractTagValueNode;
+
+final class TreeRootTagValueNode extends AbstractTagValueNode
+{
+    /**
+     * @var string
+     */
+    public const CLASS_NAME = TreeRoot::class;
+
+    public function __toString(): string
+    {
+        return '';
+    }
+}

packages/BetterPhpDocParser/src/PhpDocNode/Gedmo/TreeTagValueNode.php

@@ -0,0 +1,36 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\BetterPhpDocParser\PhpDocNode\Gedmo;
+
+use Gedmo\Mapping\Annotation\Tree;
+use Rector\BetterPhpDocParser\PhpDocNode\AbstractTagValueNode;
+
+final class TreeTagValueNode extends AbstractTagValueNode
+{
+    /**
+     * @var string
+     */
+    public const CLASS_NAME = Tree::class;
+
+    /**
+     * @var string
+     */
+    private $type;
+
+    public function __construct(string $type)
+    {
+        $this->type = $type;
+    }
+
+    public function __toString(): string
+    {
+        return sprintf('(type="%s")', $this->type);
+    }
+
+    public function getType(): string
+    {
+        return $this->type;
+    }
+}

packages/BetterPhpDocParser/src/PhpDocNodeFactory/AbstractBasicPropertyPhpDocNodeFactory.php

@@ -0,0 +1,31 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\BetterPhpDocParser\PhpDocNodeFactory;
+
+use Doctrine\ORM\Mapping\Annotation;
+use PhpParser\Node;
+use PhpParser\Node\Stmt\Property;
+use PHPStan\PhpDocParser\Ast\PhpDoc\PhpDocTagValueNode;
+
+abstract class AbstractBasicPropertyPhpDocNodeFactory extends AbstractPhpDocNodeFactory
+{
+    protected function createFromNode(Node $node): ?PhpDocTagValueNode
+    {
+        if (! $node instanceof Property) {
+            return null;
+        }
+
+        /** @var Annotation|null $annotation */
+        $annotation = $this->nodeAnnotationReader->readPropertyAnnotation($node, $this->getClass());
+        if ($annotation === null) {
+            return null;
+        }
+
+        $getTagValueNodeClass = $this->getTagValueNodeClass();
+        return new $getTagValueNodeClass();
+    }
+
+    abstract protected function getTagValueNodeClass(): string;
+}

packages/BetterPhpDocParser/src/PhpDocNodeFactory/Gedmo/TreeLeftPhpDocNodeFactory.php

@@ -0,0 +1,33 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\BetterPhpDocParser\PhpDocNodeFactory\Gedmo;
+
+use Gedmo\Mapping\Annotation\TreeLeft;
+use PhpParser\Node;
+use PHPStan\PhpDocParser\Ast\PhpDoc\PhpDocTagValueNode;
+use PHPStan\PhpDocParser\Parser\TokenIterator;
+use Rector\BetterPhpDocParser\PhpDocNode\Gedmo\TreeLeftTagValueNode;
+use Rector\BetterPhpDocParser\PhpDocNodeFactory\AbstractBasicPropertyPhpDocNodeFactory;
+
+final class TreeLeftPhpDocNodeFactory extends AbstractBasicPropertyPhpDocNodeFactory
+{
+    public function getClass(): string
+    {
+        return TreeLeft::class;
+    }
+
+    /**
+     * @return TreeLeftTagValueNode|null
+     */
+    public function createFromNodeAndTokens(Node $node, TokenIterator $tokenIterator): ?PhpDocTagValueNode
+    {
+        return $this->createFromNode($node);
+    }
+
+    protected function getTagValueNodeClass(): string
+    {
+        return TreeLeftTagValueNode::class;
+    }
+}

packages/BetterPhpDocParser/src/PhpDocNodeFactory/Gedmo/TreeLevelPhpDocNodeFactory.php

@@ -0,0 +1,33 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\BetterPhpDocParser\PhpDocNodeFactory\Gedmo;
+
+use Gedmo\Mapping\Annotation\TreeLevel;
+use PhpParser\Node;
+use PHPStan\PhpDocParser\Ast\PhpDoc\PhpDocTagValueNode;
+use PHPStan\PhpDocParser\Parser\TokenIterator;
+use Rector\BetterPhpDocParser\PhpDocNode\Gedmo\TreeLevelTagValueNode;
+use Rector\BetterPhpDocParser\PhpDocNodeFactory\AbstractBasicPropertyPhpDocNodeFactory;
+
+final class TreeLevelPhpDocNodeFactory extends AbstractBasicPropertyPhpDocNodeFactory
+{
+    public function getClass(): string
+    {
+        return TreeLevel::class;
+    }
+
+    /**
+     * @return TreeLevelTagValueNode|null
+     */
+    public function createFromNodeAndTokens(Node $node, TokenIterator $tokenIterator): ?PhpDocTagValueNode
+    {
+        return $this->createFromNode($node);
+    }
+
+    protected function getTagValueNodeClass(): string
+    {
+        return TreeLevelTagValueNode::class;
+    }
+}

packages/BetterPhpDocParser/src/PhpDocNodeFactory/Gedmo/TreeParentPhpDocNodeFactory.php

@@ -0,0 +1,33 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\BetterPhpDocParser\PhpDocNodeFactory\Gedmo;
+
+use Gedmo\Mapping\Annotation\TreeParent;
+use PhpParser\Node;
+use PHPStan\PhpDocParser\Ast\PhpDoc\PhpDocTagValueNode;
+use PHPStan\PhpDocParser\Parser\TokenIterator;
+use Rector\BetterPhpDocParser\PhpDocNode\Gedmo\TreeParentTagValueNode;
+use Rector\BetterPhpDocParser\PhpDocNodeFactory\AbstractBasicPropertyPhpDocNodeFactory;
+
+final class TreeParentPhpDocNodeFactory extends AbstractBasicPropertyPhpDocNodeFactory
+{
+    public function getClass(): string
+    {
+        return TreeParent::class;
+    }
+
+    /**
+     * @return TreeParentTagValueNode|null
+     */
+    public function createFromNodeAndTokens(Node $node, TokenIterator $tokenIterator): ?PhpDocTagValueNode
+    {
+        return $this->createFromNode($node);
+    }
+
+    protected function getTagValueNodeClass(): string
+    {
+        return TreeParentTagValueNode::class;
+    }
+}

packages/BetterPhpDocParser/src/PhpDocNodeFactory/Gedmo/TreePhpDocNodeFactory.php

@@ -0,0 +1,39 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\BetterPhpDocParser\PhpDocNodeFactory\Gedmo;
+
+use Gedmo\Mapping\Annotation\Tree;
+use PhpParser\Node;
+use PhpParser\Node\Stmt\Class_;
+use PHPStan\PhpDocParser\Ast\PhpDoc\PhpDocTagValueNode;
+use PHPStan\PhpDocParser\Parser\TokenIterator;
+use Rector\BetterPhpDocParser\PhpDocNode\Gedmo\TreeTagValueNode;
+use Rector\BetterPhpDocParser\PhpDocNodeFactory\AbstractPhpDocNodeFactory;
+
+final class TreePhpDocNodeFactory extends AbstractPhpDocNodeFactory
+{
+    public function getClass(): string
+    {
+        return Tree::class;
+    }
+
+    /**
+     * @return TreeTagValueNode|null
+     */
+    public function createFromNodeAndTokens(Node $node, TokenIterator $tokenIterator): ?PhpDocTagValueNode
+    {
+        if (! $node instanceof Class_) {
+            return null;
+        }
+
+        /** @var Tree|null $tree */
+        $tree = $this->nodeAnnotationReader->readClassAnnotation($node, $this->getClass());
+        if ($tree === null) {
+            return null;
+        }
+
+        return new TreeTagValueNode($tree->type);
+    }
+}

packages/BetterPhpDocParser/src/PhpDocNodeFactory/Gedmo/TreeRightPhpDocNodeFactory.php

@@ -0,0 +1,33 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\BetterPhpDocParser\PhpDocNodeFactory\Gedmo;
+
+use Gedmo\Mapping\Annotation\TreeRight;
+use PhpParser\Node;
+use PHPStan\PhpDocParser\Ast\PhpDoc\PhpDocTagValueNode;
+use PHPStan\PhpDocParser\Parser\TokenIterator;
+use Rector\BetterPhpDocParser\PhpDocNode\Gedmo\TreeRightTagValueNode;
+use Rector\BetterPhpDocParser\PhpDocNodeFactory\AbstractBasicPropertyPhpDocNodeFactory;
+
+final class TreeRightPhpDocNodeFactory extends AbstractBasicPropertyPhpDocNodeFactory
+{
+    public function getClass(): string
+    {
+        return TreeRight::class;
+    }
+
+    /**
+     * @return TreeRightTagValueNode|null
+     */
+    public function createFromNodeAndTokens(Node $node, TokenIterator $tokenIterator): ?PhpDocTagValueNode
+    {
+        return $this->createFromNode($node);
+    }
+
+    protected function getTagValueNodeClass(): string
+    {
+        return TreeRightTagValueNode::class;
+    }
+}

packages/BetterPhpDocParser/src/PhpDocNodeFactory/Gedmo/TreeRootPhpDocNodeFactory.php

@@ -0,0 +1,33 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\BetterPhpDocParser\PhpDocNodeFactory\Gedmo;
+
+use Gedmo\Mapping\Annotation\TreeRoot;
+use PhpParser\Node;
+use PHPStan\PhpDocParser\Ast\PhpDoc\PhpDocTagValueNode;
+use PHPStan\PhpDocParser\Parser\TokenIterator;
+use Rector\BetterPhpDocParser\PhpDocNode\Gedmo\TreeRootTagValueNode;
+use Rector\BetterPhpDocParser\PhpDocNodeFactory\AbstractBasicPropertyPhpDocNodeFactory;
+
+final class TreeRootPhpDocNodeFactory extends AbstractBasicPropertyPhpDocNodeFactory
+{
+    public function getClass(): string
+    {
+        return TreeRoot::class;
+    }
+
+    /**
+     * @return TreeRootTagValueNode|null
+     */
+    public function createFromNodeAndTokens(Node $node, TokenIterator $tokenIterator): ?PhpDocTagValueNode
+    {
+        return $this->createFromNode($node);
+    }
+
+    protected function getTagValueNodeClass(): string
+    {
+        return TreeRootTagValueNode::class;
+    }
+}