Spatie\MediaLibrary\MediaRepository::loadMedia()

Complexity

Conditions	4
Paths	3

Size

Total Lines	28
Code Lines	17

Duplication

Lines	0
Ratio	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	MediaRepository::all()	0	4	1
A	MediaRepository::getByModelType()	0	4	1

<?php

namespace Spatie\MediaLibrary;

use Illuminate\Support\Collection;
use Spatie\MediaLibrary\HasMedia\Interfaces\HasMedia;
use Illuminate\Database\Eloquent\Collection as DbCollection;

class MediaRepository
{
    /**
     * @var \Spatie\MediaLibrary\Media
     */
    protected $model;

    /**
     * @param \Spatie\MediaLibrary\Media $model
     */
    public function __construct(Media $model)
    {
        $this->model = $model;
    }

    /**
     * Get all media in the collection.
     *
     * @param HasMedia       $model
     * @param string         $collectionName
     * @param array|callable $filter
     *
     * @return Collection
     */
    public function getCollection(HasMedia $model, string $collectionName, $filter = []) : Collection
    {
        $mediaCollection = $model->loadMedia($collectionName);

        $mediaCollection = $this->applyFilterToMediaCollection($mediaCollection, $filter);

        return collect($mediaCollection);
    }

    /**
     * Apply given filters on media.
     *
     * @param \Illuminate\Support\Collection $media
     * @param array|callable                 $filter
     *
     * @return Collection
     */
    protected function applyFilterToMediaCollection(Collection $media, $filter) : Collection
    {
        if (is_array($filter)) {
            $filter = $this->getDefaultFilterFunction($filter);
        }

        return $media->filter($filter);
    }

    /**
     * Get all media.
     */
    public function all() : DbCollection
    {
        return $this->model->all();
    }

    /*
     * Get all media for the given type.
     */
    public function getByModelType(string $modelType) : DbCollection
    {
        return $this->model->where('model_type', $modelType)->get();
    }

    /*
     * Get media by ids.
     */
    public function getByIds(array $ids) : DbCollection
    {
        return $this->model->whereIn('id', $ids)->get();
    }

    /*
     * Get all media for the given type and collection name.
     */
    public function getByModelTypeAndCollectionName(string $modelType, string $collectionName) : DbCollection
    {
        return $this->model
            ->where('model_type', $modelType)
            ->where('collection_name', $collectionName)
            ->get();
    }

    /*
     * Get all media for the given type and collection name.
     */
    public function getByCollectionName(string $collectionName) : DbCollection
    {
        return $this->model
            ->where('collection_name', $collectionName)
            ->get();
    }

    /**
     * Convert the given array to a filter function.
     *
     * @param $filters
     *
     * @return \Closure
     */
    protected function getDefaultFilterFunction(array $filters)
    {
        return function (Media $media) use ($filters) {
            $customProperties = $media->custom_properties;

            foreach ($filters as $property => $value) {
                if (! isset($customProperties[$property])) {
                    return false;
                }
                if ($customProperties[$property] != $value) {
                    return false;
                }
            }

            return true;
        };
    }
}