discours.io/inbox
5
0
Fork 0
Code Pull Requests Actions Wiki Activity

formatted

Browse Source
This commit is contained in:
Untone
2023-10-14 15:59:43 +03:00
parent 4ddc424f29
commit 154633c114
15 changed files with 63 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
resolvers/__init__.py
View File

@@ -1,14 +1,13 @@
from resolvers.chats import create_chat, delete_chat, update_chat
from resolvers.load import load_chats, load_messages_by, load_recipients
from resolvers.messages import (
create_message,
delete_message,
update_message,
mark_as_read,
)
from resolvers.load import load_chats, load_messages_by, load_recipients
from resolvers.search import search_recipients
__all__ = [
# inbox
"load_chats",

4
resolvers/chats.py
View File

@@ -1,10 +1,11 @@
import json
import uuid
from datetime import datetime, timezone
from validators.inbox import Chat
from services.auth import login_required
from services.redis import redis
from services.schema import mutation
from validators.inbox import Chat
@mutation.field("updateChat")
@@ -107,4 +108,3 @@ async def delete_chat(_, info, chat_id: str):
await redis.execute("SREM", f"chats_by_author/{author_id}", chat_id)
else:
return {"error": "chat not exist"}

20
resolvers/load.py
View File

@@ -1,13 +1,15 @@
import asyncio
import json
from typing import Any, Dict, List, Optional, Union
from services.auth import login_required
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 validators.inbox import Message, Chat, ChatMember
from .chats import create_chat
from .unread import get_unread_counter
import asyncio
# NOTE: not an API handler
async def load_messages(
@@ -21,10 +23,10 @@ async def load_messages(
message_ids = [] + ids
if limit:
mids = (
await redis.lrange(
f"chats/{chat_id}/message_ids", offset, offset + limit
)
) or []
await redis.lrange(
f"chats/{chat_id}/message_ids", offset, offset + limit
)
) or []
mids = [mid for mid in mids]
message_ids += mids
if message_ids:
@@ -54,7 +56,7 @@ async def load_chats(
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)]
cids = list(cids)[offset: (offset + limit)]
chats = []
lock = asyncio.Lock()
if len(cids) == 0:
@@ -87,8 +89,8 @@ 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))
) or []
await redis.execute("SMEMBERS", "chats_by_author/" + str(author_id))
) or []
user_chats = [c for c in user_chats]
if user_chats:
messages = []

16
resolvers/messages.py
View File

@@ -1,11 +1,12 @@
import json
from datetime import datetime, timezone
from typing import List
from validators.inbox import Message
from services.auth import login_required
from services.presence import notify_message
from services.redis import redis
from services.schema import mutation
from validators.inbox import Message
@mutation.field("createMessage")
@@ -26,16 +27,15 @@ async def create_message(_, info, chat: str, body: str, reply_to=None):
else:
chat_dict = json.loads(chat_data)
print(chat_dict)
message_id = (
await redis.execute("GET", f"chats/{chat_dict['id']}/next_message_id")
) or 0
message_id = int(message_id)
message_id = await redis.execute("GET", f"chats/{chat_dict['id']}/next_message_id")
message_id = int(message_id) if message_id else 0
new_message: Message = {
"chat": chat_dict["id"],
"id": message_id,
"author": author_id,
"body": body,
"createdAt": int(datetime.now(tz=timezone.utc).timestamp()),
"updatedAt": None
}
if reply_to:
new_message["replyTo"] = reply_to
@@ -110,10 +110,10 @@ async def delete_message(_, info, chat_id: str, message_id: int):
return {"error": "chat not exist"}
chat = json.loads(chat)
message = await redis.execute("GET", f"chats/{chat_id}/messages/{str(message_id)}")
if not message:
message_data = await redis.execute("GET", f"chats/{chat_id}/messages/{str(message_id)}")
if not message_data:
return {"error": "message not exist"}
message: Message = json.loads(message)
message: Message = json.loads(message_data)
if message["author"] != author_id:
return {"error": "access denied"}

13
resolvers/search.py
View File

@@ -1,10 +1,11 @@
import json
from typing import Dict, Union, List, Any
from datetime import datetime, timezone, timedelta
from typing import Dict, Union, List, Any
from resolvers.load import load_messages
from services.auth import login_required
from services.core import get_network
from services.redis import redis
from resolvers.load import load_messages
from services.schema import query
@@ -14,11 +15,11 @@ async def search_recipients(_, info, text: str, limit: int = 50, offset: int = 0
result = []
# TODO: maybe redis scan?
author_id = info.context["author_id"]
talk_before = await redis.execute("GET", f"/chats_by_author/{author_id}")
if talk_before:
talk_before = list(json.loads(talk_before))[offset : (offset + limit)]
talk_before = list(json.loads(talk_before))[offset: (offset + limit)]
for chat_id in talk_before:
members = await redis.execute("GET", f"/chats/{chat_id}/members")
if members:
@@ -68,8 +69,8 @@ async def search_in_chats(
mmm = list(
filter(
lambda msg: int(datetime.now(tz=timezone.utc))
- int(msg["createdAt"])
< timedelta(days=days_ago),
- int(msg["createdAt"])
< timedelta(days=days_ago),
mmm,
)
)

2
resolvers/unread.py
View File

@@ -1,7 +1,7 @@
from services.redis import redis
import json
from services.auth import login_required
from services.redis import redis
async def get_unread_counter(chat_id: str, author_id: int) -> int: