<?php

namespace BeyondCode\LaravelWebSockets\WebSockets\Channels;

use ReflectionClass;
use Ratchet\ConnectionInterface;

class ChannelManager
{
    /** @var array */
    protected $channels = [];

    /** @var string */
    protected $appId;

    public function findOrCreate(string $appId, string $channelId): Channel
    {
        if (!isset($this->channels[$appId][$channelId])) {
            $this->channels[$appId][$channelId] = (new ReflectionClass($this->detectChannelClass($channelId)))
                ->newInstance($channelId);
        }

        return $this->channels[$appId][$channelId];
    }

    public function find(string $appId, string $channelId): ?Channel
    {
        return $this->channels[$appId][$channelId] ?? null;
    }

    protected function detectChannelClass(string $channelId): string
    {
        if (starts_with($channelId, 'private-')) {
            return PrivateChannel::class;
        }

        if (starts_with($channelId, 'presence-')) {
            return PresenceChannel::class;
        }
        return Channel::class;
    }

    public function getChannels(string $appId): array
    {
        return $this->channels[$appId] ?? [];
    }

    public function removeFromAllChannels(ConnectionInterface $connection)
    {
        if (! isset($connection->client)) {
            return;
        }

        collect($this->channels[$connection->client->appId])->each->unsubscribe($connection);

        collect($this->channels[$connection->client->appId])
            ->reject->hasConnections()
            ->each(function (Channel $channel, string $channelId) use ($connection) {
                unset($this->channels[$connection->client->appId][$channelId]);
            });

        if (count($this->channels[$connection->client->appId]) === 0) {
            unset($this->channels[$connection->client->appId]);
        };
    }
}