import asyncio


class FollowingResult:
    def __init__(self, event, kind, payload):
        self.event = event
        self.kind = kind
        self.payload = payload


class Following:
    queue = asyncio.Queue()

    def __init__(self, kind, uid):
        self.kind = kind  # author topic shout chat
        self.uid = uid


class FollowingManager:
    lock = asyncio.Lock()
    data = {
        'author': [],
        'topic': [],
        'shout': [],
        'chat': []
    }

    @staticmethod
    async def register(kind, uid):
        async with FollowingManager.lock:
            FollowingManager[kind].append(uid)

    @staticmethod
    async def remove(kind, uid):
        async with FollowingManager.lock:
            FollowingManager[kind].remove(uid)

    @staticmethod
    async def push(kind, payload):
        try:
            async with FollowingManager.lock:
                if kind == 'chat':
                    for chat in FollowingManager['chat']:
                        if payload.message["chatId"] == chat.uid:
                            chat.queue.put_nowait(payload)
                else:
                    for entity in FollowingManager[kind]:
                        if payload.shout['createdBy'] == entity.uid:
                            entity.queue.put_nowait(payload)
        except Exception as e:
            print(Exception(e))