Add Eloquent collection implementation

Author: tobyzerner

src/Laravel/EloquentCollection.php

@@ -0,0 +1,108 @@
+<?php
+
+namespace Tobyz\JsonApiServer\Laravel;
+
+use Illuminate\Database\Query\Builder;
+use Illuminate\Support\Facades\Auth;
+use RuntimeException;
+use Tobyz\JsonApiServer\Context;
+use Tobyz\JsonApiServer\Pagination\OffsetPagination;
+use Tobyz\JsonApiServer\Resource\CollectionInterface;
+use Tobyz\JsonApiServer\Resource\Countable;
+use Tobyz\JsonApiServer\Resource\Listable;
+use Tobyz\JsonApiServer\Resource\Paginatable;
+
+abstract class EloquentCollection implements CollectionInterface, Listable, Paginatable, Countable
+{
+    public function resource(object $model, Context $context): ?string
+    {
+        foreach ($this->eloquentResources($context) as $resource) {
+            $class = $resource->newModel($context);
+
+            if ($model instanceof $class) {
+                return $resource->type();
+            }
+        }
+
+        return null;
+    }
+
+    /**
+     * @return EloquentResource[]
+     */
+    public function eloquentResources(Context $context): array
+    {
+        return array_map(function ($type) use ($context) {
+            $resource = $context->resource($type);
+
+            if (!$resource instanceof EloquentResource) {
+                throw new RuntimeException('must be Eloquent resource');
+            }
+
+            return $resource;
+        }, $this->resources());
+    }
+
+    public function endpoints(): array
+    {
+        return [];
+    }
+
+    public function query(Context $context): object
+    {
+        $queries = array_map(function ($resource) use ($context) {
+            $keyName = $resource->newModel($context)->getQualifiedKeyName();
+            $type = $resource->type();
+
+            $query = $resource
+                ->query($context)
+                ->toBase()
+                ->select("$keyName as id")
+                ->selectRaw('? as type', [$type]);
+
+            $this->scope($query, $type);
+
+            return $query;
+        }, $this->eloquentResources($context));
+
+        return new UnionBuilder($queries);
+    }
+
+    public function scope(Builder $query, string $type): void
+    {
+    }
+
+    public function results(object $query, Context $context): array
+    {
+        $results = $query->get();
+        $types = $results->groupBy('type');
+
+        foreach ($types as $type => $rows) {
+            $model = $context->resource($type)->newModel($context);
+
+            $types[$type] = $model::findMany($rows->pluck('id'));
+        }
+
+        return $results->map(fn($row) => $types[$row->type]->find($row->id))->all();
+    }
+
+    public function filters(): array
+    {
+        return [];
+    }
+
+    public function sorts(): array
+    {
+        return [];
+    }
+
+    public function paginate(object $query, OffsetPagination $pagination): void
+    {
+        $query->take($pagination->limit)->skip($pagination->offset);
+    }
+
+    public function count(object $query, Context $context): ?int
+    {
+        return $query->count();
+    }
+}

src/Laravel/UnionBuilder.php

@@ -0,0 +1,63 @@
+<?php
+
+namespace Tobyz\JsonApiServer\Laravel;
+
+use Illuminate\Contracts\Database\Query\Builder;
+
+class UnionBuilder implements Builder
+{
+    private array $queryCalls = [];
+    private array $outerQueryCalls = [];
+
+    public function __construct(protected array $queries)
+    {
+    }
+
+    public function count($columns = '*'): int
+    {
+        return $this->buildQuery()->count($columns);
+    }
+
+    public function get($columns = ['*'])
+    {
+        return $this->buildQuery()->get($columns);
+    }
+
+    protected function buildQuery()
+    {
+        $queries = array_map(fn($query) => clone $query, $this->queries);
+
+        foreach ($queries as $query) {
+            foreach ($this->queryCalls as $call) {
+                $call($query);
+            }
+        }
+
+        $outerQuery = array_shift($queries);
+
+        foreach ($queries as $query) {
+            $outerQuery->union($query);
+        }
+
+        foreach ($this->outerQueryCalls as $call) {
+            $call($outerQuery);
+        }
+
+        return $outerQuery;
+    }
+
+    public function __call($method, $parameters)
+    {
+        $call = $this->queryCalls[] = fn($query) => $query->$method(...$parameters);
+
+        if ($method === 'orderBy') {
+            $this->queryCalls[] = fn($query) => $query->addSelect($parameters[0]);
+        }
+
+        if (in_array($method, ['take', 'limit', 'skip', 'offset', 'orderBy'])) {
+            $this->outerQueryCalls[] = $call;
+        }
+
+        return $this;
+    }
+}