trim

Author: gaultier

main.js

@@ -4,30 +4,35 @@ import * as scheme from './scheme.min.js';
 import * as x86asm from './x86asm.min.js';
 import * as dockerfile from './dockerfile.min.js';
 
-let socket = new WebSocket("ws://localhost:8001/ws", "echo");
-socket.onopen = (ev) => {
-  console.log("connected", ev);
-  window.addEventListener('beforeunload', () => {
-    console.log('closing', ev);
+function ws_connect() {
+  const socket = new WebSocket("ws://localhost:8001/ws", "echo");
+  socket.onopen = (ev) => {
+    console.log("connected", ev);
+    window.addEventListener('beforeunload', () => {
+      console.log('closing', ev);
+      socket.close();
+    });
+  }
+  socket.onmessage = function(ev) {
+    console.log('reloading', ev);
     socket.close();
-  });
-}
-socket.onmessage = function(ev) {
-  console.log('reloading', ev);
-  socket.close();
-  location.reload();
-}
-socket.onclose = function() {
-  // TODO: Reconnect?
-  console.log('closed');
+    location.reload();
+  }
+  socket.onclose = function() {
+    // TODO: Reconnect?
+    console.log('closed');
+    // Retry.
+    setTimeout(connect, 2000);
+  }
+  socket.onerror = function(ev) {
+    console.log('error', ev);
+  }
 }
-socket.onerror = function(ev) {
-  console.log('error', ev);
+
+if (!location.origin.includes("github")) {
+  ws_connect();
 }
 
-// hljs.configure({
-  // ignoreUnescapedHTML: true,
-// });
 hljs.registerLanguage("cmake", cmake.default);
 hljs.registerLanguage("scheme", scheme.default);
 hljs.registerLanguage("x86asm", x86asm.default);

src/main.rs

@@ -14,7 +14,7 @@ use std::{
     path::{Path, PathBuf},
     process::Command,
     thread,
-    time::Instant,
+    time::{Duration, Instant},
 };
 
 use std::io::Write;
@@ -1423,6 +1423,7 @@ fn main() {
                         loop {
                             if let Ok(_) = erx.recv() {
                               println!("ws event");
+                              thread::sleep(Duration::from_millis(200));
                               ws.send_text("reload").unwrap();
                               println!("ws event sent");
                             }