<?php

namespace App\Services\Media;

use Spatie\MediaLibrary\HasMedia\HasMediaTrait as BasicHasMedia;

trait HasMediaTrait
{
    use BasicHasMedia;

    public function main_image()
    {
        return $this->morphOne(Media::class, 'model')->where('collection_name', 'main_image');
    }

    public function main_images()
    {
        return $this->morphMany(Media::class, 'model')->where('collection_name', 'main_images');
    }
}