BookStack/app/Api/ListingResponseBuilder.php

164 lines
4.4 KiB
PHP
Raw Normal View History

2021-06-26 23:23:15 +08:00
<?php
namespace BookStack\Api;
2019-12-28 22:58:07 +08:00
use BookStack\Model;
2019-12-28 22:58:07 +08:00
use Illuminate\Database\Eloquent\Builder;
use Illuminate\Database\Eloquent\Collection;
use Illuminate\Http\JsonResponse;
use Illuminate\Http\Request;
2019-12-28 22:58:07 +08:00
class ListingResponseBuilder
{
protected $query;
protected $request;
2019-12-28 22:58:07 +08:00
protected $fields;
/**
* @var array<callable>
*/
protected $resultModifiers = [];
2019-12-28 22:58:07 +08:00
protected $filterOperators = [
'eq' => '=',
'ne' => '!=',
'gt' => '>',
'lt' => '<',
'gte' => '>=',
'lte' => '<=',
2021-06-26 23:23:15 +08:00
'like' => 'like',
];
2019-12-28 22:58:07 +08:00
/**
* ListingResponseBuilder constructor.
* The given fields will be forced visible within the model results.
2019-12-28 22:58:07 +08:00
*/
public function __construct(Builder $query, Request $request, array $fields)
2019-12-28 22:58:07 +08:00
{
$this->query = $query;
$this->request = $request;
2019-12-28 22:58:07 +08:00
$this->fields = $fields;
}
/**
* Get the response from this builder.
*/
public function toResponse(): JsonResponse
2019-12-28 22:58:07 +08:00
{
$filteredQuery = $this->filterQuery($this->query);
$total = $filteredQuery->count();
2022-02-08 23:29:58 +08:00
$data = $this->fetchData($filteredQuery)->each(function ($model) {
foreach ($this->resultModifiers as $modifier) {
$modifier($model);
}
});
2019-12-28 22:58:07 +08:00
return response()->json([
2021-06-26 23:23:15 +08:00
'data' => $data,
2019-12-28 22:58:07 +08:00
'total' => $total,
]);
}
/**
2022-02-08 23:29:58 +08:00
* Add a callback to modify each element of the results.
*
* @param (callable(Model)) $modifier
*/
public function modifyResults($modifier): void
{
$this->resultModifiers[] = $modifier;
}
/**
* Fetch the data to return within the response.
2019-12-28 22:58:07 +08:00
*/
protected function fetchData(Builder $query): Collection
2019-12-28 22:58:07 +08:00
{
$query = $this->countAndOffsetQuery($query);
$query = $this->sortQuery($query);
2021-06-26 23:23:15 +08:00
return $query->get($this->fields);
2019-12-28 22:58:07 +08:00
}
/**
* Apply any filtering operations found in the request.
*/
protected function filterQuery(Builder $query): Builder
{
$query = clone $query;
$requestFilters = $this->request->get('filter', []);
if (!is_array($requestFilters)) {
return $query;
}
$queryFilters = collect($requestFilters)->map(function ($value, $key) {
return $this->requestFilterToQueryFilter($key, $value);
})->filter(function ($value) {
return !is_null($value);
})->values()->toArray();
return $query->where($queryFilters);
}
/**
* Convert a request filter query key/value pair into a [field, op, value] where condition.
*/
protected function requestFilterToQueryFilter($fieldKey, $value): ?array
{
$splitKey = explode(':', $fieldKey);
$field = $splitKey[0];
$filterOperator = $splitKey[1] ?? 'eq';
if (!in_array($field, $this->fields)) {
return null;
}
if (!in_array($filterOperator, array_keys($this->filterOperators))) {
$filterOperator = 'eq';
}
$queryOperator = $this->filterOperators[$filterOperator];
2021-06-26 23:23:15 +08:00
return [$field, $queryOperator, $value];
}
2019-12-28 22:58:07 +08:00
/**
* Apply sorting operations to the query from given parameters
* otherwise falling back to the first given field, ascending.
*/
protected function sortQuery(Builder $query): Builder
2019-12-28 22:58:07 +08:00
{
$query = clone $query;
2019-12-28 22:58:07 +08:00
$defaultSortName = $this->fields[0];
$direction = 'asc';
$sort = $this->request->get('sort', '');
2019-12-28 22:58:07 +08:00
if (strpos($sort, '-') === 0) {
$direction = 'desc';
}
$sortName = ltrim($sort, '+- ');
if (!in_array($sortName, $this->fields)) {
$sortName = $defaultSortName;
}
return $query->orderBy($sortName, $direction);
2019-12-28 22:58:07 +08:00
}
/**
* Apply count and offset for paging, based on params from the request while falling
* back to system defined default, taking the max limit into account.
*/
protected function countAndOffsetQuery(Builder $query): Builder
2019-12-28 22:58:07 +08:00
{
$query = clone $query;
$offset = max(0, $this->request->get('offset', 0));
2019-12-28 22:58:07 +08:00
$maxCount = config('api.max_item_count');
$count = $this->request->get('count', config('api.default_item_count'));
2019-12-28 22:58:07 +08:00
$count = max(min($maxCount, $count), 1);
return $query->skip($offset)->take($count);
2019-12-28 22:58:07 +08:00
}
}