"""项目入口：初始化数据源、加载字典、启动持续消费 worker"""

import asyncio
import logging
import signal
import warnings

from app.core.logger import log

# 屏蔽 asyncmy INSERT IGNORE 产生的 Duplicate entry warnings
warnings.filterwarnings("ignore", message=".*Duplicate entry.*")
logging.getLogger("asyncmy").setLevel(logging.ERROR)

from app.core.database import init_db, close_db
from app.services.dict_cache_service import dict_cache
from app.scheduler.tasks import create_scheduler


async def main():
    log.info("=" * 50)
    log.info("OfferPie Job Cleaner 启动中...")
    log.info("=" * 50)

    # 初始化双数据源
    await init_db()

    # 加载字典缓存
    await dict_cache.refresh()

    # 创建并启动调度器（僵尸恢复、岗位下架等辅助任务）
    scheduler = create_scheduler()
    scheduler.start()
    log.info("调度器已启动，辅助定时任务已注册")

    # 启动持续消费任务
    from app.services.job_clean_service import run_job_clean, stop_job_clean
    from app.services.company_clean_service import run_company_clean, stop_company_clean

    job_clean_task = asyncio.create_task(run_job_clean())
    company_clean_task = asyncio.create_task(run_company_clean())
    log.info("岗位清洗 & 公司补充持续消费任务已启动")

    # 优雅关闭
    stop_event = asyncio.Event()

    def _shutdown(*args):
        log.info("收到关闭信号，正在关闭...")
        stop_job_clean()
        stop_company_clean()
        stop_event.set()

    loop = asyncio.get_running_loop()
    for sig in (signal.SIGINT, signal.SIGTERM):
        try:
            loop.add_signal_handler(sig, _shutdown)
        except (NotImplementedError, ValueError):
            pass

    try:
        await stop_event.wait()
    except KeyboardInterrupt:
        _shutdown()

    # 等待 worker 协程退出
    await asyncio.gather(job_clean_task, company_clean_task, return_exceptions=True)

    scheduler.shutdown(wait=False)
    await close_db()
    log.info("OfferPie Job Cleaner 已关闭")


if __name__ == "__main__":
    asyncio.run(main())