Improve support for Factories

src/Database/Factories/Factory.php

@@ -0,0 +1,37 @@
+<?php
+
+namespace Winter\Storm\Database\Factories;
+
+use Illuminate\Database\Eloquent\Factories\Factory as BaseFactory;
+use Illuminate\Support\Str;
+
+/**
+ * @template TModel of \Winter\Storm\Database\Model
+ *
+ * @method $this trashed()
+ */
+abstract class Factory extends BaseFactory
+{
+    /**
+     * Get the factory name for the given model name.
+     *
+     * @param  class-string<\Winter\Storm\Database\Model>  $modelName
+     * @return class-string<\Winter\Storm\Database\Factories\Factory>
+     */
+    public static function resolveFactoryName(string $modelName)
+    {
+        if (Str::contains($modelName, 'Models\\')) {
+            $pluginNamespace = trim(Str::before($modelName, 'Models'), '\\');
+            $modelClassName = trim(Str::after($modelName, 'Models'), '\\');
+        } else {
+            $pluginNamespace = '';
+            $modelClassName = $modelName;
+        }
+
+        return trim(implode('\\', [
+            $pluginNamespace,
+            trim(static::$namespace, '\\'),
+            $modelClassName.'Factory'
+        ]), '\\');
+    }
+}

src/Database/Factories/HasFactory.php

@@ -0,0 +1,30 @@
+<?php
+
+namespace Winter\Storm\Database\Factories;
+
+use Illuminate\Database\Eloquent\Factories\Factory as BaseFactory;
+
+trait HasFactory
+{
+    /**
+     * Get a new factory instance for the model.
+     */
+    public static function factory(callable|array|int|null $count = null, callable|array $state = []): BaseFactory
+    {
+        $factory = static::newFactory() ?: Factory::factoryForModel(get_called_class());
+
+        return $factory
+            ->count(is_numeric($count) ? $count : null)
+            ->state(is_callable($count) || is_array($count) ? $count : $state);
+    }
+
+    /**
+     * Create a new factory instance for the model.
+     *
+     * @return \Winter\Storm\Database\Factories\Factory<static>
+     */
+    protected static function newFactory()
+    {
+        //
+    }
+}

tests/Database/Factories/FactoryTest.php

@@ -0,0 +1,18 @@
+<?php
+
+use Winter\Storm\Database\Factories\Factory;
+
+class FactoryTest extends TestCase
+{
+    public function testResolveFactoryName()
+    {
+        $factoryClass = Factory::resolveFactoryName('Plugin\Author\Models\TestModel');
+        $this->assertEquals($factoryClass, 'Plugin\Author\Database\Factories\TestModelFactory');
+
+        $factoryClass = Factory::resolveFactoryName('Models\TestModel');
+        $this->assertEquals($factoryClass, 'Database\Factories\TestModelFactory');
+
+        $factoryClass = Factory::resolveFactoryName('TestModel');
+        $this->assertEquals($factoryClass, 'Database\Factories\TestModelFactory');
+    }
+}