implement chat message sending

backend/server.py

@@ -0,0 +1,124 @@
+from quart import Quart, Response, request, send_from_directory, websocket
+import json
+import asyncio
+
+def start_server(app):
+    server = Quart(__name__, static_folder='../frontend/build/web')
+
+    @server.route('/')
+    async def index():
+        return await send_from_directory(server.static_folder, 'index.html')
+
+    @server.route('/<path:path>')
+    async def serve_static(path):
+        return await send_from_directory(server.static_folder, path)
+
+    @server.route('/user/signup', methods=['POST'])
+    async def signup():
+        data = await request.get_json()
+
+        user_id = None
+
+        username = data['username']
+        email = data['email']
+        password = data['password']
+
+        user_id = app.create_new_user(username, email, password)
+        if user_id is None:
+            return {"error": "existing credentials"}
+        
+        app.generate_default_user_data(user_id)
+
+        # token = auth_manager.dump_token(user_id)
+        return {"userId": user_id}
+
+    @server.websocket('/ws')
+    async def ws():
+        print("incoming ws")
+
+        # Authenticate before dedicating any resources
+        auth_req = json.loads(await websocket.receive())
+        if "username" not in auth_req or "password" not in auth_req:
+            print("Invalid auth request:", auth_req)
+            await websocket.send(json.dumps({'error': 'invalid auth'}))
+            return
+
+        user_id = app.verify_user_credentials(auth_req['username'], auth_req['password'])
+        if user_id is None:
+            await websocket.send(json.dumps({'error': 'invalid credentials'}))
+            return
+
+        send_queue = asyncio.Queue()
+        app.add_user_connection(user_id, send_queue)
+
+        async def sender_task():
+            nonlocal send_queue
+            while True:
+                msg = await send_queue.get()
+                await websocket.send(json.dumps(msg))
+
+        async def send(msg):
+            await send_queue.put(msg)
+
+        asyncio.create_task(sender_task())
+
+        try:
+            while True:
+                req = json.loads(await websocket.receive())
+                action = req['action']
+                body = req.get('body', {})
+                nonce = req.get('nonce', None)
+
+                if action == "ping":
+                    print("ping")
+                    await send({'message': "pong"})
+
+                elif action == "create-agent":
+                    agent_id = app.create_agent(
+                        user_id=user_id,
+                        name=body['name'],
+                        description=body['description'],
+                        picture=body.get('picture'),
+                        kind=body['kind'],
+                        data=body['data']
+                    )
+                    await send({'message': "agent created", 'agentId': agent_id, 'nonce': nonce})
+
+                elif action == "get-agents":
+                    agents_data = app.get_agents(user_id)
+                    await send({'message': "agents retrieved", 'agents': agents_data, 'nonce': nonce})
+
+                elif action == "create-chat":
+                    chat_id = app.create_chat(
+                        user_id=user_id,
+                        name=body['name'],
+                        description=body.get('description'),
+                        picture=body.get('picture')
+                    )
+                    await send({'message': "chat created", 'chatId': chat_id, 'nonce': nonce})
+
+                elif action == "get-chats":
+                    chats_data = app.get_chats(user_id)
+                    await send({'message': "chats retrieved", 'chats': chats_data, 'nonce': nonce})
+
+                elif action == "get-chat-messages":
+                    messages_data = app.get_chat_messages(
+                        body['chatId'],
+                        earliest_index=body.get('earliestIndex', 0)
+                    )
+                    await send({'message': "chat messages retrieved", 'messages': messages_data, 'nonce': nonce})
+
+                elif action == "send-chat-message":
+                    message = app.send_chat_message(
+                        chat_id=body['chatId'],
+                        content=body['content'],
+                    )
+                    await send({'message': "message sent", 'messageId': message['id'], 'nonce': nonce})
+
+                else:
+                    print("Unknown message: " + action)
+
+        finally:
+            app.remove_user_connection(user_id, send_queue)
+
+    server.run(debug=True)