osu-web / app / Transformers / BeatmapPlaycountTransformer.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\BeatmapPlaycount;

class BeatmapPlaycountTransformer extends TransformerAbstract
{
    protected array $defaultIncludes = [
        'beatmap',
        'beatmapset',
    ];

    protected array $availableIncludes = [
        'beatmap',
        'beatmapset',
    ];

    public function transform(BeatmapPlaycount $playcount)
    {
        return [
            'beatmap_id' => $playcount->beatmap_id,
            'count' => $playcount->playcount,
        ];
    }

    public function includeBeatmap(BeatmapPlaycount $playcount)
    {
        if ($playcount->beatmap === null) {
            return;
        }

        return $this->item(
            $playcount->beatmap,
            new BeatmapCompactTransformer()
        );
    }

    public function includeBeatmapset(BeatmapPlaycount $playcount)
    {
        if ($playcount->beatmap === null) {
            return;
        }

        return $this->item(
            $playcount->beatmap->beatmapset,
            new BeatmapsetCompactTransformer()
        );
    }
}