<?php

declare(strict_types=1);

namespace Stancl\Tenancy; // todo new Overrides namespace?

use Illuminate\Broadcasting\Broadcasters\Broadcaster;
use Illuminate\Broadcasting\BroadcastManager;
use Illuminate\Contracts\Broadcasting\Broadcaster as BroadcasterContract;
use Illuminate\Contracts\Foundation\Application;

class TenancyBroadcastManager extends BroadcastManager
{
    /**
     * Names of broadcasters to always recreate using $this->resolve() (even when they're
     * cached and available in the $broadcasters property).
     *
     * The reason for recreating the broadcasters is
     * to make your app use the correct broadcaster credentials when tenancy is initialized.
     */
    public static array $tenantBroadcasters = ['pusher', 'ably'];

    /**
     * Override the get method so that the broadcasters in $tenantBroadcasters
     * always get freshly resolved even when they're cached and available in the $broadcasters property,
     * and that the resolved broadcaster will override the BroadcasterContract::class singleton.
     *
     * If there's a cached broadcaster with the same name as $name,
     * give its channels to the newly resolved bootstrapper.
     */
    protected function get($name)
    {
        if (in_array($name, static::$tenantBroadcasters)) {
            /** @var Broadcaster|null $cachedBroadcaster */
            $cachedBroadcaster = $this->drivers[$name] ?? null;

            /** @var Broadcaster $broadcaster */
            $broadcaster = $this->resolve($name);

            // If there is a cached broadcaster, give its channels to the newly resolved one
            if ($cachedBroadcaster) {
                // invade() because channels can't be retrieved through any of the broadcaster's public methods
                $cachedBroadcaster = invade($cachedBroadcaster);

                foreach ($cachedBroadcaster->channels as $channel => $callback) {
                    $broadcaster->channel($channel, $callback, $cachedBroadcaster->retrieveChannelOptions($channel));
                }
            }

            $this->app->singleton(BroadcasterContract::class, fn (Application $app) => $broadcaster);

            return $broadcaster;
        }

        return parent::get($name);
    }

    public function setDriver(string $name, BroadcasterContract $broadcaster): static
    {
        $this->drivers[$name] = $broadcaster;

        return $this;
    }
}