forked from mirror/pixelfed
325 lines
8.7 KiB
PHP
325 lines
8.7 KiB
PHP
<?php
|
|
|
|
namespace App\Services;
|
|
|
|
use App\Jobs\InternalPipeline\NotificationEpochUpdatePipeline;
|
|
use App\Notification;
|
|
use App\Transformer\Api\NotificationTransformer;
|
|
use Cache;
|
|
use Illuminate\Support\Facades\Redis;
|
|
use League\Fractal;
|
|
use League\Fractal\Serializer\ArraySerializer;
|
|
|
|
class NotificationService
|
|
{
|
|
const CACHE_KEY = 'pf:services:notifications:ids:';
|
|
|
|
const EPOCH_CACHE_KEY = 'pf:services:notifications:epoch-id:by-months:';
|
|
|
|
const ITEM_CACHE_TTL = 86400;
|
|
|
|
const MASTODON_TYPES = [
|
|
'follow',
|
|
'follow_request',
|
|
'mention',
|
|
'reblog',
|
|
'favourite',
|
|
'poll',
|
|
'status',
|
|
];
|
|
|
|
public static function get($id, $start = 0, $stop = 400)
|
|
{
|
|
$res = collect([]);
|
|
$key = self::CACHE_KEY.$id;
|
|
$stop = $stop > 400 ? 400 : $stop;
|
|
$ids = Redis::zrangebyscore($key, $start, $stop);
|
|
if (empty($ids)) {
|
|
$ids = self::coldGet($id, $start, $stop);
|
|
}
|
|
foreach ($ids as $id) {
|
|
$n = self::getNotification($id);
|
|
if ($n != null) {
|
|
$res->push($n);
|
|
}
|
|
}
|
|
|
|
return $res;
|
|
}
|
|
|
|
public static function getEpochId($months = 6)
|
|
{
|
|
$epoch = Cache::get(self::EPOCH_CACHE_KEY.$months);
|
|
if (! $epoch) {
|
|
NotificationEpochUpdatePipeline::dispatch();
|
|
|
|
return 1;
|
|
}
|
|
|
|
return $epoch;
|
|
}
|
|
|
|
public static function coldGet($id, $start = 0, $stop = 400)
|
|
{
|
|
$stop = $stop > 400 ? 400 : $stop;
|
|
$ids = Notification::where('id', '>', self::getEpochId())
|
|
->where('profile_id', $id)
|
|
->orderByDesc('id')
|
|
->skip($start)
|
|
->take($stop)
|
|
->pluck('id');
|
|
foreach ($ids as $key) {
|
|
self::set($id, $key);
|
|
}
|
|
|
|
return $ids;
|
|
}
|
|
|
|
public static function getMax($id = false, $start = 0, $limit = 10)
|
|
{
|
|
$ids = self::getRankedMaxId($id, $start, $limit);
|
|
|
|
if (empty($ids)) {
|
|
return [];
|
|
}
|
|
|
|
$res = collect([]);
|
|
foreach ($ids as $id) {
|
|
$n = self::getNotification($id);
|
|
if ($n != null) {
|
|
$res->push($n);
|
|
}
|
|
}
|
|
|
|
return $res->toArray();
|
|
}
|
|
|
|
public static function getMin($id = false, $start = 0, $limit = 10)
|
|
{
|
|
$ids = self::getRankedMinId($id, $start, $limit);
|
|
|
|
if (empty($ids)) {
|
|
return [];
|
|
}
|
|
|
|
$res = collect([]);
|
|
foreach ($ids as $id) {
|
|
$n = self::getNotification($id);
|
|
if ($n != null) {
|
|
$res->push($n);
|
|
}
|
|
}
|
|
|
|
return $res->toArray();
|
|
}
|
|
|
|
public static function getMaxMastodon($id = false, $start = 0, $limit = 10)
|
|
{
|
|
$ids = self::getRankedMaxId($id, $start, $limit);
|
|
|
|
if (empty($ids)) {
|
|
return [];
|
|
}
|
|
|
|
$res = collect([]);
|
|
foreach ($ids as $id) {
|
|
$n = self::rewriteMastodonTypes(self::getNotification($id));
|
|
if ($n != null && in_array($n['type'], self::MASTODON_TYPES)) {
|
|
if (isset($n['account'])) {
|
|
$n['account'] = AccountService::getMastodon($n['account']['id']);
|
|
}
|
|
|
|
if (isset($n['relationship'])) {
|
|
unset($n['relationship']);
|
|
}
|
|
|
|
if ($n['type'] === 'mention' && isset($n['tagged'], $n['tagged']['status_id'])) {
|
|
$n['status'] = StatusService::getMastodon($n['tagged']['status_id'], false);
|
|
unset($n['tagged']);
|
|
}
|
|
|
|
if (isset($n['status'])) {
|
|
$n['status'] = StatusService::getMastodon($n['status']['id'], false);
|
|
}
|
|
|
|
$res->push($n);
|
|
}
|
|
}
|
|
|
|
return $res->toArray();
|
|
}
|
|
|
|
public static function getMinMastodon($id = false, $start = 0, $limit = 10)
|
|
{
|
|
$ids = self::getRankedMinId($id, $start, $limit);
|
|
|
|
if (empty($ids)) {
|
|
return [];
|
|
}
|
|
|
|
$res = collect([]);
|
|
foreach ($ids as $id) {
|
|
$n = self::rewriteMastodonTypes(self::getNotification($id));
|
|
if ($n != null && in_array($n['type'], self::MASTODON_TYPES)) {
|
|
if (isset($n['account'])) {
|
|
$n['account'] = AccountService::getMastodon($n['account']['id']);
|
|
}
|
|
|
|
if (isset($n['relationship'])) {
|
|
unset($n['relationship']);
|
|
}
|
|
|
|
if ($n['type'] === 'mention' && isset($n['tagged'], $n['tagged']['status_id'])) {
|
|
$n['status'] = StatusService::getMastodon($n['tagged']['status_id'], false);
|
|
unset($n['tagged']);
|
|
}
|
|
|
|
if (isset($n['status'])) {
|
|
$n['status'] = StatusService::getMastodon($n['status']['id'], false);
|
|
}
|
|
|
|
$res->push($n);
|
|
}
|
|
}
|
|
|
|
return $res->toArray();
|
|
}
|
|
|
|
public static function getRankedMaxId($id = false, $start = null, $limit = 10)
|
|
{
|
|
if (! $start || ! $id) {
|
|
return [];
|
|
}
|
|
|
|
return array_keys(Redis::zrevrangebyscore(self::CACHE_KEY.$id, $start, '-inf', [
|
|
'withscores' => true,
|
|
'limit' => [1, $limit],
|
|
]));
|
|
}
|
|
|
|
public static function getRankedMinId($id = false, $end = null, $limit = 10)
|
|
{
|
|
if (! $end || ! $id) {
|
|
return [];
|
|
}
|
|
|
|
return array_keys(Redis::zrevrangebyscore(self::CACHE_KEY.$id, '+inf', $end, [
|
|
'withscores' => true,
|
|
'limit' => [0, $limit],
|
|
]));
|
|
}
|
|
|
|
public static function rewriteMastodonTypes($notification)
|
|
{
|
|
if (! $notification || ! isset($notification['type'])) {
|
|
return $notification;
|
|
}
|
|
|
|
if ($notification['type'] === 'comment') {
|
|
$notification['type'] = 'mention';
|
|
}
|
|
|
|
if ($notification['type'] === 'share') {
|
|
$notification['type'] = 'reblog';
|
|
}
|
|
|
|
if ($notification['type'] === 'tagged') {
|
|
$notification['type'] = 'mention';
|
|
}
|
|
|
|
return $notification;
|
|
}
|
|
|
|
public static function set($id, $val)
|
|
{
|
|
if (self::count($id) > 400) {
|
|
Redis::zpopmin(self::CACHE_KEY.$id);
|
|
}
|
|
|
|
return Redis::zadd(self::CACHE_KEY.$id, $val, $val);
|
|
}
|
|
|
|
public static function del($id, $val)
|
|
{
|
|
Cache::forget('service:notification:'.$val);
|
|
|
|
return Redis::zrem(self::CACHE_KEY.$id, $val);
|
|
}
|
|
|
|
public static function add($id, $val)
|
|
{
|
|
return self::set($id, $val);
|
|
}
|
|
|
|
public static function rem($id, $val)
|
|
{
|
|
return self::del($id, $val);
|
|
}
|
|
|
|
public static function count($id)
|
|
{
|
|
return Redis::zcount(self::CACHE_KEY.$id, '-inf', '+inf');
|
|
}
|
|
|
|
public static function getNotification($id)
|
|
{
|
|
$notification = Cache::remember('service:notification:'.$id, self::ITEM_CACHE_TTL, function () use ($id) {
|
|
$n = Notification::with('item')->find($id);
|
|
|
|
if (! $n) {
|
|
return null;
|
|
}
|
|
|
|
$account = AccountService::get($n->actor_id, true);
|
|
|
|
if (! $account) {
|
|
return null;
|
|
}
|
|
|
|
$fractal = new Fractal\Manager();
|
|
$fractal->setSerializer(new ArraySerializer());
|
|
$resource = new Fractal\Resource\Item($n, new NotificationTransformer());
|
|
|
|
return $fractal->createData($resource)->toArray();
|
|
});
|
|
|
|
if (! $notification) {
|
|
return;
|
|
}
|
|
|
|
if (isset($notification['account'])) {
|
|
$notification['account'] = AccountService::get($notification['account']['id'], true);
|
|
}
|
|
|
|
return $notification;
|
|
}
|
|
|
|
public static function setNotification(Notification $notification)
|
|
{
|
|
return Cache::remember('service:notification:'.$notification->id, self::ITEM_CACHE_TTL, function () use ($notification) {
|
|
$fractal = new Fractal\Manager();
|
|
$fractal->setSerializer(new ArraySerializer());
|
|
$resource = new Fractal\Resource\Item($notification, new NotificationTransformer());
|
|
|
|
return $fractal->createData($resource)->toArray();
|
|
});
|
|
}
|
|
|
|
public static function warmCache($id, $stop = 400, $force = false)
|
|
{
|
|
if (self::count($id) == 0 || $force == true) {
|
|
$ids = Notification::where('profile_id', $id)
|
|
->where('id', '>', self::getEpochId())
|
|
->orderByDesc('id')
|
|
->limit($stop)
|
|
->pluck('id');
|
|
foreach ($ids as $key) {
|
|
self::set($id, $key);
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
}
|