import asyncclick as click
from anyio import create_task_group
from pyslobs import StreamingService

from .cli import cli


@cli.group()
def stream():
    """Stream management commands."""


@stream.command()
@click.pass_context
async def start(ctx: click.Context):
    """Start the stream."""

    conn = ctx.obj["connection"]
    ss = StreamingService(conn)

    async def _run():
        model = await ss.get_model()
        active = model.streaming_status != "offline"

        if active:
            conn.close()
            raise click.Abort(click.style("Stream is already active.", fg="red"))

        await ss.toggle_streaming()
        click.echo("Stream started.")
        conn.close()

    async with create_task_group() as tg:
        tg.start_soon(conn.background_processing)
        tg.start_soon(_run)


@stream.command()
@click.pass_context
async def stop(ctx: click.Context):
    """Stop the stream."""

    conn = ctx.obj["connection"]
    ss = StreamingService(conn)

    async def _run():
        model = await ss.get_model()
        active = model.streaming_status != "offline"

        if not active:
            conn.close()
            raise click.Abort(click.style("Stream is already inactive.", fg="red"))

        await ss.toggle_streaming()
        click.echo("Stream stopped.")
        conn.close()

    async with create_task_group() as tg:
        tg.start_soon(conn.background_processing)
        tg.start_soon(_run)


@stream.command()
@click.pass_context
async def status(ctx: click.Context):
    """Get the current stream status."""

    conn = ctx.obj["connection"]
    ss = StreamingService(conn)

    async def _run():
        model = await ss.get_model()
        active = model.streaming_status != "offline"

        if active:
            click.echo("Stream is currently active.")
        else:
            click.echo("Stream is currently inactive.")
        conn.close()

    async with create_task_group() as tg:
        tg.start_soon(conn.background_processing)
        tg.start_soon(_run)


@stream.command()
@click.pass_context
async def toggle(ctx: click.Context):
    """Toggle the stream status."""

    conn = ctx.obj["connection"]
    ss = StreamingService(conn)

    async def _run():
        model = await ss.get_model()
        active = model.streaming_status != "offline"

        await ss.toggle_streaming()
        if active:
            click.echo("Stream stopped.")
        else:
            click.echo("Stream started.")

        conn.close()

    async with create_task_group() as tg:
        tg.start_soon(conn.background_processing)
        tg.start_soon(_run)