core/resolvers/inbox/load.py

import json
# from datetime import datetime, timedelta, timezone

from auth.authenticate import login_required
from auth.credentials import AuthCredentials
from base.redis import redis
from base.orm import local_session
from base.resolvers import query
from orm.user import User
from resolvers.zine.profile import followed_authors
from .unread import get_unread_counter


async def load_messages(chat_id: str, limit: int, offset: int):
    ''' load :limit messages for :chat_id with :offset '''
    messages = []
    # print(f'[inbox] loading messages by chat: {chat_id}[{offset}:{offset + limit}]')
    try:
        message_ids = await redis.lrange(f"chats/{chat_id}/message_ids",
                                         offset,
                                         offset + limit
                                         )

        # print(f'[inbox] message_ids: {message_ids}')
    except Exception as e:
        print(e)
    if message_ids:
        message_keys = [
            f"chats/{chat_id}/messages/{mid.decode('utf-8')}" for mid in message_ids
        ]
        # print(message_keys)
        messages = await redis.mget(*message_keys)
        messages = [json.loads(msg.decode('utf-8')) for msg in messages]
        # print('[inbox] messages \n%r' % messages)
    return messages


@query.field("loadChats")
@login_required
async def load_chats(_, info, limit: int = 50, offset: int = 0):
    """ load :limit chats of current user with :offset """
    auth: AuthCredentials = info.context["request"].auth

    cids = await redis.execute("SMEMBERS", "chats_by_user/" + str(auth.user_id))
    onliners = await redis.execute("SMEMBERS", "users-online")
    if cids:
        cids = list(cids)[offset:offset + limit]
    if not cids:
        print('[inbox.load] no chats were found')
        cids = []
    chats = []
    for cid in cids:
        cid = cid.decode("utf-8")
        c = await redis.execute("GET", "chats/" + cid)
        if c:
            c = dict(json.loads(c))
            c['messages'] = await load_messages(cid, 5, 0)
            c['unread'] = await get_unread_counter(cid, auth.user_id)
            with local_session() as session:
                c['members'] = []
                for uid in c["users"]:
                    a = session.query(User).where(User.id == uid).first()
                    if a:
                        c['members'].append({
                            "id": a.id,
                            "slug": a.slug,
                            "userpic": a.userpic,
                            "name": a.name,
                            "lastSeen": a.lastSeen,
                            "online": a.id in onliners
                        })
                chats.append(c)
    return {
        "chats": chats,
        "error": None
    }


@query.field("loadMessagesBy")
@login_required
async def load_messages_by(_, info, by, limit: int = 10, offset: int = 0):
    ''' load :limit messages of :chat_id with :offset '''

    auth: AuthCredentials = info.context["request"].auth
    userchats = await redis.execute("SMEMBERS", "chats_by_user/" + str(auth.user_id))
    userchats = [c.decode('utf-8') for c in userchats]
    # print('[inbox] userchats: %r' % userchats)
    if userchats:
        # print('[inbox] loading messages by...')
        messages = []
        by_chat = by.get('chat')
        if by_chat in userchats:
            chat = await redis.execute("GET", f"chats/{by_chat}")
            # print(chat)
            if not chat:
                return {
                    "messages": [],
                    "error": "chat not exist"
                }
            # everyone's messages in filtered chat
            messages = await load_messages(by_chat, limit, offset)
        return {
            "messages": sorted(
                list(messages),
                key=lambda m: m['createdAt']
            ),
            "error": None
        }
    else:
        return {
            "error": "Cannot access messages of this chat"
        }


@query.field("loadRecipients")
async def load_recipients(_, info, limit=50, offset=0):
    chat_users = []
    auth: AuthCredentials = info.context["request"].auth

    try:
        chat_users += await followed_authors(auth.user_id)
        limit = limit - len(chat_users)
    except Exception:
        pass
    with local_session() as session:
        chat_users += session.query(User).where(User.emailConfirmed).limit(limit).offset(offset)
    return {
        "members": chat_users,
        "error": None
    }