import argparse
import asyncio
import json
from pathlib import Path

from .config import Settings
from .jobs import Jobs
from .routing import Dispatcher
from .schemas import OCROptions


async def main(args):
    settings = Settings.from_env()
    if args.command == "probe-alias":
        from .adapters.freellm_alias import FreeLLMAlias

        backend = FreeLLMAlias(settings.freellm_url, settings.freellm_key)
        try:
            print(json.dumps(await backend.probe(), indent=2))
        finally:
            await backend.close()
        return
    from .adapters.rapidocr_cpu import RapidCPU

    cpu = RapidCPU(settings.models_dir)
    await cpu.initialize()
    dispatcher = Dispatcher(cpu, None, settings)
    dispatcher.start()
    jobs = Jobs(dispatcher, settings)
    options = OCROptions(mode="cpu_only", sensitivity=args.sensitivity)
    try:
        path = Path(args.file)
        job = jobs.submit(path.read_bytes(), path.name, options, "cli")
        await job.task
        if not job.result:
            raise RuntimeError("job_failed")
        if args.format == "markdown":
            print(job.result.markdown)
        else:
            print(job.result.model_dump_json(indent=2))
        if job.status != "completed":
            raise SystemExit(1)
    finally:
        await jobs.close()
        await dispatcher.close()


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    sub = parser.add_subparsers(dest="command", required=True)
    run = sub.add_parser("run")
    run.add_argument("file")
    run.add_argument("--sensitivity", choices=["D0", "D1", "D2"], default="D2")
    run.add_argument("--format", choices=["json", "markdown"], default="json")
    sub.add_parser("probe-alias")
    asyncio.run(main(parser.parse_args()))