pixelfed/app/Services/SearchApiV2Service.php

228 lines
5.1 KiB
PHP
Raw Normal View History

2020-02-07 00:12:20 +00:00
<?php
namespace App\Services;
use Cache;
use Illuminate\Support\Facades\Redis;
use App\{Hashtag, Profile, Status};
use App\Transformer\Api\AccountTransformer;
use App\Transformer\Api\StatusTransformer;
use League\Fractal;
use League\Fractal\Serializer\ArraySerializer;
use League\Fractal\Pagination\IlluminatePaginatorAdapter;
use App\Util\ActivityPub\Helpers;
2020-02-07 08:27:05 +00:00
use Illuminate\Support\Str;
use App\Services\AccountService;
use App\Services\HashtagService;
use App\Services\StatusService;
2020-02-07 00:12:20 +00:00
class SearchApiV2Service
{
private $query;
public static function query($query)
{
return (new self)->run($query);
}
protected function run($query)
{
$this->query = $query;
2020-02-07 08:27:05 +00:00
if($query->has('resolve') &&
$query->resolve == true &&
Helpers::validateUrl(urldecode($query->input('q')))
) {
return $this->resolve();
2020-02-07 00:12:20 +00:00
}
if($query->has('type')) {
switch ($query->input('type')) {
case 'accounts':
return [
2020-02-07 07:26:45 +00:00
'accounts' => $this->accounts(),
2020-02-07 00:12:20 +00:00
'hashtags' => [],
'statuses' => []
];
break;
case 'hashtags':
return [
'accounts' => [],
2020-02-07 07:26:45 +00:00
'hashtags' => $this->hashtags(),
2020-02-07 00:12:20 +00:00
'statuses' => []
];
break;
case 'statuses':
return [
'accounts' => [],
'hashtags' => [],
2020-02-07 07:26:45 +00:00
'statuses' => $this->statuses()
2020-02-07 00:12:20 +00:00
];
break;
}
}
if($query->has('account_id')) {
return [
'accounts' => [],
'hashtags' => [],
'statuses' => $this->statusesById()
];
}
return [
'accounts' => $this->accounts(),
'hashtags' => $this->hashtags(),
'statuses' => $this->statuses()
];
}
protected function resolve()
{
2020-02-07 08:27:05 +00:00
$query = urldecode($this->query->input('q'));
if(Str::startsWith($query, '@') == true) {
return WebfingerService::lookup($this->query->input('q'));
} else if (Str::startsWith($query, 'https://') == true) {
return $this->resolveQuery();
}
2020-02-07 00:12:20 +00:00
}
protected function accounts()
{
$user = request()->user();
$limit = $this->query->input('limit') ?? 20;
$offset = $this->query->input('offset') ?? 0;
2020-02-07 00:12:20 +00:00
$query = '%' . $this->query->input('q') . '%';
$results = Profile::select('profiles.*', 'followers.profile_id', 'followers.created_at')
->whereNull('status')
->leftJoin('followers', function($join) use($user) {
return $join->on('profiles.id', '=', 'followers.following_id')
->where('followers.profile_id', $user->profile_id);
})
2020-02-07 00:12:20 +00:00
->where('username', 'like', $query)
->orderByDesc('profiles.followers_count')
->orderByDesc('followers.created_at')
->offset($offset)
2020-02-07 00:12:20 +00:00
->limit($limit)
->get()
->map(function($res) {
return AccountService::get($res['id']);
});
2020-02-07 00:12:20 +00:00
return $results;
2020-02-07 00:12:20 +00:00
}
protected function hashtags()
{
$limit = $this->query->input('limit') ?? 20;
$offset = $this->query->input('offset') ?? 0;
2020-02-07 00:12:20 +00:00
$query = '%' . $this->query->input('q') . '%';
return Hashtag::whereIsBanned(false)
->where('name', 'like', $query)
->offset($offset)
2020-02-07 00:12:20 +00:00
->limit($limit)
->get()
->map(function($tag) {
return [
'name' => $tag->name,
'url' => $tag->url(),
'count' => HashtagService::count($tag->id),
2020-02-07 00:12:20 +00:00
'history' => []
];
});
}
protected function statuses()
{
// Removed until we provide more relevent sorting/results
return [];
2020-02-07 00:12:20 +00:00
}
protected function statusesById()
{
$accountId = $this->query->input('account_id');
$limit = $this->query->input('limit', 20);
$query = '%' . $this->query->input('q') . '%';
$results = Status::where('caption', 'like', $query)
->whereProfileId($accountId)
->limit($limit)
->get()
->map(function($status) {
return StatusService::get($status->id);
});
return $results;
2020-02-07 00:12:20 +00:00
}
2020-02-07 08:27:05 +00:00
protected function resolveQuery()
{
$query = urldecode($this->query->input('q'));
2020-02-07 08:28:13 +00:00
if(Helpers::validateLocalUrl($query)) {
2020-02-07 08:27:05 +00:00
if(Str::contains($query, '/p/')) {
return $this->resolveLocalStatus();
} else {
return $this->resolveLocalProfile();
}
} else {
return [
'accounts' => [],
'hashtags' => [],
'statuses' => []
];
}
}
protected function resolveLocalStatus()
{
$query = urldecode($this->query->input('q'));
$query = last(explode('/', $query));
$status = Status::whereNull('uri')
->whereScope('public')
->find($query);
if(!$status) {
return [
'accounts' => [],
'hashtags' => [],
'statuses' => []
];
}
$fractal = new Fractal\Manager();
$fractal->setSerializer(new ArraySerializer());
$resource = new Fractal\Resource\Item($status, new StatusTransformer());
return [
'accounts' => [],
'hashtags' => [],
'statuses' => $fractal->createData($resource)->toArray()
];
}
protected function resolveLocalProfile()
{
$query = urldecode($this->query->input('q'));
$query = last(explode('/', $query));
$profile = Profile::whereNull('status')
->whereNull('domain')
->whereUsername($query)
->first();
if(!$profile) {
return [
'accounts' => [],
'hashtags' => [],
'statuses' => []
];
}
$fractal = new Fractal\Manager();
$fractal->setSerializer(new ArraySerializer());
$resource = new Fractal\Resource\Item($profile, new AccountTransformer());
return [
'accounts' => $fractal->createData($resource)->toArray(),
'hashtags' => [],
'statuses' => []
];
}
}