GameQuery.php

<?php

namespace Boy132\PlayerCounter\Models;

use App\Models\Allocation;
use App\Models\Egg;
use Boy132\PlayerCounter\Extensions\Query\QueryTypeService;
use Illuminate\Database\Eloquent\Collection;
use Illuminate\Database\Eloquent\Model;
use Illuminate\Database\Eloquent\Relations\BelongsToMany;

/**
 * @property int $id
 * @property string $query_type
 * @property ?int $query_port_offset
 * @property Collection|Egg[] $eggs
 * @property int|null $eggs_count
 */
class GameQuery extends Model
{
    protected $fillable = [
        'query_type',
        'query_port_offset',
    ];

    protected $attributes = [
        'query_port_offset' => null,
    ];

    public function eggs(): BelongsToMany
    {
        return $this->belongsToMany(Egg::class);
    }

    /** @return ?array{hostname: string, map: string, current_players: int, max_players: int, players: ?array<array{id: string, name: string}>} */
    public function runQuery(Allocation $allocation): ?array
    {
        if (!static::canRunQuery($allocation)) {
            return null;
        }

        $ip = static::resolveIp($allocation);
        $ip = is_ipv6($ip) ? '[' . $ip . ']' : $ip;

        /** @var QueryTypeService $service */
        $service = app(QueryTypeService::class);

        $schema = $service->get($this->query_type);
        if (!$schema) {
            return null;
        }

        // Allow schema to provide its own port resolution (e.g. from server variables)
        if (method_exists($schema, 'resolvePort')) {
            $resolvedPort = $schema->resolvePort($allocation);
            if ($resolvedPort !== null) {
                return $schema->process($ip, $resolvedPort);
            }
        }

        return $schema->process($ip, $allocation->port + ($this->query_port_offset ?? 0));
    }

    public static function canRunQuery(?Allocation $allocation): bool
    {
        if (!$allocation) {
            return false;
        }

        $ip = static::resolveIp($allocation);

        return !in_array($ip, ['0.0.0.0', '::']);
    }

    protected static function resolveIp(Allocation $allocation): string
    {
        if (config('player-counter.use_alias') && !empty($allocation->ip_alias)) {
            return $allocation->ip_alias;
        }

        return $allocation->ip;
    }
}