inbox/resolvers/load.py

import json
from services.core import get_author, get_network
from services.redis import redis
from services.auth import login_required
from services.schema import query
from .chats import create_chat
from .unread import get_unread_counter
import asyncio


# NOTE: not an API handler
async def load_messages(chat_id: str, limit: int = 5, offset: int = 0, ids=None):
    """load :limit messages for :chat_id with :offset"""
    if ids is None:
        ids = []
    messages = []
    try:
        message_ids = [] + ids
        if limit:
            mids = await redis.lrange(
                f"chats/{chat_id}/message_ids", offset, offset + limit
            )
            mids = [mid for mid in mids]
            message_ids += mids
        if message_ids:
            message_keys = [f"chats/{chat_id}/messages/{mid}" for mid in message_ids]
            messages = await redis.mget(*message_keys)
            messages = [json.loads(msg.decode("utf-8")) for msg in messages]
            replies = []
            for m in messages:
                rt = m.get("replyTo")
                if rt:
                    rt = int(rt)
                    if rt not in message_ids:
                        replies.append(rt)
            if replies:
                messages += await load_messages(chat_id, offset, limit, replies)
    except Exception as e:
        print(f"Error loading messages for chat {chat_id}: {e}")
    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"""
    author_id = info.context["author_id"]
    cids = (await redis.execute("SMEMBERS", f"chats_by_author/{author_id}")) or []
    members_online = (await redis.execute("SMEMBERS", "authors-online")) or []
    cids = list(cids)[offset:(offset + limit)]
    chats = []
    lock = asyncio.Lock()
    if len(cids) == 0:
        print(f"[resolvers.load] no chats for user with id={author_id}, create one with Discours (id=2)")
        r = await create_chat(None, info, members=[2])  # member with id = 2 is discours
        print(f"[resolvers.load] created chat: {r}")
        cids.append(r["chat"]["id"])
    for cid in cids:
        async with lock:
            c = await redis.execute("GET", f"chats/{cid}")
            print(f"[resolvers.load] redis GET by {cid}: {c}")
            if c:
                c = json.loads(c)
                c["messages"] = await load_messages(cid, 5, 0)
                c["unread"] = await get_unread_counter(cid, author_id)
                member_ids = c["members"].copy()
                c["members"] = []
                for member_id in member_ids:
                    a = await get_author(member_id)
                    print(f"[resolvers.load] author with id={member_id}: {a}")
                    if a:
                        a["online"] = a.get("id") in members_online
                        c["members"].append(a)
                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"""
    author_id = info.context["author_id"]
    user_chats = await redis.execute("SMEMBERS", "chats_by_author/" + str(author_id))
    user_chats = [c for c in user_chats]
    if user_chats:
        messages = []
        by_chat = by.get("chat")
        if by_chat in user_chats:
            chat = await redis.execute("GET", f"chats/{by_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):
    """load possible chat participants"""
    onliners = (await redis.execute("SMEMBERS", "authors-online")) or []
    members = []
    all_authors = await get_network(limit, offset)
    for a in all_authors:
        members.append(
            {
                "id": a.id,
                "slug": a.slug,
                "userpic": a.userpic,
                "name": a.name,
                "lastSeen": a.lastSeen,
                "online": a.id in onliners,
            }
        )

    # NOTE: maybe sort members here

    return {"members": members, "error": None}