Support setup command traits before running (#6577)

Author: huangdijia

src/Command.php

@@ -23,6 +23,8 @@
 
 use function Hyperf\Collection\collect;
 use function Hyperf\Coroutine\run;
+use function Hyperf\Support\class_basename;
+use function Hyperf\Support\class_uses_recursive;
 use function Hyperf\Support\swoole_hook_flags;
 use function Hyperf\Tappable\tap;
 
@@ -99,6 +101,8 @@ public function run(InputInterface $input, OutputInterface $output): int
     {
         $this->output = new SymfonyStyle($input, $output);
 
+        $this->setUpTraits();
+
         return parent::run($this->input = $input, $this->output);
     }
 
@@ -203,4 +207,20 @@ protected function execute(InputInterface $input, OutputInterface $output): int
 
         return $this->exitCode >= 0 && $this->exitCode <= 255 ? $this->exitCode : self::INVALID;
     }
+
+    /**
+     * Setup traits of command.
+     */
+    protected function setUpTraits(): array
+    {
+        $uses = array_flip(class_uses_recursive(static::class));
+
+        foreach ($uses as $trait) {
+            if (method_exists($this, $method = 'setUp' . class_basename($trait))) {
+                $this->{$method}($this->input, $this->output);
+            }
+        }
+
+        return $uses;
+    }
 }

tests/Command/FooTraitCommand.php

@@ -0,0 +1,26 @@
+<?php
+
+declare(strict_types=1);
+/**
+ * This file is part of Hyperf.
+ *
+ * @link     https://www.hyperf.io
+ * @document https://hyperf.wiki
+ * @contact  group@hyperf.io
+ * @license  https://github.com/hyperf/hyperf/blob/master/LICENSE
+ */
+namespace HyperfTest\Command\Command;
+
+class FooTraitCommand extends \Hyperf\Command\Command
+{
+    use Traits\Foo;
+
+    public function __construct(string $name = null)
+    {
+        parent::__construct($name);
+    }
+
+    public function handle()
+    {
+    }
+}

tests/Command/Traits/Foo.php

@@ -0,0 +1,25 @@
+<?php
+
+declare(strict_types=1);
+/**
+ * This file is part of Hyperf.
+ *
+ * @link     https://www.hyperf.io
+ * @document https://hyperf.wiki
+ * @contact  group@hyperf.io
+ * @license  https://github.com/hyperf/hyperf/blob/master/LICENSE
+ */
+namespace HyperfTest\Command\Command\Traits;
+
+use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Output\OutputInterface;
+
+trait Foo
+{
+    private ?string $propertyFoo = null;
+
+    protected function setUpFoo(?InputInterface $input, ?OutputInterface $output)
+    {
+        $this->propertyFoo = 'foo';
+    }
+}

tests/CommandTest.php

@@ -17,6 +17,7 @@
 use HyperfTest\Command\Command\FooCommand;
 use HyperfTest\Command\Command\FooExceptionCommand;
 use HyperfTest\Command\Command\FooExitCommand;
+use HyperfTest\Command\Command\FooTraitCommand;
 use HyperfTest\Command\Command\SwooleFlagsCommand;
 use Mockery;
 use PHPUnit\Framework\Attributes\CoversNothing;
@@ -72,6 +73,10 @@ public function testExitCodeWhenThrowException()
         $command = new ClassInvoker(new FooCommand());
         $exitCode = $command->execute($input, $output);
         $this->assertSame(0, $exitCode);
+
+        $command = new FooTraitCommand();
+        $this->assertArrayHasKey(\HyperfTest\Command\Command\Traits\Foo::class, (fn () => $this->setUpTraits())->call($command));
+        $this->assertSame('foo', (fn () => $this->propertyFoo)->call($command));
     }
 
     public function testExitCodeWhenThrowExceptionInCoroutine()