main.py

import random
import string
from datetime import datetime
from flask import Flask, render_template, request, session, redirect, url_for
from flask_socketio import join_room, leave_room, emit, send, SocketIO

app = Flask(__name__)
app.config["SECRET_KEY"] = "vjkdilghjkkjhgffdghjkljfsdklndjk"
socketio = SocketIO(app)

rooms = {}


def generate_unique_code(length):
    while True:
        code = ''.join(random.choices(string.ascii_letters + string.digits, k=length)).upper()
        if code not in rooms:
            break

    return code
    

@app.route("/", methods=["GET", "POST"])
def home():
    session.clear()
    if request.method == "POST":
        name = request.form.get("name")
        code = request.form.get("code")
        join = request.form.get("join", False)
        create = request.form.get("create", False)
        
        if not name:
            return render_template(
                "home.html",
                error="Please enter a name.",
                code=code,
                name=name
            )

        if join != False and not code:
            return render_template(
                "home.html",
                error="Please enter a room code.",
                code=code,
                name=name
            )
        
        room = code
        if create != False:
            room = generate_unique_code(6)
            rooms[room] = {
                "members": 0,
                "messages": []
            }
        elif code not in rooms:
            return render_template(
                "home.html",
                error="The room does't not exists",
                code=code,
                name=name
            )
        session["room"] = room
        session["name"] = name
        return redirect(url_for('room'))

    return render_template("home.html")


@app.route("/room")
def room():
    room = session.get("room")
    name = session.get("name")
    if room is None or name is None or room not in rooms:
        return redirect(url_for("home"))

    return render_template("room.html", code=room, user=name, messages=rooms[room]["messages"])


@socketio.on('message')
def message(data):
    room = session.get("room")
    name = session.get("name")
    if room not in rooms:
        return
    content = {
        "name": name,
        "date": datetime.now().strftime("%d/%m/%Y %H:%M:%S"),
        "message": data["msg"]
    }
    
    send(content, to=room)
    rooms[room]["messages"].append(content)
    print(f"{name} said: {data['data']}")


@socketio.on('connect')
def connect(auth):
    room = session.get("room")
    name = session.get("name")
    
    if not room or not name:
        return
    
    if room not in rooms:
        leave_room(room)
        return
    
    join_room(room)
    send(
        {
            "name": name,
            "date": datetime.now().strftime("%d/%m/%Y %H:%M:%S"),
            "message": f"{name} has entered the room"
        },
        to=room
    )
    rooms[room]["members"] += 1
    print(f"{name} joined the room {room}")
    

@socketio.on("disconnect")
def disconnect():
    room = session.get("room")
    name = session.get("name")
    leave_room(room)
    
    if room in rooms:
        rooms[room]["members"] -= 1
        
    if rooms[room]["members"] <= 0:
        del rooms[room]
        
    send(
        {
            "name": name,
            "date": datetime.now().strftime("%d/%m/%Y %H:%M:%S"),
            "message": f"{name} has left the room"
        },
        to=room
    )
    print(f"{name} left the room {room}")


if __name__ == "__main__":
    socketio.run(app, debug=True, host="0.0.0.0")