import logging.config

import rich
from celery import Celery
from celery.signals import setup_logging

from app.config import settings

celery_app = Celery("resonance-engine")
celery_app.autodiscover_tasks(["app.pipeline"])

celery_app.conf.update(
    broker_url=settings.redis_url,
    result_backend=None,
    task_serializer="json",
    accept_content=["json"],
    worker_prefetch_multiplier=1,
    task_acks_late=True,
    broker_connection_retry_on_startup=True,
    worker_send_task_events=True,
    task_send_sent_event=True,
)


@setup_logging.connect
def _on_setup_logging(**kwargs: object) -> None:  # noqa: ARG001
    if settings.logging_debug:
        rich.reconfigure(force_terminal=True)

    logging.config.dictConfig(settings.logging_config)