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)