forked from navaneethkrishna30/WhisperZone
-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
160 lines (128 loc) · 5.03 KB
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
from string import ascii_uppercase
from datetime import datetime
import random
import json
import os
from flask_socketio import join_room, send, SocketIO
from flask import Flask, request, session, jsonify
from flask_cors import CORS
from pymongo import MongoClient
from redis import Redis
# Flask Config
app = Flask(__name__)
app.config["SECRET_KEY"] = os.urandom(24)
CORS(app, supports_credentials=True)
# MongoDB configuration
mongo_client = MongoClient('mongodb://mongo:27017/chat_db')
db = mongo_client['chat_db']
# Redis configuration
redis_host = 'redis'
redis = Redis(host=redis_host, port=6379, decode_responses=True)
#SocketIO config
socketio = SocketIO(app, cors_allowed_origins="*", message_queue=f'redis://redis:6379')
#room-code generator
def generate_unique_code(length):
while True:
code = "".join(random.choice(ascii_uppercase) for _ in range(length))
if not redis.exists(f"room:{code}"):
break
return code
@app.route("/api/create-room", methods=["POST"])
def create_room():
name = request.json.get("name")
if not name:
return jsonify({"error": "Name is required"}), 400
try:
room_code = generate_unique_code(6)
redis.hset(f"room:{room_code}", "members", json.dumps({}))
redis.hset(f"room:{room_code}", "messages", json.dumps([]))
session["room"] = room_code
session["name"] = name
return jsonify({"room": room_code, "name": name})
except Exception as e:
# Improved error handling: return detailed error message
return jsonify({"error": f"Failed to create room: {str(e)}"}), 500
@app.route("/api/join-room", methods=["POST"])
def join_room_api():
name = request.json.get("name")
code = request.json.get("code")
if not name:
return jsonify({"error": "Name is required"}), 400
if not redis.exists(f"room:{code}"):
return jsonify({"error": "Room does not exist"}), 400
try:
session["room"] = code
session["name"] = name
return jsonify({"room": code, "name": name})
except Exception as e:
# Improved error handling: return detailed error message
return jsonify({"error": f"Failed to join room: {str(e)}"}), 500
@app.route("/api/save-chat", methods=["POST"])
def save_chat():
room = request.json.get("room")
if not room:
return jsonify({"error": "Room ID is required"}), 400
try:
previous_messages = json.loads(redis.hget(f"room:{room}", "messages") or "[]")
chat_data = {
"room_id": room,
"messages": previous_messages,
}
db.chats.insert_one(chat_data)
return jsonify({"message": "Chat saved successfully"}), 200
except Exception as e:
# Improved error handling: return detailed error message
return jsonify({"error": f"Failed to save chat: {str(e)}"}), 500
# SockerIO routes
@socketio.on("message")
def handle_message(data):
room = session.get("room")
if not redis.exists(f"room:{room}"):
return
content = {
"name": session.get("name"),
"message": data["data"],
"timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
}
# Broadcast the message to all users in the room
send(content, to=room)
# Save the message in Redis
messages = json.loads(redis.hget(f"room:{room}", "messages"))
messages.append(content)
redis.hset(f"room:{room}", "messages", json.dumps(messages))
@socketio.on("connect")
def handle_connect():
room = session.get("room")
name = session.get("name")
if not room or not name:
return
if not redis.exists(f"room:{room}"):
return
join_room(room)
# Update member list in Redis
members = json.loads(redis.hget(f"room:{room}", "members"))
members[name] = True
redis.hset(f"room:{room}", "members", json.dumps(members))
# Notify the room about the new user, send updated member list to new user and broadcast that list to room
send({"members": list(members.keys())}, to=request.sid)
socketio.emit('members', {"members": list(members.keys())}, to=room)
# Send previous messages to the newly connected user
previous_messages = json.loads(redis.hget(f"room:{room}", "messages"))
socketio.emit('previous-messages', {"messages": previous_messages}, to=request.sid)
@socketio.on("disconnect")
def handle_disconnect():
room = session.get("room")
name = session.get("name")
if not room or not name:
return
if not redis.exists(f"room:{room}"):
return
# Remove the user from the member list in Redis
members = json.loads(redis.hget(f"room:{room}", "members"))
if name in members:
del members[name]
redis.hset(f"room:{room}", "members", json.dumps(members))
# Notify the room about the user leaving
socketio.emit('members', {"members": list(members.keys())}, to=room)
if __name__ == "__main__":
socketio.run(app)