"""module for managing studio mode in Slobs CLI."""

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

from . import console
from .cli import cli
from .errors import SlobsCliError


@cli.group()
def studiomode():
    """Manage studio mode in Slobs CLI."""


@studiomode.command()
@click.pass_context
async def enable(ctx: click.Context):
    """Enable studio mode."""
    conn = ctx.obj['connection']
    ts = TransitionsService(conn)

    async def _run():
        model = await ts.get_model()
        if model.studio_mode:
            conn.close()
            raise SlobsCliError('Studio mode is already enabled.')

        await ts.enable_studio_mode()
        console.out.print('Studio mode enabled successfully.')
        conn.close()

    try:
        async with create_task_group() as tg:
            tg.start_soon(conn.background_processing)
            tg.start_soon(_run)
    except* SlobsCliError as excgroup:
        raisable = next(iter(excgroup.exceptions))
        raise raisable


@studiomode.command()
@click.pass_context
async def disable(ctx: click.Context):
    """Disable studio mode."""
    conn = ctx.obj['connection']
    ts = TransitionsService(conn)

    async def _run():
        model = await ts.get_model()
        if not model.studio_mode:
            conn.close()
            raise SlobsCliError('Studio mode is already disabled.')

        await ts.disable_studio_mode()
        console.out.print('Studio mode disabled successfully.')
        conn.close()

    try:
        async with create_task_group() as tg:
            tg.start_soon(conn.background_processing)
            tg.start_soon(_run)
    except* SlobsCliError as excgroup:
        raisable = next(iter(excgroup.exceptions))
        raise raisable


@studiomode.command()
@click.pass_context
async def status(ctx: click.Context):
    """Check the status of studio mode."""
    conn = ctx.obj['connection']
    ts = TransitionsService(conn)

    async def _run():
        model = await ts.get_model()
        if model.studio_mode:
            console.out.print('Studio mode is currently enabled.')
        else:
            console.out.print('Studio mode is currently disabled.')
        conn.close()

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


@studiomode.command()
@click.pass_context
async def toggle(ctx: click.Context):
    """Toggle studio mode."""
    conn = ctx.obj['connection']
    ts = TransitionsService(conn)

    async def _run():
        model = await ts.get_model()
        if model.studio_mode:
            await ts.disable_studio_mode()
            console.out.print('Studio mode disabled successfully.')
        else:
            await ts.enable_studio_mode()
            console.out.print('Studio mode enabled successfully.')
        conn.close()

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


@studiomode.command()
@click.pass_context
async def force_transition(ctx: click.Context):
    """Force a transition in studio mode."""
    conn = ctx.obj['connection']
    ts = TransitionsService(conn)

    async def _run():
        model = await ts.get_model()
        if not model.studio_mode:
            conn.close()
            raise SlobsCliError('Studio mode is not enabled.')

        await ts.execute_studio_mode_transition()
        console.out.print('Forced studio mode transition.')
        conn.close()

    try:
        async with create_task_group() as tg:
            tg.start_soon(conn.background_processing)
            tg.start_soon(_run)
    except* SlobsCliError as excgroup:
        raisable = next(iter(excgroup.exceptions))
        raise raisable