barebones unauthenticated websocket

Archmonger · Archmonger · commit 923276e7d431 · 2021-05-17T22:59:37.000-07:00
diff --git a/dj_idom/asgi.py b/dj_idom/asgi.py
@@ -17,19 +17,13 @@
 # Fetch ASGI application before importing dependencies that require ORM models.
 http_asgi_app = get_asgi_application()
 
-from channels.auth import AuthMiddlewareStack
 from channels.routing import ProtocolTypeRouter, URLRouter
-from channels.security.websocket import AllowedHostsOriginValidator
 
 from .consumers import CommandConsumer
 
 application = ProtocolTypeRouter(
     {
-        # ASGI app has concurrency problems, see
-        # See https://github.com/django/channels/issues/1587
         "http": http_asgi_app,
-        "websocket": AllowedHostsOriginValidator(
-            AuthMiddlewareStack(URLRouter([url("", CommandConsumer().as_asgi())]))
-        ),
+        "websocket": URLRouter([url("", CommandConsumer().as_asgi())]),
     }
 )
diff --git a/dj_idom/settings.py b/dj_idom/settings.py
@@ -37,6 +37,7 @@
     "django.contrib.sessions",
     "django.contrib.messages",
     "django.contrib.staticfiles",
+    "channels",  # Websocket library
 ]
 
 MIDDLEWARE = [
@@ -123,3 +124,8 @@
 # https://docs.djangoproject.com/en/3.2/ref/settings/#default-auto-field
 
 DEFAULT_AUTO_FIELD = "django.db.models.BigAutoField"
+
+# Static Files (CSS, JavaScript, Images)
+STATICFILES_DIRS = [
+    os.path.join(BASE_DIR, "dj_idom", "static"),
+]
diff --git a/dj_idom/static/scripts.js b/dj_idom/static/scripts.js
@@ -0,0 +1,40 @@
+// Set up a websocket at the base endpoint
+let LOCATION = window.location;
+let WS_PROTOCOL = "";
+if (LOCATION.protocol == "https:") {
+	WS_PROTOCOL = "wss://";
+} else {
+	WS_PROTOCOL = "ws://";
+}
+let WS_ENDPOINT_URL = WS_PROTOCOL + LOCATION.host;
+let COMMAND_SOCKET = new WebSocket(WS_ENDPOINT_URL);
+
+// Receivable commands
+COMMAND_SOCKET.onmessage = function (response) {
+	// Websocket message received, parse for JSON
+	console.info(response);
+	json_response = JSON.parse(response.data);
+
+	// Check for valid commands
+	console.info("Websocket has recieved a message", json_response);
+};
+
+// Websocket open event
+COMMAND_SOCKET.onopen = function () {
+	console.info("Websocket has opened.");
+};
+
+// Websocket close event
+COMMAND_SOCKET.onclose = function () {
+	console.info("Websocket has closed.");
+};
+
+// Websocket error event
+COMMAND_SOCKET.onerror = function (error) {
+	console.error(
+		"Websocket encountered a crtical error: ",
+		error.message,
+		"Closing socket..."
+	);
+	COMMAND_SOCKET.close();
+};
diff --git a/dj_idom/templates/base.html b/dj_idom/templates/base.html
@@ -1,10 +1,12 @@
+{% load static %}
 <!DOCTYPE html>
 <html lang="en">
 
 <head>
     <meta charset="UTF-8">
     <meta http-equiv="X-UA-Compatible" content="IE=edge">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <script src="{% static 'scripts.js' %}" crossorigin="anonymous"></script>
     <title>IDOM</title>
 </head>
 
diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,4 @@
-django<4.0.0
-daphne<4.0.0
-channels<4.0.0
-idom<1.0.0
+django<4.0.0 # Django Library
+daphne<4.0.0 # Production ASGI webserver
+channels<4.0.0 # Django websocket features
+idom<1.0.0 # Python React

Original file line number	Diff line number	Diff line change
`@@ -17,19 +17,13 @@`
`17`	`17`	`# Fetch ASGI application before importing dependencies that require ORM models.`
`18`	`18`	`http_asgi_app = get_asgi_application()`
`19`	`19`
`20`		`-from channels.auth import AuthMiddlewareStack`
`21`	`20`	`from channels.routing import ProtocolTypeRouter, URLRouter`
`22`		`-from channels.security.websocket import AllowedHostsOriginValidator`
`23`	`21`
`24`	`22`	`from .consumers import CommandConsumer`
`25`	`23`
`26`	`24`	`application = ProtocolTypeRouter(`
`27`	`25`	`{`
`28`		`- # ASGI app has concurrency problems, see`
`29`		`- # See https://github.com/django/channels/issues/1587`
`30`	`26`	`"http": http_asgi_app,`
`31`		`- "websocket": AllowedHostsOriginValidator(`
`32`		`- AuthMiddlewareStack(URLRouter([url("", CommandConsumer().as_asgi())]))`
`33`		`- ),`
	`27`	`+ "websocket": URLRouter([url("", CommandConsumer().as_asgi())]),`
`34`	`28`	`}`
`35`	`29`	`)`