import sys
from typing import List, Optional, Union

from engorgio import engorgio
from rich.console import Console
import typer

from learn_sql_model.config import Config, get_config
from learn_sql_model.factories.hero import HeroFactory
from learn_sql_model.factories.pet import PetFactory
from learn_sql_model.models.hero import Hero, HeroCreate

hero_app = typer.Typer()


@hero_app.callback()
def hero():
    "model cli"


@hero_app.command()
@engorgio(typer=True)
def get(
    id: Optional[int] = typer.Argument(default=None),
    config: Config = None,
) -> Union[Hero, List[Hero]]:
    "get one hero"
    config.init()
    hero = Hero().get(id=id)
    Console().print(hero)
    return hero


@hero_app.command()
@engorgio(typer=True)
def list(
    config: Config = None,
) -> Union[Hero, List[Hero]]:
    "get one hero"
    hero = Hero().get()
    Console().print(hero)
    return hero


@hero_app.command()
@engorgio(typer=True)
def create(
    hero: HeroCreate,
    config: Config = None,
) -> Hero:
    "read all the heros"
    config.init()
    hero = hero.post(config=config)
    Console().print(hero)


@hero_app.command()
@engorgio(typer=True)
def populate(
    hero: Hero,
    n: int = 10,
) -> Hero:
    "read all the heros"
    config = get_config()
    if config.env == "prod":
        Console().print("populate is not supported in production")
        sys.exit(1)

    for hero in HeroFactory().batch(n):
        pet = PetFactory().build()
        hero.pet = pet
        Console().print(hero)
        hero.post(config=config)