[PHP 8.1] Keep use stmts in MyCLabsClassToEnumRector enum (#2696)

rules-tests/Php81/Rector/Class_/MyCLabsClassToEnumRector/Fixture/keep_used_trait.php.inc

@@ -0,0 +1,39 @@
+<?php
+
+namespace Rector\Tests\Php81\Rector\Class_\MyCLabsClassToEnumRector\Fixture;
+
+use MyCLabs\Enum\Enum;
+use Rector\Tests\Php81\Rector\Class_\MyCLabsClassToEnumRector\Source\ComparingTrait;
+
+final class KeepUsedTrait extends Enum
+{
+    use ComparingTrait;
+
+    /**
+     * Some comment
+     */
+    private const VIEW = 'view';
+
+    private const EDIT = 'edit';
+}
+
+?>
+-----
+<?php
+
+namespace Rector\Tests\Php81\Rector\Class_\MyCLabsClassToEnumRector\Fixture;
+
+use MyCLabs\Enum\Enum;
+use Rector\Tests\Php81\Rector\Class_\MyCLabsClassToEnumRector\Source\ComparingTrait;
+
+enum KeepUsedTrait : string
+{
+    use ComparingTrait;
+    /**
+     * Some comment
+     */
+    case VIEW = 'view';
+    case EDIT = 'edit';
+}
+
+?>

rules-tests/Php81/Rector/Class_/MyCLabsClassToEnumRector/Source/ComparingTrait.php

@@ -0,0 +1,18 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Rector\Tests\Php81\Rector\Class_\MyCLabsClassToEnumRector\Source;
+
+use MyCLabs\Enum\Enum;
+
+trait ComparingTrait
+{
+    /**
+     * @param Enum
+     */
+    public function equalsEnum(self $other): bool
+    {
+        return $other->getValue() === $this->getValue();
+    }
+}

rules/Php81/NodeFactory/EnumFactory.php

@@ -35,6 +35,8 @@ public function createFromClass(Class_ $class): Enum_
 
         $constants = $class->getConstants();
 
+        $enum->stmts = $class->getTraitUses();
+
         if ($constants !== []) {
             $value = $this->valueResolver->getValue($constants[0]->consts[0]->value);
             $enum->scalarType = is_string($value)