pixelfed/app/Services/NotificationService.php

<?php

namespace App\Services;

use Cache;
use Illuminate\Support\Facades\Redis;
use App\{
	Notification,
	Profile
};
use App\Transformer\Api\NotificationTransformer;
use League\Fractal;
use League\Fractal\Serializer\ArraySerializer;
use League\Fractal\Pagination\IlluminatePaginatorAdapter;

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)
	{
		return Cache::remember(self::EPOCH_CACHE_KEY . $months, 1209600, function() use($months) {
			return Notification::where('created_at', '>', now()->subMonths($months))->first()->id;
		});
	}

	public static function coldGet($id, $start = 0, $stop = 400)
	{
		$stop = $stop > 400 ? 400 : $stop;
		$ids = Notification::where('id', '>', self::getEpochId())
			->where('profile_id', $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(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(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';
		}

		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())
				->limit($stop)
				->pluck('id');
			foreach($ids as $key) {
				self::set($id, $key);
			}
			return 1;
		}
		return 0;
	}
}