osu-web / app / Libraries / Beatmapset / ChangeBeatmapOwners.php

<?php

// Copyright (c) ppy Pty Ltd <[e-mail]>. Licensed under the GNU Affero General Public License v3.0.
// See the LICENCE file in the repository root for full licence text.

declare(strict_types=1);

namespace App\Libraries\Beatmapset;

use App\Exceptions\InvariantException;
use App\Jobs\Notifications\BeatmapOwnerChange;
use App\Models\Beatmap;
use App\Models\BeatmapOwner;
use App\Models\BeatmapsetEvent;
use App\Models\User;
use Ds\Set;

class ChangeBeatmapOwners
{
    private Set $userIds;

    public function __construct(private Beatmap $beatmap, array $newUserIds, private User $source)
    {
        priv_check_user($source, 'BeatmapUpdateOwner', $beatmap->beatmapset)->ensureCan();

        $this->userIds = new Set($newUserIds);

        if ($this->userIds->count() > $GLOBALS['cfg']['osu']['beatmaps']['owners_max']) {
            throw new InvariantException(osu_trans('beatmaps.change_owner.too_many'));
        }

        if ($this->userIds->isEmpty()) {
            throw new InvariantException('user_ids must be specified');
        }
    }

    public function handle(): void
    {
        $currentOwners = new Set($this->beatmap->getOwners()->pluck('user_id'));
        if ($currentOwners->xor($this->userIds)->isEmpty()) {
            return;
        }

        $newUserIds = $this->userIds->diff($currentOwners);

        if (User::whereIn('user_id', $newUserIds->toArray())->default()->withoutBots()->withoutNoProfile()->count() !== $newUserIds->count()) {
            throw new InvariantException('invalid user_id');
        }

        $this->beatmap->getConnection()->transaction(function () {
            $params = array_map(
                fn ($userId) => ['beatmap_id' => $this->beatmap->getKey(), 'user_id' => $userId],
                $this->userIds->toArray()
            );

            $this->beatmap->fill(['user_id' => $this->userIds->first()])->saveOrExplode();
            $this->beatmap->beatmapOwners()->delete();
            BeatmapOwner::insert($params);

            $this->beatmap->refresh();

            $newUsers = $this->beatmap->getOwners()->select('id', 'username')->all();
            $beatmapset = $this->beatmap->beatmapset;
            $firstMapper = $newUsers[0];

            BeatmapsetEvent::log(BeatmapsetEvent::BEATMAP_OWNER_CHANGE, $this->source, $beatmapset, [
                'beatmap_id' => $this->beatmap->getKey(),
                'beatmap_version' => $this->beatmap->version,
                // TODO: mainly for compatibility during dev when switching branches, can be removed after deployed.
                'new_user_id' => $firstMapper['id'],
                'new_user_username' => $firstMapper['username'],
                'new_users' => $newUsers,
            ])->saveOrExplode();

            $beatmapset->update(['eligible_main_rulesets' => null]);
        });

        (new BeatmapOwnerChange($this->beatmap, $this->source))->dispatch();
    }
}