ModelGenerator.php

<?php

namespace App\Support;

use Blueprint\Models\Model;
use Illuminate\Support\Str;
use Blueprint\Generators\ModelGenerator as BaseModelGenerator;

class ModelGenerator extends BaseModelGenerator
{
    protected function populateStub(string $stub, Model $model)
    {
        $stub = parent::populateStub($stub, $model);

        if ($model->name() === 'User') {
            $stub = Str::replaceFirst('extends Model', 'extends \Illuminate\Foundation\Auth\User', $stub);
        }

        if ($model->name() === 'Team') {
            $stub = Str::replaceFirst('extends Model', 'extends \Laravel\Jetstream\Team', $stub);
        }

        return $stub;
    }

    protected function addTraits(Model $model, $stub)
    {
        $stub = parent::addTraits($model, $stub);

        $behaviour = "{$model->name()}Behaviour";

        if (trait_exists("App\Models\\$behaviour")) {
            $stub = Str::replaceFirst('use HasFactory', "use HasFactory, $behaviour", $stub);
        }

        return $stub;
    }
}