144 lines
4.9 KiB
Python
144 lines
4.9 KiB
Python
import threading
|
|
import json
|
|
from flask import Flask, jsonify, request
|
|
import sqlite3
|
|
import socket
|
|
|
|
app = Flask(__name__)
|
|
|
|
socket_server = socket.socket()
|
|
socket_server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
|
|
|
active_users = {}
|
|
chat_connections = []
|
|
|
|
def get_db_connection():
|
|
conn = sqlite3.connect("usr.db")
|
|
conn.row_factory = sqlite3.Row
|
|
return conn
|
|
|
|
def isuserxist(name):
|
|
cn = get_db_connection()
|
|
csr = cn.cursor()
|
|
csr.execute('SELECT * FROM users WHERE name = ?', (name,))
|
|
rst = csr.fetchone()
|
|
cn.close()
|
|
if rst is not None:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def ispsswdright(name,passwd):
|
|
cn = get_db_connection()
|
|
csr = cn.cursor()
|
|
csr.execute("SELECT COUNT(*) FROM users WHERE name=?", (name,))
|
|
row_count = csr.fetchone()[0]
|
|
password = None
|
|
if row_count > 0:
|
|
csr.execute("SELECT passwd FROM users WHERE name=?", (name,))
|
|
password = csr.fetchone()[0]
|
|
if password == passwd:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def register_user(usr, pwd):
|
|
conn = get_db_connection()
|
|
csr2 = conn.cursor()
|
|
csr2.execute('SELECT * FROM users WHERE name = ?', (usr,))
|
|
result = csr2.fetchone()
|
|
if result is not None:
|
|
return {"success": False, "message": "Username already exists"}
|
|
else:
|
|
try:
|
|
cursor = conn.cursor()
|
|
cursor.execute("INSERT INTO users (name, passwd) VALUES (?, ?)", (usr, pwd))
|
|
conn.commit()
|
|
return {"success": True, "message": "User registered successfully"}
|
|
except sqlite3.Error as e:
|
|
return {"success": False, "message": str(e)}
|
|
finally:
|
|
conn.close()
|
|
|
|
@app.route("/api/register", methods=['POST'])
|
|
def register1():
|
|
vl = request.get_json()
|
|
usr = vl.get('username')
|
|
pwd = vl.get('password')
|
|
result = register_user(usr, pwd)
|
|
if result['success']:
|
|
return jsonify(result)
|
|
else:
|
|
return jsonify(result), 403 if result['message'] == "Username already exists" else 500
|
|
|
|
def broadcast_message(message, sender=None):
|
|
for conn in chat_connections:
|
|
try:
|
|
conn.sendall(json.dumps(message).encode())
|
|
except:
|
|
chat_connections.remove(conn)
|
|
|
|
def handle_socket_message(data, addr, conn):
|
|
try:
|
|
action = data.get('type')
|
|
if action == 'register':
|
|
result = register_user(data.get('username'), data.get('password'))
|
|
return {"type": "register","status": "success" if result['success'] else "error", "message": result['message']}
|
|
elif action == 'login':
|
|
if isuserxist(data['username']):
|
|
if ispsswdright(data['username'], data['password']):
|
|
active_users[addr[0]] = data['username']
|
|
chat_connections.append(conn)
|
|
return {"type": "login", "status": "success", "message": "Login successful"}
|
|
return {"type": "login", "status": "error", "message": "Invalid credentials"}
|
|
elif action == 'chat':
|
|
if addr[0] in active_users:
|
|
message = {
|
|
"type": "chat",
|
|
"user": active_users[addr[0]],
|
|
"message": data['message']
|
|
}
|
|
broadcast_message(message)
|
|
return {"type": "chat", "status": "success", }
|
|
return {"type": "chat", "status": "error", "message": "Not logged in"}
|
|
except Exception as e:
|
|
return {"status": "error", "message": str(e)}
|
|
|
|
def run_socket_server():
|
|
socket_server.bind(("localhost", 8889))
|
|
socket_server.listen()
|
|
print("Socket server running on port 8889")
|
|
while True:
|
|
conn, addr = socket_server.accept()
|
|
print(f"Socket client connected: {addr}")
|
|
try:
|
|
while True:
|
|
data = conn.recv(1024)
|
|
if not data:
|
|
if addr[0] in active_users:
|
|
del active_users[addr[0]]
|
|
break
|
|
|
|
try:
|
|
json_data = json.loads(data.decode())
|
|
response = handle_socket_message(json_data, addr, conn)
|
|
conn.sendall(json.dumps(response).encode())
|
|
except json.JSONDecodeError:
|
|
conn.sendall(json.dumps(
|
|
{"status": "error", "message": "Invalid JSON"}
|
|
).encode())
|
|
except ConnectionResetError:
|
|
if addr[0] in active_users:
|
|
del active_users[addr[0]]
|
|
print(f"Client {addr} disconnected")
|
|
finally:
|
|
if conn in chat_connections:
|
|
chat_connections.remove(conn)
|
|
conn.close()
|
|
|
|
if __name__ == '__main__':
|
|
with get_db_connection() as conn:
|
|
conn.execute('''CREATE TABLE IF NOT EXISTS users
|
|
(name TEXT, passwd TEXT)''')
|
|
threading.Thread(target=run_socket_server).start()
|
|
app.run(port=5001) |