"""module containing commands for manipulating scene collections."""

from typing import Annotated

import typer
from rich.table import Table

from . import console, validate
from .alias import SubTyperAliasGroup

app = typer.Typer(cls=SubTyperAliasGroup)


@app.callback()
def main():
    """Control scene collections in OBS."""


@app.command('list | ls')
def list_(ctx: typer.Context):
    """List all scene collections."""
    resp = ctx.obj.get_scene_collection_list()

    table = Table(title='Scene Collections', padding=(0, 2))
    table.add_column('Scene Collection Name', justify='left', style='cyan')

    for scene_collection_name in resp.scene_collections:
        table.add_row(scene_collection_name)

    console.out.print(table)


@app.command('current | get')
def current(ctx: typer.Context):
    """Get the current scene collection."""
    resp = ctx.obj.get_scene_collection_list()
    console.out.print(resp.current_scene_collection_name)


@app.command('switch | set')
def switch(
    ctx: typer.Context,
    scene_collection_name: Annotated[
        str, typer.Argument(..., help='Name of the scene collection to switch to')
    ],
):
    """Switch to a scene collection."""
    if not validate.scene_collection_in_scene_collections(ctx, scene_collection_name):
        console.err.print(
            f'Scene collection [yellow]{scene_collection_name}[/yellow] not found.'
        )
        raise typer.Exit(1)

    current_scene_collection = (
        ctx.obj.get_scene_collection_list().current_scene_collection_name
    )
    if scene_collection_name == current_scene_collection:
        console.err.print(
            f'Scene collection [yellow]{scene_collection_name}[/yellow] is already active.'
        )
        raise typer.Exit(1)

    ctx.obj.set_current_scene_collection(scene_collection_name)
    console.out.print(
        f'Switched to scene collection [green]{scene_collection_name}[/green].'
    )


@app.command('create | new')
def create(
    ctx: typer.Context,
    scene_collection_name: Annotated[
        str, typer.Argument(..., help='Name of the scene collection to create')
    ],
):
    """Create a new scene collection."""
    if validate.scene_collection_in_scene_collections(ctx, scene_collection_name):
        console.err.print(
            f'Scene collection [yellow]{scene_collection_name}[/yellow] already exists.'
        )
        raise typer.Exit(1)

    ctx.obj.create_scene_collection(scene_collection_name)
    console.out.print(
        f'Created scene collection [green]{scene_collection_name}[/green].'
    )