from django.db.models.signals import post_save
from django.dispatch import receiver
from django.utils.functional import LazyObject

from common.decorators import on_transaction_commit
from common.utils import get_logger
from common.utils.connection import RedisPubSub
from ..const import TaskNameType
from ..models import Task, Session
from ..utils import DBPortManager

db_port_manager: DBPortManager

logger = get_logger(__file__)


class ComponentEventChan(LazyObject):
    def _setup(self):
        self._wrapped = RedisPubSub('fm.component_event_chan')


component_event_chan = ComponentEventChan()


@receiver(post_save, sender=Task)
@on_transaction_commit
def on_task_created(sender, instance: Task, created, **kwargs):
    if not created and instance.is_finished:
        # 当组件完成 task 时，修改 session 的 lock 状态
        session_id = instance.args
        name = instance.name
        if name == TaskNameType.lock_session:
            Session.lock_session(session_id)
        elif name == TaskNameType.unlock_session:
            Session.unlock_session(session_id)
        logger.debug(f"signal task post save: {instance.name}, "
                     f"session: {session_id}, "
                     f"is_finished: {instance.is_finished}")
        return
    event = {
        "type": instance.name,
        "payload": {
            "id": str(instance.id),
        },
    }
    component_event_chan.publish(event)