<?php

namespace FootyRoom\Queries;

use FootyRoom\Http\Models\StageStats;
use FootyRoom\Support\MongoClient;

class StageStatsQuery
{
    /**
     * @var \FootyRoom\Support\MongoClient
     */
    protected $mongo;

    /**
     * Constructor.
     *
     * @param \FootyRoom\Support\MongoClient $mongo
     */
    public function __construct(MongoClient $mongo)
    {
        $this->mongo = $mongo->footyroom;
    }

    /**
     * Find stage stats by stage id.
     *
     * @param int $stageId
     *
     * @return \FootyRoom\Http\Models\StageStats|null
     */
    public function findByStageId($stageId)
    {
        $options = [
            'projection' => [
                '_id' => 0,
                'goalsByPlayer' => 1,
                'assistsByPlayer' => 1,
                'yellowCardsByPlayer' => 1,
                'redCardsByPlayer' => 1,
            ],
        ];

        $statsData = $this->mongo->stages->findOne(['stageId' => $stageId], $options);

        if (!$statsData) {
            return null;
        }

        $stats = new StageStats();

        $empty = true;

        foreach ($statsData as $key => $value) {
            $stats->{$key} = $value;

            if ($empty && count($value) > 0) {
                $empty = false;
            }
        }

        if ($empty) {
            return null;
        }

        return $stats;
    }
}