osu-web / app / Transformers / BeatmapCompactTransformer.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.

namespace App\Transformers;

use App\Models\Beatmap;
use App\Models\BeatmapFailtimes;
use App\Models\DeletedUser;
use App\Models\User;

class BeatmapCompactTransformer extends TransformerAbstract
{
    protected array $availableIncludes = [
        'beatmapset',
        'checksum',
        'failtimes',
        'max_combo',
        'owners',
        'top_tag_ids',
        'user',
    ];

    protected $beatmapsetTransformer = BeatmapsetCompactTransformer::class;

    public function transform(Beatmap $beatmap)
    {
        return [
            'beatmapset_id' => $beatmap->beatmapset_id,
            'difficulty_rating' => $beatmap->difficultyrating,
            'id' => $beatmap->beatmap_id,
            'mode' => $beatmap->mode,
            'status' => $beatmap->status(),
            'total_length' => $beatmap->total_length,
            'user_id' => $beatmap->user_id,
            'version' => $beatmap->version,
        ];
    }

    public function includeBeatmapset(Beatmap $beatmap)
    {
        $beatmapset = $beatmap->beatmapset;

        return $beatmapset === null
            ? $this->primitive(null)
            : $this->item($beatmap->beatmapset, new $this->beatmapsetTransformer());
    }

    public function includeChecksum(Beatmap $beatmap)
    {
        return $this->primitive($beatmap->checksum);
    }

    public function includeFailtimes(Beatmap $beatmap)
    {
        $failtimes = $beatmap->failtimes;

        $result = [];

        foreach ($failtimes as $failtime) {
            $result[$failtime->type] = $failtime->data;
        }

        foreach (['exit', 'fail'] as $type) {
            if (!isset($result[$type])) {
                $result[$type] = (new BeatmapFailtimes())->data;
            }
        }

        return $this->primitive($result);
    }

    public function includeMaxCombo(Beatmap $beatmap)
    {
        return $this->primitive($beatmap->maxCombo());
    }

    public function includeOwners(Beatmap $beatmap)
    {
        return $this->collection($beatmap->getOwners(), fn (User $user) => [
            'id' => $user->getKey(),
            'username' => $user->username,
        ]);
    }

    public function includeTopTagIds(Beatmap $beatmap)
    {
        return $this->primitive($beatmap->topTagIds());
    }

    public function includeUser(Beatmap $beatmap)
    {
        return $this->item(
            $beatmap->user ?? new DeletedUser(),
            new UserCompactTransformer(),
        );
    }
}