chore: add uses replcaement in make:model command

swooletw · Sep 20, 2024 · 1632601 · 1632601
1 parent 75466dc
commit 1632601
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/src/devtool/src/Generator/ModelCommand.php b/src/devtool/src/Generator/ModelCommand.php
@@ -53,6 +53,18 @@ public function execute(InputInterface $input, OutputInterface $output): int
         return 0;
     }
 
+    /**
+     * Replace the class name for the given stub.
+     */
+    protected function replaceClass(string $stub, string $name): string
+    {
+        $stub = parent::replaceClass($stub, $name);
+
+        $uses = $this->getConfig()['uses'] ?? \SwooleTW\Hyperf\Database\Eloquent\Model::class;
+
+        return str_replace('%USES%', $uses, $stub);
+    }
+
     protected function getStub(): string
     {
         return $this->getConfig()['stub'] ?? __DIR__ . '/stubs/model.stub';

diff --git a/src/devtool/src/Generator/stubs/model.stub b/src/devtool/src/Generator/stubs/model.stub
@@ -4,7 +4,7 @@ declare(strict_types=1);
 
 namespace %NAMESPACE%;
 
-use SwooleTW\Hyperf\Database\Eloquent\Model;
+use %USES%;
 
 class %CLASS% extends Model
 {