package multiplexer_test

import (
	"context"
	"testing"
	"time"

	"git.netflux.io/rob/termstream/container"
	"git.netflux.io/rob/termstream/mediaserver"
	"git.netflux.io/rob/termstream/multiplexer"
	"git.netflux.io/rob/termstream/testhelpers"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

const component = "multiplexer"

func TestMultiplexer(t *testing.T) {
	ctx, cancel := context.WithCancel(context.Background())
	t.Cleanup(cancel)

	logger := testhelpers.NewTestLogger()
	containerClient, err := container.NewClient(ctx, logger)
	require.NoError(t, err)
	t.Cleanup(func() { require.NoError(t, containerClient.Close()) })

	running, err := containerClient.ContainerRunning(ctx, map[string]string{"component": component})
	require.NoError(t, err)
	assert.False(t, running)

	srv := mediaserver.StartActor(ctx, mediaserver.StartActorParams{
		RTMPPort:                  1936,
		APIPort:                   9998,
		FetchIngressStateInterval: 250 * time.Millisecond,
		ContainerClient:           containerClient,
		ChanSize:                  1,
		Logger:                    logger,
	})
	defer srv.Close()
	testhelpers.ChanDiscard(srv.C())

	time.Sleep(2 * time.Second)
	testhelpers.StreamFLV(t, srv.State().URL)

	require.Eventually(
		t,
		func() bool { return srv.State().Live },
		time.Second*10,
		time.Second,
		"source not live",
	)

	mp := multiplexer.NewActor(ctx, multiplexer.NewActorParams{
		SourceURL:       srv.State().URL,
		ChanSize:        1,
		ContainerClient: containerClient,
		Logger:          logger,
	})
	defer mp.Close()
	testhelpers.ChanDiscard(mp.C())

	requireListeners(t, srv, 0)

	mp.ToggleDestination("rtmp://localhost:1936/destination/test1")
	mp.ToggleDestination("rtmp://localhost:1936/destination/test2")
	mp.ToggleDestination("rtmp://localhost:1936/destination/test3")
	requireListeners(t, srv, 3)

	mp.ToggleDestination("rtmp://localhost:1936/destination/test3")
	requireListeners(t, srv, 2)

	mp.ToggleDestination("rtmp://localhost:1936/destination/test2")
	mp.ToggleDestination("rtmp://localhost:1936/destination/test1")
	requireListeners(t, srv, 0)
}

func requireListeners(t *testing.T, srv *mediaserver.Actor, expected int) {
	require.Eventually(
		t,
		func() bool { return srv.State().Listeners == expected },
		time.Second*10,
		time.Second,
		"expected %d listeners", expected,
	)
}