import argparse
import asyncio
import logging

from app.core.config import get_settings
from app.infrastructure.db import engine
from app.worker.runtime import WorkerRuntime


async def serve(*, once: bool = False) -> None:
    settings = get_settings()
    runtime = WorkerRuntime(settings=settings)
    try:
        while True:
            worked = await runtime.run_once()
            if once:
                return
            if not worked:
                await asyncio.sleep(settings.worker_poll_seconds)
    finally:
        await engine.dispose()


def main() -> None:
    parser = argparse.ArgumentParser(description="Agent 底座独立 Worker")
    parser.add_argument("--once", action="store_true", help="消费一次后退出")
    args = parser.parse_args()
    logging.basicConfig(level=get_settings().log_level)
    try:
        asyncio.run(serve(once=args.once))
    except KeyboardInterrupt:
        pass


if __name__ == "__main__":
    main()