from __future__ import annotations

from typing import TYPE_CHECKING

from .base import RedisSubscriber

if TYPE_CHECKING:
    from app.router.chat.server import ChatServer


JOIN_CHANNEL = "chat:room:joined"
EXIT_CHANNEL = "chat:room:left"


class ChatSubscriber(RedisSubscriber):
    def __init__(self):
        super().__init__()
        self.room_subscriber: dict[int, list[int]] = {}
        self.chat_server: "ChatServer | None" = None

    async def start_subscribe(self):
        await self.subscribe(JOIN_CHANNEL)
        self.add_handler(JOIN_CHANNEL, self.on_join_room)
        await self.subscribe(EXIT_CHANNEL)
        self.add_handler(EXIT_CHANNEL, self.on_leave_room)
        self.start()

    async def on_join_room(self, c: str, s: str):
        channel_id, user_id = s.split(":")
        if self.chat_server is None:
            return
        await self.chat_server.join_room_channel(int(channel_id), int(user_id))

    async def on_leave_room(self, c: str, s: str):
        channel_id, user_id = s.split(":")
        if self.chat_server is None:
            return
        await self.chat_server.leave_room_channel(int(channel_id), int(user_id))