130 lines
4.5 KiB
Python
130 lines
4.5 KiB
Python
import json
|
|
from datetime import datetime, timezone
|
|
|
|
from auth.authenticate import login_required
|
|
from auth.credentials import AuthCredentials
|
|
from base.redis import redis
|
|
from base.resolvers import mutation
|
|
from services.following import FollowingManager, FollowingResult
|
|
|
|
|
|
@mutation.field("createMessage")
|
|
@login_required
|
|
async def create_message(_, info, chat: str, body: str, replyTo=None):
|
|
"""create message with :body for :chat_id replying to :replyTo optionally"""
|
|
auth: AuthCredentials = info.context["request"].auth
|
|
|
|
chat = await redis.execute("GET", f"chats/{chat}")
|
|
if not chat:
|
|
return {"error": "chat is not exist"}
|
|
else:
|
|
chat_dict = dict(json.loads(chat))
|
|
message_id = await redis.execute("GET", f"chats/{chat_dict['id']}/next_message_id")
|
|
message_id = int(message_id)
|
|
new_message = {
|
|
"chatId": chat_dict["id"],
|
|
"id": message_id,
|
|
"author": auth.user_id,
|
|
"body": body,
|
|
"createdAt": int(datetime.now(tz=timezone.utc).timestamp()),
|
|
}
|
|
if replyTo:
|
|
new_message["replyTo"] = replyTo
|
|
chat_dict["updatedAt"] = new_message["createdAt"]
|
|
await redis.execute("SET", f"chats/{chat_dict['id']}", json.dumps(chat))
|
|
print(f"[inbox] creating message {new_message}")
|
|
await redis.execute(
|
|
"SET", f"chats/{chat_dict['id']}/messages/{message_id}", json.dumps(new_message)
|
|
)
|
|
await redis.execute("LPUSH", f"chats/{chat_dict['id']}/message_ids", str(message_id))
|
|
await redis.execute("SET", f"chats/{chat_dict['id']}/next_message_id", str(message_id + 1))
|
|
|
|
users = chat_dict["users"]
|
|
for user_slug in users:
|
|
await redis.execute(
|
|
"LPUSH", f"chats/{chat_dict['id']}/unread/{user_slug}", str(message_id)
|
|
)
|
|
|
|
result = FollowingResult("NEW", "chat", new_message)
|
|
await FollowingManager.push("chat", result)
|
|
|
|
return {"message": new_message, "error": None}
|
|
|
|
|
|
@mutation.field("updateMessage")
|
|
@login_required
|
|
async def update_message(_, info, chat_id: str, message_id: int, body: str):
|
|
auth: AuthCredentials = info.context["request"].auth
|
|
|
|
chat = await redis.execute("GET", f"chats/{chat_id}")
|
|
if not chat:
|
|
return {"error": "chat not exist"}
|
|
|
|
message = await redis.execute("GET", f"chats/{chat_id}/messages/{message_id}")
|
|
if not message:
|
|
return {"error": "message not exist"}
|
|
|
|
message = json.loads(message)
|
|
if message["author"] != auth.user_id:
|
|
return {"error": "access denied"}
|
|
|
|
message["body"] = body
|
|
message["updatedAt"] = int(datetime.now(tz=timezone.utc).timestamp())
|
|
|
|
await redis.execute("SET", f"chats/{chat_id}/messages/{message_id}", json.dumps(message))
|
|
|
|
result = FollowingResult("UPDATED", "chat", message)
|
|
await FollowingManager.push("chat", result)
|
|
|
|
return {"message": message, "error": None}
|
|
|
|
|
|
@mutation.field("deleteMessage")
|
|
@login_required
|
|
async def delete_message(_, info, chat_id: str, message_id: int):
|
|
auth: AuthCredentials = info.context["request"].auth
|
|
|
|
chat = await redis.execute("GET", f"chats/{chat_id}")
|
|
if not chat:
|
|
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:
|
|
return {"error": "message not exist"}
|
|
message = json.loads(message)
|
|
if message["author"] != auth.user_id:
|
|
return {"error": "access denied"}
|
|
|
|
await redis.execute("LREM", f"chats/{chat_id}/message_ids", 0, str(message_id))
|
|
await redis.execute("DEL", f"chats/{chat_id}/messages/{str(message_id)}")
|
|
|
|
users = chat["users"]
|
|
for user_id in users:
|
|
await redis.execute("LREM", f"chats/{chat_id}/unread/{user_id}", 0, str(message_id))
|
|
|
|
result = FollowingResult("DELETED", "chat", message)
|
|
await FollowingManager.push(result)
|
|
|
|
return {}
|
|
|
|
|
|
@mutation.field("markAsRead")
|
|
@login_required
|
|
async def mark_as_read(_, info, chat_id: str, messages: [int]):
|
|
auth: AuthCredentials = info.context["request"].auth
|
|
|
|
chat = await redis.execute("GET", f"chats/{chat_id}")
|
|
if not chat:
|
|
return {"error": "chat not exist"}
|
|
|
|
chat = json.loads(chat)
|
|
users = set(chat["users"])
|
|
if auth.user_id not in users:
|
|
return {"error": "access denied"}
|
|
|
|
for message_id in messages:
|
|
await redis.execute("LREM", f"chats/{chat_id}/unread/{auth.user_id}", 0, str(message_id))
|
|
|
|
return {"error": None}
|