inbox/resolvers/search.py

import json
from datetime import datetime, timedelta, timezone
from typing import Any, Dict, List, Union

from resolvers.load import load_messages
from services.auth import login_required
from services.core import authors_by_id
from services.rediscache import redis
from services.schema import query


@query.field("search_recipients")
@login_required
async def search_recipients(_, info, text: str, limit: int = 50, offset: int = 0):
    result = set([])

    # TODO: maybe redis scan?

    author_id = info.context["author_id"]
    existed_chats = await redis.execute("SMEMBERS", f"/chats_by_author/{author_id}")
    if existed_chats:
        for chat_id in list(json.loads(existed_chats))[offset : (offset + limit)]:
            members_ids = await redis.execute("GET", f"/chats/{chat_id}/members")
            for member_id in members_ids:
                author = authors_by_id.get(member_id)
                if author:
                    if author["name"].startswith(text):
                        result.add(author)

    more_amount = limit - len(result)
    if more_amount > 0:
        result.update(authors_by_id.values()[0:more_amount])
    return {"members": list(result), "error": None}


@query.field("search_messages")
@login_required
async def search_messages(
    _, info, by: Dict[str, Union[str, int]], limit: int, offset: int
) -> Dict[str, Union[List[Dict[str, Any]], None]]:
    author_id = info.context["author_id"]
    lookup_chats = set((await redis.execute("SMEMBERS", f"chats_by_author/{author_id}")) or [])
    messages_set = set([])

    by_member = by.get("author")
    body_like = by.get("body")
    days_ago = by.get("days")

    # pre-filter lookup chats
    if by_member:
        lookup_chats = filter(
            lambda ca: by_member in ca["members"],
            list(lookup_chats),
        )

    # load the messages from lookup chats
    for c in lookup_chats:
        chat_id = c.decode()
        mmm = await load_messages(chat_id, limit, offset)
        if by_member:
            mmm = list(filter(lambda mx: mx["author"] == by_member, mmm))
        if body_like:
            mmm = list(filter(lambda mx: body_like in mx["body"], mmm))
        if days_ago:
            mmm = list(
                filter(
                    lambda msg: int(datetime.now(tz=timezone.utc)) - int(msg["created_at"])
                    < int(timedelta(days=days_ago)),
                    mmm,
                )
            )

        messages_set.union(set(mmm))

    messages_sorted = sorted(list(messages_set))
    return {"messages": messages_sorted, "error": None}