pixelfed/app/Services/AvatarService.php

<?php

namespace App\Services;

use App\Avatar;
use App\Jobs\AvatarPipeline\AvatarStorageLargePurge;
use App\Profile;
use Cache;
use Exception;
use Illuminate\Support\Str;
use League\Flysystem\UnableToCheckDirectoryExistence;
use League\Flysystem\UnableToRetrieveMetadata;
use Storage;

class AvatarService
{
    public static function get($profile_id)
    {
        $exists = Cache::get('avatar:'.$profile_id);
        if ($exists) {
            return $exists;
        }

        $profile = Profile::find($profile_id);
        if (! $profile) {
            return config('app.url').'/storage/avatars/default.jpg';
        }

        return $profile->avatarUrl();
    }

    public static function disk()
    {
        $storage = [
            'cloud' => boolval(config_cache('pixelfed.cloud_storage')),
            'local' => boolval(config_cache('federation.avatars.store_local')),
        ];

        if (! $storage['cloud'] && ! $storage['local']) {
            return false;
        }

        $driver = $storage['cloud'] == false ? 'local' : config('filesystems.cloud');
        $disk = Storage::disk($driver);

        return $disk;
    }

    public static function storage(Avatar $avatar)
    {
        $disk = self::disk();

        if (! $disk) {
            return;
        }

        $storage = [
            'cloud' => boolval(config_cache('pixelfed.cloud_storage')),
            'local' => boolval(config_cache('federation.avatars.store_local')),
        ];

        $base = ($storage['cloud'] == false ? 'public/cache/' : 'cache/').'avatars/';

        return $disk->allFiles($base.$avatar->profile_id);
    }

    public static function cleanup($avatar, $confirm = false)
    {
        if (! $avatar || ! $confirm) {
            return;
        }

        if ($avatar->cdn_url == null) {
            return;
        }

        $storage = [
            'cloud' => boolval(config_cache('pixelfed.cloud_storage')),
            'local' => boolval(config_cache('federation.avatars.store_local')),
        ];

        if (! $storage['cloud'] && ! $storage['local']) {
            return;
        }

        $disk = self::disk();

        if (! $disk) {
            return;
        }

        $base = ($storage['cloud'] == false ? 'public/cache/' : 'cache/').'avatars/';

        try {
            $exists = $disk->directoryExists($base.$avatar->profile_id);
        } catch (
            UnableToRetrieveMetadata|
            UnableToCheckDirectoryExistence|
            Exception $e
        ) {
            return;
        }

        if (! $exists) {
            return;
        }

        $files = collect($disk->allFiles($base.$avatar->profile_id));

        if (! $files || ! $files->count() || $files->count() === 1) {
            return;
        }

        if ($files->count() > 5) {
            AvatarStorageLargePurge::dispatch($avatar)->onQueue('mmo');

            return;
        }

        $curFile = Str::of($avatar->cdn_url)->explode('/')->last();

        $files = $files->filter(function ($f) use ($curFile) {
            return ! $curFile || ! str_ends_with($f, $curFile);
        })->each(function ($name) use ($disk) {
            $disk->delete($name);
        });

    }
}