From d628f2765fa78eafb9d60bd78aa8813df27dd336 Mon Sep 17 00:00:00 2001
From: Jaehun You <jaehun0121@snu.ac.kr>
Date: Thu, 21 Nov 2024 14:24:46 +0900
Subject: [PATCH] Setup ASGI config

See also: #17
---
 wlm_server/wlm_server/asgi.py | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/wlm_server/wlm_server/asgi.py b/wlm_server/wlm_server/asgi.py
index 03d5e6e..033872c 100644
--- a/wlm_server/wlm_server/asgi.py
+++ b/wlm_server/wlm_server/asgi.py
@@ -10,7 +10,19 @@
 import os
 
 from django.core.asgi import get_asgi_application
+from channels.auth import AuthMiddlewareStack
+from channels.routing import ProtocolTypeRouter, URLRouter
+from channels.security.websocket import AllowedHostsOriginValidator
 
 os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'wlm_server.settings')
 
-application = get_asgi_application()
+application = ProtocolTypeRouter(
+    {
+        'http': get_asgi_application(),
+        'websocket': AllowedHostsOriginValidator(
+            AuthMiddlewareStack(
+                URLRouter([])
+            )
+        ),
+    }
+)