app.py

import os
from flask import Flask
from flask_sock import Sock
from dotenv import load_dotenv

load_dotenv()

sock = Sock()

def create_app(config: dict | None = None) -> Flask:
    app = Flask(__name__)
    app.config["SECRET_KEY"] = os.environ.get("FLASK_SECRET_KEY", "dev")
    if config:
        app.config.update(config)

    sock.init_app(app)

    _ensure_indexes(app)

    from routes.views import views_bp
    from routes.decks import decks_bp
    from routes.rooms import rooms_bp
    from routes.battle import battle_bp

    app.register_blueprint(views_bp)
    app.register_blueprint(decks_bp)
    app.register_blueprint(rooms_bp)
    app.register_blueprint(battle_bp)

    from sockets.stream import register_stream
    register_stream(sock)

    return app


def _ensure_indexes(app: Flask) -> None:
    if app.config.get("TESTING"):
        return

    from db import get_db
    from pymongo import ASCENDING

    with app.app_context():
        try:
            db = get_db()
            db.rooms.create_index([("room_code", ASCENDING)], unique=True)
            db.player_card_stats.create_index(
                [("room_id", ASCENDING), ("player_username", ASCENDING), ("card_id", ASCENDING)],
                unique=True,
            )
        except Exception:
            pass


if __name__ == "__main__":
    create_app().run(debug=True, port=8080)