import os

os.environ.setdefault("DJANGO_SETTINGS_MODULE", "config.settings")

from channels.auth import AuthMiddlewareStack
from channels.routing import ProtocolTypeRouter, URLRouter
from django.core.asgi import get_asgi_application
from django.urls import path

django_asgi_application = get_asgi_application()

from common.consumers import HealthConsumer
from contest.routing import websocket_urlpatterns

application = ProtocolTypeRouter(
    {
        "http": django_asgi_application,
        "websocket": AuthMiddlewareStack(
            URLRouter(
                [
                    path("ws/health/", HealthConsumer.as_asgi()),
                    *websocket_urlpatterns,
                ]
            )
        ),
    }
)