|
| 1 | +<?php |
| 2 | + |
| 3 | +namespace SkoreLabs\JsonApi\Http\Resources; |
| 4 | + |
| 5 | +use Illuminate\Http\Resources\MissingValue; |
| 6 | +use Illuminate\Support\Str; |
| 7 | +use Illuminate\Pagination\AbstractPaginator; |
| 8 | +use Illuminate\Pagination\LengthAwarePaginator; |
| 9 | +use Illuminate\Support\Facades\Auth; |
| 10 | + |
| 11 | +trait CollectsResources |
| 12 | +{ |
| 13 | + /** |
| 14 | + * Map the given collection resource into its individual resources. |
| 15 | + * |
| 16 | + * @param mixed $resource |
| 17 | + * @return mixed |
| 18 | + */ |
| 19 | + protected function collectResource($resource) |
| 20 | + { |
| 21 | + if ($resource instanceof MissingValue) { |
| 22 | + return $resource; |
| 23 | + } |
| 24 | + |
| 25 | + $collects = $this->collects(); |
| 26 | + |
| 27 | + $this->collection = $collects && ! $resource->first() instanceof $collects |
| 28 | + ? $this->getFiltered($resource, $collects) |
| 29 | + : $resource->toBase(); |
| 30 | + |
| 31 | + return $resource instanceof AbstractPaginator |
| 32 | + ? $this->refreshPaginator($resource) |
| 33 | + : $this->collection; |
| 34 | + } |
| 35 | + |
| 36 | + /** |
| 37 | + * Undocumented function |
| 38 | + * |
| 39 | + * @param \Illuminate\Pagination\AbstractPaginator $resource |
| 40 | + * @return void |
| 41 | + */ |
| 42 | + protected function refreshPaginator(AbstractPaginator $resource) |
| 43 | + { |
| 44 | + return (new LengthAwarePaginator( |
| 45 | + $this->collection, |
| 46 | + $this->collection->count(), |
| 47 | + $resource->perPage(), |
| 48 | + $resource->currentPage(), |
| 49 | + $resource->getOptions() |
| 50 | + ))->setPageName($resource->getPageName()); |
| 51 | + } |
| 52 | + |
| 53 | + /** |
| 54 | + * Get resource collection filtered by authorization. |
| 55 | + * |
| 56 | + * @param mixed $resource |
| 57 | + * @param mixed $collects |
| 58 | + * @return \Illuminate\Support\Collection |
| 59 | + */ |
| 60 | + protected function getFiltered($resource, $collects) |
| 61 | + { |
| 62 | + /** @var \Illuminate\Support\Collection $collection */ |
| 63 | + $collection = $resource->map(function ($item) use ($collects) { |
| 64 | + $authorize = $this->authorize; |
| 65 | + |
| 66 | + if (gettype($this->authorize) !== 'boolean') { |
| 67 | + $authorize = Auth::user()->can('viewAny', class_basename($item)); |
| 68 | + } |
| 69 | + |
| 70 | + return new $collects($item, $authorize); |
| 71 | + }); |
| 72 | + |
| 73 | + return $collection->filter(function (JsonApiResource $item) { |
| 74 | + return ! $item->resource instanceof MissingValue; |
| 75 | + }); |
| 76 | + } |
| 77 | + |
| 78 | + /** |
| 79 | + * Get the resource that this resource collects. |
| 80 | + * |
| 81 | + * @return string|null |
| 82 | + */ |
| 83 | + protected function collects() |
| 84 | + { |
| 85 | + if ($this->collects) { |
| 86 | + return $this->collects; |
| 87 | + } |
| 88 | + |
| 89 | + if (Str::endsWith(class_basename($this), 'Collection') && |
| 90 | + class_exists($class = Str::replaceLast('Collection', '', get_class($this)))) { |
| 91 | + return $class; |
| 92 | + } |
| 93 | + } |
| 94 | + |
| 95 | + /** |
| 96 | + * Get an iterator for the resource collection. |
| 97 | + * |
| 98 | + * @return \ArrayIterator |
| 99 | + */ |
| 100 | + public function getIterator() |
| 101 | + { |
| 102 | + return $this->collection->getIterator(); |
| 103 | + } |
| 104 | +} |
0 commit comments