pixelfed/app/Services/InstanceService.php

<?php

namespace App\Services;

use Cache;
use App\Instance;
use App\Util\Blurhash\Blurhash;
use App\Services\ConfigCacheService;

class InstanceService
{
    const CACHE_KEY_BY_DOMAIN = 'pf:services:instance:by_domain:';
	const CACHE_KEY_BANNED_DOMAINS = 'instances:banned:domains';
	const CACHE_KEY_UNLISTED_DOMAINS = 'instances:unlisted:domains';
	const CACHE_KEY_NSFW_DOMAINS = 'instances:auto_cw:domains';
	const CACHE_KEY_STATS = 'pf:services:instances:stats';
	const CACHE_KEY_BANNER_BLURHASH = 'pf:services:instance:header-blurhash:v1';

	public function __construct()
	{
		ini_set('memory_limit', config('pixelfed.memory_limit', '1024M'));
	}

	public static function getByDomain($domain)
	{
		return Cache::remember(self::CACHE_KEY_BY_DOMAIN.$domain, 3600, function() use($domain) {
			return Instance::whereDomain($domain)->first();
		});
	}

	public static function getBannedDomains()
	{
		return Cache::remember(self::CACHE_KEY_BANNED_DOMAINS, 1209600, function() {
			return Instance::whereBanned(true)->pluck('domain')->toArray();
		});
	}

	public static function getUnlistedDomains()
	{
		return Cache::remember(self::CACHE_KEY_UNLISTED_DOMAINS, 1209600, function() {
			return Instance::whereUnlisted(true)->pluck('domain')->toArray();
		});
	}

	public static function getNsfwDomains()
	{
		return Cache::remember(self::CACHE_KEY_NSFW_DOMAINS, 1209600, function() {
			return Instance::whereAutoCw(true)->pluck('domain')->toArray();
		});
	}

	public static function software($domain)
	{
		$key = 'instances:software:' . strtolower($domain);
		return Cache::remember($key, 86400, function() use($domain) {
			$instance = Instance::whereDomain($domain)->first();
			if(!$instance) {
				return;
			}
			return $instance->software;
		});
	}

	public static function stats()
	{
		return Cache::remember(self::CACHE_KEY_STATS, 86400, function() {
			return [
				'total_count' => Instance::count(),
				'new_count' => Instance::where('created_at', '>', now()->subDays(14))->count(),
				'banned_count' => Instance::whereBanned(true)->count(),
				'nsfw_count' => Instance::whereAutoCw(true)->count()
			];
		});
	}

    public static function refresh()
    {
        Cache::forget(self::CACHE_KEY_BANNED_DOMAINS);
        Cache::forget(self::CACHE_KEY_UNLISTED_DOMAINS);
        Cache::forget(self::CACHE_KEY_NSFW_DOMAINS);
        Cache::forget(self::CACHE_KEY_STATS);

        self::getBannedDomains();
        self::getUnlistedDomains();
        self::getNsfwDomains();

        return true;
    }

    public static function headerBlurhash()
    {
    	return Cache::rememberForever(self::CACHE_KEY_BANNER_BLURHASH, function() {
    		if(str_ends_with(config_cache('app.banner_image'), 'headers/default.jpg')) {
    			return 'UzJR]l{wHZRjM}R%XRkCH?X9xaWEjZj]kAjt';
    		}
    		$cached = config_cache('instance.banner.blurhash');

    		if($cached && $cached !== 'UzJR]l{wHZRjM}R%XRkCH?X9xaWEjZj]kAjt') {
    			return $cached;
    		}

			$file = config_cache('app.banner_image') ?? url(Storage::url('public/headers/default.jpg'));

			$image = imagecreatefromstring(file_get_contents($file));
			if(!$image) {
				return 'UzJR]l{wHZRjM}R%XRkCH?X9xaWEjZj]kAjt';
			}
			$width = imagesx($image);
			$height = imagesy($image);

			$pixels = [];
			for ($y = 0; $y < $height; ++$y) {
				$row = [];
				for ($x = 0; $x < $width; ++$x) {
					$index = imagecolorat($image, $x, $y);
					$colors = imagecolorsforindex($image, $index);

					$row[] = [$colors['red'], $colors['green'], $colors['blue']];
				}
				$pixels[] = $row;
			}

			$components_x = 4;
			$components_y = 4;
			$blurhash = Blurhash::encode($pixels, $components_x, $components_y);
			if(strlen($blurhash) > 191) {
				return 'UzJR]l{wHZRjM}R%XRkCH?X9xaWEjZj]kAjt';
			}

			ConfigCacheService::put('instance.banner.blurhash', $blurhash);

			return $blurhash;
    	});
    }
}