# -*- coding: utf-8 -*-

import asyncio

import typer
from uvicorn import Config, Server

from felicity.almigrator import FelicityMigrator

alembic_service = FelicityMigrator()
app = typer.Typer()


@app.command()
def runserver(host: str = "0.0.0.0", port: int = 8000, workers: int = 1, reload: bool = False, colors: bool = True):
    """Felicity LIMS Server
    """
    config = Config(
        app="felicity.main:felicity",
        host=host,
        port=port,
        workers=workers,
        reload=reload,
        log_level="info",
        use_colors=colors
    )
    Server(config).run()


@app.command()
def upgrade(revision: str = typer.Option("head", help="Target revision to upgrade to")):
    """Upgrade to a specified revision."""
    alembic_service.upgrade(revision)
    typer.echo(f"Upgraded to revision: {revision}")


@app.command()
def downgrade(revision: str = typer.Argument(..., help="Target revision to downgrade to")):
    """Downgrade to a specified revision."""
    alembic_service.downgrade(revision)
    typer.echo(f"Downgraded to revision: {revision}")


@app.command()
def revision(message: str = typer.Argument(..., help="Message for the new revision")):
    """Create a new Alembic revision with a message."""
    alembic_service.create_revision(message)
    typer.echo(f"Created new revision with message: {message}")


@app.command()
def current():
    """Show the current database revision."""
    alembic_service.current()


@app.command()
def history():
    """Show the revision history."""
    alembic_service.history()


@app.command()
def updates():
    """Check for updates"""
    asyncio.run(alembic_service.check_for_updates())


def main(): app()