websocket deplyment

pycob · Mar 22, 2023 · 68bfedc · 68bfedc
1 parent e5cd5ff
commit 68bfedc
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/sample-apps/websockets/main.py b/sample-apps/websockets/main.py
@@ -13,11 +13,18 @@ def to_html(self):
         return """
         <div id="log"></div>
         <script>
+            var httpProtocol = 'http://'; 
+            var wsProtocol = 'ws://';
+            if (window.location.protocol === 'https:') {
+                httpProtocol = 'http://';
+                wsProtocol = 'wss://';
+            }
+
             const log = (text, color) => {
                 document.getElementById('log').innerHTML += `<span style="color: ${color}">${text}</span><br>`;
             };
 
-            const socket = new WebSocket('ws://' + location.host + '""" + self.path_to_websocket + """');
+            const socket = new WebSocket(wsProtocol + location.host + '""" + self.path_to_websocket + """');
             socket.addEventListener('message', ev => {
                 document.getElementById('log').innerHTML += ev.data;
             });
@@ -49,7 +56,7 @@ def to_html(self):
 
 @app.route('/')
 def index():
-    page = pv.Page('Websocket Test')
+    page = pv.Page('Websocket Test', description='WebSocket proof of concept using Flask-Sock and PyVibe')
 
     page.add_header("Websocket Test")
 
@@ -62,7 +69,8 @@ def index():
             card.add_header('Receive')
             card.add_component(WebSocketReceiverComponent('/echo'))
 
-    page.add_emgithub("https://github.com/pycob/pyvibe/blob/weboscket/sample-apps/websockets/main.py")
+    page.add_header("Source Code")
+    page.add_emgithub("https://github.com/pycob/pyvibe/blob/main/sample-apps/websockets/main.py")
 
     return page.to_html()
 

diff --git a/sample-apps/websockets/requirements.txt b/sample-apps/websockets/requirements.txt
@@ -1,4 +1,5 @@
 pyvibe
+pycob
 flask
 flask-sock
 gunicorn